feat: add pattern/yaml mode toggle

Author: HerringtonDarkholme

media/pattern-dark.svg

@@ -81,6 +81,22 @@
           "dark": "media/expand-dark.svg"
         }
       },
+      {
+        "command": "ast-grep.enableYAML",
+        "title": "Use YAML",
+        "icon": {
+          "light": "media/yaml-light.svg",
+          "dark": "media/yaml-dark.svg"
+        }
+      },
+      {
+        "command": "ast-grep.enablePattern",
+        "title": "Use Pattern",
+        "icon": {
+          "light": "media/pattern-light.svg",
+          "dark": "media/pattern-dark.svg"
+        }
+      },
       {
         "command": "ast-grep.searchByCode",
         "title": "ast-grep: Search by Code"
@@ -103,6 +119,16 @@
         }
       ],
       "view/title": [
+        {
+          "command": "ast-grep.enableYAML",
+          "when": "view == ast-grep.search.input && !ast-grep.yamlMode",
+          "group": "navigation"
+        },
+        {
+          "command": "ast-grep.enablePattern",
+          "when": "view == ast-grep.search.input && ast-grep.yamlMode",
+          "group": "navigation"
+        },
         {
           "command": "ast-grep.collapseAll",
           "when": "view == ast-grep.search.input && !ast-grep.searchDefaultCollapse",

media/pattern-light.svg

@@ -19,6 +19,8 @@ export function activateWebview(context: vscode.ExtensionContext) {
     commands.registerCommand('ast-grep.clearSearchResults', clearSearchResults),
     commands.registerCommand('ast-grep.expandAll', toggleAllSearch),
     commands.registerCommand('ast-grep.collapseAll', toggleAllSearch),
+    commands.registerCommand('ast-grep.enableYAML', enableYAML),
+    commands.registerCommand('ast-grep.enablePattern', enablePattern),
   )
 }
 
@@ -122,6 +124,22 @@ function clearSearchResults() {
   parentPort.postMessage('clearSearchResults', {})
 }
 
+function enableYAML() {
+  vscode.commands.executeCommand(
+    'setContext',
+    'ast-grep.yamlMode',
+    true,
+  )
+}
+
+function enablePattern() {
+  vscode.commands.executeCommand(
+    'setContext',
+    'ast-grep.yamlMode',
+    false,
+  )
+}
+
 let defaultCollapse = false
 
 // reset default collapse when starting a new search