feat: adding clickble symbol render in chat panel

clients/tabby-chat-panel/src/index.ts

@@ -53,7 +53,14 @@ export interface ServerApi {
   updateTheme: (style: string, themeClass: string) => void
   updateActiveSelection: (context: Context | null) => void
 }
-
+export interface KeywordInfo {
+  sourceFile: string
+  sourceLine: number
+  sourceChar: number
+  targetFile: string
+  targetLine: number
+  targetChar: number
+}
 export interface ClientApi {
   navigate: (context: Context, opts?: NavigateOpts) => void
   refresh: () => Promise<void>
@@ -69,6 +76,10 @@ export interface ClientApi {
   onCopy: (content: string) => void
 
   onKeyboardEvent: (type: 'keydown' | 'keyup' | 'keypress', event: KeyboardEventInit) => void
+  onRenderLsp: (filepaths: string[], keywords: string[]) => Promise<Record<
+  string,
+  KeywordInfo
+>>
 }
 
 export interface ChatMessage {
@@ -94,6 +105,7 @@ export function createClient(target: HTMLIFrameElement, api: ClientApi): ServerA
       onLoaded: api.onLoaded,
       onCopy: api.onCopy,
       onKeyboardEvent: api.onKeyboardEvent,
+      onRenderLsp: api.onRenderLsp,
     },
   })
 }

clients/vscode/src/chat/WebviewHelper.ts

@@ -10,6 +10,8 @@ import {
   Webview,
   ColorThemeKind,
   ProgressLocation,
+  commands,
+  LocationLink,
 } from "vscode";
 import type { ServerApi, ChatMessage, Context, NavigateOpts, OnLoadedParams } from "tabby-chat-panel";
 import { TABBY_CHAT_PANEL_API_VERSION } from "tabby-chat-panel";
@@ -23,6 +25,7 @@ import { createClient } from "./chatPanel";
 import { ChatFeature } from "../lsp/ChatFeature";
 import { isBrowser } from "../env";
 import { getFileContextFromSelection, showFileContext } from "./fileContext";
+import path from "path";
 
 export class WebviewHelper {
   webview?: Webview;
@@ -548,6 +551,89 @@ export class WebviewHelper {
         this.logger.debug(`Dispatching keyboard event: ${type} ${JSON.stringify(event)}`);
         this.webview?.postMessage({ action: "dispatchKeyboardEvent", type, event });
       },
+      onRenderLsp: async (filepaths: string[], keywords: string[]) => {
+        const foundLocations: Record<
+          string,
+          {
+            sourceFile: string;
+            sourceLine: number;
+            sourceChar: number;
+            targetFile: string;
+            targetLine: number;
+            targetChar: number;
+          }
+        > = {};
+
+        if (!keywords.length || !filepaths.length || filepaths.length === 0) {
+          this.logger.info("No keywords or filepaths provided");
+          return foundLocations;
+        }
+
+        const foundKeywords = new Set<string>();
+
+        try {
+          const workspaceRoot = workspace.workspaceFolders?.[0];
+          if (!workspaceRoot) {
+            this.logger.error("No workspace folder found");
+            return foundLocations;
+          }
+          const rootPath = workspaceRoot.uri;
+
+          for (const filepath of filepaths) {
+            if (foundKeywords.size === keywords.length) {
+              this.logger.info("All keywords found, stopping search");
+              break;
+            }
+
+            const normalizedPath = filepath.startsWith("/") ? filepath.slice(1) : filepath;
+            const fullPath = path.join(rootPath.path, normalizedPath);
+            const fileUri = Uri.file(fullPath);
+            const document = await workspace.openTextDocument(fileUri);
+
+            const remainingKeywords = keywords.filter((kw) => !foundKeywords.has(kw));
+
+            for (const keyword of remainingKeywords) {
+              if (foundLocations[keyword]) continue;
+
+              const content = document.getText();
+              const pos = content.indexOf(keyword);
+
+              if (pos !== -1) {
+                const position = document.positionAt(pos);
+                const locations = await commands.executeCommand<LocationLink[]>(
+                  "vscode.executeDefinitionProvider",
+                  fileUri,
+                  position,
+                );
+
+                if (locations && locations.length > 0) {
+                  // TODO: handle multiple locations
+                  const location = locations[0];
+                  if (location) {
+                    foundLocations[keyword] = {
+                      sourceFile: filepath,
+                      sourceLine: position.line + 1,
+                      sourceChar: position.character,
+                      targetFile: workspace.asRelativePath(location.targetUri.path),
+                      targetLine: location.targetRange.start.line,
+                      targetChar: location.targetRange.start.character,
+                    };
+                    foundKeywords.add(keyword);
+                  }
+                }
+              }
+            }
+          }
+
+          // TODO: test
+          for (const [keyword, location] of Object.entries(foundLocations)) {
+            this.logger.info(`Found locations for ${keyword}:`, location);
+          }
+        } catch (error) {
+          this.logger.error("Error in onRenderLsp:", error);
+        }
+        return foundLocations;
+      },
     });
   }
 }

