feat: 增加刷新反向引用信息的功能

Littlefean · Littlefean · commit 3816e6f21ffb · 2025-11-22T12:56:07.000+08:00
diff --git a/app/src/core/service/dataManageService/textNodeSmartTools.tsx b/app/src/core/service/dataManageService/textNodeSmartTools.tsx
@@ -1,14 +1,11 @@
 import { Dialog } from "@/components/ui/dialog";
-import { loadAllServicesBeforeInit } from "@/core/loadAllServices";
 import { Project } from "@/core/Project";
-import { RecentFileManager } from "@/core/service/dataFileService/RecentFileManager";
 import { Edge } from "@/core/stage/stageObject/association/Edge";
 import { LineEdge } from "@/core/stage/stageObject/association/LineEdge";
 import { CollisionBox } from "@/core/stage/stageObject/collisionBox/collisionBox";
 import { ReferenceBlockNode } from "@/core/stage/stageObject/entity/ReferenceBlockNode";
 import { TextNode } from "@/core/stage/stageObject/entity/TextNode";
 import { DetailsManager } from "@/core/stage/stageObject/tools/entityDetailsManager";
-import { PathString } from "@/utils/pathString";
 import { averageColors, Color, Vector } from "@graphif/data-structures";
 import { Rectangle } from "@graphif/shapes";
 import { toast } from "sonner";
@@ -545,62 +542,6 @@ export namespace TextNodeSmartTools {
 
     project.stageManager.add(referenceBlock);
     project.stageManager.delete(selectedNode); // TODO: 直接删除原有节点有隐患
-
-    // 更新被引用文件的reference.msgpack
-    const currentFileName = PathString.getFileNameFromPath(project.uri.path);
-    if (!currentFileName) return;
-
-    try {
-      // 根据文件名查找被引用文件
-      const recentFiles = await RecentFileManager.getRecentFiles();
-      const referencedFile = recentFiles.find(
-        (file) =>
-          PathString.getFileNameFromPath(file.uri.path) === fileName ||
-          PathString.getFileNameFromPath(file.uri.fsPath) === fileName,
-      );
-      if (!referencedFile) return;
-
-      // 创建被引用文件的Project实例
-      const referencedProject = new Project(referencedFile.uri);
-
-      // 初始化项目
-      loadAllServicesBeforeInit(referencedProject);
-      await referencedProject.init();
-
-      // 更新引用
-      if (sectionName) {
-        // 引用特定Section的情况
-        if (!referencedProject.references.sections[sectionName]) {
-          referencedProject.references.sections[sectionName] = [];
-        }
-
-        // 确保数组中没有重复的文件名
-        const index = referencedProject.references.sections[sectionName].indexOf(currentFileName);
-        if (index === -1) {
-          referencedProject.references.sections[sectionName].push(currentFileName);
-          // 保存更新
-          await referencedProject.save();
-        }
-      } else {
-        // 引用整个文件的情况
-        if (!referencedProject.references.files) {
-          referencedProject.references.files = [];
-        }
-
-        // 确保数组中没有重复的文件名
-        const index = referencedProject.references.files.indexOf(currentFileName);
-        if (index === -1) {
-          referencedProject.references.files.push(currentFileName);
-          // 保存更新
-          await referencedProject.save();
-        }
-      }
-
-      // 关闭项目
-      await referencedProject.dispose();
-      // TODO: 存在隐患，欠考虑如果引用已经被当前软件打开的情况。
-    } catch (error) {
-      toast.error("Failed to update reference:" + String(error));
-    }
+    await project.referenceManager.insertRefDataToSourcePrgFile(fileName, sectionName);
   }
 }
diff --git a/app/src/core/stage/stageManager/concreteMethods/StageReferenceManager.tsx b/app/src/core/stage/stageManager/concreteMethods/StageReferenceManager.tsx
@@ -8,6 +8,7 @@ import { onOpenFile } from "@/core/service/GlobalMenu";
 import { ReferenceBlockNode } from "../../stageObject/entity/ReferenceBlockNode";
 import { RectangleLittleNoteEffect } from "@/core/service/feedbackService/effectEngine/concrete/RectangleLittleNoteEffect";
 import { SectionReferencePanel } from "@/sub/ReferencesWindow";
