feat: add search in folder command

part of #186

Author: HerringtonDarkholme

package.json

@@ -32,8 +32,21 @@
         "command": "ast-grep.restartLanguageServer",
         "title": "restart Language Server",
         "category": "ast-grep"
+      },
+      {
+        "command": "ast-grep.searchInFolder",
+        "title": "Find ast-grep Pattern in Folder"
       }
     ],
+    "menus": {
+      "explorer/context": [
+        {
+          "command": "ast-grep.searchInFolder",
+          "group": "4_search",
+          "when": "explorerResourceIsFolder"
+        }
+      ]
+    },
     "configuration": {
       "type": "object",
       "title": "ast-grep",

src/extension.ts

@@ -7,7 +7,7 @@ import {
   Executable
 } from 'vscode-languageclient/node'
 
-import { activate as activateWebview } from './view'
+import { activate as activateWebview, findInFolder } from './view'
 
 let client: LanguageClient
 const diagnosticCollectionName = 'ast-grep-diagnostics'
@@ -35,6 +35,7 @@ function getExecutable(isDebug: boolean): Executable {
 
 function activateLsp(context: ExtensionContext) {
   context.subscriptions.push(
+    commands.registerCommand('ast-grep.searchInFolder', findInFolder),
     commands.registerCommand('ast-grep.restartLanguageServer', async () => {
       console.log(
         'Restart the ast-grep language server by ast-grep.restart command'

src/view.ts

@@ -1,3 +1,4 @@
+import path from 'path'
 import type {
   Definition,
   ParentPort,
@@ -7,14 +8,14 @@ import type {
 } from './types'
 import { Unport, ChannelMessage } from 'unport'
 import * as vscode from 'vscode'
-import { workspace } from 'vscode'
+import { workspace, window } from 'vscode'
 import { type ChildProcessWithoutNullStreams, spawn } from 'node:child_process'
 
 export function activate(context: vscode.ExtensionContext) {
   const provider = new SearchSidebarProvider(context.extensionUri)
 
   context.subscriptions.push(
-    vscode.window.registerWebviewViewProvider(
+    window.registerWebviewViewProvider(
       SearchSidebarProvider.viewType,
       provider,
       { webviewOptions: { retainContextWhenHidden: true } }
@@ -288,3 +289,14 @@ function getNonce() {
   }
   return text
 }
+
+export function findInFolder(data: any) {
+  const workspacePath = workspace.workspaceFolders?.[0].uri.fsPath
+  // compute relative path to the workspace folder
+  const relative = workspacePath && path.relative(workspacePath, data.fsPath)
+  if (!relative) {
+    window.showErrorMessage('ast-grep Error: folder is not in the workspace')
+    return
+  }
+  window.showInformationMessage('ast-grep: searching in ' + relative)
+}