feat(gpt-runner-vscode): add context menu for ask codes

Author: 2214962083

packages/gpt-runner-shared/src/common/types/eventemitter.ts

@@ -27,6 +27,7 @@ export interface ClientEventData {
 
   [ClientEventName.UpdateUserSelectedText]: {
     text: string
+    insertInputPrompt?: boolean
   }
 
   [ClientEventName.OpenFileInIde]: {

packages/gpt-runner-vscode/package.json

@@ -81,6 +81,11 @@
         "title": "Open GPT Runner Chat",
         "icon": "res/logo.svg",
         "category": "GPT Runner"
+      },
+      {
+        "command": "gpt-runner.askSelection",
+        "title": "💬 Ask GPT Runner",
+        "category": "GPT Runner"
       }
     ],
     "configuration": {
@@ -110,6 +115,13 @@
           "group": "navigation",
           "icon": "res/logo.svg"
         }
+      ],
+      "editor/context": [
+        {
+          "command": "gpt-runner.askSelection",
+          "group": "gpt-runner",
+          "when": "editorHasSelection"
+        }
       ]
     },
     "jsonValidation": [

packages/gpt-runner-vscode/src/constant.ts

@@ -8,6 +8,7 @@ export enum Commands {
   OpenInBrowser = `${EXT_NAME}.openInBrowser`,
   InsertCodes = `${EXT_NAME}.insertCodes`,
   DiffCodes = `${EXT_NAME}.diffCodes`,
+  AskSelection = `${EXT_NAME}.askSelection`,
 }
 
 export const URI_SCHEME = 'gpt-runner'

packages/gpt-runner-vscode/src/register/sync-select-text.ts

@@ -3,6 +3,7 @@ import { ClientEventName, debounce } from '@nicepkg/gpt-runner-shared/common'
 import type { ContextLoader } from '../contextLoader'
 import { state } from '../state'
 import { emitter } from '../emitter'
+import { Commands } from '../constant'
 
 export async function registerSyncSelectText(
   cwd: string,
@@ -37,6 +38,38 @@ export async function registerSyncSelectText(
       debounceUpdateSelectText(editor)
     }))
 
+    disposes.push(vscode.commands.registerCommand(Commands.AskSelection, async () => {
+      if (vscode.window.activeTextEditor)
+        state.activeEditor = vscode.window.activeTextEditor
+
+      const selectText = vscode.window.activeTextEditor?.document.getText(vscode.window.activeTextEditor.selection)
+      if (!selectText)
+        return
+
+      const handleSidebarOpenAndInit = () => {
+        // focus
+        state.sidebarWebviewView?.show(true)
+
+        // send selectText to webview
+        emitter.emit(ClientEventName.UpdateUserSelectedText, {
+          text: selectText,
+          insertInputPrompt: true,
+        })
+      }
+
+      // open sidebar webview when it's not open
+      if (!state.sidebarWebviewView?.visible) {
+        await vscode.commands.executeCommand('gpt-runner.chatView.focus')
+        state.sidebarWebviewView?.webview.onDidReceiveMessage(({ eventName }: { eventName: ClientEventName }) => {
+          if (eventName === ClientEventName.InitSuccess)
+            handleSidebarOpenAndInit()
+        })
+      }
+      else {
+        handleSidebarOpenAndInit()
+      }
+    }))
+
     return vscode.Disposable.from({
       dispose,
     })

packages/gpt-runner-web/client/src/hooks/use-emit-bind.hook.ts

@@ -2,10 +2,21 @@ import { useEffect } from 'react'
 import { ClientEventName, toUnixPath } from '@nicepkg/gpt-runner-shared/common'
 import { emitter } from '../helpers/emitter'
 import { useTempStore } from '../store/zustand/temp'
+import { useGlobalStore } from '../store/zustand/global'
 import { useOn } from './use-on.hook'
+import { useChatInstance } from './use-chat-instance.hook'
 
 export function useEmitBind(deps: any[] = []) {
-  const { updateIdeSelectedText, updateIdeOpeningFilePaths, updateIdeActiveFilePath } = useTempStore()
+  const {
+    updateIdeSelectedText,
+    updateIdeOpeningFilePaths,
+    updateIdeActiveFilePath,
+  } = useTempStore()
+
+  const { activeChatId } = useGlobalStore()
+  const { updateCurrentChatInstance } = useChatInstance({
+    chatId: activeChatId,
+  })
 
   useEffect(() => {
     emitter.emit(ClientEventName.InitSuccess)
@@ -35,9 +46,17 @@ export function useEmitBind(deps: any[] = []) {
 
   useOn({
     eventName: ClientEventName.UpdateUserSelectedText,
-    listener: ({ text }) => {
-      updateIdeSelectedText(text)
+    listener: ({ text, insertInputPrompt }) => {
+      if (!insertInputPrompt) {
+        updateIdeSelectedText(text)
+      }
+      else {
+        updateIdeSelectedText('')
+        updateCurrentChatInstance({
+          inputtingPrompt: text,
+        })
+      }
     },
-    deps: [...deps, updateIdeSelectedText],
+    deps: [...deps, updateIdeSelectedText, updateCurrentChatInstance],
   })
 }