#!/bin/bash
# Windows 用户请使用 Git Bash 运行此脚本

echo "================================"
echo "开始获取 grok2api 源代码..."
echo "================================"
echo ""

# 检查当前目录
if [ ! -f "pyproject.toml" ]; then
    echo "错误: 请在 grok2api 配置目录中运行此脚本"
    exit 1
fi

# 检查 app 目录是否已存在
if [ -d "app" ]; then
    echo "警告: app 目录已存在"
    read -p "是否覆盖? (y/N) " -n 1 -r
    echo
    if [[ ! $REPLY =~ ^[Yy]$ ]]; then
        echo "已取消"
        exit 0
    fi
    rm -rf app
fi

echo "正在克隆原项目..."
git clone --depth 1 https://github.com/chenyme/grok2api.git temp_grok2api

if [ $? -ne 0 ]; then
    echo "错误: 克隆失败，请检查网络连接"
    exit 1
fi

echo "正在复制源代码..."
cp -r temp_grok2api/app ./

echo "正在清理临时文件..."
rm -rf temp_grok2api

if [ -d "app" ]; then
    echo ""
    echo "================================"
    echo "✅ 源代码获取成功！"
    echo "================================"
    echo ""
    echo "接下来的步骤："
    echo "1. 创建 Hugging Face Space (Docker SDK)"
    echo "2. 推送代码到 HF Space"
    echo "3. 配置环境变量"
    echo "4. 启用 Persistent Storage"
    echo "5. 设置保活项目"
    echo ""
    echo "详细说明请查看 DEPLOY_GUIDE.md"
else
    echo "错误: 源代码复制失败"
    exit 1
fi