refactor: improve file match logic

Author: erha19

packages/ai-native/src/browser/components/ChatEditor.tsx

@@ -254,29 +254,31 @@ const CodeBlock = ({
               let lastIndex = 0;
               const fragments: (string | React.ReactNode)[] = [];
 
-              // 处理文件标记
-              fileMatches.forEach((match, matchIndex) => {
-                if (match.index !== undefined) {
-                  const spanText = processedText.slice(lastIndex, match.index);
-                  if (spanText) {
-                    fragments.push(<span key={`${index}-${matchIndex}`}>{spanText}</span>);
+              // 通用处理函数
+              const processMatches = (matches: RegExpMatchArray[], isFolder: boolean) => {
+                matches.forEach((match, matchIndex) => {
+                  if (match.index !== undefined) {
+                    const spanText = processedText.slice(lastIndex, match.index);
+                    if (spanText) {
+                      fragments.push(
+                        <span key={`${index}-${matchIndex}-${isFolder ? 'folder' : 'file'}`}>{spanText}</span>,
+                      );
+                    }
+                    fragments.push(
+                      renderAttachment(
+                        match[1],
+                        isFolder,
+                        `${index}-tag-${matchIndex}-${isFolder ? 'folder' : 'file'}`,
+                      ),
+                    );
+                    lastIndex = match.index + match[0].length;
                   }
-                  fragments.push(renderAttachment(match[1], false, `${index}-tag-${matchIndex}`));
-                  lastIndex = match.index + match[0].length;
-                }
-              });
+                });
+              };
 
-              // 处理文件夹标记
-              folderMatches.forEach((match, matchIndex) => {
-                if (match.index !== undefined) {
-                  const spanText = processedText.slice(lastIndex, match.index);
-                  if (spanText) {
-                    fragments.push(<span key={`${index}-${matchIndex}`}>{spanText}</span>);
-                  }
-                  fragments.push(renderAttachment(match[1], true, `${index}-tag-${matchIndex}`));
-                  lastIndex = match.index + match[0].length;
-                }
-              });
+              // 处理文件标记
+              processMatches(fileMatches, false);
+              processMatches(folderMatches, true);
 
               fragments.push(processedText.slice(lastIndex));
               renderedContent.push(...fragments);

packages/ai-native/src/browser/context/llm-context.service.ts

@@ -218,31 +218,35 @@ export class LLMContextServiceImpl extends WithEventBus implements LLMContextSer
 
   private async getPartiaFolderStructure(folder: string, level = 2): Promise<string[]> {
     const result: string[] = [];
-    const stat = await this.fileService.getFileStat(folder);
-
-    for (const child of stat?.children || []) {
-      const relativePath = new URI(folder).relative(new URI(child.uri))!.toString();
-
-      if (child.isSymbolicLink) {
-        // 处理软链接
-        const target = await this.fileService.getFileStat(child.realUri || child.uri);
-        if (target) {
-          result.push(`${relativePath} -> ${target} (symbolic link)`);
-        } else {
-          result.push(`${relativePath} (broken symbolic link)`);
+    try {
+      const stat = await this.fileService.getFileStat(folder);
+
+      for (const child of stat?.children || []) {
+        const relativePath = new URI(folder).relative(new URI(child.uri))!.toString();
+
+        if (child.isSymbolicLink) {
+          // 处理软链接
+          const target = await this.fileService.getFileStat(child.realUri || child.uri);
+          if (target) {
+            result.push(`${relativePath} -> ${target} (symbolic link)`);
+          } else {
+            result.push(`${relativePath} (broken symbolic link)`);
+          }
+          continue;
         }
-        continue;
-      }
 
-      if (child.type === FileType.Directory) {
-        result.push(`${relativePath}/`);
-        if (level > 1) {
-          const subDirStructure = await this.getPartiaFolderStructure(child.uri, level - 1);
-          result.push(...subDirStructure.map((subEntry) => `${relativePath}/${subEntry}`));
+        if (child.type === FileType.Directory) {
+          result.push(`${relativePath}/`);
+          if (level > 1) {
+            const subDirStructure = await this.getPartiaFolderStructure(child.uri, level - 1);
+            result.push(...subDirStructure.map((subEntry) => `${relativePath}/${subEntry}`));
+          }
+        } else if (child.type === FileType.File) {
+          result.push(relativePath);
         }
-      } else if (child.type === FileType.File) {
-        result.push(relativePath);
       }
+    } catch {
+      return result;
     }
 
     return result;