#!/bin/bash
# ============================================================
# autoClaude-TGbot 一键部署脚本
# 用法: chmod +x deploy.sh && sudo ./deploy.sh
# ============================================================

set -e

# --- 颜色 ---
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
CYAN='\033[0;36m'
NC='\033[0m'

info()  { echo -e "${CYAN}[INFO]${NC} $1"; }
ok()    { echo -e "${GREEN}[✔]${NC} $1"; }
warn()  { echo -e "${YELLOW}[!]${NC} $1"; }
err()   { echo -e "${RED}[✘]${NC} $1"; exit 1; }

# --- 检查 root ---
if [ "$EUID" -ne 0 ]; then
    err "请使用 sudo 运行: sudo ./deploy.sh"
fi

# --- 变量 ---
APP_NAME="autoclaude-tgbot"
APP_DIR="$(cd "$(dirname "$0")" && pwd)"
SERVICE_FILE="/etc/systemd/system/${APP_NAME}.service"
RUN_USER="${SUDO_USER:-$(whoami)}"
RUN_GROUP="$(id -gn "$RUN_USER")"

echo ""
echo -e "${CYAN}╔══════════════════════════════════════════╗${NC}"
echo -e "${CYAN}║   autoClaude-TGbot 一键部署              ║${NC}"
echo -e "${CYAN}╚══════════════════════════════════════════╝${NC}"
echo ""
info "项目目录: ${APP_DIR}"
info "运行用户: ${RUN_USER}"
echo ""

# ============================================================
# 1. 安装系统依赖
# ============================================================
info "检查系统依赖..."

# 安装 uv（如果不存在）
if ! command -v uv &> /dev/null; then
    info "安装 uv..."
    curl -LsSf https://astral.sh/uv/install.sh | sh
    export PATH="$HOME/.local/bin:$PATH"
    ok "uv 已安装"
else
    ok "uv 已存在 ($(uv --version))"
fi

# ============================================================
# 2. 安装 Python 依赖
# ============================================================
info "安装 Python 依赖..."
cd "$APP_DIR"
sudo -u "$RUN_USER" uv sync 2>/dev/null || sudo -u "$RUN_USER" uv pip install -r pyproject.toml 2>/dev/null || true
ok "依赖安装完成"

# ============================================================
# 3. 检查配置文件
# ============================================================
if [ ! -f "${APP_DIR}/config.toml" ]; then
    warn "config.toml 不存在，从模板复制..."
    cp "${APP_DIR}/config.toml.example" "${APP_DIR}/config.toml"
    chown "$RUN_USER:$RUN_GROUP" "${APP_DIR}/config.toml"
    echo ""
    echo -e "${YELLOW}════════════════════════════════════════════${NC}"
    echo -e "${YELLOW}  ⚠️  请编辑 config.toml 填入实际配置：${NC}"
    echo -e "${YELLOW}  nano ${APP_DIR}/config.toml${NC}"
    echo -e "${YELLOW}════════════════════════════════════════════${NC}"
    echo ""
    read -p "编辑完成后按 Enter 继续，或 Ctrl+C 退出..." _
else
    ok "config.toml 已存在"
fi

# ============================================================
# 4. 获取 uv 和 python 路径
# ============================================================
UV_PATH="$(sudo -u "$RUN_USER" bash -c 'which uv')"
info "uv 路径: ${UV_PATH}"

# ============================================================
# 5. 创建 systemd 服务
# ============================================================
info "创建 systemd 服务: ${APP_NAME}"

cat > "$SERVICE_FILE" <<EOF
[Unit]
Description=autoClaude Telegram Bot
After=network.target
Wants=network-online.target

[Service]
Type=simple
User=${RUN_USER}
Group=${RUN_GROUP}
WorkingDirectory=${APP_DIR}
ExecStart=${UV_PATH} run python bot.py
Restart=on-failure
RestartSec=10
StartLimitIntervalSec=60
StartLimitBurst=3

# 环境
Environment=PYTHONUNBUFFERED=1

# 日志
StandardOutput=journal
StandardError=journal
SyslogIdentifier=${APP_NAME}

# 安全加固
NoNewPrivileges=true
ProtectSystem=strict
ProtectHome=read-only
ReadWritePaths=${APP_DIR}
PrivateTmp=true

[Install]
WantedBy=multi-user.target
EOF

ok "服务文件已创建: ${SERVICE_FILE}"

# ============================================================
# 6. 启用并启动服务
# ============================================================
info "重载 systemd 配置..."
systemctl daemon-reload

info "启用开机自启..."
systemctl enable "$APP_NAME"

info "启动服务..."
systemctl restart "$APP_NAME"

# 等一会检查状态
sleep 2

if systemctl is-active --quiet "$APP_NAME"; then
    ok "服务已启动！"
else
    warn "服务启动可能失败，请检查日志"
fi

# ============================================================
# 7. 完成
# ============================================================
echo ""
echo -e "${GREEN}╔══════════════════════════════════════════╗${NC}"
echo -e "${GREEN}║   ✅ 部署完成！                          ║${NC}"
echo -e "${GREEN}╚══════════════════════════════════════════╝${NC}"
echo ""
echo -e "  ${CYAN}常用命令：${NC}"
echo ""
echo -e "  查看状态   ${GREEN}systemctl status ${APP_NAME}${NC}"
echo -e "  查看日志   ${GREEN}journalctl -u ${APP_NAME} -f${NC}"
echo -e "  重启服务   ${GREEN}systemctl restart ${APP_NAME}${NC}"
echo -e "  停止服务   ${GREEN}systemctl stop ${APP_NAME}${NC}"
echo -e "  编辑配置   ${GREEN}nano ${APP_DIR}/config.toml${NC}"
echo ""
echo -e "  ${YELLOW}修改配置后记得重启: systemctl restart ${APP_NAME}${NC}"
echo ""