feat: select code to search

fix #225

Author: HerringtonDarkholme

package.json

@@ -80,6 +80,10 @@
           "light": "media/expand-light.svg",
           "dark": "media/expand-dark.svg"
         }
+      },
+      {
+        "command": "ast-grep.searchByCode",
+        "title": "ast-grep: Search by Code"
       }
     ],
     "menus": {
@@ -90,6 +94,14 @@
           "when": "explorerResourceIsFolder"
         }
       ],
+
+      "editor/context": [
+        {
+          "command": "ast-grep.searchByCode",
+          "when": "editorHasSelection",
+          "group": "navigation"
+        }
+      ],
       "view/title": [
         {
           "command": "ast-grep.collapseAll",

src/extension/search.ts

@@ -11,6 +11,7 @@ import type { SgSearch, DisplayResult, SearchQuery } from '../types'
 export function activateSearch(context: ExtensionContext) {
   context.subscriptions.push(
     commands.registerCommand('ast-grep.searchInFolder', findInFolder),
+    commands.registerCommand('ast-grep.searchByCode', searchByCode),
   )
 }
 
@@ -176,3 +177,14 @@ parentPort.onMessage('search', async payload => {
   })
   parentPort.postMessage('searchEnd', payload)
 })
+
+function searchByCode() {
+  const editor = window.activeTextEditor
+  if (!editor) {
+    return
+  }
+  const selection = editor.selection
+  const text = editor.document.getText(selection)
+  commands.executeCommand('ast-grep.search.input.focus')
+  parentPort.postMessage('searchByCode', { text })
+}

src/types.ts

@@ -69,6 +69,7 @@ export interface ParentToChild {
   refreshAllSearch: unknown
   clearSearchResults: unknown
   toggleAllSearch: unknown
+  searchByCode: { text: string }
 }
 
 export interface Diff {

src/webview/hooks/useQuery.tsx

@@ -41,6 +41,17 @@ export function useSearchField(key: keyof typeof LS_KEYS) {
   useEffect(() => {
     searchQuery[key] = field
   }, [field, key])
+  // this is really BAD code :(
+  useEffect(() => {
+    if (key !== 'pattern') {
+      return
+    }
+    childPort.onMessage('searchByCode', ({ text }) => {
+      searchQuery[key] = field
+      setField(text)
+      refreshResult()
+    })
+  }, [key])
   useDebounce(refreshResult, 150, [field])
   return [field, setField] as const
 }