+import { loadAllServicesBeforeInit } from "@/core/loadAllServices";
 
 interface parserResult {
   /**
@@ -32,6 +33,12 @@ interface parserResult {
 export class ReferenceManager {
   constructor(private readonly project: Project) {}
 
+  /**
+   * 保险检查函数
+   * 解析用户在文本节点中输入的引用格式文本，防止直接退出编辑模式后触发转换，导致引用块内容被错误解析
+   * @param text 引用块文本
+   * @returns
+   */
   public static referenceBlockTextParser(text: string): parserResult {
     if (!text.startsWith("[[") || !text.endsWith("]]")) {
       return {
@@ -103,6 +110,136 @@ export class ReferenceManager {
     }
   }
 
+  /**
+   * 更新当前项目的引用信息
+   * （清理无效的引用）
+   */
+  public async updateCurrentProjectReference() {
+    const recentFiles = await RecentFileManager.getRecentFiles();
+
+    // 遍历当前项目的每一个被引用的Section框
+    for (const sectionName in this.project.references.sections) {
+      const fileNameList = this.project.references.sections[sectionName];
+      const fileNameListNew = [];
+      for (const fileName of fileNameList) {
+        const file = recentFiles.find(
+          (file) =>
+            PathString.getFileNameFromPath(file.uri.path) === fileName ||
+            PathString.getFileNameFromPath(file.uri.fsPath) === fileName,
+        );
+        if (file) {
+          // 即使文件存在，也要打开看一看引用块是否在那个文件中。
+          const thatProject = new Project(file.uri);
+          loadAllServicesBeforeInit(thatProject);
+          await thatProject.init();
+          if (this.checkReferenceBlockInProject(thatProject, fileName, sectionName)) {
+            fileNameListNew.push(fileName);
+          }
+          thatProject.dispose();
+        }
+      }
+      if (fileNameListNew.length === 0) {
+        // 直接把这个章节从引用列表中删除
+        delete this.project.references.sections[sectionName];
+      } else {
+        this.project.references.sections[sectionName] = fileNameListNew;
+      }
+    }
+
+    // 遍历每一个直接引用自己整个文件的文件
+    const fileNameListNew = [];
+    for (const fileName of this.project.references.files) {
+      const file = recentFiles.find(
+        (file) =>
+          PathString.getFileNameFromPath(file.uri.path) === fileName ||
+          PathString.getFileNameFromPath(file.uri.fsPath) === fileName,
+      );
+      if (file) {
+        // 即使文件存在，也要打开看一看引用块是否在那个文件中。
+        const thatProject = new Project(file.uri);
+        loadAllServicesBeforeInit(thatProject);
+        await thatProject.init();
+        if (this.checkReferenceBlockInProject(thatProject, fileName, "")) {
+          fileNameListNew.push(fileName);
+        }
+        thatProject.dispose();
+      }
+    }
+    this.project.references.files = fileNameListNew;
+  }
+
+  public checkReferenceBlockInProject(project: Project, fileName: string, sectionName: string) {
+    const referenceBlocks = project.stage
+      .filter((object) => object instanceof ReferenceBlockNode)
+      .filter(
+        (referenceBlockNode) =>
+          referenceBlockNode.fileName === fileName && referenceBlockNode.sectionName === sectionName,
+      );
+    if (referenceBlocks.length > 0) {
+      return true;
+    }
+    return false;
+  }
+
+  public async insertRefDataToSourcePrgFile(fileName: string, sectionName: string) {
+    // 更新被引用文件的reference.msgpack
+    const currentFileName = PathString.getFileNameFromPath(this.project.uri.path);
+    if (!currentFileName) return;
+
+    try {
+      // 根据文件名查找被引用文件
+      const recentFiles = await RecentFileManager.getRecentFiles();
+      const referencedFile = recentFiles.find(
+        (file) =>
+          PathString.getFileNameFromPath(file.uri.path) === fileName ||
+          PathString.getFileNameFromPath(file.uri.fsPath) === fileName,
+      );
+      if (!referencedFile) return;
+
+      // 创建被引用文件的Project实例
+      const referencedProject = new Project(referencedFile.uri);
+
+      // 初始化项目
+      loadAllServicesBeforeInit(referencedProject);
+      await referencedProject.init();
+
+      // 更新引用
+      if (sectionName) {
+        // 引用特定Section的情况
+        if (!referencedProject.references.sections[sectionName]) {
+          referencedProject.references.sections[sectionName] = [];
+        }
+
+        // 确保数组中没有重复的文件名
+        const index = referencedProject.references.sections[sectionName].indexOf(currentFileName);
+        if (index === -1) {
+          referencedProject.references.sections[sectionName].push(currentFileName);
+          // 保存更新
+          await referencedProject.save();
+        }
+      } else {
+        // 引用整个文件的情况
+        if (!referencedProject.references.files) {
+          referencedProject.references.files = [];
+        }
+
+        // 确保数组中没有重复的文件名
+        const index = referencedProject.references.files.indexOf(currentFileName);
+        if (index === -1) {
+          referencedProject.references.files.push(currentFileName);
+          // 保存更新
+          await referencedProject.save();
+        }
+      }
+
+      // 关闭项目
+      await referencedProject.dispose();
+      // TODO: 存在隐患，欠考虑如果引用已经被当前软件打开的情况。
+    } catch (error) {
+      toast.error("更新reference.msgpack失败：" + String(error));
+    }
+  }
+
   /**
    * 从源头 跳转到引用位置
    * @param section
diff --git a/app/src/sub/ReferencesWindow.tsx b/app/src/sub/ReferencesWindow.tsx
@@ -15,14 +15,17 @@ export default function ReferencesWindow(props: { currentProjectFileName: string
   if (!project) return <></>;
 
   const [references, setReferences] = useState(project.references);
+  const [isUpdating, setIsUpdating] = useState(false);
 
-  function refresh() {
+  async function refresh() {
+    setIsUpdating(true);
+    await project?.referenceManager.updateCurrentProjectReference();
     setReferences({ ...project!.references });
-    // 现在这个刷新作用很小，不能检测已经失效的引用
+    setIsUpdating(false);
   }
 
   useEffect(() => {
-    refresh();
+    setReferences({ ...project!.references });
   }, []);
 
   return (
@@ -33,60 +36,65 @@ export default function ReferencesWindow(props: { currentProjectFileName: string
           刷新
         </Button>
       </div>
-
-      {/* 引用信息展示 */}
-      <div className="flex-1 overflow-y-auto">
-        <div className="mb-4">
-          <h3 className="mb-2 text-lg font-semibold">直接引用{currentProjectFileName}的文件</h3>
-          {references.files.length === 0 ? (
-            <p className="text-muted-foreground text-sm">当前项目中没有引用{currentProjectFileName}的文件</p>
-          ) : (
-            <div className="space-y-1">
-              {references.files.map((filePath) => {
-                const fileName = PathString.getFileNameFromPath(filePath);
-                return (
-                  <div
-                    key={filePath}
-                    className="text-select-option-text flex cursor-pointer items-center gap-2 rounded p-1 text-sm *:cursor-pointer hover:ring"
-                    onClick={() => project.referenceManager.jumpToReferenceLocation(fileName, "")}
-                  >
-                    <span className="font-medium">{fileName}</span>
-                    <span className="text-muted-foreground text-xs">{filePath}</span>
-                  </div>
-                );
-              })}
-            </div>
-          )}
-        </div>
-
-        <div>
-          <h3 className="mb-2 text-lg font-semibold">引用此文件中一些Section框的文件</h3>
-          {Object.keys(references.sections).length === 0 ? (
-            <p className="text-muted-foreground text-sm">{currentProjectFileName}中没有被引用的Section</p>
-          ) : (
-            <div className="space-y-3">
-              {Object.entries(references.sections).map(([referencedSectionName, sections]) => (
-                <div key={referencedSectionName}>
-                  <div className="text-select-option-text rounded p-1 font-medium">{referencedSectionName}</div>
-                  <div className="my-1 ml-4 space-y-1">
-                    {sections.map((fileName) => (
+      {isUpdating ? (
+        <div className="text-muted-foreground text-sm">正在刷新中...</div>
+      ) : (
+        <>
+          {/* 引用信息展示 */}
+          <div className="flex-1 overflow-y-auto">
+            <div className="mb-4">
+              <h3 className="mb-2 text-lg font-semibold">直接引用{currentProjectFileName}的文件</h3>
+              {references.files.length === 0 ? (
+                <p className="text-muted-foreground text-sm">当前项目中没有引用{currentProjectFileName}的文件</p>
+              ) : (
+                <div className="space-y-1">
+                  {references.files.map((filePath) => {
+                    const fileName = PathString.getFileNameFromPath(filePath);
+                    return (
                       <div
-                        onClick={() =>
-                          project.referenceManager.jumpToReferenceLocation(fileName, referencedSectionName)
-                        }
-                        key={fileName}
-                        className="border-muted text-select-option-text cursor-pointer rounded border-l-2 p-1 pl-2 text-sm hover:ring"
+                        key={filePath}
+                        className="text-select-option-text flex cursor-pointer items-center gap-2 rounded p-1 text-sm *:cursor-pointer hover:ring"
+                        onClick={() => project.referenceManager.jumpToReferenceLocation(fileName, "")}
                       >
-                        {fileName}
+                        <span className="font-medium">{fileName}</span>
+                        <span className="text-muted-foreground text-xs">{filePath}</span>
                       </div>
-                    ))}
-                  </div>
+                    );
+                  })}
                 </div>
-              ))}
+              )}
             </div>
-          )}
-        </div>
-      </div>
+
+            <div>
+              <h3 className="mb-2 text-lg font-semibold">引用此文件中一些Section框的文件</h3>
+              {Object.keys(references.sections).length === 0 ? (
+                <p className="text-muted-foreground text-sm">{currentProjectFileName}中没有被引用的Section</p>
+              ) : (
+                <div className="space-y-3">
+                  {Object.entries(references.sections).map(([referencedSectionName, sections]) => (
+                    <div key={referencedSectionName}>
+                      <div className="text-select-option-text rounded p-1 font-medium">{referencedSectionName}</div>
+                      <div className="my-1 ml-4 space-y-1">
+                        {sections.map((fileName) => (
+                          <div
+                            onClick={() =>
+                              project.referenceManager.jumpToReferenceLocation(fileName, referencedSectionName)
+                            }
+                            key={fileName}
+                            className="border-muted text-select-option-text cursor-pointer rounded border-l-2 p-1 pl-2 text-sm hover:ring"
+                          >
+                            {fileName}
+                          </div>
+                        ))}
+                      </div>
+                    </div>
+                  ))}
+                </div>
+              )}
+            </div>
+          </div>
+        </>
+      )}
     </div>
   );
 }
