start.sh

#!/bin/bash

# Artify 一键启动脚本
# 使用方法: chmod +x start.sh && ./start.sh

set -e

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

# 项目根目录
PROJECT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# 设置 PATH 确保能找到 npm/node (支持 homebrew 和常见安装位置)
export PATH="/opt/homebrew/bin:/usr/local/bin:$HOME/.nvm/versions/node/$(ls $HOME/.nvm/versions/node 2>/dev/null | tail -1)/bin:$PATH" 2>/dev/null || true

echo -e "${BLUE}"
echo "╔══════════════════════════════════════════════════════════╗"
echo "║              Artify 一键启动脚本                          ║"
echo "╚══════════════════════════════════════════════════════════╝"
echo -e "${NC}"

# 检查必要的命令是否存在
check_commands() {
    # 检查 Python
    if ! command -v python3 &> /dev/null; then
        echo -e "${RED}❌ 未找到 python3，请先安装 Python 3${NC}"
        exit 1
    fi
    
    # 检查 Node.js 和 npm
    if ! command -v npm &> /dev/null; then
        echo -e "${RED}❌ 未找到 npm，请先安装 Node.js${NC}"
        echo -e "   推荐使用 homebrew: brew install node"
        echo -e "   或者访问: https://nodejs.org/"
        exit 1
    fi
    
    if ! command -v node &> /dev/null; then
        echo -e "${RED}❌ 未找到 node，请先安装 Node.js${NC}"
        exit 1
    fi
}

# 检查 .env 文件
check_env() {
    if [ ! -f "$PROJECT_DIR/.env" ]; then
        echo -e "${YELLOW}⚠️  未检测到 .env 文件${NC}"
        echo -e "正在从 .env.example 创建..."
        
        if [ -f "$PROJECT_DIR/.env.example" ]; then
            cp "$PROJECT_DIR/.env.example" "$PROJECT_DIR/.env"
            echo -e "${RED}❌ 请编辑 .env 文件，填入您的 API Key 后重新运行${NC}"
            echo -e "   vim $PROJECT_DIR/.env"
            exit 1
        else
            echo -e "${RED}❌ .env.example 文件不存在${NC}"
            exit 1
        fi
    fi
    echo -e "${GREEN}✓ .env 配置文件已就绪${NC}"
}

# 检查 Python 虚拟环境
check_python_env() {
    if [ ! -d "$PROJECT_DIR/venv" ]; then
        echo -e "${YELLOW}📦 创建 Python 虚拟环境...${NC}"
        python3 -m venv "$PROJECT_DIR/venv"
    fi
    
    source "$PROJECT_DIR/venv/bin/activate"
    
    # 检查依赖
    if ! pip show uvicorn > /dev/null 2>&1; then
        echo -e "${YELLOW}📦 安装 Python 依赖...${NC}"
        pip install -r "$PROJECT_DIR/backend/requirements.txt" -q
    fi
    
    echo -e "${GREEN}✓ Python 环境已就绪${NC}"
}

# 检查 Node.js 环境
check_node_env() {
    cd "$PROJECT_DIR/frontend"
    
    # 检查 node_modules 是否存在且完整
    NEED_INSTALL=false
    
    if [ ! -d "node_modules" ]; then
        NEED_INSTALL=true
    elif [ ! -f "node_modules/.bin/vite" ]; then
        # vite 不存在，说明 node_modules 不完整
        echo -e "${YELLOW}⚠️  检测到 node_modules 不完整，重新安装...${NC}"
        rm -rf node_modules package-lock.json
        NEED_INSTALL=true
    fi
    
    if [ "$NEED_INSTALL" = true ]; then
        echo -e "${YELLOW}📦 安装前端依赖...${NC}"
        npm install
    fi
    
    # 验证 vite 是否可用
    if [ ! -f "node_modules/.bin/vite" ]; then
        echo -e "${RED}❌ 前端依赖安装失败，请手动运行: cd frontend && npm install${NC}"
        exit 1
    fi
    
    echo -e "${GREEN}✓ Node.js 环境已就绪${NC}"
    cd "$PROJECT_DIR"
}

# 停止已有服务
stop_services() {
    echo -e "${YELLOW}🔄 停止已有服务...${NC}"
    
    # 停止后端
    lsof -ti:8001 | xargs kill -9 2>/dev/null || true
    
    # 停止前端
    lsof -ti:3000 | xargs kill -9 2>/dev/null || true
    
    sleep 1
}

# 启动后端
start_backend() {
    echo -e "${BLUE}🚀 启动后端服务 (端口 8001)...${NC}"
    
    cd "$PROJECT_DIR"
    source venv/bin/activate
    
    nohup python -m uvicorn backend.app.main:app --host 0.0.0.0 --port 8001 > "$PROJECT_DIR/backend.log" 2>&1 &
    
    # 等待后端启动
    sleep 3
    
    if curl -s http://localhost:8001/health > /dev/null 2>&1; then
        echo -e "${GREEN}✓ 后端服务启动成功${NC}"
    else
        echo -e "${RED}❌ 后端服务启动失败，查看日志: tail -f $PROJECT_DIR/backend.log${NC}"
        exit 1
    fi
}

# 启动前端
start_frontend() {
    echo -e "${BLUE}🚀 启动前端服务 (端口 3000)...${NC}"
    
    cd "$PROJECT_DIR/frontend"
    
    # 使用 node 直接运行 vite，避免 PATH 问题
    nohup node node_modules/.bin/vite > "$PROJECT_DIR/frontend.log" 2>&1 &
    
    # 等待前端启动
    sleep 5
    
    if curl -s http://localhost:3000 > /dev/null 2>&1; then
        echo -e "${GREEN}✓ 前端服务启动成功${NC}"
    else
        echo -e "${YELLOW}⚠️  前端服务正在启动中...${NC}"
    fi
    
    cd "$PROJECT_DIR"
}

# 显示服务状态
show_status() {
    echo ""
    echo -e "${GREEN}╔══════════════════════════════════════════════════════════╗${NC}"
    echo -e "${GREEN}║                    🎉 启动成功！                         ║${NC}"
    echo -e "${GREEN}╠══════════════════════════════════════════════════════════╣${NC}"
    echo -e "${GREEN}║  前端地址: ${NC}${BLUE}http://localhost:3000${NC}${GREEN}                        ║${NC}"
    echo -e "${GREEN}║  后端地址: ${NC}${BLUE}http://localhost:8001${NC}${GREEN}                        ║${NC}"
    echo -e "${GREEN}║  API 文档: ${NC}${BLUE}http://localhost:8001/docs${NC}${GREEN}                   ║${NC}"
    echo -e "${GREEN}╠══════════════════════════════════════════════════════════╣${NC}"
    echo -e "${GREEN}║  日志查看:                                               ║${NC}"
    echo -e "${GREEN}║    后端: tail -f backend.log                             ║${NC}"
    echo -e "${GREEN}║    前端: tail -f frontend.log                            ║${NC}"
    echo -e "${GREEN}╠══════════════════════════════════════════════════════════╣${NC}"
    echo -e "${GREEN}║  停止服务: ./stop.sh                                     ║${NC}"
    echo -e "${GREEN}╚══════════════════════════════════════════════════════════╝${NC}"
    echo ""
}

# 主流程
main() {
    check_commands
    check_env
    check_python_env
    check_node_env
    stop_services
    start_backend
    start_frontend
    show_status
}

main