Merge pull request #274 from codefori/feature/run_from_uri

Ability to run a statement via URI
codefori · Sep 10, 2024 · b2c5afb · b2c5afb
2 parents d524f89 + 84475dc
commit b2c5afb
Show file tree

Hide file tree

Showing 4 changed files with 112 additions and 11 deletions.
diff --git a/package.json b/package.json
@@ -317,6 +317,11 @@
         "title": "Generate JSON SQL parser",
         "category": "Db2 for i"
       },
+      {
+        "command": "vscode-db2i.getStatementUri",
+        "title": "Copy sharable statement URI",
+        "category": "Db2 for i"
+      },
       {
         "command": "vscode-db2i.refreshSchemaBrowser",
         "title": "Refresh Schema Browser",
@@ -667,19 +672,11 @@
       }
     ],
     "menus": {
-      "editor/context": [
+      "commandPalette": [
         {
-          "command": "vscode-db2i.json.pasteGenerator",
-          "group": "sql@1",
+          "command": "vscode-db2i.getStatementUri",
           "when": "editorLangId == sql"
         },
-        {
-          "command": "vscode-db2i.json.pasteParser",
-          "group": "sql@2",
-          "when": "editorLangId == sql"
-        }
-      ],
-      "commandPalette": [
         {
           "command": "vscode-db2i.setSchemaFilter",
           "when": "never"
@@ -757,6 +754,23 @@
           "when": "never"
         }
       ],
+      "editor/context": [
+        {
+          "command": "vscode-db2i.getStatementUri",
+          "group": "sql@0",
+          "when": "editorLangId == sql"
+        },
+        {
+          "command": "vscode-db2i.json.pasteGenerator",
+          "group": "sql@1",
+          "when": "editorLangId == sql"
+        },
+        {
+          "command": "vscode-db2i.json.pasteParser",
+          "group": "sql@2",
+          "when": "editorLangId == sql"
+        }
+      ],
       "view/title": [
         {
           "command": "vscode-db2i.refreshSchemaBrowser",

diff --git a/src/contributes.json b/src/contributes.json
@@ -94,10 +94,26 @@
         "command": "vscode-db2i.json.pasteParser",
         "title": "Generate JSON SQL parser",
         "category": "Db2 for i"
+      },
+      {
+        "command": "vscode-db2i.getStatementUri",
+        "title": "Copy sharable statement URI",
+        "category": "Db2 for i"
       }
     ],
     "menus": {
+      "commandPalette": [
+        {
+          "command": "vscode-db2i.getStatementUri",
+          "when": "editorLangId == sql"
+        }
+      ],
       "editor/context": [
+        {
+          "command": "vscode-db2i.getStatementUri",
+          "group": "sql@0",
+          "when": "editorLangId == sql"
+        },
         {
           "command": "vscode-db2i.json.pasteGenerator",
           "group": "sql@1",

diff --git a/src/extension.ts b/src/extension.ts
@@ -20,6 +20,7 @@ import { notebookInit } from "./notebooks/IBMiSerializer";
 import { SelfTreeDecorationProvider, selfCodesResultsView } from "./views/jobManager/selfCodes/selfCodesResultsView";
 import Configuration from "./configuration";
 import { JDBCOptions } from "@ibm/mapepire-js/dist/src/types";
+import { Db2iUriHandler, getStatementUri } from "./uriHandler";
 
 export interface Db2i {
   sqlJobManager: SQLJobManager,
@@ -66,7 +67,9 @@ export function activate(context: vscode.ExtensionContext): Db2i {
     ),
     vscode.window.registerFileDecorationProvider(
       new SelfTreeDecorationProvider()
-    ) 
+    ),
+    vscode.window.registerUriHandler(new Db2iUriHandler()),
+    getStatementUri
   );
 
   JSONServices.initialise(context);

diff --git a/src/uriHandler.ts b/src/uriHandler.ts
@@ -0,0 +1,68 @@
+import { commands, env, Selection, Uri, UriHandler, window, workspace } from "vscode";
+import querystring from "querystring";
+import Document from "./language/sql/document";
+import { ServerComponent } from "./connection/serverComponent";
+import { remoteAssistIsEnabled } from "./language/providers/available";
+
+export class Db2iUriHandler implements UriHandler {
+  handleUri(uri: Uri) {
+    const path = uri.path;
+
+    switch (path) {
+      case '/sql':
+        const queryData = querystring.parse(uri.query);
+        const content = String(queryData.content).trim();
+
+        if (content) {
+          const asLower = content.toLowerCase();
+          const isValid = asLower.startsWith(`select `) || asLower.startsWith(`with `);
+
+          if (isValid) {
+            const run = queryData.run === `true`;
+
+            if (run) {
+              if (remoteAssistIsEnabled()) {
+                commands.executeCommand(`vscode-db2i.runEditorStatement`, {
+                  content,
+                  qualifier: `statement`,
+                  open: true,
+                });
+              } else {
+                window.showErrorMessage(`You must be connected to a system to run a statement.`);
+              }
+            } else {
+              workspace.openTextDocument({ language: `sql`, content }).then(textDoc => {
+                window.showTextDocument(textDoc);
+              });
+            }
+          } else {
+            window.showErrorMessage(`Only SELECT or WITH statements are supported.`);
+          }
+        }
+
+        break;
+    }
+  }
+}
+
+export const getStatementUri = commands.registerCommand(`vscode-db2i.getStatementUri`, async () => {
+  const editor = window.activeTextEditor;
+
+  if (editor) {
+    const content = editor.document.getText();
+
+    const sqlDocument = new Document(content);
+    const cursor = editor.document.offsetAt(editor.selection.active);
+
+    const currentStmt = sqlDocument.getGroupByOffset(cursor);
+
+    if (currentStmt) {
+      const stmtContent = content.substring(currentStmt.range.start, currentStmt.range.end);
+      editor.selection = new Selection(editor.document.positionAt(currentStmt.range.start), editor.document.positionAt(currentStmt.range.end));
+      const uri = `vscode://halcyontechltd.vscode-db2i/sql?content=${encodeURIComponent(stmtContent)}`;
+
+      env.clipboard.writeText(uri);
+      window.showInformationMessage(`Statement URI copied to clipboard.`);
+    }
+  }
+});