fix: fix several bugs

Author: HerringtonDarkholme

src/extension/codelens.ts

@@ -14,7 +14,7 @@ export function activateCodeLens(context: vscode.ExtensionContext) {
   // Register the command that the CodeLens will execute
   const runRuleCommand = vscode.commands.registerCommand(
     'ast-grep.runRule',
-    (ruleId: string, text: string) => {
+    (text: string) => {
       parentPort.postMessage('searchByYAML', {
         text,
       })
@@ -29,27 +29,21 @@ class CodelensProvider implements vscode.CodeLensProvider {
     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))
+    // Look for rule definitions in YAML files
+    const isAstGrepRule = /^rule:/m.test(text) && /^id:/m.test(text) && /^language:/m.test(text)
+    if (!isAstGrepRule) {
+      return []
     }
 
-    return codeLenses
+    const range = new vscode.Range(0, 0, 0, 0)
+    const command: vscode.Command = {
+      title: `Run ast-grep rule`,
+      command: 'ast-grep.runRule',
+      arguments: [text],
+    }
+    return [new vscode.CodeLens(range, command)]
   }
 
   resolveCodeLens(codeLens: vscode.CodeLens, _token: vscode.CancellationToken): vscode.CodeLens {

src/webview/hooks/useSearch.tsx

@@ -177,7 +177,7 @@ function getSnapshot() {
 
 function queryHasRewrite() {
   if ('yaml' in queryInFlight) {
-    return /^fix:/.test(queryInFlight.yaml)
+    return /^fix:/m.test(queryInFlight.yaml)
   }
   return !!queryInFlight.rewrite
 }