added ability to save cell output (#1576)

* added ability to export cell output

* added support for output only to be exported.

* added the option for the user to choose HTML, HTML + Log.

* changes to export only ODS and Log

* fixed feature to meet specs.

* fixes

* references name changing so export works.

* comment resolution.

* comment resolution.

* DCO Remediation Commit for Kishan Patel <[email protected]>

I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: feaa5e9
I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: 2711500
I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: 8da0a71
I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: 39a4585
I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: 47f30ea
I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: 27f60fb
I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: b770fef
I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: dc0338c
I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: a936bb8

Signed-off-by: Kishan Patel <[email protected]>

* comment resolution.

* DCO Remediation Commit for Kishan Patel <[email protected]>

I, Kishan Patel <[email protected]>, hereby add my Signed-off-by to this commit: 6ed0193

Signed-off-by: Kishan Patel <[email protected]>

---------

Signed-off-by: Kishan Patel <[email protected]>
Signed-off-by: Kishan Patel <[email protected]>
Co-authored-by: Kishan Patel <[email protected]>

Author: kpatl1

client/src/components/notebook/exporters/index.ts

@@ -1,6 +1,6 @@
 // Copyright © 2025, SAS Institute Inc., Cary, NC, USA.  All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
-import { Uri, window, workspace } from "vscode";
+import { Uri, l10n, window, workspace } from "vscode";
 import type { LanguageClient } from "vscode-languageclient/node";
 
 import path from "path";
@@ -30,3 +30,108 @@ export const exportNotebook = async (client: LanguageClient) => {
 
   workspace.fs.writeFile(uri, Buffer.from(content));
 };
+
+export const saveOutput = async () => {
+  const notebook = window.activeNotebookEditor?.notebook;
+  const activeCell = window.activeNotebookEditor?.selection?.start;
+
+  if (!notebook || activeCell === undefined) {
+    return;
+  }
+
+  const cell = notebook.cellAt(activeCell);
+  if (!cell) {
+    return;
+  }
+
+  let odsItem = null;
+  let logItem = null;
+
+  for (const output of cell.outputs) {
+    if (!odsItem) {
+      odsItem = output.items.find(
+        (item) => item.mime === "application/vnd.sas.ods.html5",
+      );
+    }
+    if (!logItem) {
+      logItem = output.items.find(
+        (item) => item.mime === "application/vnd.sas.compute.log.lines",
+      );
+    }
+
+    if (odsItem && logItem) {
+      break;
+    }
+  }
+
+  const choices: Array<{
+    label: string;
+    outputType: "html" | "log";
+  }> = [];
+
+  if (odsItem) {
+    choices.push({
+      label: l10n.t("Save ODS HTML"),
+      outputType: "html",
+    });
+  }
+
+  if (logItem) {
+    choices.push({
+      label: l10n.t("Save Log"),
+      outputType: "log",
+    });
+  }
+
+  const exportChoice = await window.showQuickPick(choices, {
+    placeHolder: l10n.t("Choose output type to save"),
+    ignoreFocusOut: true,
+  });
+
+  if (!exportChoice) {
+    return;
+  }
+
+  let content = "";
+  let fileExtension = "";
+  let fileName = "";
+  try {
+    if (exportChoice.outputType === "html" && odsItem) {
+      content = odsItem.data.toString();
+      fileExtension = "html";
+      fileName = `${path.basename(notebook.uri.path, ".sasnb")}_${l10n.t("output")}_${
+        activeCell + 1
+      }.html`;
+    } else if (exportChoice.outputType === "log" && logItem) {
+      const logs: Array<{ line: string; type: string }> = JSON.parse(
+        logItem.data.toString(),
+      );
+      content = logs.map((log) => log.line).join("\n");
+      fileExtension = "log";
+      fileName = `${path.basename(notebook.uri.path, ".sasnb")}_${l10n.t("output")}_${
+        activeCell + 1
+      }.log`;
+    }
+  } catch (error) {
+    window.showErrorMessage(
+      l10n.t("Failed to extract output content." + error),
+    );
+    return;
+  }
+
+  const filters: { [name: string]: string[] } = {};
+  filters[fileExtension.toUpperCase()] = [fileExtension];
+
+  const uri = await window.showSaveDialog({
+    filters,
+    defaultUri: Uri.parse(fileName),
+  });
+
+  if (!uri) {
+    return;
+  }
+
+  await workspace.fs.writeFile(uri, Buffer.from(content));
+
+  window.showInformationMessage(l10n.t("Saved to {0}", uri.fsPath));
+};

client/src/node/extension.ts

@@ -54,7 +54,7 @@ import { LogTokensProvider, legend } from "../components/logViewer";
 import { sasDiagnostic } from "../components/logViewer/sasDiagnostics";
 import { NotebookController } from "../components/notebook/Controller";
 import { NotebookSerializer } from "../components/notebook/Serializer";
-import { exportNotebook } from "../components/notebook/exporters";
+import { exportNotebook, saveOutput } from "../components/notebook/exporters";
 import { ConnectionType } from "../components/profile";
 import { SasTaskProvider } from "../components/tasks/SasTaskProvider";
 import { SAS_TASK_TYPE } from "../components/tasks/SasTasks";
@@ -202,6 +202,7 @@ export function activate(context: ExtensionContext) {
     commands.registerCommand("SAS.notebook.export", () =>
       exportNotebook(client),
     ),
+    commands.registerCommand("SAS.notebook.saveOutput", saveOutput),
     tasks.registerTaskProvider(SAS_TASK_TYPE, new SasTaskProvider()),
     ...sasDiagnostic.getSubscriptions(),
     commands.registerTextEditorCommand("SAS.toggleLineComment", (editor) => {

package.json

@@ -825,6 +825,11 @@
         "title": "%commands.SAS.notebook.export%",
         "category": "SAS Notebook"
       },
+      {
+        "command": "SAS.notebook.saveOutput",
+        "title": "%commands.SAS.notebook.saveOutput%",
+        "category": "SAS Notebook"
+      },
       {
         "command": "SAS.file.new",
         "shortTitle": "%commands.SAS.file.new.short%",
@@ -1097,6 +1102,12 @@
           "command": "SAS.notebook.export"
         }
       ],
+      "notebook/cell/title": [
+        {
+          "command": "SAS.notebook.saveOutput",
+          "when": "notebookType == 'sas-notebook' && notebookCellHasOutputs"
+        }
+      ],
       "commandPalette": [
         {
           "when": "editorLangId == sas && !SAS.hideRunMenuItem",

package.nls.json

@@ -16,6 +16,7 @@
   "commands.SAS.file.new": "New SAS File",
   "commands.SAS.file.new.short": "SAS File",
   "commands.SAS.notebook.export": "Export",
+  "commands.SAS.notebook.saveOutput": "Save Output",
   "commands.SAS.notebook.new": "New SAS Notebook",
   "commands.SAS.notebook.new.short": "SAS Notebook",
   "commands.SAS.refresh": "Refresh",