feat: implment find in folder

fix #186

Author: HerringtonDarkholme

src/types.ts

@@ -50,6 +50,9 @@ export type Definition = {
       id: number
       error: Error
     }
+    setIncludeFile: {
+      includeFile: string
+    }
   }
   child2parent: {
     search: SearchQuery & {

src/view.ts

@@ -177,9 +177,9 @@ function openFile({
   })
 }
 
-function setupParentPort(webviewView: vscode.WebviewView) {
-  const parentPort: ParentPort = new Unport()
+const parentPort: ParentPort = new Unport()
 
+function setupParentPort(webviewView: vscode.WebviewView) {
   parentPort.implementChannel({
     async send(message) {
       webviewView.webview.postMessage(message)
@@ -298,5 +298,8 @@ export function findInFolder(data: any) {
     window.showErrorMessage('ast-grep Error: folder is not in the workspace')
     return
   }
-  window.showInformationMessage('ast-grep: searching in ' + relative)
+  vscode.commands.executeCommand('ast-grep.search.input.focus')
+  parentPort.postMessage('setIncludeFile', {
+    includeFile: relative
+  })
 }

src/webview/SearchSidebar/SearchWidgetContainer/comps/SearchOptions.tsx

@@ -1,6 +1,5 @@
 import type { CSSProperties } from 'react'
 import { VscEllipsis } from 'react-icons/vsc'
-import { useBoolean } from 'react-use'
 import IncludeFile from './IncludeFile'
 
 const buttonStyle: CSSProperties = {
@@ -29,14 +28,17 @@ interface SearchOptionsProps {
   includeFile: string
   setIncludeFile: (value: string) => void
   refreshResult: () => void
+  showOptions: boolean
+  toggleOptions: () => void
 }
 
 export default function SearchOptions({
   includeFile,
   setIncludeFile,
-  refreshResult
+  refreshResult,
+  showOptions,
+  toggleOptions
 }: SearchOptionsProps) {
-  const [showOptions, toggleOptions] = useBoolean(false)
   return (
     <div style={optionsStyle}>
       <button style={buttonStyle} onClick={toggleOptions}>
@@ -49,7 +51,7 @@ export default function SearchOptions({
             setIncludeFile={setIncludeFile}
             refreshResult={refreshResult}
           />
-          {/* TODO: add file include*/}
+          {/* TODO: add file exclude*/}
         </div>
       )}
     </div>

src/webview/SearchSidebar/SearchWidgetContainer/index.tsx

@@ -12,6 +12,8 @@ const SearchWidgetContainer = ({ onQueryChange }: Props) => {
     setInputValue,
     includeFile,
     setIncludeFile,
+    showOptions,
+    toggleOptions,
     refreshResult
   } = useSearchQuery(onQueryChange)
 
@@ -26,6 +28,8 @@ const SearchWidgetContainer = ({ onQueryChange }: Props) => {
         includeFile={includeFile}
         setIncludeFile={setIncludeFile}
         refreshResult={refreshResult}
+        showOptions={showOptions}
+        toggleOptions={toggleOptions}
       />
     </div>
   )

src/webview/hooks/useQuery.tsx

@@ -1,7 +1,7 @@
-import { useCallback } from 'react'
-import { useLocalStorage } from 'react-use'
-import { useDebounce } from 'react-use'
+import { useCallback, useEffect } from 'react'
+import { useLocalStorage, useDebounce, useBoolean } from 'react-use'
 import { SearchQuery } from '../../types'
+import { childPort } from '../postMessage'
 export { SearchQuery }
 
 export function useSearchQuery(startSearch: (query: SearchQuery) => void) {
@@ -13,13 +13,21 @@ export function useSearchQuery(startSearch: (query: SearchQuery) => void) {
     'ast-grep-search-panel-include-value',
     ''
   )
+  const [showOptions, toggleOptions] = useBoolean(Boolean(includeFile))
   const refreshResult = useCallback(() => {
     startSearch({
       inputValue,
       includeFile
     })
   }, [inputValue, includeFile, startSearch])
 
+  useEffect(() => {
+    childPort.onMessage('setIncludeFile', val => {
+      setIncludeFile(val.includeFile)
+      toggleOptions(true)
+    })
+  }, [])
+
   // auto refresh result when input value changes
   useDebounce(refreshResult, 100, [inputValue, includeFile])
 
@@ -28,6 +36,8 @@ export function useSearchQuery(startSearch: (query: SearchQuery) => void) {
     setInputValue,
     includeFile,
     setIncludeFile,
-    refreshResult
+    refreshResult,
+    showOptions,
+    toggleOptions
   }
 }