#!/bin/bash

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

# 获取脚本所在目录
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# 显示帮助信息
show_help() {
    echo "Gost转发面板 Docker Compose 部署脚本"
    echo ""
    echo "用法: $0 [命令] [选项]"
    echo ""
    echo "命令:"
    echo "  start     启动所有服务"
    echo "  stop      停止所有服务"
    echo "  restart   重启所有服务"
    echo "  build     重新构建前端镜像"
    echo "  logs      查看服务日志"
    echo "  status    查看服务状态"
    echo "  clean     清理未使用的容器和镜像"
    echo ""
    echo "选项:"
    echo "  -h, --help    显示此帮助信息"
    echo ""
    echo "环境变量配置:"
    echo "  SERVER_HOST       服务器地址 (默认: localhost:6365)"
    echo "  VUE_APP_API_URL   前端API地址 (默认: http://localhost:6365/api/v1)"
    echo "  VUE_APP_WS_URL    WebSocket地址 (默认: ws://localhost:6365)"
    echo ""
    echo "示例:"
    echo "  # 启动服务"
    echo "  $0 start"
    echo ""
    echo "  # 使用服务器IP启动"
    echo "  SERVER_HOST=192.168.1.100:6365 $0 start"
    echo ""
    echo "  # 自定义完整地址启动"
    echo "  VUE_APP_API_URL=http://192.168.1.100:6365/api/v1 VUE_APP_WS_URL=ws://192.168.1.100:6365 $0 start"
}

# 检测可用的Docker Compose命令
get_docker_compose_cmd() {
    if command -v docker-compose &> /dev/null; then
        echo "docker-compose"
    elif docker compose version &> /dev/null; then
        echo "docker compose"
    else
        echo ""
    fi
}

# 检查docker和docker-compose是否安装
check_dependencies() {
    if ! command -v docker &> /dev/null; then
        echo -e "${RED}错误: Docker 未安装${NC}"
        echo "请先安装Docker: https://docs.docker.com/get-docker/"
        exit 1
    fi
    
    DOCKER_COMPOSE_CMD=$(get_docker_compose_cmd)
    if [ -z "$DOCKER_COMPOSE_CMD" ]; then
        echo -e "${RED}错误: Docker Compose 未安装${NC}"
        echo "请先安装Docker Compose: https://docs.docker.com/compose/install/"
        echo "或者更新Docker到最新版本（包含Compose插件）"
        exit 1
    fi
    
    echo -e "${GREEN}检测到Docker Compose命令: $DOCKER_COMPOSE_CMD${NC}"
}

