fix: 完善自我修正函数

Author: yefansky

src/cvbManager.ts

@@ -262,38 +262,63 @@ export class TCVB {
   public static autoFixTCVBContent(original: string): string {
     let content = original;
 
-    // Remove leading spaces before ## directives
+    // 移除##指令前的空格
     content = content.replace(/^(\s+)##/gm, '##');
-  
-    // Close unclosed code blocks, inspired by the file-ending reference regex
-    content = content.replace(/(```[^\n]*\n[\s\S]*?)(?=^##|\n## END_TCVB|(?![\s\S]))/gm, '$1\n```');
-  
-    // Ensure END_TCVB is present
+
+    // 规则4/10：修复代码块包裹问题（精确处理需要代码块的操作）
+    content = content.replace(
+      /(## (?:OLD_CONTENT|NEW_CONTENT|OPERATION:CREATE)\n)([\s\S]*?)(?=\n## |\n## END_TCVB|$)/gis,
+      (match, directive, codeSection) => {
+          // 检测代码块是否被正确包裹
+          const trimmed = codeSection.trim();
+          const hasStart = trimmed.startsWith('```');
+          const hasEnd = trimmed.endsWith('```');
+
+          // 情况1：完全未包裹的代码块
+          if (!hasStart && !hasEnd) {
+              return `${directive}\n\`\`\`\n${codeSection.trim()}\n\`\`\``;
+          }
+
+          // 情况2：只有开始标记
+          if (hasStart && !hasEnd) {
+              return `${directive}\n${codeSection}\n\`\`\``;
+          }
+
+          // 情况3：只有结束标记（异常情况）
+          if (!hasStart && hasEnd) {
+              return `${directive}\n\`\`\`\n${codeSection}`;
+          }
+
+          return match;
+      }
+    );
+
+    // 确保存在END_TCVB标记
     if (!/^##\s*END_TCVB\s*$/m.test(content)) {
-      content += '\n## END_TCVB';
+        content += '\n## END_TCVB';
     }
-  
-    // Fix GLOBAL-REPLACE operations
+
+    // 严格标准化END_TCVB标记（处理中间/前后空格）
+    content = content.replace(/^##\s*END_TCVB\s*$/gm, '## END_TCVB');
+
+    // 修复GLOBAL-REPLACE操作
     content = content.replace(
-      /## OPERATION:GLOBAL-REPLACE\n([\s\S]*?)(?=\n## OPERATION:|\n## FILE:|\n## END_TCVB|$)/g,
-      (match) => {
-        const hasOldContent = /## OLD_CONTENT/.test(match);
-        const hasNewContent = /## NEW_CONTENT/.test(match);
+        /## OPERATION:GLOBAL-REPLACE\n([\s\S]*?)(?=\n## OPERATION:|\n## FILE:|\n## END_TCVB|$)/g,
+        (match) => {
+            const hasOldContent = /## OLD_CONTENT/.test(match);
+            const hasNewContent = /## NEW_CONTENT/.test(match);
 
-        if (!hasOldContent && hasNewContent) {
-          // Convert to CREATE if only NEW_CONTENT is present
-          const newContentMatch = match.match(/## NEW_CONTENT\n```([\s\S]*?)```/);
-          const newContentCode = newContentMatch ? newContentMatch[1] : '';
-          return `## OPERATION:CREATE\n\`\`\`\n${newContentCode}\n\`\`\``;
-        } else if (hasOldContent && !hasNewContent) {
-          // Add empty NEW_CONTENT if only OLD_CONTENT is present
-          return match + '\n## NEW_CONTENT\n```\n```';
+            if (!hasOldContent && hasNewContent) {
+                const newContentMatch = match.match(/## NEW_CONTENT\n```([\s\S]*?)```/);
+                const newContentCode = newContentMatch ? newContentMatch[1] : '';
+                return `## OPERATION:CREATE\n\`\`\`\n${newContentCode}\n\`\`\``;
+            } else if (hasOldContent && !hasNewContent) {
+                return match + '\n## NEW_CONTENT\n```\n```';
+            }
+            return match;
         }
-        // Return unchanged if both are present or both are missing (assuming minimal valid structure)
-        return match;
-      }
     );
-  
+
     return content;
   }