yg-team-cli 2.4.5 → 2.4.7

package/dist/cli.js

@@ -3548,46 +3548,57 @@ async function updateAiMemory(featureName, featureSlug) {
     return;
   }
   let content = await FileUtils.read(aiMemoryFile);
-  const newRow = `| ${featureName} | ${featureSlug}.md | \u25CB \u672A\u5F00\u59CB | 0/0 | - | |`;
+  const featureDisplay = featureName.replace(/[-_]/g, " ").split(" ").map((word) => word.charAt(0).toUpperCase() + word.slice(1).toLowerCase()).join(" ");
+  const newRow = `| ${featureDisplay} | ${featureSlug}.md | \u25CB \u672A\u5F00\u59CB | 0/0 | - | |`;
   if (!content.includes("## \u529F\u80FD\u6E05\u5355")) {
-    const lastHeaderMatch = content.match(/^(#{1,6}\s+.+)$/m);
-    if (lastHeaderMatch) {
-      const featureList = `
-
+    content += `
 ## \u529F\u80FD\u6E05\u5355 (Feature Inventory)
 
 | \u529F\u80FD | Spec \u6587\u4EF6 | \u72B6\u6001 | \u8FDB\u5EA6 | \u5B8C\u6210\u65E5\u671F | \u5907\u6CE8 |
 |------|----------|------|------|---------|------|
 ${newRow}
 `;
-      content = content.replace(lastHeaderMatch[0], lastHeaderMatch[0] + featureList);
+  } else {
+    const lines = content.split("\n");
+    let featureInventorySection = false;
+    let insertIndex = -1;
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i];
+      if (line.includes("## \u529F\u80FD\u6E05\u5355")) {
+        featureInventorySection = true;
+        continue;
+      }
+      if (featureInventorySection && line.startsWith("## ") && !line.includes("\u529F\u80FD\u6E05\u5355")) {
+        break;
+      }
+      if (featureInventorySection && /^\|[-]+\|/.test(line.trim())) {
+        insertIndex = i + 1;
+        break;
+      }
+    }
+    if (insertIndex !== -1) {
+      lines.splice(insertIndex, 0, newRow);
+      content = lines.join("\n");
     } else {
-      content += `
-## \u529F\u80FD\u6E05\u5355 (Feature Inventory)
-
-| \u529F\u80FD | Spec \u6587\u4EF6 | \u72B6\u6001 | \u8FDB\u5EA6 | \u5B8C\u6210\u65E5\u671F | \u5907\u6CE8 |
-|------|----------|------|------|---------|------|
+      const sectionIndex = lines.findIndex((line) => line.includes("## \u529F\u80FD\u6E05\u5355"));
+      if (sectionIndex !== -1) {
+        const tableLines = [
+          "",
+          "| \u529F\u80FD | Spec \u6587\u4EF6 | \u72B6\u6001 | \u8FDB\u5EA6 | \u5B8C\u6210\u65E5\u671F | \u5907\u6CE8 |",
+          "|------|----------|------|------|---------|------|",
+          newRow,
+          ""
+        ];
+        lines.splice(sectionIndex + 1, 0, ...tableLines);
+        content = lines.join("\n");
+      } else {
+        content += `
 ${newRow}
 `;
-    }
-  } else {
-    const tableHeaderPattern = /(\| 功能 \| Spec 文件 \| 状态 \| 进度 \| 完成日期 \| 备注 \|(?:\r?\n)?)/;
-    if (tableHeaderPattern.test(content)) {
-      content = content.replace(tableHeaderPattern, `$1
-${newRow}$2`);
-    } else {
-      content = content.replace(
-        /## 功能清单 \(Feature Inventory\)/,
-        `## \u529F\u80FD\u6E05\u5355 (Feature Inventory)
-
-| \u529F\u80FD | Spec \u6587\u4EF6 | \u72B6\u6001 | \u8FDB\u5EA6 | \u5B8C\u6210\u65E5\u671F | \u5907\u6CE8 |
-|------|----------|------|------|---------|------|
-${newRow}`
-      );
+      }
     }
   }
   await FileUtils.write(aiMemoryFile, content);
-  logger.success("AI_MEMORY.md \u5DF2\u66F4\u65B0");
 }
 async function showSpecPreview(specFile) {
   logger.newLine();