Upload 4 files

.gitignore

@@ -0,0 +1,25 @@
+# 临时文件
+*.log
+*.tmp
+.DS_Store
+
+# 运行时数据
+/data/
+/config/
+/uploads/
+/cache/
+/backups/
+/thumbnails/
+
+# Node.js
+node_modules/
+npm-debug.log
+
+# 编辑器配置文件
+.idea/
+.vscode/
+*.swp
+*.swo
+
+# 本地环境变量
+.env 

Dockerfile

@@ -0,0 +1,67 @@
+FROM node:19.1.0-alpine3.16
+
+# 设置应用目录
+ARG APP_HOME=/home/node/app
+
+# 安装系统依赖
+RUN apk add gcompat tini git python3 py3-pip
+
+# 安装HuggingFace Hub
+RUN pip3 install --no-cache-dir huggingface_hub
+
+# 确保正确处理内核信号
+ENTRYPOINT [ "tini", "--" ]
+
+# 创建应用目录
+WORKDIR ${APP_HOME}
+
+# 设置NODE_ENV为production
+ENV NODE_ENV=production
+
+# 设置登录凭证环境变量
+ENV USERNAME="admin"
+ENV PASSWORD="password"
+
+# 克隆官方SillyTavern仓库（最新版本）
+RUN git clone https://github.com/SillyTavern/SillyTavern.git .
+
+# 安装依赖
+RUN echo "*** 安装npm包 ***" && \
+    npm install && npm cache clean --force
+
+# 添加启动脚本和数据同步脚本
+COPY launch.sh sync_data.sh ./
+RUN chmod +x launch.sh sync_data.sh
+
+# 安装生产依赖
+RUN echo "*** 安装生产npm包 ***" && \
+    npm i --no-audit --no-fund --loglevel=error --no-progress --omit=dev && npm cache clean --force
+
+# 创建配置目录
+RUN mkdir -p "config" || true && \
+    rm -f "config.yaml" || true && \
+    ln -s "./config/config.yaml" "config.yaml" || true
+
+# 清理不必要的文件
+RUN echo "*** 清理 ***" && \
+    mv "./docker/docker-entrypoint.sh" "./" && \
+    rm -rf "./docker" && \
+    echo "*** 使docker-entrypoint.sh可执行 ***" && \
+    chmod +x "./docker-entrypoint.sh" && \
+    echo "*** 转换行尾为Unix格式 ***" && \
+    dos2unix "./docker-entrypoint.sh" || true
+
+# 修改入口脚本，添加自定义启动脚本
+RUN sed -i 's/# Start the server/.\/launch.sh/g' docker-entrypoint.sh
+
+# 创建临时备份目录
+RUN mkdir -p /tmp/sillytavern_backup
+
+# 设置权限
+RUN chmod -R 777 ${APP_HOME}
+
+# 暴露端口
+EXPOSE 8000
+
+# 启动命令
+CMD [ "./docker-entrypoint.sh" ] 

launch.sh

@@ -0,0 +1,55 @@
+#!/bin/sh
+
+BASE=/home/node/app
+USERNAME=$(printenv USERNAME)
+PASSWORD=$(printenv PASSWORD)
+HF_TOKEN=$(printenv HF_TOKEN)
+DATASET_ID=$(printenv DATASET_ID)
+SYNC_INTERVAL=$(printenv SYNC_INTERVAL)
+
+# 如果没有设置用户名和密码，使用默认值
+if [[ -z "${USERNAME}" ]]; then
+  USERNAME="admin"
+fi
+
+if [[ -z "${PASSWORD}" ]]; then
+  PASSWORD="password"
+fi
+
+echo
+echo "用户名: ${USERNAME}"
+echo "密码: ${PASSWORD}"
+echo
+
+# 确保配置目录存在
+mkdir -p "${BASE}/config"
+
+# 如果配置文件不存在，从默认目录复制
+if [ ! -e "${BASE}/config/config.yaml" ]; then
+  echo "配置文件不存在，从默认目录复制: config.yaml"
+  cp -r "${BASE}/default/config.yaml" "${BASE}/config/config.yaml"
+fi
+
+# 修改配置文件中的用户名和密码
+sed -i "s/username: .*/username: \"${USERNAME}\"/" ${BASE}/config/config.yaml
+sed -i "s/password: .*/password: \"${PASSWORD}\"/" ${BASE}/config/config.yaml
+
+# 启用基本认证模式，禁用白名单模式
+sed -i "s/whitelistMode: true/whitelistMode: false/" ${BASE}/config/config.yaml
+sed -i "s/basicAuthMode: false/basicAuthMode: true/" ${BASE}/config/config.yaml
+
+# 显示配置文件内容以便验证
+echo "配置文件内容:"
+cat ${BASE}/config/config.yaml
+
+# 启动数据同步服务（如果提供了必要的环境变量）
+if [[ ! -z "${HF_TOKEN}" ]] && [[ ! -z "${DATASET_ID}" ]]; then
+  echo "启动数据同步服务..."
+  nohup ${BASE}/sync_data.sh > ${BASE}/sync_data.log 2>&1 &
+  echo "数据同步服务已在后台启动"
+else
+  echo "未提供HF_TOKEN或DATASET_ID，不启动数据同步服务"
+fi
+
+# 正常启动服务器
+exec node server.js --listen "$@" 

