File size: 4,198 Bytes
55ba4a3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 | #!/usr/bin/env bash
set -euo pipefail
# Preserve all existing environment variables (from HF Space / os.environ)
# before sourcing env files, so HF Space always takes full priority.
declare -A _PRESERVED_ENV
while IFS='=' read -r _key _value; do
_PRESERVED_ENV["$_key"]="$_value"
done < <(env)
unset _key _value
# 恢复日志配置
RESTORE_LOG_DIR="${OPENCLAW_RESTORE_LOG_DIR:-/var/log/openclaw}"
RESTORE_LOG_FILE="${RESTORE_LOG_DIR}/restore.log"
# 确保日志目录存在
mkdir -p "$RESTORE_LOG_DIR"
# 日志函数
restore_log() {
local level="$1"
shift
local message="$*"
local timestamp
timestamp=$(date -u +"%Y-%m-%dT%H:%M:%SZ")
echo "[$timestamp] [$level] $message" | timeout 1 tee -a "$RESTORE_LOG_FILE" 2>/dev/null || echo "[$timestamp] [$level] $message"
}
restore_log_info() {
restore_log "INFO" "$@"
}
restore_log_warn() {
restore_log "WARN" "$@"
}
restore_log_error() {
restore_log "ERROR" "$@" >&2
}
restore_log_info "=== Starting Backup Restore ==="
restore_log_info "Restore started at: $(date -u +"%Y-%m-%d %H:%M:%S UTC")"
BACKUP_ENV_FILE_PATH="${OPENCLAW_BACKUP_ENV_FILE_PATH:-/root/.env.d/openclaw-backup.env}"
if [[ -f "$BACKUP_ENV_FILE_PATH" ]]; then
# shellcheck disable=SC1091
source "$BACKUP_ENV_FILE_PATH"
restore_log_info "Loaded backup environment from: $BACKUP_ENV_FILE_PATH"
else
restore_log_warn "Backup environment file not found: $BACKUP_ENV_FILE_PATH"
fi
# Restore all previously existing variables (HF Space takes priority over backup env files)
for _var in "${!_PRESERVED_ENV[@]}"; do
export "$_var"="${_PRESERVED_ENV[$_var]}"
done
unset _PRESERVED_ENV _var
# 记录恢复配置信息
restore_log_info "Restore Configuration:"
# 多数据集支持：优先使用 OPENCLAW_RESTORE_DATASET_REPO，否则使用 OPENCLAW_BACKUP_DATASET_REPO
RESTORE_DATASET_REPO="${OPENCLAW_RESTORE_DATASET_REPO:-${OPENCLAW_BACKUP_DATASET_REPO:-}}"
if [[ -z "${RESTORE_DATASET_REPO:-}" ]]; then
restore_log_warn "Restore skipped: no dataset configured"
restore_log_warn " OPENCLAW_RESTORE_DATASET_REPO: ${OPENCLAW_RESTORE_DATASET_REPO:-<not set>}"
restore_log_warn " OPENCLAW_BACKUP_DATASET_REPO: ${OPENCLAW_BACKUP_DATASET_REPO:-<not set>}"
exit 2
fi
restore_log_info " Restore Source Dataset: ${RESTORE_DATASET_REPO}"
restore_log_info " Backup Dataset: ${OPENCLAW_BACKUP_DATASET_REPO:-<not set>}"
restore_log_info " State Dir: ${OPENCLAW_STATE_DIR:-/root/.openclaw}"
restore_log_info " Backup Source Dir: ${OPENCLAW_BACKUP_SOURCE_DIR:-<not set>}"
restore_log_info " Path Prefix: ${OPENCLAW_BACKUP_PATH_PREFIX:-backups}"
# 检查数据集配置
if [[ -z "${OPENCLAW_BACKUP_DATASET_REPO:-}" ]]; then
restore_log_warn "Restore skipped: OPENCLAW_BACKUP_DATASET_REPO is not set"
restore_log_warn "Backup target dataset is required for restore"
exit 2
fi
# 注意：容器重启/重建时总是执行恢复，这是设计决定的
restore_log_info ""
restore_log_info "Starting restore process from dataset: ${RESTORE_DATASET_REPO}"
restore_log_info "----------------------------------------"
# Run restore with real-time logging to restore.log
# Using tee to write output to log file immediately as it's produced
restore_exit_code=0
python3 /opt/openclaw-hf/openclaw_hf/backup.py restore 2>&1 | tee -a "$RESTORE_LOG_FILE" || restore_exit_code=$?
if [[ $restore_exit_code -eq 0 ]]; then
restore_log_info "----------------------------------------"
restore_log_info "=== Backup Restore Completed Successfully ==="
restore_log_info "Restore source: ${RESTORE_DATASET_REPO}"
restore_log_info "Restore finished at: $(date -u +"%Y-%m-%d %H:%M:%S UTC")"
# 创建恢复成功标记，标记恢复完成后才允许备份
RESTORE_FLAG_FILE="/opt/restore-completed"
mkdir -p "$(dirname "$RESTORE_FLAG_FILE")"
date -u +"%Y-%m-%dT%H:%M:%SZ" > "$RESTORE_FLAG_FILE"
restore_log_info "Created restore completed flag: $RESTORE_FLAG_FILE"
exit 0
else
restore_log_error "----------------------------------------"
restore_log_error "=== Backup Restore Failed ==="
restore_log_error "Restore source: ${RESTORE_DATASET_REPO}"
restore_log_error "Restore failed at: $(date -u +"%Y-%m-%d %H:%M:%S UTC")"
exit 1
fi
|