modify scripts

docker/paperless/plugins/docker_patch.sh

@@ -1,73 +1,149 @@
 #!/bin/bash
 
-# 定义文件路径
-SOURCE_PARSE_FILENAME="/usr/src/paperless/git_scripts/parse_filename.py"
-DEST_PARSE_FILENAME="/usr/src/paperless/scripts/parse_filename.py"
+# 定义文件替换对（数组形式，格式：源文件 目标文件）
+# 可按需添加/删除行，每行一组 源文件 目标文件
+FILE_PAIRS=(
+    "/usr/src/paperless/scripts/parsers.py" "/usr/src/paperless/src/paperless_tesseract/parsers.py"
+    # 示例：新增更多文件对
+    #"/usr/src/paperless/git_scripts/parse_filename.py" "/usr/src/paperless/scripts/parse_filename.py"
+    # "/path/to/source/file3" "/path/to/dest/file3"
+    # "/path/to/source/file4" "/path/to/dest/file4"
+)
 
-SOURCE_PARSERS="/usr/src/paperless/git_scripts/parsers.py"
-DEST_PARSERS="/usr/src/paperless/src/paperless_tesseract/parsers.py"
-
-# 检查文件是否存在
+# 检查所有文件是否存在（仅检查replace/check操作需要的文件）
 check_files_exist() {
     local missing=0
-    for file in "$SOURCE_PARSE_FILENAME" "$DEST_PARSE_FILENAME" "$SOURCE_PARSERS" "$DEST_PARSERS"; do
-        if [ ! -f "$file" ]; then
-            echo "错误：文件不存在 - $file"
-            missing=1
+    local pair_count=${#FILE_PAIRS[@]}
+    
+    # 遍历文件对（步长2：源文件、目标文件为一组）
+    for ((i=0; i<pair_count; i+=2)); do
+        local source="${FILE_PAIRS[$i]}"
+        local dest="${FILE_PAIRS[$i+1]}"
+        
+        # 根据操作类型检查对应文件
+        if [ "$1" = "replace" ] || [ "$1" = "check" ]; then
+            if [ ! -f "$source" ]; then
+                echo "错误：源文件不存在 - $source"
+                missing=1
+            fi
+            if [ ! -f "$dest" ]; then
+                echo "错误：目标文件不存在 - $dest"
+                missing=1
+            fi
+        elif [ "$1" = "rollback" ]; then
+            if [ ! -f "$dest.bak" ]; then
+                echo "警告：备份文件不存在（未执行过替换？） - $dest.bak"
+                missing=1
+            fi
         fi
     done
-    if [ $missing -eq 1 ]; then
+    
+    if [ $missing -eq 1 ] && [ "$1" != "rollback" ]; then
+        echo "错误：关键文件缺失，无法继续执行"
         exit 1
     fi
 }
 
-# 显示文件差异
+# 显示所有文件对的差异
 show_diffs() {
-    echo "=== 检查 parse_filename.py 差异 ==="
-    diff -u "$DEST_PARSE_FILENAME" "$SOURCE_PARSE_FILENAME" || true  # 差异不存在时不报错
+    local pair_count=${#FILE_PAIRS[@]}
+    echo "=== 开始检查文件差异 ==="
     
-    echo -e "\n=== 检查 parsers.py 差异 ==="
-    diff -u "$DEST_PARSERS" "$SOURCE_PARSERS" || true
+    for ((i=0; i<pair_count; i+=2)); do
+        local source="${FILE_PAIRS[$i]}"
+        local dest="${FILE_PAIRS[$i+1]}"
+        
+        echo -e "\n--- 检查 $dest <-> $source 的差异 ---"
+        diff -u "$dest" "$source" || true  # 无差异时不报错
+    done
 }
 
-# 备份并替换文件
-replace_files() {
-    # 备份目标文件
-    backup_file "$DEST_PARSE_FILENAME"
-    backup_file "$DEST_PARSERS"
-
-    # 执行替换
-    cp -f "$SOURCE_PARSE_FILENAME" "$DEST_PARSE_FILENAME"
-    cp -f "$SOURCE_PARSERS" "$DEST_PARSERS"
-
-    echo -e "\n=== 替换完成，以下是替换后的差异（应无差异） ==="
-    show_diffs
-}
-
-# 备份文件（添加 .bak 后缀，保留原权限）
+# 备份单个文件（添加 .bak 后缀，保留原权限）
 backup_file() {
     local file="$1"
     local backup="$file.bak"
+    
     if [ -f "$backup" ]; then
-        # 若已有备份，先删除旧备份（避免累积过多）
+        echo "提示：旧备份文件已存在，将覆盖 - $backup"
         rm -f "$backup"
     fi
-    cp -a "$file" "$backup"  # -a 保留权限和属性
+    
+    cp -a "$file" "$backup"  # -a 保留权限、属性、时间戳等
     echo "已备份：$file -> $backup"
 }
 
-# 主逻辑
-check_files_exist
-
-if [ "$1" = "check" ]; then
-    echo "=== 执行差异检查（不修改文件） ==="
+# 替换所有文件对
+replace_files() {
+    local pair_count=${#FILE_PAIRS[@]}
+    echo "=== 开始替换文件（先备份目标文件） ==="
+    
+    for ((i=0; i<pair_count; i+=2)); do
+        local source="${FILE_PAIRS[$i]}"
+        local dest="${FILE_PAIRS[$i+1]}"
+        
+        echo -e "\n--- 处理文件对：$source -> $dest ---"
+        backup_file "$dest"
+        cp -f "$source" "$dest"
+        echo "已替换：$source 覆盖 $dest"
+    done
+    
+    echo -e "\n=== 替换完成，验证最终差异（应无差异） ==="
     show_diffs
-elif [ "$1" = "replace" ]; then
-    echo "=== 执行文件替换（先备份） ==="
-    replace_files
-else
-    echo "用法：$0 [check|replace]"
-    echo "  check   - 仅检查文件差异，不做修改"
-    echo "  replace - 备份目标文件并替换，然后显示最终差异"
-    exit 1
-fi
+}
+
+# 回滚替换操作（恢复 .bak 备份文件）
+rollback_files() {
+    local pair_count=${#FILE_PAIRS[@]}
+    echo "=== 开始回滚替换操作 ==="
+    
+    for ((i=0; i<pair_count; i+=2)); do
+        local dest="${FILE_PAIRS[$i+1]}"
+        local backup="$dest.bak"
+        
+        echo -e "\n--- 处理回滚：$backup -> $dest ---"
+        if [ -f "$backup" ]; then
+            # 先备份当前文件（防止回滚出错）
+            cp -a "$dest" "$dest.rollback_temp" 2>/dev/null || true
+            # 恢复备份文件
+            mv -f "$backup" "$dest"
+            echo "已回滚：$dest 恢复为备份版本"
+            # 删除临时文件
+            rm -f "$dest.rollback_temp" 2>/dev/null || true
+        else
+            echo "跳过：备份文件不存在 - $backup"
+        fi
+    done
+    
+    echo -e "\n=== 回滚操作执行完成 ==="
+}
+
+# 主逻辑
+main() {
+    case "$1" in
+        check)
+            echo "=== 执行文件差异检查（不修改文件） ==="
+            check_files_exist "check"
+            show_diffs
+            ;;
+        replace)
+            echo "=== 执行文件替换操作（自动备份） ==="
+            check_files_exist "replace"
+            replace_files
+            ;;
+        rollback)
+            echo "=== 执行文件回滚操作（恢复备份） ==="
+            check_files_exist "rollback"
+            rollback_files
+            ;;
+        *)
+            echo "用法：$0 [check|replace|rollback]"
+            echo "  check   - 仅检查所有文件对的差异，不做修改"
+            echo "  replace - 备份所有目标文件并执行替换，完成后验证差异"
+            echo "  rollback - 回滚替换操作（恢复 .bak 备份文件）"
+            exit 1
+            ;;
+    esac
+}
+
+# 启动主逻辑
+main "$1"