support sql workbooks

Author: ecmel

package.json

@@ -213,6 +213,15 @@
           "light": "./resources/light/add-scratchpad-python.svg"
         }
       },
+      {
+        "category": "KX",
+        "command": "kdb.scratchpad.sql.create",
+        "title": "New Workbook (SQL)",
+        "icon": {
+          "dark": "./resources/dark/sql.svg",
+          "light": "./resources/light/sql.svg"
+        }
+      },
       {
         "category": "KX",
         "command": "kdb.connections.content.selectView",
@@ -686,6 +695,10 @@
           "command": "kdb.scratchpad.python.create",
           "when": "true"
         },
+        {
+          "command": "kdb.scratchpad.sql.create",
+          "when": "true"
+        },
         {
           "command": "kdb.connections.content.selectView",
           "when": "false"
@@ -915,10 +928,15 @@
           "group": "navigation@2"
         },
         {
-          "command": "kdb.scratchpad.explorer.refresh",
+          "command": "kdb.scratchpad.sql.create",
           "when": "view == kdb-scratchpad-explorer",
           "group": "navigation@3"
         },
+        {
+          "command": "kdb.scratchpad.explorer.refresh",
+          "when": "view == kdb-scratchpad-explorer",
+          "group": "navigation@4"
+        },
         {
           "command": "kdb.queryHistory.clear",
           "when": "view == kdb-query-history",

ref_card.md

@@ -6,7 +6,8 @@
 | KX: Install KDB-X      | KX: New Notebook          | KX: Focus on Datasources view       |
 | KX: Start REPL         | KX: New Workbook (q)      | KX: Focus on Workbooks view         |
 | KX: Import Connections | KX: New Workbook (Python) | KX: Focus on Query History view     |
-| KX: Export Connections | KX: New Datasource        | KX: Focus on Help and Feedback view |
+| KX: Export Connections | KX: New Workbook (SQL)    | KX: Focus on Help and Feedback view |
+|                        | KX: New Datasource        |                                     |
 
 ## Keybindings
 
@@ -60,10 +61,10 @@
 | :--------- | :--: | :--: | :---: | :------: | :----: | :----: | :------------: |
 | File `q`   |  ✓   |  ✓   |   ✓   |    ✓     |        |        |       ✓        |
 | File `py`  |  ✓   |  ✓   |   ✓   |    ✓     |        |        |       ✓        |
-| File `sql` |  ✓   |      |       |    ✓     |        |        |       ✓        |
+| File `sql` |  ✓   |  ✓   |       |    ✓     |        |        |       ✓        |
 | Cell `q`   |  ✓   |  ✓   |   ✓   |    ✓     |        |        |       ✓        |
 | Cell `py`  |  ✓   |  ✓   |   ✓   |    ✓     |        |        |       ✓        |
-| Cell `sql` |  ✓   |      |       |    ✓     |        |        |       ✓        |
+| Cell `sql` |  ✓   |  ✓   |       |    ✓     |        |        |       ✓        |
 | Datasource |      |      |       |    ✓     |   ✓    |   ✓    |       ✓        |
 
 `REPL` and `My q` requires [PyKX](https://github.com/KxSystems/kx-vscode/wiki/Use-PyKX-Within-REPL) for Python support.

resources/dark/sql-active.svg

@@ -78,6 +78,7 @@ import {
   formatScratchpadStacktrace,
   resultToBase64,
   needsScratchpad,
+  getSQLWrapper,
 } from "../utils/queryUtils";
 import { openUrl } from "../utils/uriUtils";
 import {
@@ -1126,8 +1127,12 @@ export async function runQuery(
     variable = await inputVariable();
   }
 
+  if (isSql && !isInsights) {
+    query = getSQLWrapper(query);
+  }
+
   const runner = Runner.create((_, token) => {
-    return target || isSql
+    return target || (isSql && isInsights)
       ? variable
         ? populateScratchpad(
             getPartialDatasourceFile(query, target, isSql, isPython),

resources/dark/sql-connected.svg

@@ -449,7 +449,12 @@ function isPython(uri: Uri | undefined) {
 }
 
 function isWorkbook(uri: Uri | undefined) {
-  return uri && (uri.path.endsWith(".kdb.q") || uri.path.endsWith(".kdb.py"));
+  return (
+    uri &&
+    (uri.path.endsWith(".kdb.q") ||
+      uri.path.endsWith(".kdb.py") ||
+      uri.path.endsWith(".kdb.sql"))
+  );
 }
 
 function isDataSource(uri: Uri | undefined) {
@@ -538,15 +543,6 @@ export async function runActiveEditor(type?: ExecutionTypes) {
     const target = isInsights ? getTargetForUri(uri) : undefined;
     const isSql = executorName.endsWith(".sql");
 
-    if (isSql && !isInsights) {
-      notify(
-        `SQL execution is not supported on ${conn.connLabel}.`,
-        MessageKind.ERROR,
-        { logger },
-      );
-      return;
-    }
-
     if (type === ExecutionTypes.PopulateScratchpad && !isInsights) {
       notify(
         `Populating scratchpad is not supported on ${conn.connLabel}.`,