Ability to generate SQL from JSON

Author: worksofliam

package.json

@@ -105,9 +105,20 @@
         "title": "Generate SQL",
         "category": "Db2 for i",
         "icon": "$(add)"
-      }
+      },
+      {
+				"command": "vscode-db2i.pasteGenerator",
+				"title": "Paste JSON as SQL"
+			}
     ],
     "menus": {
+      "editor/context": [
+				{
+					"command": "vscode-db2i.pasteGenerator",
+					"group": "1_sql",
+					"when": "editorLangId == sql"
+				}
+			],
       "view/title": [
         {
           "command": "vscode-db2i.addSchemaToSchemaBrowser",

src/extension.js

@@ -3,8 +3,7 @@
 const vscode = require(`vscode`);
 const schemaBrowser = require(`./views/schemaBrowser`);
 
-const Configuration = require(`./configuration`);
-
+const JSONServices = require(`./language/json`);
 const languageProvider = require(`./language/provider`);
 
 // this method is called when your extension is activated
@@ -26,6 +25,7 @@ function activate(context) {
     ),
   );
 
+  JSONServices.initialise(context);
   languageProvider.initialise(context);
 }
 

src/language/json.js

@@ -0,0 +1,65 @@
+const vscode = require(`vscode`);
+
+const Statement = require(`../database/statement`);
+
+/**
+ * 
+ * @param {vscode.ExtensionContext} context 
+ */
+exports.initialise = (context) => {
+  context.subscriptions.push(
+    vscode.commands.registerCommand(`vscode-db2i.pasteGenerator`, async () => {
+      try {
+        const clipboard_content = await vscode.env.clipboard.readText(); 
+        const parsedData = JSON.parse(clipboard_content);
+
+        const sql = exports.generateSQL(parsedData);
+        const formatted = Statement.format(sql);
+
+        if (vscode.window.activeTextEditor) {
+          vscode.window.activeTextEditor.edit((edit) => {
+            edit.insert(vscode.window.activeTextEditor.selection.active, formatted);
+          });
+        }
+      } catch (e) {
+        vscode.window.showErrorMessage(`Error: ${e.message}`);
+      }
+    }),
+  )
+}
+
+exports.generateSQL = (jsonIn) => {
+  if (Array.isArray(jsonIn)) {
+    return generateArray(jsonIn);
+  } else {
+    return generateObject(jsonIn);
+  }
+}
+
+const generateArray = (elementIn) => {
+  const firstValue = Array.isArray(elementIn) ? elementIn[0] : elementIn;
+  if (typeof firstValue === `object`) {
+    return `(SELECT json_arrayagg(${generateObject(firstValue)}) from SYSIBM.SYSDUMMY1)`
+  } else {
+    return `(SELECT json_arrayagg(${typeof firstValue === `string` ? `'${firstValue}'` : firstValue}) from SYSIBM.SYSDUMMY1)`
+  }
+}
+
+const generateObject = (objIn) => {
+  const items = [];
+
+  Object.keys(objIn).forEach((key) => {
+    const value = objIn[key];
+    if (typeof value === `object`) {
+      if (Array.isArray(value)) {
+        items.push(`'${key}': ${generateArray(value[0])} format json`);
+      } else {
+        items.push(`'${key}': ${generateObject(value)}`);
+      }
+    } else {
+      items.push(`'${key}': ${typeof value === `string` ? `'${value}'` : value}`);
+    }
+  });
+
+  return `json_object(${items.join(`, `)})`;
+}