Condition json output on opt-in settings flag

Author: cmoog

package.json

@@ -98,6 +98,11 @@
           "default": false,
           "description": "(Unstable) Use embedded language server for intelligent completion and hover information."
         },
+        "SQLNotebook.outputJSON": {
+          "type": "boolean",
+          "default": false,
+          "description": "Output JSON in addition to markdown. Other extensions may use this output type to render an interactive table."
+        },
         "SQLNotebook.queryTimeout": {
           "type": "number",
           "default": 30000,

src/controller.ts

@@ -92,10 +92,13 @@ export class SQLNotebookController {
 
     writeSuccess(
       execution,
-      result.map((item) => [
-        text(resultToMarkdownTable(item), 'text/markdown'),
-        json(item),
-      ])
+      result.map((item) => {
+        const outputs = [text(resultToMarkdownTable(item), 'text/markdown')];
+        if (outputJsonMimeType()) {
+          outputs.push(json(item));
+        }
+        return outputs;
+      })
     );
   }
 }
@@ -118,3 +121,9 @@ function writeSuccess(
   );
   execution.end(true, Date.now());
 }
+
+function outputJsonMimeType(): boolean {
+  return (
+    vscode.workspace.getConfiguration('SQLNotebook').get('outputJSON') ?? false
+  );
+}