# 从docker-compose.yml读取端口配置
get_service_port() {
    local service=$1
    local port_line=$(grep -A 10 "^  $service:" docker-compose.yml | grep -E "^\s+ports:" -A 1 | tail -1)
    if [[ $port_line =~ \"([0-9]+):[0-9]+\" ]]; then
        echo "${BASH_REMATCH[1]}"
    elif [[ $port_line =~ ([0-9]+):[0-9]+ ]]; then
        echo $(echo $port_line | cut -d':' -f1 | tr -d ' "')
    fi
}

# 从docker-compose.yml读取数据库配置
get_db_config() {
    local config_key=$1
    local config_value=$(grep -A 20 "environment:" docker-compose.yml | grep "$config_key:" | head -1 | cut -d':' -f2 | cut -d'#' -f1 | xargs)
    echo "$config_value"
}

# 启动服务
start_services() {
    echo -e "${GREEN}启动Gost转发面板服务...${NC}"
    
    # 如果设置了 SERVER_HOST，自动生成 API 和 WS 地址
    if [ -n "$SERVER_HOST" ]; then
        export VUE_APP_API_URL="http://${SERVER_HOST}/api/v1"
        export VUE_APP_WS_URL="ws://${SERVER_HOST}"
    else
        # 设置默认环境变量
        export VUE_APP_API_URL=${VUE_APP_API_URL:-"http://localhost:6365/api/v1"}
        export VUE_APP_WS_URL=${VUE_APP_WS_URL:-"ws://localhost:6365"}
    fi
    
    echo -e "${YELLOW}环境变量配置:${NC}"
    if [ -n "$SERVER_HOST" ]; then
        echo "SERVER_HOST: $SERVER_HOST"
    fi
    echo "VUE_APP_API_URL: $VUE_APP_API_URL"
    echo "VUE_APP_WS_URL: $VUE_APP_WS_URL"
    echo ""
    
    # 切换到脚本目录
    cd "$SCRIPT_DIR"
    
    # 生成前端配置文件
    echo -e "${YELLOW}生成前端配置文件...${NC}"
    cat > dist/config.js << EOF
window.APP_CONFIG = {
  VUE_APP_API_URL: "$VUE_APP_API_URL",
  VUE_APP_WS_URL: "$VUE_APP_WS_URL"
}
EOF
    echo -e "${GREEN}配置文件已生成: dist/config.js${NC}"
    
    $DOCKER_COMPOSE_CMD up -d
    
    if [ $? -eq 0 ]; then
        echo -e "${GREEN}服务启动成功!${NC}"
        echo ""
        
        # 从配置文件读取端口信息
        FRONTEND_PORT=$(get_service_port "frontend")
        BACKEND_PORT=$(get_service_port "backend") 
        MYSQL_PORT=$(get_service_port "mysql")
        REDIS_PORT=$(get_service_port "redis")
        
        # 从配置文件读取数据库配置
        DB_USER=$(get_db_config "MYSQL_USER")
        DB_PASSWORD=$(get_db_config "MYSQL_PASSWORD")
        
        echo -e "${YELLOW}访问地址:${NC}"
        echo "  前端界面: http://localhost:${FRONTEND_PORT:-80}"
        echo "  后端API: http://localhost:${BACKEND_PORT:-6365}"
        echo "  数据库: localhost:${MYSQL_PORT:-3306} (用户名: ${DB_USER:-gost}, 密码: ${DB_PASSWORD:-pwd})"
        echo "  Redis: localhost:${REDIS_PORT:-6379}"
        echo ""
        echo -e "${YELLOW}默认管理员账号:${NC}"
        echo "  用户名: admin_user"
        echo "  密码: admin_user"
        echo ""
        echo -e "${GREEN}部署完成！${NC}"
    else
        echo -e "${RED}服务启动失败!${NC}"
        exit 1
    fi
}

# 停止服务
stop_services() {
    echo -e "${YELLOW}停止所有服务...${NC}"
    cd "$SCRIPT_DIR"
    $DOCKER_COMPOSE_CMD down
    echo -e "${GREEN}服务已停止${NC}"
}

# 重启服务
restart_services() {
    echo -e "${YELLOW}重启所有服务...${NC}"
    cd "$SCRIPT_DIR"
    $DOCKER_COMPOSE_CMD restart
    echo -e "${GREEN}服务重启完成${NC}"
}

# 重新构建前端
build_frontend() {
    echo -e "${YELLOW}重新构建前端镜像...${NC}"
    cd "$SCRIPT_DIR"
    $DOCKER_COMPOSE_CMD build frontend --no-cache
    echo -e "${GREEN}前端镜像构建完成${NC}"
}

# 查看日志
show_logs() {
    cd "$SCRIPT_DIR"
    $DOCKER_COMPOSE_CMD logs -f
}

# 查看状态
show_status() {
    cd "$SCRIPT_DIR"
    $DOCKER_COMPOSE_CMD ps
}

# 清理未使用的容器和镜像
clean_docker() {
    echo -e "${YELLOW}清理未使用的Docker资源...${NC}"
    docker system prune -f
    echo -e "${GREEN}清理完成${NC}"
}

# 完全清理（包括数据卷）
clean_all() {
    echo -e "${RED}警告: 这将删除所有数据，包括数据库！${NC}"
    read -p "确定要继续吗？(y/N): " -n 1 -r
    echo
    if [[ $REPLY =~ ^[Yy]$ ]]; then
        cd "$SCRIPT_DIR"
        $DOCKER_COMPOSE_CMD down -v
        docker system prune -f
        echo -e "${GREEN}完全清理完成${NC}"
    else
        echo "操作已取消"
    fi
}

# 主程序
main() {
    check_dependencies
    
    # 设置全局的Docker Compose命令变量
    DOCKER_COMPOSE_CMD=$(get_docker_compose_cmd)
    
    case "${1:-start}" in
        start)
            start_services
            ;;
        stop)
            stop_services
            ;;
        restart)
            restart_services
            ;;
        build)
            build_frontend
            ;;
        logs)
            show_logs
            ;;
        status)
            show_status
            ;;
        clean)
            clean_docker
            ;;
        clean-all)
            clean_all
            ;;
        -h|--help)
            show_help
            ;;
        *)
            echo -e "${RED}未知命令: $1${NC}"
            show_help
            exit 1
            ;;
    esac
}

main "$@"