抱脸部署酒馆

#!/usr/bin/env bash
set -euo pipefail
set -a

APP_BASE_DIR="/home/node/app"
USERNAME="${USERNAME:-admin}"
PASSWORD="${PASSWORD:-password}"
HF_TOKEN="${HF_TOKEN:-}"
DATASET_ID="${DATASET_ID:-}"
KEEP_BACKUPS="${KEEP_BACKUPS:-10}"
SYNC_INTERVAL="${SYNC_INTERVAL:-3600}"
DIRS_TO_BACKUP=( "data" "config" "plugins" "public/scripts/extensions/third-party" )

log() {
  echo "[$(date +'%F %T')] $*"
}

init_backup() {
  if [[ -n "${DATASET_ID:-}" ]]; then
    log "📁 使用提供的 DATASET_ID=${DATASET_ID}"
    return 0
  fi
  if [[ -z "${HF_TOKEN:-}" ]]; then
    log "⚠️ 未设置 HF_TOKEN，跳过同步。"
    return 1
  fi

  USER_ID=$(python3 <<'PY'
import os, sys
from huggingface_hub import HfApi
try:
    print(HfApi(token=os.getenv("HF_TOKEN")).whoami().get("name"))
except Exception:
    sys.exit(1)
PY
  )
  if [[ -z "$USER_ID" ]]; then
    log "❌ 无法从 HF_TOKEN 获取用户名，跳过同步。"
    return 1
  fi

  DATASET_ID="${USER_ID}/data"
  log "✅ 默认 DATASET_ID 已设置为 ${DATASET_ID}"
}

prep_repo() {
  log "🔍 准备 HuggingFace 仓库..."
  python3 <<'PY'
import os
from huggingface_hub import HfApi
from huggingface_hub.utils import HfHubHTTPError

api = HfApi(token=os.getenv("HF_TOKEN"))
repo_id = os.getenv("DATASET_ID")
try:
    api.repo_info(repo_id=repo_id, repo_type="dataset")
except HfHubHTTPError as e:
    if e.response.status_code == 404:
        print(f"仓库 '{repo_id}' 未找到。将创建为私有仓库。")
        api.create_repo(repo_id=repo_id, repo_type="dataset", private=True)
    else:
        raise
PY
  log "✅ 仓库已准备就绪。"
}

restore_latest() {
  log "🔄 检查要恢复的备份..."
  python3 <<'PY'
import os, sys, tarfile, tempfile
from huggingface_hub import HfApi

api = HfApi(token=os.getenv("HF_TOKEN"))
repo_id = os.getenv("DATASET_ID")

files = api.list_repo_files(repo_id=repo_id, repo_type="dataset")
backups = sorted([f for f in files if f.endswith(".tar.gz")])

if not backups:
    print("未找到远程备份。跳过恢复。")
    sys.exit(0)

latest_backup = backups[-1]
print(f"找到最新备份: {latest_backup}。正在恢复...")

with tempfile.TemporaryDirectory() as temp_dir:
    path = api.hf_hub_download(
        repo_id=repo_id,
        filename=latest_backup,
        repo_type="dataset",
        local_dir=temp_dir
    )
    with tarfile.open(path, "r:gz") as t:
        t.extractall(os.getenv("APP_BASE_DIR"))
PY
  log "✅ 恢复检查完成。"
}

do_backup() {
  local valid_dirs=()
  for dir in "${DIRS_TO_BACKUP[@]}"; do
    if [[ -d "${APP_BASE_DIR}/${dir}" ]]; then
      valid_dirs+=("$dir")
    fi
  done

  if [[ ${#valid_dirs[@]} -eq 0 ]]; then
    log "⚠️ 没有有效的目录可供备份，跳过此周期。"
    return
  fi

  log "⬆️ 开始备份过程..."
  local ts; ts=$(date +%Y%m%d_%H%M%S)
  local fname="sillytavern_backup_${ts}.tar.gz"
  local tmp_file; tmp_file=$(mktemp)

  tar -czf "$tmp_file" -C "$APP_BASE_DIR" "${valid_dirs[@]}"
  
  python3 <<PY
import os
from huggingface_hub import HfApi
api = HfApi(token=os.getenv("HF_TOKEN"))
repo_id = os.getenv("DATASET_ID")
print(f"正在上传 {os.path.basename("$fname")}...")
api.upload_file(
    path_or_fileobj="$tmp_file",
    path_in_repo="$fname",
    repo_id=repo_id,
    repo_type="dataset"
)
print("正在清理旧备份...")
files = api.list_repo_files(repo_id=repo_id, repo_type="dataset")
backups = sorted([f for f in files if f.endswith(".tar.gz")])
keep = int(os.getenv("KEEP_BACKUPS", "10"))
if len(backups) > keep:
    for old_backup in backups[:-keep]:
        print(f"正在删除旧备份: {old_backup}")
        api.delete_file(path_in_repo=old_backup, repo_id=repo_id, repo_type="dataset")
    api.super_squash_history(repo_id=repo_id, repo_type="dataset")
PY
  rm -f "$tmp_file"
  log "✅ 备份周期完成。"
}

sync_loop() {
  while true; do
    do_backup
    log "⏳ 下次同步将在 ${SYNC_INTERVAL} 秒后进行。"
    sleep "${SYNC_INTERVAL}"
  done
}

main() {
  cd "$APP_BASE_DIR" || { log "❌ 致命错误: 无法访问 APP_BASE_DIR 位于 $APP_BASE_DIR"; exit 1; }
  
  trap 'kill 0' SIGTERM SIGINT

  log "🚀 sillytavern 启动器正在启动..."

  if [[ -f "config/config.yaml" ]]; then
    sed -i -e "s/username: .*/username: \"${USERNAME}\"/" \
           -e "s/password: .*/password: \"${PASSWORD}\"/" "config/config.yaml"
  fi

  if init_backup; then
    log "🔗 已为仓库启用同步服务: $DATASET_ID"
    prep_repo
    restore_latest
    sync_loop &
  fi

  log "🟢 正在启动 sillytavern 服务器..."
  exec node server.js --listen "$@"
}

main "$@"