fix: update shell

server/app/shell/reinstall-os.sh

@@ -2,6 +2,8 @@
 
 set -euo pipefail
 
+TTY_DEVICE="/dev/tty"
+
 if [ "${EUID:-$(id -u)}" -ne 0 ]; then
   echo "请使用 root 身份运行此脚本"
   exit 1
@@ -25,6 +27,31 @@ need_cmd() {
   command -v "$1" >/dev/null 2>&1
 }
 
+has_tty() {
+  [ -r "$TTY_DEVICE" ] && [ -w "$TTY_DEVICE" ]
+}
+
+read_from_tty() {
+  local __var_name="$1"
+  local __prompt="$2"
+  local __silent="${3:-false}"
+  local __input
+
+  if has_tty; then
+    if [ "$__silent" = "true" ]; then
+      read -r -s -p "$__prompt" __input < "$TTY_DEVICE"
+      printf '\n' > "$TTY_DEVICE"
+    else
+      read -r -p "$__prompt" __input < "$TTY_DEVICE"
+    fi
+  else
+    warn "当前没有可交互终端，无法读取用户输入"
+    return 1
+  fi
+
+  printf -v "$__var_name" '%s' "$__input"
+}
+
 install_basic_tools() {
   if need_cmd apt-get; then
     apt-get update -y
@@ -51,13 +78,14 @@ download_reinstall_script() {
 }
 
 confirm_reinstall() {
+  local answer
   echo "--------------------------------"
   echo "该脚本仅封装 bin456789/reinstall 项目。"
   echo "重装会清空目标硬盘数据，存在失联风险。"
   echo "请提前备份重要数据，并确认可以接受自动重启。"
   echo "如果误操作，可在重启前执行: bash reinstall.sh reset"
   echo "--------------------------------"
-  read -r -p "确认继续吗？(Y/N): " answer
+  read_from_tty answer "确认继续吗？(Y/N): "
   case "$answer" in
     [Yy]) ;;
     *)
@@ -72,8 +100,7 @@ read_password_args() {
   local user_label="$2"
   local pass
   echo "默认用户名: $user_label"
-  read -r -s -p "$prompt" pass
-  echo
+  read_from_tty pass "$prompt" true
   if [ -n "$pass" ]; then
     PASSWORD_ARGS=(--password "$pass")
   else
@@ -139,7 +166,7 @@ main() {
 
   confirm_reinstall
   show_menu
-  read -r -p "请选择功能: " choice
+  read_from_tty choice "请选择功能: "
 
   case "$choice" in
     1)
@@ -237,7 +264,7 @@ main() {
       ;;
 
     51)
-      read -r -p "请输入 Raw/VHD 镜像链接: " img_url
+      read_from_tty img_url "请输入 Raw/VHD 镜像链接: "
       if [ -z "$img_url" ]; then
         echo "镜像链接不能为空"
         exit 1

server/app/shell/swap.sh

@@ -0,0 +1,90 @@
+#!/usr/bin/env bash
+#Blog:https://www.moerats.com/
+
+Green="\033[32m"
+Font="\033[0m"
+Red="\033[31m"
+
+#root权限
+root_need(){
+    if [[ $EUID -ne 0 ]]; then
+        echo -e "${Red}Error:This script must be run as root!${Font}"
+        exit 1
+    fi
+}
+
+#检测ovz
+ovz_no(){
+    if [[ -d "/proc/vz" ]]; then
+        echo -e "${Red}Your VPS is based on OpenVZ，not supported!${Font}"
+        exit 1
+    fi
+}
+
+add_swap(){
+echo -e "${Green}请输入需要添加的swap，建议为内存的2倍！${Font}"
+read -p "请输入swap数值(MB):" swapsize
+
+#检查是否存在swapfile
+grep -q "swapfile" /etc/fstab
+
+#如果不存在将为其创建swap
+if [ $? -ne 0 ]; then
+	echo -e "${Green}swapfile未发现，正在为其创建swapfile${Font}"
+	fallocate -l ${swapsize}M /swapfile
+	chmod 600 /swapfile
+	mkswap /swapfile
+	swapon /swapfile
+	echo '/swapfile none swap defaults 0 0' >> /etc/fstab
+         echo -e "${Green}swap创建成功，并查看信息：${Font}"
+         cat /proc/swaps
+         cat /proc/meminfo | grep Swap
+else
+	echo -e "${Red}swapfile已存在，swap设置失败，请先运行脚本删除swap后重新设置！${Font}"
+fi
+}
+
+del_swap(){
+#检查是否存在swapfile
+grep -q "swapfile" /etc/fstab
+
+#如果存在就将其移除
+if [ $? -eq 0 ]; then
+	echo -e "${Green}swapfile已发现，正在将其移除...${Font}"
+	sed -i '/swapfile/d' /etc/fstab
+	echo "3" > /proc/sys/vm/drop_caches
+	swapoff -a
+	rm -f /swapfile
+    echo -e "${Green}swap已删除！${Font}"
+else
+	echo -e "${Red}swapfile未发现，swap删除失败！${Font}"
+fi
+}
+
+#开始菜单
+main(){
+root_need
+ovz_no
+clear
+echo -e "———————————————————————————————————————"
+echo -e "${Green}Linux VPS一键添加/删除swap脚本${Font}"
+echo -e "${Green}1、添加swap${Font}"
+echo -e "${Green}2、删除swap${Font}"
+echo -e "———————————————————————————————————————"
+read -p "请输入数字 [1-2]:" num
+case "$num" in
+    1)
+    add_swap
+    ;;
+    2)
+    del_swap
+    ;;
+    *)
+    clear
+    echo -e "${Green}请输入正确数字 [1-2]${Font}"
+    sleep 2s
+    main
+    ;;
+    esac
+}
+main