Add Source Current File button

Author: georgestagg

ggsql-vscode/package.json

@@ -55,6 +55,27 @@
         "languageId": "ggsql"
       }
     ],
+    "commands": [
+      {
+        "command": "ggsql.sourceCurrentFile",
+        "category": "ggsql",
+        "title": "Source Current File",
+        "icon": "$(play)"
+      }
+    ],
+    "menus": {
+      "editor/title/run": [
+        {
+          "command": "ggsql.sourceCurrentFile",
+          "group": "navigation@0",
+          "when": "resourceLangId == ggsql && !isInDiffEditor"
+        },
+        {
+          "command": "workbench.action.positronConsole.executeCode",
+          "when": "resourceLangId == ggsql && !isInDiffEditor"
+        }
+      ]
+    },
     "configuration": {
       "type": "object",
       "title": "ggsql",

ggsql-vscode/src/extension.ts

@@ -52,6 +52,21 @@ export function activate(context: vscode.ExtensionContext): void {
     }
 
     log(`Registered ${drivers.length} connection drivers`);
+
+    // Register "Source Current File" command for the editor run button
+    context.subscriptions.push(
+        vscode.commands.registerCommand('ggsql.sourceCurrentFile', async () => {
+            const editor = vscode.window.activeTextEditor;
+            if (!editor || editor.document.languageId !== 'ggsql') {
+                return;
+            }
+            const code = editor.document.getText();
+            if (code.trim().length === 0) {
+                return;
+            }
+            positronApi.runtime.executeCode('ggsql', code, true);
+        })
+    );
 }
 
 /**