Update scripts/openclaw-entrypoint.sh

scripts/openclaw-entrypoint.sh

@@ -1,13 +1,129 @@
+
+
 #!/usr/bin/env bash
 set -euo pipefail
 
-# 设置环境变量
+# OpenClaw HF Spaces 启动脚本 - 包含数据恢复和备份
+# 确保聊天数据不丢失
+
+OPENCLAW_USER="${OPENCLAW_USER:-root}"
 OPENCLAW_STATE_DIR="${OPENCLAW_STATE_DIR:-/root/.openclaw}"
+OPENCLAW_WORKSPACE_DIR="${OPENCLAW_WORKSPACE_DIR:-${OPENCLAW_STATE_DIR}/workspace}"
 OPENCLAW_GATEWAY_PORT="${OPENCLAW_GATEWAY_PORT:-7860}"
 
+# 备份配置
+BACKUP_ENABLED="${OPENCLAW_BACKUP_ENABLED:-true}"
+HF_TOKEN="${HF_TOKEN:-${HUGGINGFACE_HUB_TOKEN:-}}"
+BACKUP_DATASET_REPO="${OPENCLAW_BACKUP_DATASET_REPO:-}"
+
+# 创建日志目录
+mkdir -p /var/log/openclaw
+
 # 创建必要的目录结构
-mkdir -p "${OPENCLAW_STATE_DIR}"/{workspace,agents/main/agent,agents/main/sessions,logs}
+mkdir -p "${OPENCLAW_STATE_DIR}"/{workspace,agents/main/agent,agents/main/sessions,logs,identity}
+
+log_info() {
+    echo "[$(date -u +%Y-%m-%dT%H:%M:%SZ)] [ENTRYPOINT] $1" | tee -a /var/log/openclaw/gateway.stdout.log
+}
+
+# 恢复数据（如果存在备份）
+restore_if_exists() {
+    if [[ -z "${HF_TOKEN}" ]] || [[ -z "${BACKUP_DATASET_REPO}" ]]; then
+        log_info "未配置备份，跳过恢复步骤"
+        return 0
+    fi
+    
+    log_info "检查备份数据..."
+    
+    # 创建临时目录用于下载
+    RESTORE_TEMP=$(mktemp -d)
+    
+    # 尝试下载最新备份
+    if huggingface-cli download "${BACKUP_DATASET_REPO}" \
+        --repo-type dataset \
+        --token "${HF_TOKEN}" \
+        --local-dir "${RESTORE_TEMP}" \
+        2>/dev/null; then
+        
+        log_info "找到备份，开始恢复数据..."
+        
+        # 恢复工作区数据
+        if [[ -d "${RESTORE_TEMP}/workspace" ]]; then
+            cp -r "${RESTORE_TEMP}/workspace/"* "${OPENCLAW_WORKSPACE_DIR}/" 2>/dev/null || true
+        fi
+        
+        # 恢复代理数据
+        if [[ -d "${RESTORE_TEMP}/agents" ]]; then
+            cp -r "${RESTORE_TEMP}/agents/"* "${OPENCLAW_STATE_DIR}/agents/" 2>/dev/null || true
+        fi
+        
+        # 恢复身份数据
+        if [[ -d "${RESTORE_TEMP}/identity" ]]; then
+            cp -r "${RESTORE_TEMP}/identity/"* "${OPENCLAW_STATE_DIR}/identity/" 2>/dev/null || true
+        fi
+        
+        log_info "数据恢复完成"
+    else
+        log_info "未找到备份或下载失败，跳过恢复"
+    fi
+    
+    rm -rf "${RESTORE_TEMP}"
+}
+
+# 备份数据（启动时备份一次）
+backup_on_start() {
+    if [[ "${BACKUP_ENABLED}" != "true" ]]; then
+        return 0
+    fi
+    
+    if [[ -z "${HF_TOKEN}" ]] || [[ -z "${BACKUP_DATASET_REPO}" ]]; then
+        log_info "未配置备份，跳过启动备份"
+        return 0
+    fi
+    
+    log_info "执行启动备份..."
+    
+    # 创建临时备份目录
+    BACKUP_TEMP=$(mktemp -d)
+    
+    # 复制数据到临时目录
+    cp -r "${OPENCLAW_WORKSPACE_DIR}" "${BACKUP_TEMP}/" 2>/dev/null || true
+    cp -r "${OPENCLAW_STATE_DIR}/agents" "${BACKUP_TEMP}/" 2>/dev/null || true
+    cp -r "${OPENCLAW_STATE_DIR}/identity" "${BACKUP_TEMP}/" 2>/dev/null || true
+    
+    # 创建备份信息
+    echo "{\"backup_time\": \"$(date -u +%Y-%m-%dT%H:%M:%SZ)\", \"trigger\": \"startup\"}" > "${BACKUP_TEMP}/backup_info.json"
+    
+    # 上传到Hugging Face
+    if huggingface-cli upload "${BACKUP_DATASET_REPO}" \
+        "${BACKUP_TEMP}/" \
+        --repo-type dataset \
+        --token "${HF_TOKEN}" \
+        --commit-message "Auto-backup $(date -u +%Y-%m-%dT%H:%M:%SZ)" \
+        2>/dev/null; then
+        log_info "启动备份完成"
+    else
+        log_info "启动备份失败（继续启动）"
+    fi
+    
+    rm -rf "${BACKUP_TEMP}"
+}
 
-# 启动OpenClaw网关
-exec openclaw gateway --bind 0.0.0.0 --port "${OPENCLAW_GATEWAY_PORT}"
+# 主启动流程
+main() {
+    log_info "===== OpenClaw 启动 ====="
+    log_info "状态目录: ${OPENCLAW_STATE_DIR}"
+    log_info "工作区目录: ${OPENCLAW_WORKSPACE_DIR}"
+    
+    # 1. 尝试恢复数据
+    restore_if_exists
+    
+    # 2. 启动时备份
+    backup_on_start
+    
+    # 3. 启动 OpenClaw
+    log_info "启动 OpenClaw 网关..."
+    exec openclaw gateway --bind 0.0.0.0 --port "${OPENCLAW_GATEWAY_PORT}"
+}
 
+main "$@"