docker-deploy.sh

#!/usr/bin/env bash
# ============================================================
# AirGate Core - Docker Compose 部署准备脚本
#
# 这个脚本只负责"准备文件"：
#   - 在指定目录里创建 ./data 子目录
#   - 下载 docker-compose.yml
#   - 用 openssl 生成三个随机密钥并写入 .env
#
# 它故意不替你跑 docker compose up -d。检查完文件后用户自己执行：
#
#   docker compose up -d
#   docker compose logs -f core
#
# 优点：可审计 / 可在 dry-run 与正式启动之间做任意定制 / 与 CI 友好。
#
# 用法：
#   mkdir airgate && cd airgate
#   curl -sSL https://raw.githubusercontent.com/DevilGenius/airgate-core/master/deploy/docker-deploy.sh | bash
#
# 或带参数：
#   curl -sSL .../docker-deploy.sh -o docker-deploy.sh
#   bash docker-deploy.sh
#
# 环境变量覆盖（可选）：
#   AIRGATE_DIR     安装目录            默认: 当前目录
#   AIRGATE_PORT    HTTP 端口           默认: 9517
#   AIRGATE_BRANCH  从 GitHub 拉哪个分支 默认: master
#   AIRGATE_TAG     固定镜像版本         默认: latest
#   NON_INTERACTIVE 跳过所有交互（CI 用） 默认: 0
#
# 想要纯二进制 + systemd（不带 Docker）？请改用 deploy/install.sh。
# ============================================================

set -euo pipefail

# ---- 颜色输出 ----
#
# 使用 bash ANSI-C quoting `$'...'` 让变量直接保存真正的 ESC 字符（0x1B），
# 这样 printf / heredoc 都能正确显示。之前用 '\033[...]' 字面量在 heredoc
# 里会被原样输出乱码（printf '%b' 能解释，但 cat <<DONE 不能）。
if [[ -t 1 ]]; then
  C_RESET=$'\033[0m'; C_BOLD=$'\033[1m'; C_DIM=$'\033[2m'
  C_RED=$'\033[31m'; C_GREEN=$'\033[32m'; C_YELLOW=$'\033[33m'; C_BLUE=$'\033[34m'; C_CYAN=$'\033[36m'
else
  C_RESET=''; C_BOLD=''; C_DIM=''; C_RED=''; C_GREEN=''; C_YELLOW=''; C_BLUE=''; C_CYAN=''
fi

info()    { printf '%s\n' "${C_CYAN}==>${C_RESET} $*"; }
ok()      { printf '%s\n' "${C_GREEN}✓${C_RESET} $*"; }
warn()    { printf '%s\n' "${C_YELLOW}!${C_RESET} $*"; }
err()     { printf '%s\n' "${C_RED}✗${C_RESET} $*" >&2; }
section() { printf '\n%s\n' "${C_BOLD}${C_BLUE}── $* ──${C_RESET}"; }

# ---- 横幅 ----
cat <<'BANNER'

    ╔═══════════════════════════════════════════╗
    ║       AirGate Core - Docker Deploy        ║
    ║     prepares files but does NOT start     ║
    ╚═══════════════════════════════════════════╝

BANNER

# ---- 默认参数 ----
AIRGATE_DIR="${AIRGATE_DIR:-.}"
AIRGATE_PORT="${AIRGATE_PORT:-9517}"
AIRGATE_BRANCH="${AIRGATE_BRANCH:-master}"
AIRGATE_TAG="${AIRGATE_TAG:-latest}"
NON_INTERACTIVE="${NON_INTERACTIVE:-0}"

REPO_RAW_URL="https://raw.githubusercontent.com/DevilGenius/airgate-core/${AIRGATE_BRANCH}/deploy"

# ---- 依赖检查 ----
section "环境检查"

require_cmd() {
  if ! command -v "$1" >/dev/null 2>&1; then
    err "未找到命令: $1${2:+，请先安装 $2}"
    exit 1
  fi
  ok "$1"
}

require_cmd docker "Docker"

# 兼容 docker-compose 老版本（v1）和 docker compose 新插件（v2）
if docker compose version >/dev/null 2>&1; then
  COMPOSE="docker compose"
  ok "docker compose (v2)"
elif command -v docker-compose >/dev/null 2>&1; then
  COMPOSE="docker-compose"
  ok "docker-compose (v1, 兼容模式)"
else
  err "未找到 docker compose / docker-compose，请安装 Docker Compose"
  exit 1
fi

require_cmd curl
require_cmd openssl

# ---- 交互式参数 ----
prompt() {
  local var=$1 msg=$2 default=$3 reply
  if [[ "$NON_INTERACTIVE" == "1" ]]; then
    eval "$var=\"\$default\""
    return
  fi
  printf '%b' "${C_DIM}?${C_RESET} ${msg} ${C_DIM}[${default}]${C_RESET}: " > /dev/tty
  read -r reply < /dev/tty || reply=""
  eval "$var=\"\${reply:-\$default}\""
}

