feat(vscode): add at feature with AT file implementation in VSCode (#…

…3602) * feat(chat): add symbol and file information retrieval methods * feat(chat): add listFileInWorkspace and readFileContent APIs * feat(chat): enhance editor listing by including open tabs in absence of query * refactor(chat): simplify uriToListFileItem and update file content retrieval logic * fix: remove all vscode special tab like shortcut tab * chore: allow in case sensetive search * Update clients/vscode/src/chat/webview.ts Co-authored-by: Zhiming Ma <[email protected]> * update: remove label * update * update * [autofix.ci] apply automated fixes --------- Co-authored-by: aliang <[email protected]> Co-authored-by: Zhiming Ma <[email protected]> Co-authored-by: liangfung <[email protected]> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
TabbyML · Jan 17, 2025 · 130b2c1 · 130b2c1
1 parent 2d5c7c7
commit 130b2c1
Show file tree

Hide file tree

Showing 3 changed files with 69 additions and 1 deletion.
diff --git a/clients/vscode/src/chat/createClient.ts b/clients/vscode/src/chat/createClient.ts
@@ -38,6 +38,8 @@ export function createClient(webview: Webview, api: ClientApiMethods): ServerApi
       getActiveEditorSelection: api.getActiveEditorSelection,
       fetchSessionState: api.fetchSessionState,
       storeSessionState: api.storeSessionState,
+      listFileInWorkspace: api.listFileInWorkspace,
+      readFileContent: api.readFileContent,
     },
   });
 }
diff --git a/clients/vscode/src/chat/utils.ts b/clients/vscode/src/chat/utils.ts
@@ -1,11 +1,12 @@
 import path from "path";
-import { TextEditor, Position as VSCodePosition, Range as VSCodeRange, Uri, workspace } from "vscode";
+import { Position as VSCodePosition, Range as VSCodeRange, Uri, workspace, TextEditor } from "vscode";
 import type {
   Filepath,
   Position as ChatPanelPosition,
   LineRange,
   PositionRange,
   Location,
+  ListFileItem,
   FilepathInGitRepository,
 } from "tabby-chat-panel";
 import type { GitProvider } from "../git/GitProvider";
@@ -221,3 +222,14 @@ export function generateLocalNotebookCellUri(notebook: Uri, handle: number): Uri
   const fragment = `${p}${s}s${Buffer.from(notebook.scheme).toString("base64")}`;
   return notebook.with({ scheme: DocumentSchemes.vscodeNotebookCell, fragment });
 }
+
+export function uriToListFileItem(uri: Uri, gitProvider: GitProvider): ListFileItem {
+  return {
+    filepath: localUriToChatPanelFilepath(uri, gitProvider),
+  };
+}
+
+export function escapeGlobPattern(query: string): string {
+  // escape special glob characters: * ? [ ] { } ( ) ! @
+  return query.replace(/[*?[\]{}()!@]/g, "\\$&");
+}
diff --git a/clients/vscode/src/chat/webview.ts b/clients/vscode/src/chat/webview.ts
@@ -14,6 +14,7 @@ import {
   ProgressLocation,
   Location,
   LocationLink,
+  TabInputText,
 } from "vscode";
 import { TABBY_CHAT_PANEL_API_VERSION } from "tabby-chat-panel";
 import type {
@@ -26,6 +27,9 @@ import type {
   FileLocation,
   GitRepository,
   EditorFileContext,
+  ListFilesInWorkspaceParams,
+  ListFileItem,
+  FileRange,
 } from "tabby-chat-panel";
 import * as semver from "semver";
 import { v4 as uuid } from "uuid";
@@ -43,6 +47,8 @@ import {
   vscodeRangeToChatPanelPositionRange,
   chatPanelLocationToVSCodeRange,
   isValidForSyncActiveEditorSelection,
+  uriToListFileItem,
+  escapeGlobPattern,
 } from "./utils";
 import mainHtml from "./html/main.html";
 import errorHtml from "./html/error.html";
@@ -492,6 +498,54 @@ export class ChatWebview {
           ...state,
         };
       },
+
+      listFileInWorkspace: async (params: ListFilesInWorkspaceParams): Promise<ListFileItem[]> => {
+        const maxResults = params.limit || 50;
+        const searchQuery = params.query?.trim();
+
+        if (!searchQuery) {
+          const openTabs = window.tabGroups.all
+            .flatMap((group) => group.tabs)
+            .filter((tab) => tab.input && (tab.input as TabInputText).uri);
+
+          this.logger.info(`No query provided, listing ${openTabs.length} opened editors.`);
+          return openTabs.map((tab) => uriToListFileItem((tab.input as TabInputText).uri, this.gitProvider));
+        }
+
+        try {
+          const caseInsensitivePattern = searchQuery
+            .split("")
+            .map((char) => {
+              if (char.toLowerCase() !== char.toUpperCase()) {
+                return `{${char.toLowerCase()},${char.toUpperCase()}}`;
+              }
+              return escapeGlobPattern(char);
+            })
+            .join("");
+
+          const globPattern = `**/${caseInsensitivePattern}*`;
+
+          this.logger.info(`Searching files with pattern: ${globPattern}, limit: ${maxResults}`);
+
+          const files = await workspace.findFiles(globPattern, undefined, maxResults);
+          this.logger.info(`Found ${files.length} files.`);
+          return files.map((uri) => uriToListFileItem(uri, this.gitProvider));
+        } catch (error) {
+          this.logger.warn("Failed to find files:", error);
+          window.showErrorMessage("Failed to find files.");
+          return [];
+        }
+      },
+
+      readFileContent: async (info: FileRange): Promise<string | null> => {
+        const uri = chatPanelFilepathToLocalUri(info.filepath, this.gitProvider);
+        if (!uri) {
+          this.logger.warn(`Could not resolve URI from filepath: ${JSON.stringify(info.filepath)}`);
+          return null;
+        }
+        const document = await workspace.openTextDocument(uri);
+        return document.getText(chatPanelLocationToVSCodeRange(info.range) ?? undefined);
+      },
     });
   }