sync_data.sh

@@ -0,0 +1,140 @@
+#!/bin/bash
+
+# 检查环境变量
+if [[ -z "$HF_TOKEN" ]] || [[ -z "$DATASET_ID" ]]; then
+    echo "未启用备份功能 - 缺少HF_TOKEN或DATASET_ID环境变量"
+    exit 0
+fi
+
+# 创建临时目录
+TEMP_DIR="/tmp/sillytavern_backup"
+mkdir -p $TEMP_DIR
+
+# 安装python和huggingface_hub
+if ! command -v python3 &> /dev/null; then
+    echo "正在安装Python..."
+    apk add --no-cache python3 py3-pip
+fi
+
+if ! python3 -c "import huggingface_hub" &> /dev/null; then
+    echo "正在安装huggingface_hub..."
+    pip3 install --no-cache-dir huggingface_hub
+fi
+
+# 上传备份
+upload_backup() {
+    file_path="$1"
+    file_name="$2"
+    token="$HF_TOKEN"
+    repo_id="$DATASET_ID"
+
+    python3 -c "
+from huggingface_hub import HfApi
+import sys
+import os
+def manage_backups(api, repo_id, max_files=10):
+    files = api.list_repo_files(repo_id=repo_id, repo_type='dataset')
+    backup_files = [f for f in files if f.startswith('sillytavern_backup_') and f.endswith('.tar.gz')]
+    backup_files.sort()
+    
+    if len(backup_files) >= max_files:
+        files_to_delete = backup_files[:(len(backup_files) - max_files + 1)]
+        for file_to_delete in files_to_delete:
+            try:
+                api.delete_file(path_in_repo=file_to_delete, repo_id=repo_id, repo_type='dataset')
+                print(f'已删除旧备份: {file_to_delete}')
+            except Exception as e:
+                print(f'删除 {file_to_delete} 时出错: {str(e)}')
+api = HfApi(token='$token')
+try:
+    api.upload_file(
+        path_or_fileobj='$file_path',
+        path_in_repo='$file_name',
+        repo_id='$repo_id',
+        repo_type='dataset'
+    )
+    print(f'成功上传 $file_name')
+    
+    manage_backups(api, '$repo_id')
+except Exception as e:
+    print(f'上传文件时出错: {str(e)}')
+"
+}
+
+# 下载最新备份
+download_latest_backup() {
+    token="$HF_TOKEN"
+    repo_id="$DATASET_ID"
+    data_dir="/home/node/app/data"
+    
+    # 确保数据目录存在
+    mkdir -p $data_dir
+
+    python3 -c "
+from huggingface_hub import HfApi
+import sys
+import os
+import tarfile
+import tempfile
+api = HfApi(token='$token')
+try:
+    files = api.list_repo_files(repo_id='$repo_id', repo_type='dataset')
+    backup_files = [f for f in files if f.startswith('sillytavern_backup_') and f.endswith('.tar.gz')]
+    
+    if not backup_files:
+        print('未找到备份文件')
+        sys.exit()
+        
+    latest_backup = sorted(backup_files)[-1]
+    
+    with tempfile.TemporaryDirectory() as temp_dir:
+        filepath = api.hf_hub_download(
+            repo_id='$repo_id',
+            filename=latest_backup,
+            repo_type='dataset',
+            local_dir=temp_dir
+        )
+        
+        if filepath and os.path.exists(filepath):
+            with tarfile.open(filepath, 'r:gz') as tar:
+                tar.extractall('$data_dir')
+            print(f'成功从 {latest_backup} 恢复备份')
+        
+except Exception as e:
+    print(f'下载备份时出错: {str(e)}')
+"
+}
+
+# 首次启动时下载最新备份
+echo "正在从HuggingFace下载最新备份..."
+download_latest_backup
+
+# 同步函数
+sync_data() {
+    while true; do
+        echo "开始同步进程，时间: $(date)"
+        
+        data_dir="/home/node/app/data"
+        if [ -d "$data_dir" ]; then
+            timestamp=$(date +%Y%m%d_%H%M%S)
+            backup_file="sillytavern_backup_${timestamp}.tar.gz"
+            
+            # 压缩数据目录
+            tar -czf "${TEMP_DIR}/${backup_file}" -C "$data_dir" .
+            
+            echo "正在上传备份到HuggingFace..."
+            upload_backup "${TEMP_DIR}/${backup_file}" "${backup_file}"
+            
+            rm -f "${TEMP_DIR}/${backup_file}"
+        else
+            echo "数据目录尚不存在，等待下次同步..."
+        fi
+        
+        SYNC_INTERVAL=${SYNC_INTERVAL:-3600}
+        echo "下次同步将在 ${SYNC_INTERVAL} 秒后进行..."
+        sleep $SYNC_INTERVAL
+    done
+}
+
+# 启动同步进程
+sync_data