feat: add yaml codelens

Author: HerringtonDarkholme

package.json

@@ -100,6 +100,10 @@
       {
         "command": "ast-grep.searchByCode",
         "title": "ast-grep: Search by Code"
+      },
+      {
+        "command": "ast-grep.runRule",
+        "title": "Run ast-grep Rule"
       }
     ],
     "menus": {

src/extension/codelens.ts

@@ -0,0 +1,58 @@
+import * as vscode from 'vscode'
+import { parentPort } from './common'
+
+export function activateCodeLens(context: vscode.ExtensionContext) {
+  // Register the CodeLens provider for YAML files
+  const codelensProvider = new CodelensProvider()
+  context.subscriptions.push(
+    vscode.languages.registerCodeLensProvider(
+      { language: 'yaml', pattern: '**/*.yml' }, // Only trigger for YAML files
+      codelensProvider,
+    ),
+  )
+
+  // Register the command that the CodeLens will execute
+  const runRuleCommand = vscode.commands.registerCommand(
+    'ast-grep.runRule',
+    (ruleId: string, text: string) => {
+      parentPort.postMessage('searchByYAML', {
+        text,
+      })
+    },
+  )
+
+  context.subscriptions.push(runRuleCommand)
+}
+
+class CodelensProvider implements vscode.CodeLensProvider {
+  provideCodeLenses(
+    document: vscode.TextDocument,
+    _token: vscode.CancellationToken,
+  ): vscode.CodeLens[] {
+    const codeLenses: vscode.CodeLens[] = []
+    const text = document.getText()
+
+    // Look for rule definitions in YAML files (sgconfig.yml)
+    const ruleMatches = text.matchAll(/^(\s*)id:\s*(.+)$/gm)
+
+    for (const match of ruleMatches) {
+      const ruleId = match[2].trim()
+      const line = document.positionAt(match.index!).line
+      const range = new vscode.Range(line, 0, line, 0)
+
+      const command: vscode.Command = {
+        title: `Run rule: ${ruleId}`,
+        command: 'ast-grep.runRule',
+        arguments: [ruleId, text],
+      }
+
+      codeLenses.push(new vscode.CodeLens(range, command))
+    }
+
+    return codeLenses
+  }
+
+  resolveCodeLens(codeLens: vscode.CodeLens, _token: vscode.CancellationToken): vscode.CodeLens {
+    return codeLens
+  }
+}

src/extension/index.ts

@@ -1,4 +1,5 @@
 import type { ExtensionContext } from 'vscode'
+import { activateCodeLens } from './codelens'
 import { detectDefaultBinaryAtStart } from './common'
 import { activateLsp } from './lsp'
 import { activatePreview } from './preview'
@@ -11,4 +12,5 @@ export async function activate(context: ExtensionContext) {
   activateWebview(context)
   activateSearch(context)
   activatePreview(context)
+  activateCodeLens(context)
 }

src/types.ts

@@ -71,6 +71,7 @@ export interface ParentToChild {
   toggleAllSearch: unknown
   searchByCode: { text: string }
   enableYAML: boolean
+  searchByYAML: { text: string }
 }
 
 export interface Diff {

src/webview/SearchSidebar/SearchWidgetContainer/YamlWidget.tsx

@@ -22,26 +22,6 @@ export default function YamlWidget() {
   const [value, setValue] = useState('')
   return (
     <div {...stylex.props(styles.yaml)}>
-      <VSCodeTextArea
-        {...stylex.props(styles.editor)}
-        value={value}
-        placeholder="YAML configuration"
-        resize="vertical"
-        onInput={(e: any) => {
-          const newValue = e.target.value || ''
-          setValue(newValue)
-        }}
-      />
-      <VSCodeButton
-        {...stylex.props(styles.searchButton)}
-        appearance="primary"
-        onClick={() => {
-          // Handle save or apply logic here
-          postYAML({ yaml: value, includeFile: '' })
-        }}
-      >
-        Search
-      </VSCodeButton>
     </div>
   )
 }

src/webview/hooks/useSearch.tsx

@@ -125,6 +125,12 @@ childPort.onMessage('refreshSearchResult', event => {
   grouped.sort(byFilePath)
   notify()
 })
+childPort.onMessage('searchByYAML', event => {
+  postYAML({
+    yaml: event.text,
+    includeFile: '',
+  })
+})
 
 function groupBy(matches: DisplayResult[]) {
   const groups = new Map<string, DisplayResult[]>()