clients/vscode/src/chat/chatPanel.ts

@@ -32,6 +32,7 @@ export function createClient(webview: Webview, api: ClientApi): ServerApi {
       onCopy: api.onCopy,
       onLoaded: api.onLoaded,
       onKeyboardEvent: api.onKeyboardEvent,
+      onRenderLsp: api.onRenderLsp,
     },
   });
 }

ee/tabby-ui/app/chat/page.tsx

@@ -370,6 +370,7 @@ export default function ChatPage() {
         onCopyContent={isInEditor && server?.onCopy}
         onSubmitMessage={isInEditor && server?.onSubmitMessage}
         onApplyInEditor={isInEditor && server?.onApplyInEditor}
+        onRenderLsp={isInEditor && server?.onRenderLsp}
       />
     </ErrorBoundary>
   )

ee/tabby-ui/components/chat/chat.tsx

@@ -1,6 +1,11 @@
 import React, { RefObject } from 'react'
 import { compact, findIndex, isEqual, some, uniqWith } from 'lodash-es'
-import type { Context, FileContext, NavigateOpts } from 'tabby-chat-panel'
+import type {
+  Context,
+  FileContext,
+  KeywordInfo,
+  NavigateOpts
+} from 'tabby-chat-panel'
 
 import { ERROR_CODE_NOT_FOUND } from '@/lib/constants'
 import {
@@ -46,6 +51,10 @@ type ChatContextValue = {
     content: string,
     opts?: { languageId: string; smart: boolean }
   ) => void
+  onRenderLsp?: (
+    filepaths: string[],
+    keywords: string[]
+  ) => Promise<Record<string, KeywordInfo>>
   relevantContext: Context[]
   activeSelection: Context | null
   removeRelevantContext: (index: number) => void
@@ -84,6 +93,10 @@ interface ChatProps extends React.ComponentProps<'div'> {
     content: string,
     opts?: { languageId: string; smart: boolean }
   ) => void
+  onRenderLsp?: (
+    filepaths: string[],
+    keywords: string[]
+  ) => Promise<Record<string, KeywordInfo>>
   chatInputRef: RefObject<HTMLTextAreaElement>
 }
 
@@ -104,6 +117,7 @@ function ChatRenderer(
     onCopyContent,
     onSubmitMessage,
     onApplyInEditor,
+    onRenderLsp,
     chatInputRef
   }: ChatProps,
   ref: React.ForwardedRef<ChatRef>
@@ -522,6 +536,7 @@ function ChatRenderer(
         container,
         onCopyContent,
         onApplyInEditor,
+        onRenderLsp,
         relevantContext,
         removeRelevantContext,
         chatInputRef,

ee/tabby-ui/components/chat/question-answer.tsx

@@ -252,7 +252,7 @@ function AssistantMessageCard(props: AssistantMessageCardProps) {
     enableRegenerating,
     ...rest
   } = props
-  const { onNavigateToContext, onApplyInEditor, onCopyContent } =
+  const { onNavigateToContext, onApplyInEditor, onCopyContent, onRenderLsp } =
     React.useContext(ChatContext)
   const [relevantCodeHighlightIndex, setRelevantCodeHighlightIndex] =
     React.useState<number | undefined>(undefined)
@@ -389,6 +389,8 @@ function AssistantMessageCard(props: AssistantMessageCardProps) {
               onCodeCitationMouseEnter={onCodeCitationMouseEnter}
               onCodeCitationMouseLeave={onCodeCitationMouseLeave}
               canWrapLongLines={!isLoading}
+              onRenderLsp={onRenderLsp}
+              onNavigateToContext={onNavigateToContext}
             />
             {!!message.error && <ErrorMessageBlock error={message.error} />}
           </>