Positron: Improve execution of ggsql code (#296)

* Respond with null to unknown comms to avoid RPC timeouts

* cargo fmt

* Add Source Current File button

Author: georgestagg

ggsql-jupyter/src/kernel.rs

@@ -648,9 +648,27 @@ impl KernelServer {
                 self.handle_data_explorer_rpc(method, rpc_id, comm_id, parent, identities)
                     .await?;
             }
-            // Unknown comm
+            // Unknown comm — still respond to avoid RPC timeouts
             else {
-                tracing::warn!("Message for unknown comm_id: {}", comm_id);
+                tracing::warn!(
+                    "JSON-RPC request for unknown comm_id: {}, method: {}",
+                    comm_id,
+                    method
+                );
+                self.send_shell_reply(
+                    "comm_msg",
+                    json!({
+                        "comm_id": comm_id,
+                        "data": {
+                            "jsonrpc": "2.0",
+                            "id": rpc_id,
+                            "result": null
+                        }
+                    }),
+                    parent,
+                    identities,
+                )
+                .await?;
             }
         }
 

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);
+        })
+    );
 }
 
 /**