#!/bin/bash
# KVM Manager 启动脚本

set -e

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
BACKEND_DIR="$SCRIPT_DIR/backend"
FRONTEND_DIR="$SCRIPT_DIR/frontend"
LOG_DIR="/tmp/kvm-manager-logs"

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

usage() {
    echo "用法: $0 [选项]"
    echo ""
    echo "选项:"
    echo "  -b, --backend     只启动后端服务"
    echo "  -f, --frontend     只启动前端服务"
    echo "  -a, --all          启动所有服务 (默认)"
    echo "  -s, --stop         停止所有服务"
    echo "  -r, --restart      重启所有服务"
    echo "  -h, --help         显示帮助"
    echo ""
    echo "示例:"
    echo "  $0                 # 启动所有服务"
    echo "  $0 -b             # 只启动后端"
    echo "  $0 -s             # 停止所有服务"
}

# 创建日志目录
mkdir -p "$LOG_DIR"

# 检查端口是否被占用
check_port() {
    local port=$1
    if lsof -i:$port &>/dev/null; then
        return 1  # 端口被占用
    fi
    return 0  # 端口空闲
}

# 停止服务
stop_services() {
    echo -e "${YELLOW}停止 KVM Manager 服务...${NC}"
    
    # 停止后端
    if [ -f "$LOG_DIR/backend.pid" ]; then
        local pid=$(cat "$LOG_DIR/backend.pid")
        if ps -p "$pid" &>/dev/null; then
            kill "$pid" 2>/dev/null || true
            echo "  后端服务已停止 (PID: $pid)"
        fi
        rm -f "$LOG_DIR/backend.pid"
    fi
    
    # 停止前端
    if [ -f "$LOG_DIR/frontend.pid" ]; then
        local pid=$(cat "$LOG_DIR/frontend.pid")
        if ps -p "$pid" &>/dev/null; then
            kill "$pid" 2>/dev/null || true
            echo "  前端服务已停止 (PID: $pid)"
        fi
        rm -f "$LOG_DIR/frontend.pid"
    fi
    
    # 强制停止残留进程
    pkill -f "uvicorn.*app.main:app" 2>/dev/null || true
    pkill -f "vite" 2>/dev/null || true
    
    echo -e "${GREEN}所有服务已停止${NC}"
}

# 启动后端
start_backend() {
    echo -e "${YELLOW}启动后端服务...${NC}"
    
    if ! check_port 8004; then
        echo -e "${RED}错误: 端口 8004 已被占用${NC}"
        return 1
    fi
    
    cd "$BACKEND_DIR"
    
    # 检查虚拟环境
    if [ ! -d "venv" ]; then
        echo -e "${RED}错误: 未找到虚拟环境，请先运行: cd backend && python -m venv venv${NC}"
        return 1
    fi
    
    source "$BACKEND_DIR/venv/bin/activate"
    
    nohup uvicorn app.main:app --host 0.0.0.0 --port 8004 > "$LOG_DIR/backend.log" 2>&1 &
    local pid=$!
    echo $pid > "$LOG_DIR/backend.pid"
    
    sleep 2
    
    if ps -p "$pid" &>/dev/null; then
        echo -e "${GREEN}✓ 后端服务已启动 (PID: $pid, 端口: 8004)${NC}"
        echo "  日志: $LOG_DIR/backend.log"
    else
        echo -e "${RED}✗ 后端服务启动失败${NC}"
        tail -20 "$LOG_DIR/backend.log"
        return 1
    fi
}

# 启动前端
start_frontend() {
    echo -e "${YELLOW}启动前端服务...${NC}"
    
    if ! check_port 8005; then
        echo -e "${RED}错误: 端口 8005 已被占用${NC}"
        return 1
    fi
    
    cd "$FRONTEND_DIR"
    
    # 检查 node_modules
    if [ ! -d "node_modules" ]; then
        echo -e "${YELLOW}首次运行，需要安装依赖...${NC}"
        npm install
    fi
    
    nohup npm run dev -- --host 0.0.0.0 --port 8005 > "$LOG_DIR/frontend.log" 2>&1 &
    local pid=$!
    echo $pid > "$LOG_DIR/frontend.pid"
    
    sleep 5
    
    if ps -p "$pid" &>/dev/null; then
        echo -e "${GREEN}✓ 前端服务已启动 (PID: $pid, 端口: 8005)${NC}"
        echo "  日志: $LOG_DIR/frontend.log"
    else
        echo -e "${RED}✗ 前端服务启动失败${NC}"
        tail -20 "$LOG_DIR/frontend.log"
        return 1
    fi
}

# 查看状态
show_status() {
    echo ""
    echo "========== KVM Manager 服务状态 =========="
    echo ""
    
    # 后端状态
    if [ -f "$LOG_DIR/backend.pid" ]; then
        local pid=$(cat "$LOG_DIR/backend.pid")
        if ps -p "$pid" &>/dev/null; then
            echo -e "${GREEN}✓ 后端服务${NC} - 运行中 (PID: $pid, 端口: 8004)"
        else
            echo -e "${RED}✗ 后端服务${NC} - 未运行 (PID 文件过期)"
        fi
    else
        echo -e "${YELLOW}○ 后端服务${NC} - 未启动"
    fi
    
    # 前端状态
    if [ -f "$LOG_DIR/frontend.pid" ]; then
        local pid=$(cat "$LOG_DIR/frontend.pid")
        if ps -p "$pid" &>/dev/null; then
            echo -e "${GREEN}✓ 前端服务${NC} - 运行中 (PID: $pid, 端口: 8005)"
        else
            echo -e "${RED}✗ 前端服务${NC} - 未运行 (PID 文件过期)"
        fi
    else
        echo -e "${YELLOW}○ 前端服务${NC} - 未启动"
    fi
    
    echo ""
    echo "=========================================="
    echo ""
    echo "访问地址:"
    echo "  前端: http://localhost:8005"
    echo "  API:  http://localhost:8004"
    echo "  文档: http://localhost:8004/docs"
    echo ""
}

# 主程序
main() {
    case "${1:-all}" in
        -b|--backend)
            start_backend
            ;;
        -f|--frontend)
            start_frontend
            ;;
        -a|--all)
            start_backend
            start_frontend
            show_status
            ;;
        -s|--stop)
            stop_services
            ;;
        -r|--restart)
            stop_services
            sleep 1
            start_backend
            start_frontend
            show_status
            ;;
        -h|--help)
            usage
            ;;
        status)
            show_status
            ;;
        *)
            usage
            exit 1
            ;;
    esac
}

main "$@"