Upload start.sh

start.sh

@@ -0,0 +1,78 @@
+#!/bin/bash
+
+# 设置端口 (HF 必须监听 7860)
+export PORT=7860
+DB_PATH="/data/one-api.db"
+
+echo "=== 启动环境检查 ==="
+
+# 1. 检查并下载数据库
+if [ -n "$DATASET_REPO" ] && [ -n "$HF_TOKEN" ]; then
+    echo "正在从 Dataset: $DATASET_REPO 下载数据库..."
+    python3 -c "
+import os
+from huggingface_hub import hf_hub_download
+try:
+    hf_hub_download(
+        repo_id=os.environ['DATASET_REPO'],
+        filename='one-api.db',
+        local_dir='/data',
+        repo_type='dataset',
+        token=os.environ['HF_TOKEN']
+    )
+    print('数据库下载成功')
+except Exception as e:
+    print(f'提示：暂无历史数据库备份，将初始化新数据库。')
+"
+else
+    echo "警告：未设置 DATASET_REPO 或 HF_TOKEN，数据将无法持久化！"
+fi
+
+# 2. 自动查找可执行文件
+# 尝试查找名为 new-api 或 one-api 的程序
+TARGET_BIN=$(which new-api || which one-api || find / -name "new-api" -executable -type f 2>/dev/null | head -n 1 || find / -name "one-api" -executable -type f 2>/dev/null | head -n 1)
+
+if [ -z "$TARGET_BIN" ]; then
+    echo "错误：找不到可执行程序 (new-api 或 one-api)！"
+    exit 1
+fi
+
+echo "找到程序路径: $TARGET_BIN"
+
+# 3. 定义上传备份函数
+upload_db() {
+    if [ -f "$DB_PATH" ]; then
+        echo "正在备份数据库到 HF Dataset..."
+        python3 -c "
+import os
+from huggingface_hub import HfApi
+api = HfApi()
+try:
+    api.upload_file(
+        path_or_fileobj='$DB_PATH',
+        path_in_repo='one-api.db',
+        repo_id=os.environ['DATASET_REPO'],
+        repo_type='dataset',
+        token=os.environ['HF_TOKEN']
+    )
+    print('备份成功')
+except Exception as e:
+    print(f'备份失败: {e}')
+"
+    fi
+}
+
+# 4. 后台定时同步 (每 15 分钟)
+(
+    while true; do
+        sleep 900
+        upload_db
+    done
+) &
+
+# 5. 捕捉退出信号
+trap upload_db EXIT
+
+# 6. 运行主程序
+echo "=== 正在启动 New-API ==="
+exec $TARGET_BIN