section "安装参数"
prompt AIRGATE_DIR  "安装目录"           "$AIRGATE_DIR"
prompt AIRGATE_PORT "HTTP 监听端口"       "$AIRGATE_PORT"
prompt AIRGATE_TAG  "镜像版本 (latest 或 v0.x.y)" "$AIRGATE_TAG"

# 把 ~ 展开
AIRGATE_DIR="${AIRGATE_DIR/#\~/$HOME}"

# ---- 准备目录 ----
section "准备目录 ${AIRGATE_DIR}"

if [[ -e "$AIRGATE_DIR" ]] && [[ -n "$(ls -A "$AIRGATE_DIR" 2>/dev/null || true)" ]]; then
  warn "目录 ${AIRGATE_DIR} 已存在且非空"
  if [[ -f "$AIRGATE_DIR/docker-compose.yml" ]]; then
    if [[ "$NON_INTERACTIVE" == "1" ]]; then
      err "已存在 docker-compose.yml，请先备份或删除目录后重试（NON_INTERACTIVE 模式不会覆盖）"
      exit 1
    fi
    printf '%b' "${C_YELLOW}?${C_RESET} 检测到已有部署，是否覆盖 docker-compose.yml？.env 与 ./data 不会被动 [y/N]: " > /dev/tty
    read -r ans < /dev/tty || ans=""
    if [[ "$ans" != "y" && "$ans" != "Y" ]]; then
      info "已取消安装。如需重新部署，请清空 ${AIRGATE_DIR} 后重试。"
      exit 0
    fi
  fi
fi

mkdir -p "$AIRGATE_DIR"
cd "$AIRGATE_DIR"
mkdir -p data/postgres data/redis data/plugins data/uploads data/assets

ok "目录就绪：$(pwd)"

# ---- 下载 docker-compose.yml ----
section "下载 docker-compose.yml"
curl -fsSL "${REPO_RAW_URL}/docker-compose.yml" -o docker-compose.yml
ok "已下载到 $(pwd)/docker-compose.yml"

# ---- 生成 / 复用 .env ----
section "生成 .env"

if [[ -f .env ]]; then
  warn ".env 已存在，沿用现有密钥（不会覆盖）"
else
  DB_PASSWORD="$(openssl rand -hex 24)"
  REDIS_PASSWORD="$(openssl rand -hex 24)"
  JWT_SECRET="$(openssl rand -hex 32)"

  cat > .env <<EOF
# AirGate Core 部署配置（由 install.sh 自动生成于 $(date '+%Y-%m-%d %H:%M:%S')）
# 任何修改需要 docker compose up -d 重启生效
# 数据库 / Redis 密码相关说明见 README.md

AIRGATE_IMAGE=ghcr.io/devilgenius/airgate-core
AIRGATE_IMAGE_TAG=${AIRGATE_TAG}

PORT=${AIRGATE_PORT}
BIND_HOST=0.0.0.0
TZ=Asia/Shanghai

DB_PASSWORD=${DB_PASSWORD}
REDIS_PASSWORD=${REDIS_PASSWORD}
JWT_SECRET=${JWT_SECRET}

API_KEY_SECRET=
PLUGINS_MARKETPLACE_GITHUB_TOKEN=
EOF
  chmod 600 .env
  ok "已生成 .env（所有密钥随机，权限 600）"
fi

if grep -Eq '^AIRGATE_IMAGE=ghcr\.io/DevilGenius/airgate-core[[:space:]]*$' .env; then
  sed -i -E 's#^AIRGATE_IMAGE=ghcr\.io/DevilGenius/airgate-core[[:space:]]*$#AIRGATE_IMAGE=ghcr.io/devilgenius/airgate-core#' .env
  ok "已修正 .env 中的 AIRGATE_IMAGE 为小写 GHCR 路径"
fi

# ---- 完成（注意：本脚本不自动 up -d） ----
cat <<DONE

${C_GREEN}${C_BOLD}✓ 文件已就绪${C_RESET}

  部署目录 : ${C_CYAN}$(pwd)${C_RESET}
  数据目录 : ${C_CYAN}$(pwd)/data${C_RESET}
  环境变量 : ${C_CYAN}$(pwd)/.env${C_RESET}  ${C_DIM}(权限 600，含随机密钥)${C_RESET}

${C_BOLD}下一步：手动启动容器${C_RESET}
  ${C_CYAN}$COMPOSE up -d${C_RESET}
  ${C_CYAN}$COMPOSE logs -f core${C_RESET}

启动后访问 ${C_CYAN}http://<your-host>:${AIRGATE_PORT}${C_RESET}：
  - 安装向导会检测到 .env 已配置好 DB / Redis，自动跳过这两步
  - 只需要建立管理员账号
  - 进入管理后台 → 插件管理 → 插件市场 → 按需安装插件

${C_BOLD}常用命令${C_RESET}
  $COMPOSE logs -f core              # 查看日志
  $COMPOSE restart                   # 重启
  $COMPOSE down                      # 停止
  $COMPOSE pull && $COMPOSE up -d    # 升级镜像

${C_DIM}文档：https://github.com/DevilGenius/airgate-core${C_RESET}

DONE