Ability to hide SQL in export

Signed-off-by: worksofliam <[email protected]>

Author: worksofliam

src/notebooks/logic/export.ts

@@ -70,11 +70,14 @@ export const exportNotebookAsHtml = vscode.commands.registerCommand(`vscode-db2i
                         const eol = cell.document.eol === vscode.EndOfLine.CRLF ? `\r\n` : `\n`;
 
                         let content = cell.document.getText().trim();
-                        let chartDetail: ChartDetail | undefined;
 
-                        ({ chartDetail, content } = getStatementDetail(content, eol));
+                        const detail = getStatementDetail(content, eol);
 
-                        cellContents.push(`<pre>${content}</pre>`);
+                        content = detail.content;
+
+                        if (detail.settings.hideStatement !== `true`) {
+                          cellContents.push(`<pre>${content}</pre>`);
+                        }
 
                         // Execute the query
                         const query = selected.job.query(content);
@@ -98,6 +101,7 @@ export const exportNotebookAsHtml = vscode.commands.registerCommand(`vscode-db2i
 
                           let fallbackToTable = true;
 
+                          const { chartDetail } = detail;
                           if (chartDetail.type) {
                             if (chartJsTypes.includes(chartDetail.type as ChartJsType)) {
                               const possibleChart = generateChart(executionId, chartDetail, columns, table, generateChartHTMLEmbedded);

src/notebooks/logic/statement.ts

@@ -1,8 +1,17 @@
 import { ChartDetail, chartTypes } from "./chart";
 import { ChartJsType, chartJsTypes } from "./chartJs";
 
+export interface StatementSettings {
+  chart?: ChartJsType;
+  title?: string;
+  y?: string;
+  hideStatement?: string;
+  [key: string]: string
+};
+
 export function getStatementDetail(content: string, eol: string) {
   let chartDetail: ChartDetail = {};
+  let settings: StatementSettings = {};
 
   // Strip out starting comments
   if (content.startsWith(`--`)) {
@@ -12,8 +21,6 @@ export function getStatementDetail(content: string, eol: string) {
     const startingComments = lines.slice(0, firstNonCommentLine).map(line => line.substring(2).trim());
     content = lines.slice(firstNonCommentLine).join(eol);
 
-    let settings = {};
-
     for (let comment of startingComments) {
       const sep = comment.indexOf(`:`);
       const key = comment.substring(0, sep).trim();
@@ -46,5 +53,5 @@ export function getStatementDetail(content: string, eol: string) {
     chartDetail.type = chartType;
     content = content.substring(chartType.length + 1);
   }
-  return { chartDetail, content };
+  return { chartDetail, content, settings };
 }