Update tools/hf-backup.sh

tools/hf-backup.sh

@@ -1,21 +1,90 @@
-#!/usr/bin/env sh
+#!/bin/sh
+
 set -e
 
-if [ -z "${HF_TOKEN:-}" ] || [ -z "${DATASET_ID:-}" ]; then
-  echo "[backup] HF_TOKEN/DATASET_ID 未设置，跳过备份功能"
-  exit 0
-fi
+TS(){ date "+%Y-%m-%d %H:%M:%S"; }
+log(){ echo "[$(TS)] [hf-backup] $*"; }
+
+HF_TOKEN="${HF_TOKEN:-}"
+DATASET_ID="${DATASET_ID:-}"
 
 SYNC_INTERVAL="${SYNC_INTERVAL:-3600}"
-export HF_TOKEN DATASET_ID MAX_BACKUPS BACKUP_PATHS BACKUP_PREFIX
+MAX_BACKUPS="${MAX_BACKUPS:-10}"
+BACKUP_PREFIX="${BACKUP_PREFIX:-crs_backup}"
+BACKUP_PATHS="${BACKUP_PATHS:-/app/data}"
+BACKUP_EXCLUDE="${BACKUP_EXCLUDE:-/app/data/init.json,/app/data/users.json,*.db}"
+
+HF_PY="/app/tools/hf_backup.py"
+TMP_DIR="/tmp/crs_backup"
+mkdir -p "${TMP_DIR}"
+
+need_hf() {
+  if [ -z "$HF_TOKEN" ] || [ -z "$DATASET_ID" ]; then
+    log "HF_TOKEN/DATASET_ID 缺失，无法使用备份功能"; exit 1
+  fi
+}
+
+mk_archive() {
+  ts="$(date +%Y%m%d_%H%M%S)"
+  archive="${TMP_DIR}/${BACKUP_PREFIX}_${ts}.tar.gz"
+
+  EX=#
+  IFS=','; for x in $BACKUP_EXCLUDE; do EX="$EX --exclude=$x"; done; unset IFS
+
+  IFS=','; set -- $BACKUP_PATHS; unset IFS
+
+  log "创建归档：$archive"
+  # 使用 -C / 以绝对路径打包，解压时可直接 -C / 还原
+  # shellcheck disable=SC2086
+  tar -czf "$archive" $EX -C / "$@"
+  echo "$archive"
+}
+
+cmd_upload_once() {
+  need_hf
+  archive="$(mk_archive)"
+  log "上传：$archive"
+  python3 "$HF_PY" upload \
+    --token "$HF_TOKEN" \
+    --repo  "$DATASET_ID" \
+    --file  "$archive" \
+    --prefix "$BACKUP_PREFIX" \
+    --max   "$MAX_BACKUPS"
+  rm -f "$archive" || true
+  log "上传完成"
+}
 
-python3 /app/tools/hf_backup.py restore || true
+cmd_restore() {
+  need_hf
+  out="$(python3 "$HF_PY" restore --token "$HF_TOKEN" --repo "$DATASET_ID" --prefix "$BACKUP_PREFIX" 2>/dev/null || true)"
+  if [ -z "$out" ]; then
+    log "未找到可恢复的备份，跳过。"
+    exit 0
+  fi
+  # python 仅输出本地文件路径
+  latest_file="$out"
+  if [ ! -f "$latest_file" ]; then
+    log "本地文件不存在：$latest_file"; exit 1
+  fi
+  log "解压：$latest_file -> /"
+  tar -xzf "$latest_file" -C /
+  log "恢复完成"
+}
 
-trap 'echo "[backup] 收到停止信号，执行最后一次备份"; python3 /app/tools/hf_backup.py backup || true; exit 0' TERM INT
+cmd_daemon() {
+  need_hf
+  while :; do
+    log "开始周期备份（间隔 ${SYNC_INTERVAL}s，前缀 ${BACKUP_PREFIX}，保留 ${MAX_BACKUPS}）"
+    if ! cmd_upload_once; then
+      log "本轮备份失败（忽略，等待下次循环）"
+    fi
+    sleep "$SYNC_INTERVAL"
+  done
+}
 
-while true; do
-  echo "[backup] 执行备份..."
-  python3 /app/tools/hf_backup.py backup || true
-  echo "[backup] 休眠 ${SYNC_INTERVAL}s"
-  sleep "${SYNC_INTERVAL}"
-done
+case "${1:-}" in
+  restore) cmd_restore ;;
+  once)    cmd_upload_once ;;
+  daemon)  cmd_daemon ;;
+  *) echo "用法: $0 {restore|once|daemon}"; exit 1 ;;
+esac