#!/bin/bash
# ============================
#  FTP Server 启动/停止脚本 (Linux/macOS)
# ============================

APP_NAME="ftp-server"
APP_DIR="$(cd "$(dirname "$0")" && pwd)"
APP_BIN="$APP_DIR/ftp-server"
CONFIG="$APP_DIR/config.json"
LOG_FILE="$APP_DIR/ftp-server.log"
PID_FILE="$APP_DIR/ftp-server.pid"

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

info()  { echo -e "${GREEN}[信息]${NC} $1"; }
warn()  { echo -e "${YELLOW}[提示]${NC} $1"; }
error() { echo -e "${RED}[错误]${NC} $1"; }

# 获取 PID
get_pid() {
    if [ -f "$PID_FILE" ]; then
        local pid=$(cat "$PID_FILE" 2>/dev/null)
        if [ -n "$pid" ] && kill -0 "$pid" 2>/dev/null; then
            echo "$pid"
            return 0
        fi
        # PID 文件过期，清理
        rm -f "$PID_FILE"
    fi
    # 尝试通过进程名查找
    pgrep -f "$APP_BIN" 2>/dev/null | head -1
}

# ---------- 启动 ----------
do_start() {
    if [ ! -f "$APP_BIN" ]; then
        error "未找到 $APP_BIN"
        echo "请先编译: go build -o ftp-server ./cmd/"
        exit 1
    fi

    local pid=$(get_pid)
    if [ -n "$pid" ]; then
        warn "$APP_NAME 已经在运行中 (PID: $pid)"
        exit 0
    fi

    info "正在启动 $APP_NAME ..."

    nohup "$APP_BIN" -config "$CONFIG" >> "$LOG_FILE" 2>&1 &
    local new_pid=$!
    echo "$new_pid" > "$PID_FILE"

    sleep 2

    if kill -0 "$new_pid" 2>/dev/null; then
        info "$APP_NAME 已启动 (PID: $new_pid)"
        echo "       Web 管理面板: http://localhost:8080"
        echo "       FTP 端口: 2121"
        echo "       日志文件: $LOG_FILE"
    else
        error "$APP_NAME 启动失败，请查看日志: $LOG_FILE"
        rm -f "$PID_FILE"
        exit 1
    fi
}

# ---------- 停止 ----------
do_stop() {
    local pid=$(get_pid)
    if [ -z "$pid" ]; then
        warn "$APP_NAME 未在运行"
        exit 0
    fi

    info "正在停止 $APP_NAME (PID: $pid) ..."
    kill "$pid" 2>/dev/null

    # 等待进程退出
    local count=0
    while kill -0 "$pid" 2>/dev/null && [ $count -lt 10 ]; do
        sleep 1
        count=$((count + 1))
    done

    # 如果还没退出，强制杀掉
    if kill -0 "$pid" 2>/dev/null; then
        warn "正常停止超时，强制终止..."
        kill -9 "$pid" 2>/dev/null
    fi

    rm -f "$PID_FILE"
    info "$APP_NAME 已停止"
}

# ---------- 状态 ----------
do_status() {
    local pid=$(get_pid)
    if [ -n "$pid" ]; then
        info "$APP_NAME 正在运行 (PID: $pid)"
    else
        warn "$APP_NAME 未运行"
    fi
}

# ---------- 帮助 ----------
usage() {
    echo ""
    echo "  用法: $0 {start|stop|restart|status}"
    echo ""
    echo "  命令:"
    echo "    start    启动 FTP Server"
    echo "    stop     停止 FTP Server"
    echo "    restart  重启 FTP Server"
    echo "    status   查看运行状态"
    echo ""
}

# ---------- 主入口 ----------
case "$1" in
    start)   do_start  ;;
    stop)    do_stop   ;;
    restart) do_stop; sleep 2; do_start ;;
    status)  do_status ;;
    *)       usage; exit 1 ;;
esac