feat: add library support for SASPy profile

Signed-off-by: Shuguang Sun <[email protected]>

client/src/commands/authorize.ts

@@ -43,6 +43,10 @@ export const checkProfileAndAuthorize =
         commands.executeCommand("setContext", "SAS.librariesDisplayed", true);
         libraryNavigator.refresh();
         return finishAuthorization(profileConfig);
+      case ConnectionType.SASPY:
+        commands.executeCommand("setContext", "SAS.librariesDisplayed", true);
+        libraryNavigator.refresh();
+        return finishAuthorization(profileConfig);
       default:
         return finishAuthorization(profileConfig);
     }

client/src/components/LibraryNavigator/LibraryAdapterFactory.ts

@@ -2,6 +2,7 @@
 // SPDX-License-Identifier: Apache-2.0
 import ItcLibraryAdapter from "../../connection/itc/ItcLibraryAdapter";
 import RestLibraryAdapter from "../../connection/rest/RestLibraryAdapter";
+import SaspyLibraryAdapter from "../../connection/saspy/SaspyLibraryAdapter";
 import { ConnectionType } from "../profile";
 import { LibraryAdapter } from "./types";
 
@@ -11,6 +12,8 @@ class LibraryAdapterFactory {
       case ConnectionType.IOM:
       case ConnectionType.COM:
         return new ItcLibraryAdapter();
+      case ConnectionType.SASPY:
+        return new SaspyLibraryAdapter();
       case ConnectionType.Rest:
       default:
         return new RestLibraryAdapter();

client/src/components/profile.ts

@@ -20,7 +20,7 @@ enum ConnectionOptions {
   SAS9COM = "SAS 9.4 (local)",
   SAS9IOM = "SAS 9.4 (remote - IOM)",
   SAS9SSH = "SAS 9.4 (remote - SSH)",
-  SAS9SASPY = "SASPY",
+  SAS9SASPY = "SAS 9.4 (SASPy)",
   SASViya = "SAS Viya",
 }
 
@@ -95,7 +95,6 @@ export interface SSHProfile extends BaseProfile {
 
 export interface SASPYProfile extends BaseProfile {
   connectionType: ConnectionType.SASPY;
-  // saspath: string;
   cfgname: string;
   pythonpath: string;
 }
@@ -112,7 +111,12 @@ export interface IOMProfile extends BaseProfile {
   port: number;
 }
 
-export type Profile = ViyaProfile | SSHProfile | SASPYProfile | COMProfile | IOMProfile;
+export type Profile =
+  | ViyaProfile
+  | SSHProfile
+  | SASPYProfile
+  | COMProfile
+  | IOMProfile;
 
 export enum AutoExecType {
   File = "file",
@@ -601,24 +605,8 @@ export class ProfileConfig {
 
       await this.upsertProfile(name, profileClone);
     } else if (profileClone.connectionType === ConnectionType.SASPY) {
-      // profileClone.pythonpath = await createInputTextBox(
-      //   ProfilePromptType.PYTHONPath,
-      //   profileClone.pythonpath,
-      // );
-      // if (profileClone.pythonpath === undefined) {
-      //   return;
-      // }
-
-      // profileClone.cfgname = await createInputTextBox(
-      //   ProfilePromptType.Cfgname,
-      //   profileClone.cfgname,
-      // );
-      // if (profileClone.cfgname === undefined) {
-      //   return;
-      // }
-
       profileClone.cfgname = "";
-      // profileClone.sasOptions = [];        
+      // profileClone.sasOptions = [];
       await this.upsertProfile(name, profileClone);
     } else if (profileClone.connectionType === ConnectionType.COM) {
       profileClone.sasOptions = [];
@@ -697,7 +685,7 @@ export enum ProfilePromptType {
   Port,
   Username,
   Cfgname,
-  PYTHONpath
+  PYTHONpath,
 }
 
 /**

client/src/connection/index.ts

@@ -17,9 +17,9 @@ import {
   LogLine as ComputeLogLine,
   LogLineTypeEnum as ComputeLogLineTypeEnum,
 } from "./rest/api/compute";
+import { getSession as getSASPYSession } from "./saspy";
 import { Session } from "./session";
 import { getSession as getSSHSession } from "./ssh";
-import { getSession as getSASPYSession } from "./saspy";
 
 let profileConfig: ProfileConfig;
 

client/src/connection/saspy/CodeRunner.ts

@@ -0,0 +1,72 @@
+// Copyright © 2024, SAS Institute Inc., Cary, NC, USA.  All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+import { commands } from "vscode";
+
+import { SASPYSession } from ".";
+import { LogLine, getSession } from "..";
+import { useRunStore } from "../../store";
+
+class CodeRunner {
+  public async runCode(
+    code: string,
+    startTag: string = "",
+    endTag: string = "",
+  ): Promise<string> {
+    // If we're already executing code, lets wait for it
+    // to finish up.
+    let unsubscribe;
+    if (useRunStore.getState().isExecutingCode) {
+      await new Promise((resolve) => {
+        unsubscribe = useRunStore.subscribe(
+          (state) => state.isExecutingCode,
+          (isExecutingCode) => !isExecutingCode && resolve(true),
+        );
+      });
+    }
+
+    const { setIsExecutingCode } = useRunStore.getState();
+    setIsExecutingCode(true);
+    commands.executeCommand("setContext", "SAS.running", true);
+    const session = getSession();
+
+    let logText = "";
+    const onExecutionLogFn = session.onExecutionLogFn;
+    const outputLines = [];
+    const addLine = (logLines: LogLine[]) =>
+      outputLines.push(...logLines.map(({ line }) => line));
+
+    try {
+      await session.setup(true);
+
+      // Lets capture output to use it on
+      session.onExecutionLogFn = addLine;
+
+      // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+      await (session as SASPYSession).run(code);
+
+      const logOutput = outputLines.filter((line) => line.trim()).join("");
+
+      logText =
+        startTag && endTag
+          ? logOutput
+              .slice(
+                logOutput.lastIndexOf(startTag),
+                logOutput.lastIndexOf(endTag),
+              )
+              .replace(startTag, "")
+              .replace(endTag, "")
+          : logOutput;
+    } finally {
+      unsubscribe && unsubscribe();
+      // Lets update our session to write to the log
+      session.onExecutionLogFn = onExecutionLogFn;
+
+      setIsExecutingCode(false);
+      commands.executeCommand("setContext", "SAS.running", false);
+    }
+
+    return logText;
+  }
+}
+
+export default CodeRunner;