deploy.sh

#!/bin/bash

# 股票代码API部署脚本
# 适用于另一台电脑的快速部署

set -e

echo "🚀 股票代码API部署脚本"
echo "================================"

# 检查Docker是否安装
if ! command -v docker &> /dev/null; then
    echo "❌ Docker未安装，请先安装Docker"
    echo "   访问: https://docs.docker.com/get-docker/"
    exit 1
fi

# 检查Docker Compose是否安装
if ! command -v docker-compose &> /dev/null; then
    echo "❌ Docker Compose未安装，请先安装Docker Compose"
    echo "   访问: https://docs.docker.com/compose/install/"
    exit 1
fi

# 检查.env文件
if [ ! -f ".env" ]; then
    echo "📝 创建环境变量配置文件..."
    cat > .env << EOF
# 股票代码API环境变量配置
# 请填入你的API密钥

# Tushare Pro API密钥 (必需)
# 获取地址: https://tushare.pro/
TUSHARE_TOKEN=your_tushare_token_here

# DeepSeek API密钥 (必需)
# 获取地址: https://platform.deepseek.com/
DEEPSEEK_API_KEY=your_deepseek_api_key_here

# DeepSeek API配置 (可选)
DEEPSEEK_BASE_URL=https://api.deepseek.com/v1
DEEPSEEK_MODEL=deepseek-chat

# 日志级别 (可选)
LOG_LEVEL=INFO

# 最大推荐股票数量 (可选)
MAX_RECOMMENDED_STOCKS=20
EOF
    
    echo "✅ 已创建.env文件"
    echo "⚠️  请编辑.env文件，填入你的API密钥"
    echo "   然后重新运行此脚本"
    exit 1
fi

# 检查API密钥是否已配置
if grep -q "your_tushare_token_here" .env || grep -q "your_deepseek_api_key_here" .env; then
    echo "❌ 请先在.env文件中配置API密钥"
    echo "   1. 编辑.env文件"
    echo "   2. 填入TUSHARE_TOKEN和DEEPSEEK_API_KEY"
    echo "   3. 重新运行此脚本"
    exit 1
fi

echo "✅ 环境配置检查通过"

# 创建日志目录
mkdir -p logs

# 构建并启动服务
echo "🔨 构建Docker镜像..."
docker-compose -f docker-compose-simple.yml build

echo "🚀 启动API服务..."
docker-compose -f docker-compose-simple.yml up -d

# 等待服务启动
echo "⏳ 等待服务启动..."
sleep 10

# 检查服务状态
echo "🔍 检查服务状态..."
if curl -f http://localhost:8000/health > /dev/null 2>&1; then
    echo "✅ API服务启动成功!"
    echo ""
    echo "📊 API服务信息:"
    echo "   - 服务地址: http://localhost:8000"
    echo "   - API文档: http://localhost:8000/docs"
    echo "   - 健康检查: http://localhost:8000/health"
    echo ""
    echo "📋 可用接口:"
    echo "   - GET /stock-codes/recommended    - 获取推荐股票代码"
    echo "   - GET /stock-codes/{industry}     - 获取指定行业股票代码"
    echo "   - GET /stock-codes/all           - 获取所有股票代码"
    echo "   - GET /industries                - 获取行业列表"
    echo ""
    echo "💡 使用示例:"
    echo "   curl http://localhost:8000/stock-codes/recommended"
    echo "   curl http://localhost:8000/stock-codes/银行"
    echo ""
    echo "🛠️  管理命令:"
    echo "   - 查看日志: docker-compose -f docker-compose-simple.yml logs -f"
    echo "   - 停止服务: docker-compose -f docker-compose-simple.yml down"
    echo "   - 重启服务: docker-compose -f docker-compose-simple.yml restart"
else
    echo "❌ 服务启动失败，请检查日志:"
    docker-compose -f docker-compose-simple.yml logs
    exit 1
fi