[优化] 部署脚本优化

scripts/uninstall-agent.sh

@@ -3,6 +3,7 @@ set -euo pipefail
 
 INSTALL_DIR="/opt/openflare-agent"
 SERVICE_NAME="openflare-agent"
+INSTALL_METHOD=""
 
 usage() {
   cat <<EOF
@@ -14,29 +15,102 @@ Usage:
 Options:
   --install-dir DIR         Installation directory (default: /opt/openflare-agent)
   --service-name NAME       systemd service name (default: openflare-agent)
+  --docker                  Uninstall Docker container agent
+  --method METHOD           Uninstall method: 'local' or 'docker' (default: local)
   -h, --help                Show this help message
 
 Behavior:
   1. Stop the agent service/process and remove the entire installation directory
   2. Remove the systemd service definition when present
-  3. Leave the local OpenResty installation untouched
+  3. Uninstall Docker container when --docker is specified
+  4. Leave the local OpenResty installation untouched
 
 Examples:
+  # Interactive uninstall (prompts for options)
   uninstall-agent.sh
-  uninstall-agent.sh --install-dir /srv/openflare-agent
+
+  # Automated local uninstall
+  uninstall-agent.sh --install-dir /opt/openflare-agent
+
+  # Automated Docker uninstall
+  uninstall-agent.sh --docker
 EOF
   exit 0
 }
 
+HAS_ARGS="false"
+if [[ $# -gt 0 ]]; then
+  HAS_ARGS="true"
+fi
+
 while [[ $# -gt 0 ]]; do
   case "$1" in
     --install-dir)  INSTALL_DIR="$2"; shift 2 ;;
     --service-name) SERVICE_NAME="$2"; shift 2 ;;
+    --docker)       INSTALL_METHOD="docker"; shift ;;
+    --method)       INSTALL_METHOD="$2"; shift 2 ;;
     -h|--help)      usage ;;
     *) echo "Unknown option: $1"; exit 1 ;;
   esac
 done
 
+if [[ "$HAS_ARGS" == "false" ]]; then
+  echo "=================================================="
+  echo "欢迎使用 OpenFlare Agent 卸载脚本"
+  echo "Welcome to the OpenFlare Agent Uninstaller"
+  echo "=================================================="
+  echo "请选择卸载方式 / Please choose uninstall method:"
+  echo "  1) Local  (本地卸载: 停止服务并删除二进制及配置)"
+  echo "  2) Docker (容器卸载: 停止并删除 Docker 容器)"
+  read -p "请输入序号 [1-2] (默认 1): " method_choice
+  method_choice=${method_choice:-1}
+  case "$method_choice" in
+    2) INSTALL_METHOD="docker" ;;
+    *) INSTALL_METHOD="local" ;;
+  esac
+else
+  if [[ -z "$INSTALL_METHOD" ]]; then
+    INSTALL_METHOD="local"
+  fi
+fi
+
+if [[ "$INSTALL_METHOD" == "docker" ]]; then
+  echo "正在卸载 OpenFlare Agent (Docker)..."
+
+  # Check if container exists
+  if command -v docker >/dev/null 2>&1; then
+    if docker ps -a --format '{{.Names}}' | grep -Eq "^openflare-agent$"; then
+      echo "停止并移除 openflare-agent 容器..."
+      docker stop openflare-agent || true
+      docker rm -f openflare-agent || true
+      echo "已成功移除 openflare-agent 容器。"
+    else
+      echo "未检测到 openflare-agent 容器。"
+    fi
+
+    # In interactive mode, ask if user wants to remove docker image
+    if [[ "$HAS_ARGS" == "false" ]]; then
+      read -p "是否删除 OpenFlare Agent 镜像 (ghcr.io/rain-kl/openflare-agent:latest)? (y/n) [n]: " remove_image_choice
+      remove_image_choice=${remove_image_choice:-n}
+      case "$remove_image_choice" in
+        [yY])
+          echo "正在删除 Docker 镜像..."
+          docker rmi ghcr.io/rain-kl/openflare-agent:latest || true
+          ;;
+        *)
+          echo "保留 Docker 镜像。"
+          ;;
+      esac
+    fi
+  else
+    echo "Error: 未检测到 Docker 运行环境，无法进行 Docker 方式的卸载。"
+    exit 1
+  fi
+
+  echo "OpenFlare Agent (Docker) 卸载完成。"
+  exit 0
+fi
+
 if [[ -z "$INSTALL_DIR" || "$INSTALL_DIR" == "/" || "$INSTALL_DIR" == "." ]]; then
   echo "Refusing to remove unsafe install directory: '${INSTALL_DIR}'"
   exit 1