![](\"website/static/images/featuresGlance.png\"/)
\n\n## Support\n\n### FAQs\n\nPlease check the [FAQ](https://sassoftware.github.io/vscode-sas-extension/faq) page for some common questions.\n\n### SAS Programming Documentation\n\n[SAS Programming documentation](https://go.documentation.sas.com/doc/en/pgmsascdc/v_048/lepg/titlepage.htm)\n\n### SAS Communities\n\nFor usage questions, tips, and workarounds, interact with other SAS users to ask questions and get answers on the [SAS Programmers Community site](https://communities.sas.com/t5/SAS-Programming/bd-p/programming).\n\n### SAS Technical Support\n\nSAS Technical Support provides standard support for the current release of the [SAS Extension for Visual Studio Code](https://marketplace.visualstudio.com/items?itemName=SAS.sas-lsp) available through the [Visual Studio Marketplace](https://marketplace.visualstudio.com/items), in accordance with the [Policies for All SAS Products | SAS Support](https://support.sas.com/en/technical-support/services-policies/policies-for-sas-products.html). If you have been unable to solve a problem or find answers using self‑help resources, open a case in the [Customer Service Portal](https://service.sas.com/csm) to get technical support for the SAS Extension for VS Code.\n\n### Reporting Issues\n\nSubmit a [GitHub issue](https://github.com/sassoftware/vscode-sas-extension/issues) for tracking bugs, feature requests, or questions regarding open‑source contributions.\n\n## Contributing to the SAS Extension\n\nWe welcome your contributions! Please read [CONTRIBUTING.md](/CONTRIBUTING.md) for details on how to submit contributions to this project.\n\n## License\n\nThis project is subject to the Apache License Version 2.0, a copy of which is included as [LICENSE](LICENSE)\n"
},
{
"path": "SUPPORT.md",
"content": "## Support\n\nSupport is provided through multiple channels, depending on your situation:\n\n- For usage questions, tips, and workarounds, interact with other SAS users to ask questions and get answers on the [SAS Programmers Community site](https://communities.sas.com/t5/SAS-Programming/bd-p/programming).\n\n- SAS Technical Support provides standard support for the current release of the [SAS Extension for Visual Studio Code](https://marketplace.visualstudio.com/items?itemName=SAS.sas-lsp) available through the [Visual Studio Marketplace](https://marketplace.visualstudio.com), in accordance with the [Policies for All SAS Products | SAS Support](https://support.sas.com/en/technical-support/services-policies/policies-for-sas-products.html). If you have been unable to solve a problem or find answers using self‑help resources, open a case in the [Customer Service Portal](https://service.sas.com/csm) to get technical support for the SAS Extension for VS Code.\n\n- Submit a [GitHub issue](https://github.com/sassoftware/vscode-sas-extension/issues) for tracking bugs, feature requests, or questions regarding open‑source contributions.\n"
},
{
"path": "client/package.json",
"content": "{\n \"name\": \"sas-lsp-client\",\n \"description\": \"VS Code client for SAS language server\",\n \"author\": \"SAS Institute Inc.\",\n \"license\": \"Apache-2.0\",\n \"version\": \"0.0.1\",\n \"publisher\": \"SAS\",\n \"engines\": {\n \"vscode\": \"^1.89.0\"\n },\n \"dependencies\": {\n \"ag-grid-community\": \"^35.2.1\",\n \"ag-grid-react\": \"^35.2.1\",\n \"axios\": \"^1.15.2\",\n \"highlight.js\": \"^11.11.1\",\n \"katex\": \"^0.16.45\",\n \"marked\": \"^17.0.5\",\n \"marked-katex-extension\": \"^5.1.7\",\n \"media-typer\": \"^1.1.0\",\n \"react\": \"^19.2.5\",\n \"react-dom\": \"^19.2.5\",\n \"ssh2\": \"^1.17.0\",\n \"uuid\": \"^14.0.0\",\n \"vscode-languageclient\": \"^10.0.0-next.2\",\n \"zustand\": \"^5.0.12\"\n },\n \"devDependencies\": {\n \"@types/react\": \"^19.2.14\",\n \"@types/react-dom\": \"^19.2.3\",\n \"@types/ssh2\": \"^1.15.5\",\n \"@types/uuid\": \"^11.0.0\",\n \"@types/vscode\": \"1.82.0\",\n \"@types/vscode-notebook-renderer\": \"^1.72.4\",\n \"@vscode/test-electron\": \"^2.5.2\"\n }\n}\n"
},
{
"path": "client/src/browser/extension.ts",
"content": "// Copyright © 2022, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { ExtensionContext, Uri, commands } from \"vscode\";\nimport { LanguageClientOptions } from \"vscode-languageclient\";\nimport { LanguageClient } from \"vscode-languageclient/browser\";\n\nlet client: LanguageClient;\n\n// this method is called when vs code is activated\nexport function activate(context: ExtensionContext): void {\n commands.executeCommand(\"setContext\", \"SAS.hideRunMenuItem\", true);\n\n // Options to control the language client\n const clientOptions: LanguageClientOptions = {\n // Register the server for sas file\n documentSelector: [{ language: \"sas\" }],\n };\n\n client = createWorkerLanguageClient(context, clientOptions);\n\n client.start();\n}\n\nfunction createWorkerLanguageClient(\n context: ExtensionContext,\n clientOptions: LanguageClientOptions,\n) {\n // Create a worker. The worker main file implements the language server.\n const serverMain = Uri.joinPath(\n context.extensionUri,\n \"server/dist/browser/server.js\",\n );\n const worker = new Worker(serverMain.toString());\n\n // create the language server client to communicate with the server running in the worker\n return new LanguageClient(\n \"sas-lsp\",\n \"SAS Language Server\",\n worker,\n clientOptions,\n );\n}\n\nexport function deactivate(): Thenable\\s*.*)/;\n const match = codeLine.match(capturingRegExp);\n\n return match !== null && match.groups !== undefined\n ? {\n lineNumber: parseInt(match.groups.lineNum),\n code: match.groups.code,\n }\n : null;\n}\n\nfunction getLastElement(arr: T[]): T | null {\n const lastIndex = arr.length - 1;\n return lastIndex >= 0 ? arr[lastIndex] : null;\n}\n\n/* there are two kind of beginning log lines indicating new problem\n 1) problem with this kind of beginning has location indicator in one of previous log line.\n \"22\" will be extracted as problem number in below log line.\n \"ERROR 22-322: Syntax error, expecting one of the following: ;, CANCEL, \"\n 2) problem with this kind of beginning has no location indicator and it is located to the previous closest source code.\n \"undefined\" will be returned in below log line, which means a general problem location will be used.\n \"WARNING: Variable POP_100 not found in data set WORK.UNIVOUT.\"\n*/\nfunction getProblemNumberFromProblemLogLine(\n logLine: string,\n): string | undefined {\n const regExp = /(?<=^(warning|error)\\s+)\\d+(?=-\\d+:\\s.*)/gi;\n const match = logLine.match(regExp);\n return match?.[0] ?? undefined;\n}\n\nfunction processLocations(\n line: string,\n from: \"indicator\" | \"problemNumber\",\n): ProblemLocation[] | null {\n switch (from) {\n case \"indicator\":\n return processLocationsFromIndicatorLine(line);\n case \"problemNumber\":\n return processLocationFromProblemNumberLine(line);\n }\n}\n\nfunction processLocationsFromIndicatorLine(\n indicatorLogLine: string,\n): ProblemLocation[] | null {\n // example: \" ---- --- --\"\n const regExp = /[-_]+/g;\n const locations = Array.from(indicatorLogLine.matchAll(regExp), (match) => ({\n startColumn: match.index,\n endColumn: match.index + match[0].length,\n }));\n\n return locations.length === 0 ? null : locations;\n}\n\n// problem number log line example:\n// 22 79\nfunction processLocationFromProblemNumberLine(\n problemNumberLine: string,\n): ProblemLocation[] | null {\n const regExp = /\\d+/g;\n const locations: ProblemLocation[] = Array.from(\n problemNumberLine.matchAll(regExp),\n (match): ProblemLocation => ({\n startColumn: match.index,\n problemNumber: match[0],\n }),\n );\n\n return locations.length === 0 ? null : locations;\n}\n\nfunction reviseLocationsFromIndicatorLine(\n indicatorLogLine: string,\n baseLocations: ProblemLocation[],\n): ProblemLocation[] {\n const arr = Array.from(indicatorLogLine);\n baseLocations.forEach((location) => {\n arr[location.startColumn] = \"=\";\n });\n\n const regExp = /=[-_]*/g;\n const locations = Array.from(arr.join(\"\").matchAll(regExp), (match) => ({\n startColumn: match.index,\n endColumn: match.index + match[0].length,\n }));\n\n return locations.length === 0 ? null : locations;\n}\n\nfunction processRawLocationDescs(\n rawLocationDescs: RawLocationDesc[],\n offset: LocationOffset,\n): ProblemLocation[] {\n const locations: ProblemLocation[] = [];\n rawLocationDescs.forEach((rawLocationDesc) =>\n locations.push(...processRawLocationDesc(rawLocationDesc, offset)),\n );\n return locations;\n}\n\nfunction processSourceCodeLines(\n sourceCodeLines: string[],\n columnOffset: number,\n): { lineNumber?: number; columnCorrection?: number } {\n if (!sourceCodeLines || sourceCodeLines.length === 0) {\n return {};\n }\n\n const lineNumber = decomposeCodeLogLine(sourceCodeLines[0]).lineNumber;\n\n // the source code log line may be separated into multiple lines, and the error occurred at the latter line,\n // in this case, must take account of the previous lines length to make start column correct.\n const columnCorrection = sourceCodeLines.reduce(\n (accumulator, line, index, lines) => {\n // it is possible the beginning and end spaces are trimmed, which makes the column correction wrong.\n // so in such case, 1 additional column (suppose only one space between two lines) will be considered.\n const isLastLineEndWithSpace =\n index === 0 || lines[index - 1].endsWith(\" \");\n const isCurrentLineStartWithSpace = lines[index]\n .substring(columnOffset)\n .startsWith(\" \");\n // excludes the length of last line\n const currentLineLength =\n lines.length - index === 1 ? 0 : line.length - columnOffset;\n\n return (\n accumulator +\n currentLineLength +\n (!(isLastLineEndWithSpace || isCurrentLineStartWithSpace) ? 1 : 0)\n );\n },\n 0,\n );\n\n return { lineNumber, columnCorrection };\n}\n\nfunction processRawLocationDesc(\n rawLocationDesc: RawLocationDesc,\n offset: LocationOffset,\n): ProblemLocation[] {\n const { lineNumber, columnCorrection } = processSourceCodeLines(\n rawLocationDesc.sourceCodeLines,\n offset.columnOffset,\n );\n\n if (lineNumber === undefined || columnCorrection === undefined) {\n return [];\n }\n\n const locations: ProblemLocation[] = [];\n const groups = rawLocationDesc.rawLocationGroups;\n groups.forEach((group) => {\n locations.push(...processRawLocationGroup(group));\n });\n\n locations.map((location) => {\n location.lineNumber = lineNumber;\n location.startColumn =\n location.startColumn - offset.columnOffset + columnCorrection;\n location.endColumn =\n location.endColumn - offset.columnOffset + columnCorrection;\n });\n\n return locations;\n}\n\n/* \n to handle the case below:\n\n 87 data CUSTOMERS (label=\"Customer data for geocoding\");\n 88 infile datalines dlm=#' dlm=#' dlm=#;\n _ _\n 24 24\n 24 24\n ERROR 24-322: Variable name is not valid.\n ERROR 24-2: Invalid value for the DLM option.\n 89 length address $ 24 city $ 24 state $ 2;\n\n adjust the item order in list make the items which have same startColumn placed continuously\n*/\nfunction adjustAppearanceOrder(\n locations: ProblemLocation[],\n): ProblemLocation[] {\n if (locations.length < 3) {\n return locations;\n }\n\n const findAllMatched = (startColumn: number, problemNumber: string) => {\n const found = [];\n locations.map((location, index) => {\n if (\n location &&\n location.startColumn === startColumn &&\n location.problemNumber === problemNumber\n ) {\n locations[index] = undefined;\n found.push(location);\n }\n });\n return found;\n };\n\n const orderedLocations: ProblemLocation[] = [];\n locations.forEach((location, index) => {\n if (!location) {\n return;\n }\n orderedLocations.push(location);\n locations[index] = undefined;\n const found = findAllMatched(location.startColumn, location.problemNumber);\n orderedLocations.push(...found);\n });\n\n return orderedLocations;\n}\n\nfunction processRawLocationGroup(\n group: RawLocationGroup,\n): ProblemLocation[] | null {\n if (\n !group.indicatorLine ||\n !group.problemNumberLines ||\n group.problemNumberLines.length === 0\n ) {\n return null;\n }\n\n let indicatorLocations = processLocations(group.indicatorLine, \"indicator\");\n\n const problemNumberLocations = processLocations(\n group.problemNumberLines[0],\n \"problemNumber\",\n );\n\n // to handle this case:\n // _____\n // 1 22\n if (problemNumberLocations.length > indicatorLocations.length) {\n indicatorLocations = reviseLocationsFromIndicatorLine(\n group.indicatorLine,\n problemNumberLocations,\n );\n }\n\n // TODO\n if (problemNumberLocations.length < indicatorLocations.length) {\n // is it possible to have such case:\n // - -\n // 221\n // is it possible to have such case, below includes two problems:\n // ---\n // 221\n }\n\n const locations: ProblemLocation[] = [];\n group.problemNumberLines.forEach((line) => {\n const locationsWithProblemNumber = processLocations(line, \"problemNumber\");\n locationsWithProblemNumber.forEach((location1) => {\n const found = indicatorLocations.find(\n (location2) => location2.startColumn === location1.startColumn,\n );\n\n if (found) {\n locations.push({\n startColumn: found.startColumn,\n endColumn: found.endColumn,\n problemNumber: location1.problemNumber,\n });\n }\n });\n });\n\n return adjustAppearanceOrder(locations);\n}\n\nfunction processGeneralLocation(\n sourceCodeLines: string[],\n offset: LocationOffset,\n): ProblemLocation | null {\n if (!sourceCodeLines || sourceCodeLines.length === 0 || !offset) {\n return null;\n }\n\n const lineNumber = decomposeCodeLogLine(sourceCodeLines[0]).lineNumber;\n const wholeLine = sourceCodeLines.reduce(\n (accumulator, line) => accumulator + line.substring(offset.columnOffset),\n );\n\n const startColumn = getFirstCharacterIndex(wholeLine) - offset.columnOffset;\n const endColumn =\n startColumn + wholeLine.substring(startColumn).trim().length - 1;\n return { lineNumber, startColumn, endColumn };\n}\n\nfunction getFirstCharacterIndex(logLine: string): number {\n if (logLine.trim() === \"\") {\n return -1;\n }\n\n /*\n below are 3 cases that source code may be presented in log lines.\n 8 ...\n \n 8 ! ...\n \n 12! ...\n */\n const regExp = /(?<=^\\d+\\s*!?\\s+)[^\\s!]/;\n const match = logLine.match(regExp);\n\n return match === null ? -1 : match.index;\n}\n\n/* \n below are 3 cases that source code may be presented in log lines.\n 8 ...\n\n 8 ! ...\n\n 8! ...\n*/\nexport function isSourceCodeLineAfterLineWrapping(logLine: string): boolean {\n return /^(?\\d+)\\s*!\\s(?\\s*.*)/.test(logLine);\n}\n"
},
{
"path": "client/src/components/logViewer/index.ts",
"content": "// Copyright © 2024, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport {\n DocumentSemanticTokensProvider,\n OutputChannel,\n SemanticTokensBuilder,\n l10n,\n window,\n} from \"vscode\";\n\nimport type { OnLogFn } from \"../../connection\";\nimport { useLogStore, useRunStore } from \"../../store\";\nimport { logSelectors, runSelectors } from \"../../store/selectors\";\nimport {\n clearLogOnExecutionStart,\n showLogOnExecutionFinish,\n showLogOnExecutionStart,\n} from \"../utils/settings\";\n\nconst { setProducedExecutionLogOutput } = useLogStore.getState();\n\nlet outputChannel: OutputChannel;\nlet data: string[] = [];\nlet fileName = \"\";\n\nexport const legend = {\n tokenTypes: [\"error\", \"warning\", \"note\"],\n tokenModifiers: [],\n};\n\nexport const LogTokensProvider: DocumentSemanticTokensProvider = {\n provideDocumentSemanticTokens: (document) => {\n if (document.getText() === \"\") {\n data = [];\n }\n const tokensBuilder = new SemanticTokensBuilder(legend);\n\n for (let i = 0; i < data.length; i++) {\n if (legend.tokenTypes.includes(data[i])) {\n tokensBuilder.push(document.lineAt(i).range, data[i]);\n }\n }\n return tokensBuilder.build();\n },\n};\n\n/**\n * Handles log lines generated for the SAS session startup.\n * @param logs array of log lines to write.\n */\nexport const appendSessionLogFn: OnLogFn = (logLines) => {\n appendLogLines(logLines);\n};\n\n/**\n * Handles log lines generated for the SAS session execution.\n * @param logs array of log lines to write.\n */\nexport const appendExecutionLogFn: OnLogFn = (logLines) => {\n appendLogLines(logLines);\n\n if (!useLogStore.getState().producedExecutionOutput) {\n setProducedExecutionLogOutput(true);\n }\n};\n\nexport const appendLogToken = (type: string): void => {\n data.push(type);\n};\n\nexport const setFileName = (name: string) => {\n fileName = name;\n};\n\nconst appendLogLines: OnLogFn = (logs) => {\n if (!outputChannel) {\n const name = clearLogOnExecutionStart()\n ? l10n.t(\"SAS Log: {name}\", { name: fileName })\n : l10n.t(\"SAS Log\");\n outputChannel = window.createOutputChannel(name, \"sas-log\");\n }\n for (const line of logs) {\n line.line\n .trimEnd()\n .split(\"\\n\")\n .forEach((text) => {\n appendLogToken(line.type);\n outputChannel.appendLine(text);\n });\n }\n};\n\nuseLogStore.subscribe(\n logSelectors.selectProducedExecutionOutput,\n (producedOutput, prevProducedOutput) => {\n if (producedOutput && !prevProducedOutput) {\n if (showLogOnExecutionStart()) {\n outputChannel?.show(true);\n }\n }\n },\n);\n\nuseRunStore.subscribe(\n runSelectors.selectIsExecutingCode,\n (isExecuting, prevIsExecuting) => {\n if (\n !isExecuting &&\n prevIsExecuting &&\n useLogStore.getState().producedExecutionOutput\n ) {\n if (showLogOnExecutionFinish()) {\n outputChannel?.show(true);\n }\n } else if (isExecuting && !prevIsExecuting) {\n setProducedExecutionLogOutput(false);\n\n if (\n clearLogOnExecutionStart() &&\n outputChannel &&\n useRunStore.getState().isUserExecuting\n ) {\n outputChannel.dispose();\n outputChannel = undefined;\n data = [];\n }\n }\n },\n);\n"
},
{
"path": "client/src/components/logViewer/logParser.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { LogLine } from \"../../connection\";\nimport {\n LocationOffset,\n Problem,\n ProblemProcessor,\n decomposeCodeLogLine,\n isSourceCodeLineAfterLineWrapping,\n} from \"./ProblemProcessor\";\n\nexport function parseLog(\n logs: LogLine[],\n logStartFlag: string,\n): [Problem[], string[]] {\n if (logs.length === 0 || logStartFlag.trim() === \"\") {\n return [[], []];\n }\n\n // logs cleaning\n const latestLogs = getTheLatestLogs(logs, logStartFlag);\n const problemRelatedLogs = getProblemRelatedLogs(latestLogs);\n\n let problemProcessor: ProblemProcessor = new ProblemProcessor();\n let offset: LocationOffset;\n const problems: Problem[] = [];\n const codeLines: string[] = [];\n\n problemRelatedLogs.forEach((logLine) => {\n if (isProblemTypeLog(logLine)) {\n if (isNewProblemLogLine(logLine.line)) {\n problemProcessor.addProblemLogLine(logLine);\n return;\n }\n\n if (isLocationIndicatorLogLine(logLine.line)) {\n problemProcessor.addLocationIndicatorLogLine(logLine);\n return;\n }\n\n if (isProblemNumberLogLine(logLine.line)) {\n problemProcessor.addProblemNumberLogLine(logLine);\n return;\n }\n\n problemProcessor.appendProblemLogLine(logLine);\n return;\n } else {\n codeLines.push(logLine.line);\n\n if (!isValidSourceCodeLog(logLine)) {\n return;\n }\n\n const currentSourceCodeLine = logLine.line;\n if (!offset) {\n offset = calculateLocationOffset(currentSourceCodeLine, logStartFlag);\n }\n\n const isWrappedLine = isSourceCodeLineAfterLineWrapping(\n currentSourceCodeLine,\n );\n\n const previousSourceCodeLines = problemProcessor.getSourceCodeLines();\n const isSameAsPrevious = areSameLines(\n currentSourceCodeLine,\n previousSourceCodeLines[previousSourceCodeLines.length - 1],\n );\n\n if (problemProcessor.isReady() && !isSameAsPrevious) {\n problems.push(...problemProcessor.processProblems(offset));\n const unclaimedLocations = problemProcessor.getUnclaimedLocations();\n problemProcessor = isWrappedLine\n ? new ProblemProcessor(previousSourceCodeLines, unclaimedLocations)\n : new ProblemProcessor(undefined, unclaimedLocations);\n }\n\n problemProcessor.setSourceCodeLine(currentSourceCodeLine);\n }\n });\n\n if (problemProcessor.isReady()) {\n problems.push(...problemProcessor.processProblems(offset));\n }\n\n problemProcessor = null;\n\n return [problems, cleanCodeLines(codeLines)];\n}\n\nfunction getTheLatestLogs(logs: LogLine[], firstCodeLine: string): LogLine[] {\n let beginningIndex = -1;\n logs.forEach((logLine, index) => {\n if (logLine.type !== \"source\") {\n return;\n }\n\n const code = decomposeCodeLogLine(logLine.line)?.code ?? null;\n if (code !== null && firstCodeLine === code.trim()) {\n beginningIndex = index;\n }\n });\n\n return beginningIndex === -1 ? [] : logs.slice(beginningIndex);\n}\n\nfunction getProblemRelatedLogs(logs: LogLine[]): LogLine[] {\n return logs.filter((logLine) => {\n return [\"error\", \"warning\", \"source\"].includes(logLine.type);\n });\n}\n\nfunction calculateLocationOffset(\n codeLogLine: string,\n firstCodeLine: string,\n): { columnOffset: number; lineOffset: number } {\n const codeInfo = decomposeCodeLogLine(codeLogLine);\n\n // there may be a log kept when finishing running selected code,\n // the kept log will be sent out in following running code.\n // that log line number should not be used for calculating line offset.\n if (codeInfo === null || firstCodeLine !== codeInfo.code.trim()) {\n return { lineOffset: -1, columnOffset: -1 };\n }\n\n const lineOffset = codeInfo.lineNumber;\n const columnOffset = codeLogLine.indexOf(firstCodeLine);\n\n return { columnOffset, lineOffset };\n}\n\nfunction isProblemTypeLog(logLine: LogLine): boolean {\n return logLine.type === \"error\" || logLine.type === \"warning\";\n}\n\nfunction isSourceTypeLog(logLine: LogLine): boolean {\n return logLine.type === \"source\";\n}\n\nfunction isEmptyCodeLogLine(logLine: string): boolean {\n return /^\\d+\\s*$/.test(logLine);\n}\n\n/* there are two kind of beginning log lines indicating new problem\n 1) problem with this kind of beginning has location indicator in one of previous log line.\n \"ERROR 22-322: Syntax error, expecting one of the following: ;, CANCEL, \"\n 2) problem with this kind of beginning has no location indicator and it is located to the previous closest source code\n \"WARNING: Variable POP_100 not found in data set WORK.UNIVOUT.\"\n*/\nfunction isNewProblemLogLine(line: string): boolean {\n return /^(?error|warning)(?\\s*\\d+-\\d+)?:\\s(?.*)/i.test(\n line,\n );\n}\n\n/*\nthe continuous hyphens/underscores means a location indicator. \nin below logs, the 2nd & 5th lines are location indicator log lines.\nbelow are part of logs:\n18 call call symputx('mac', quote(strip(emple))); \n ------- -\n 22 79\n 68\n ----\n 251\n*/\nfunction isLocationIndicatorLogLine(logLine: string): boolean {\n return /^(?\\s+)(?[-_]+\\s*)+$/.test(logLine);\n}\n\n/*\nthe number below the continuous hyphens/underscores means a problem number.\nin below logs, the 3rd & 4th & 6th lines are problem number log lines.\nbelow are part of logs:\n18 call call symputx('mac', quote(strip(emple))); \n ------- -\n 22 79\n 68\n ----\n 251\n*/\nfunction isProblemNumberLogLine(line: string): boolean {\n return /^(?\\s+)(?\\d+\\s*)+$/.test(line);\n}\n\nfunction isValidSourceCodeLog(logLine: LogLine): boolean {\n return isSourceTypeLog(logLine) && !isEmptyCodeLogLine(logLine.line);\n}\n\nfunction areSameLines(line1: string, line2: string) {\n return line1 === line2;\n}\n\n// keep each line number only once\nfunction cleanCodeLines(codeLines: string[]): string[] {\n if (codeLines.length === 0) {\n return codeLines;\n }\n\n let previousLineNumber = -1;\n const result = [];\n codeLines.forEach((line) => {\n const lineNumber = decomposeCodeLogLine(line)?.lineNumber ?? -1;\n if (lineNumber === previousLineNumber) {\n return;\n } else {\n previousLineNumber = lineNumber;\n result.push(line);\n }\n });\n\n return result;\n}\n"
},
{
"path": "client/src/components/logViewer/sasDiagnostics.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport {\n Diagnostic,\n DiagnosticCollection,\n DiagnosticSeverity,\n Disposable,\n Range,\n Uri,\n commands,\n languages,\n window,\n workspace,\n} from \"vscode\";\n\nimport { LogLine, OnLogFn } from \"../../connection\";\nimport { useRunStore } from \"../../store\";\nimport { runSelectors } from \"../../store/selectors\";\nimport { SASCodeDocument } from \"../utils/SASCodeDocument\";\nimport { isShowProblemsFromSASLogEnabled } from \"../utils/settings\";\nimport { DiagnosticCodeActionProvider } from \"./DiagnosticCodeActionProvider\";\nimport { Problem } from \"./ProblemProcessor\";\nimport { parseLog } from \"./logParser\";\n\nexport const diagnosticSource = \"sas log\";\n\nlet diagnosticCollection: DiagnosticCollection;\n\nenum DiagnosticCommands {\n IgnoreCommand = \"SAS.diagnostic.ignore\",\n IgnoreAllWarningCommand = \"SAS.diagnostic.ignoreAllWarning\",\n IgnoreAllErrorCommand = \"SAS.diagnostic.ignoreAllError\",\n IgnoreAllCommand = \"SAS.diagnostic.ignoreAll\",\n}\n\nfunction ignore(diagnosticsToRemove: Diagnostic[], uri: Uri): void {\n const diagnostics = getSasDiagnosticCollection().get(uri);\n const newDiagnostics = diagnostics.filter((diagnostic) => {\n return !diagnosticsToRemove.includes(diagnostic);\n });\n\n getSasDiagnosticCollection().set(uri, newDiagnostics);\n}\n\nfunction ignoreAll(uri: Uri, severity?: DiagnosticSeverity): void {\n if (severity === undefined) {\n getSasDiagnosticCollection().delete(uri);\n } else {\n const diagnostics = getSasDiagnosticCollection().get(uri);\n const newDiagnostics = diagnostics.filter((diagnostic) => {\n return diagnostic.severity !== severity;\n });\n getSasDiagnosticCollection().set(uri, newDiagnostics);\n }\n}\n\nfunction updateDiagnosticUri(oldUri: Uri, newUri: Uri): void {\n const diagnosticCollection = getSasDiagnosticCollection();\n const diagnostics = diagnosticCollection.get(oldUri);\n diagnosticCollection.delete(oldUri);\n diagnosticCollection.set(newUri, diagnostics);\n}\n\nfunction getSasDiagnosticCollection(): DiagnosticCollection {\n if (diagnosticCollection === undefined) {\n diagnosticCollection = languages.createDiagnosticCollection(\"sas\");\n }\n return diagnosticCollection;\n}\n\nasync function updateDiagnostics(\n logs: LogLine[],\n codeDoc: SASCodeDocument,\n): Promise {\n if (!isShowProblemsFromSASLogEnabled()) {\n getSasDiagnosticCollection().clear();\n return;\n }\n\n const [problems, codeLinesInLog] = parseLog(\n logs,\n codeDoc.wrappedCodeLineAt(0),\n );\n\n await updateProblemLocation(problems, codeDoc, codeLinesInLog);\n\n const problemsWithValidLocation = problems.filter((problem) => {\n const { lineNumber, startColumn, endColumn } = problem;\n return lineNumber * startColumn * endColumn >= 0;\n });\n\n const diagnostics = constructDiagnostics(problemsWithValidLocation);\n\n getSasDiagnosticCollection().set(\n Uri.parse(codeDoc.getUri()),\n diagnostics.length > 0 ? diagnostics : undefined,\n );\n}\n\nasync function updateProblemLocation(\n problems: Problem[],\n codeDoc: SASCodeDocument,\n codeLinesInLog: string[],\n) {\n for (const problem of problems) {\n const { lineNumber, startColumn, endColumn } = problem;\n const {\n lineNumber: actualLineNumber,\n startColumn: actualStartColumn,\n endColumn: actualEndColumn,\n } = await codeDoc.getLocationInRawCode(\n {\n lineNumber,\n startColumn,\n endColumn,\n },\n codeLinesInLog,\n );\n\n problem.lineNumber = actualLineNumber;\n problem.startColumn = actualStartColumn;\n problem.endColumn = actualEndColumn;\n }\n}\n\nfunction constructDiagnostics(problems: Problem[]): Diagnostic[] {\n const diagnostics = problems.map((problem) => {\n const { lineNumber, startColumn, endColumn, message, type } = problem;\n const range = new Range(lineNumber, startColumn, lineNumber, endColumn);\n const diagnostic = new Diagnostic(\n range,\n message,\n type === \"error\" ? DiagnosticSeverity.Error : DiagnosticSeverity.Warning,\n );\n diagnostic.source = diagnosticSource;\n return diagnostic;\n });\n\n return diagnostics;\n}\n\nfunction generateLogFn(\n codeDoc: SASCodeDocument,\n originLogFn?: OnLogFn,\n): OnLogFn {\n const uri = codeDoc.getUri();\n if (uri === undefined || uri.trim() === \"\") {\n return originLogFn;\n }\n\n const receivedLogs = [];\n const additionalLogFn: OnLogFn = (logs) => {\n receivedLogs.push(...logs);\n };\n\n const unsubscribe = useRunStore.subscribe(\n runSelectors.selectIsExecutingCode,\n (isExecuting) => {\n if (!isExecuting) {\n updateDiagnostics(receivedLogs, codeDoc);\n unsubscribe();\n }\n },\n );\n\n return (logs) => {\n originLogFn?.(logs);\n additionalLogFn(logs);\n };\n}\n\nfunction getSubscriptions(): Disposable[] {\n return [\n getSasDiagnosticCollection(),\n\n commands.registerCommand(DiagnosticCommands.IgnoreCommand, ignore),\n commands.registerCommand(\n DiagnosticCommands.IgnoreAllWarningCommand,\n ignoreAll,\n ),\n commands.registerCommand(\n DiagnosticCommands.IgnoreAllErrorCommand,\n ignoreAll,\n ),\n commands.registerCommand(DiagnosticCommands.IgnoreAllCommand, ignoreAll),\n languages.registerCodeActionsProvider(\n \"sas\",\n new DiagnosticCodeActionProvider(),\n ),\n workspace.onDidRenameFiles((e) => {\n e.files.forEach((file) => updateDiagnosticUri(file.oldUri, file.newUri));\n }),\n\n workspace.onDidDeleteFiles((e) => {\n e.files.forEach((uri) => {\n ignoreAll(uri);\n });\n }),\n workspace.onDidSaveTextDocument((e) => {\n // the new file\n const uri = window.activeTextEditor.document.uri;\n const isNewFileSaved =\n uri.scheme === \"untitled\" && e.languageId === \"sas\";\n if (isNewFileSaved) {\n // clear diagnostics on new file\n ignoreAll(uri);\n // if the new file is saved, e indicates the file which new file saved to.\n // no matter if it override a existing file, it is ok to clear its diagnostics.\n ignoreAll(e.uri);\n }\n }),\n workspace.onDidCloseTextDocument((e) => {\n const uri = e.uri;\n // if the new file is saved, the onDidSaveTextDocument is invoked, then this is invoked.\n // if the new file is not saved, only this is invoked.\n const isNewFileClosed = uri.scheme === \"untitled\";\n if (isNewFileClosed) {\n ignoreAll(uri);\n }\n }),\n ];\n}\n\nexport const sasDiagnostic = {\n DiagnosticCommands,\n generateLogFn,\n getSubscriptions,\n updateDiagnosticUri,\n ignoreAll,\n};\n"
},
{
"path": "client/src/components/notebook/Controller.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport * as vscode from \"vscode\";\n\nimport { getSession } from \"../../connection\";\nimport { SASCodeDocument } from \"../utils/SASCodeDocument\";\nimport { getCodeDocumentConstructionParameters } from \"../utils/SASCodeDocumentHelper\";\nimport { Deferred, deferred } from \"../utils/deferred\";\n\nexport class NotebookController {\n readonly controllerId = \"sas-notebook-controller-id\";\n readonly notebookType = \"sas-notebook\";\n readonly label = \"SAS Notebook\";\n readonly supportedLanguages = [\"sas\", \"sql\", \"python\", \"r\"];\n\n private readonly _controller: vscode.NotebookController;\n private _executionOrder = 0;\n private _interrupted: Deferred | undefined;\n\n constructor() {\n this._controller = vscode.notebooks.createNotebookController(\n this.controllerId,\n this.notebookType,\n this.label,\n );\n\n this._controller.supportedLanguages = this.supportedLanguages;\n this._controller.supportsExecutionOrder = true;\n this._controller.executeHandler = this._execute.bind(this);\n this._controller.interruptHandler = this._interrupt.bind(this);\n }\n\n dispose(): void {\n this._controller.dispose();\n }\n\n private async _execute(cells: vscode.NotebookCell[]): Promise {\n this._interrupted = undefined;\n\n try {\n const session = getSession();\n await session.setup();\n } catch (err) {\n vscode.window.showErrorMessage(\n err.response?.data ? JSON.stringify(err.response.data) : err.message,\n );\n return;\n }\n\n for (const cell of cells) {\n await this._doExecution(cell);\n }\n }\n\n private async _doExecution(cell: vscode.NotebookCell): Promise {\n if (this._interrupted) {\n return;\n }\n\n const execution = this._controller.createNotebookCellExecution(cell);\n execution.executionOrder = ++this._executionOrder;\n execution.start(Date.now()); // Keep track of elapsed time to execute cell.\n execution.clearOutput();\n\n const session = getSession();\n session.onExecutionLogFn = (logLines) => {\n logs = logs.concat(logLines);\n };\n\n const parameters = getCodeDocumentConstructionParameters(cell.document);\n const codeDoc = new SASCodeDocument(parameters);\n\n let logs = [];\n try {\n const result = await session.run(codeDoc.getWrappedCode());\n\n execution.replaceOutput([\n new vscode.NotebookCellOutput([\n ...(result.html5?.length\n ? [\n vscode.NotebookCellOutputItem.text(\n result.html5,\n \"application/vnd.sas.ods.html5\",\n ),\n ]\n : []),\n vscode.NotebookCellOutputItem.json(\n logs,\n \"application/vnd.sas.compute.log.lines\",\n ),\n ]),\n ]);\n execution.end(true, Date.now());\n } catch (error) {\n execution.replaceOutput([\n new vscode.NotebookCellOutput([\n vscode.NotebookCellOutputItem.error(error),\n ]),\n ]);\n execution.end(false, Date.now());\n if (!this._interrupted) {\n this._interrupted = deferred();\n }\n }\n if (this._interrupted) {\n this._interrupted.resolve();\n }\n }\n\n private _interrupt() {\n if (this._interrupted) {\n return;\n }\n this._interrupted = deferred();\n vscode.window.withProgress(\n {\n location: vscode.ProgressLocation.Notification,\n title: vscode.l10n.t(\"Cancelling job...\"),\n },\n () => this._interrupted.promise,\n );\n const session = getSession();\n session.cancel?.();\n }\n}\n"
},
{
"path": "client/src/components/notebook/Serializer.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport * as vscode from \"vscode\";\n\nimport { TextDecoder, TextEncoder } from \"util\";\n\ninterface RawNotebookCell {\n language: string;\n value: string;\n kind: vscode.NotebookCellKind;\n outputs?: {\n items: {\n data: string;\n mime: string;\n }[];\n }[];\n}\n\nexport class NotebookSerializer implements vscode.NotebookSerializer {\n private readonly _decoder = new TextDecoder();\n private readonly _encoder = new TextEncoder();\n\n async deserializeNotebook(content: Uint8Array): Promise {\n const contents = this._decoder.decode(content);\n\n let raw: RawNotebookCell[];\n try {\n // eslint-disable-next-line @typescript-eslint/consistent-type-assertions\n raw = JSON.parse(contents);\n } catch {\n raw = [];\n }\n\n const cells = raw.map((item) => {\n const cell = new vscode.NotebookCellData(\n item.kind,\n item.value,\n item.language,\n );\n if (item.outputs) {\n cell.outputs = item.outputs.map(\n (output) =>\n new vscode.NotebookCellOutput(\n output.items.map((item) =>\n vscode.NotebookCellOutputItem.text(item.data, item.mime),\n ),\n ),\n );\n }\n return cell;\n });\n\n return new vscode.NotebookData(cells);\n }\n\n async serializeNotebook(data: vscode.NotebookData): Promise {\n const contents: RawNotebookCell[] = [];\n\n for (const cell of data.cells) {\n const content: RawNotebookCell = {\n kind: cell.kind,\n language: cell.languageId,\n value: cell.value,\n };\n if (cell.outputs) {\n content.outputs = cell.outputs.map((output) => ({\n items: output.items.map((item) => ({\n data: this._decoder.decode(item.data),\n mime: item.mime,\n })),\n }));\n }\n contents.push(content);\n }\n\n return this._encoder.encode(JSON.stringify(contents));\n }\n}\n"
},
{
"path": "client/src/components/notebook/exporters/index.ts",
"content": "// Copyright © 2025, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { Uri, l10n, window, workspace } from \"vscode\";\nimport type { LanguageClient } from \"vscode-languageclient/node\";\n\nimport path from \"path\";\n\nimport { exportToHTML } from \"./toHTML\";\nimport { exportToSAS } from \"./toSAS\";\n\nexport const exportNotebook = async (client: LanguageClient) => {\n const notebook = window.activeNotebookEditor?.notebook;\n\n if (!notebook) {\n return;\n }\n\n // Show QuickPick for export format selection\n const formatChoices = [\n {\n label: l10n.t(\"HTML\"),\n description: l10n.t(\"Export as HTML file\"),\n format: \"html\" as const,\n extension: \"html\",\n },\n {\n label: l10n.t(\"SAS Code\"),\n description: l10n.t(\"Export as SAS program file\"),\n format: \"sas\" as const,\n extension: \"sas\",\n },\n ];\n\n const formatChoice = await window.showQuickPick(formatChoices, {\n placeHolder: l10n.t(\"Select export format\"),\n ignoreFocusOut: true,\n });\n\n if (!formatChoice) {\n return;\n }\n\n // Show save dialog with appropriate file extension\n const defaultFileName =\n path.basename(notebook.uri.path, \".sasnb\") + `.${formatChoice.extension}`;\n\n const filters: { [name: string]: string[] } = {};\n filters[formatChoice.extension.toUpperCase()] = [formatChoice.extension];\n\n const uri = await window.showSaveDialog({\n filters,\n defaultUri: Uri.parse(defaultFileName),\n });\n\n if (!uri) {\n return;\n }\n\n try {\n let content: string | Uint8Array;\n\n // Generate content based on selected format\n switch (formatChoice.format) {\n case \"html\":\n content = await exportToHTML(notebook, client);\n await workspace.fs.writeFile(uri, new TextEncoder().encode(content));\n break;\n case \"sas\":\n content = exportToSAS(notebook);\n await workspace.fs.writeFile(uri, new TextEncoder().encode(content));\n break;\n }\n\n window.showInformationMessage(\n l10n.t(\"Notebook exported to {0}\", uri.fsPath),\n );\n } catch (error) {\n window.showErrorMessage(\n l10n.t(\"Failed to export notebook: {0}\", error.message || error),\n );\n }\n};\n\nexport const saveOutput = async () => {\n const notebook = window.activeNotebookEditor?.notebook;\n const activeCell = window.activeNotebookEditor?.selection?.start;\n\n if (!notebook || activeCell === undefined) {\n return;\n }\n\n const cell = notebook.cellAt(activeCell);\n if (!cell) {\n return;\n }\n\n let odsItem = null;\n let logItem = null;\n\n for (const output of cell.outputs) {\n if (!odsItem) {\n odsItem = output.items.find(\n (item) => item.mime === \"application/vnd.sas.ods.html5\",\n );\n }\n if (!logItem) {\n logItem = output.items.find(\n (item) => item.mime === \"application/vnd.sas.compute.log.lines\",\n );\n }\n\n if (odsItem && logItem) {\n break;\n }\n }\n\n const choices: Array<{\n label: string;\n outputType: \"html\" | \"log\";\n }> = [];\n\n if (odsItem) {\n choices.push({\n label: l10n.t(\"Save ODS HTML\"),\n outputType: \"html\",\n });\n }\n\n if (logItem) {\n choices.push({\n label: l10n.t(\"Save Log\"),\n outputType: \"log\",\n });\n }\n\n const exportChoice = await window.showQuickPick(choices, {\n placeHolder: l10n.t(\"Choose output type to save\"),\n ignoreFocusOut: true,\n });\n\n if (!exportChoice) {\n return;\n }\n\n let content = \"\";\n let fileExtension = \"\";\n let fileName = \"\";\n try {\n if (exportChoice.outputType === \"html\" && odsItem) {\n content = odsItem.data.toString();\n fileExtension = \"html\";\n fileName = `${path.basename(notebook.uri.path, \".sasnb\")}_${l10n.t(\"output\")}_${\n activeCell + 1\n }.html`;\n } else if (exportChoice.outputType === \"log\" && logItem) {\n const logs: Array<{ line: string; type: string }> = JSON.parse(\n logItem.data.toString(),\n );\n content = logs.map((log) => log.line).join(\"\\n\");\n fileExtension = \"log\";\n fileName = `${path.basename(notebook.uri.path, \".sasnb\")}_${l10n.t(\"output\")}_${\n activeCell + 1\n }.log`;\n }\n } catch (error) {\n window.showErrorMessage(\n l10n.t(\"Failed to extract output content.\" + error),\n );\n return;\n }\n\n const filters: { [name: string]: string[] } = {};\n filters[fileExtension.toUpperCase()] = [fileExtension];\n\n const uri = await window.showSaveDialog({\n filters,\n defaultUri: Uri.parse(fileName),\n });\n\n if (!uri) {\n return;\n }\n\n await workspace.fs.writeFile(uri, new TextEncoder().encode(content));\n\n window.showInformationMessage(l10n.t(\"Saved to {0}\", uri.fsPath));\n};\n"
},
{
"path": "client/src/components/notebook/exporters/templates/dark.css",
"content": "body {\n background-color: #03233a;\n color: #cccccc;\n --border-color: #b0b7bb;\n --highlight-background-color: #9fb6c633;\n}\n.markdown-cell a:link {\n color: #3794ff;\n}\n.hljs {\n background: #021727;\n color: #dddfe4;\n}\n\n/* Syntax highlighting */\n.hljs-operator,\n.hljs-punctuation {\n color: #dddfe4;\n}\n\n.hljs-keyword {\n color: #4398f9;\n}\n\n.hljs-comment {\n color: #97c03f;\n}\n\n.hljs-string {\n color: #f17e70;\n}\n\n.hljs-number {\n color: #54b6a4;\n}\n\n.hljs-title {\n color: #92c3fc;\n font-weight: bold;\n}\n\n.hljs-built_in {\n color: #faaa6b;\n}\n\n/* SAS Syntax */\n.sas-syntax-sep {\n color: #dddfe4;\n}\n\n.sas-syntax-keyword,\n.sas-syntax-macro-keyword {\n color: #4398f9;\n}\n\n.sas-syntax-sec-keyword,\n.sas-syntax-proc-name {\n color: #92c3fc;\n font-weight: bold;\n}\n\n.sas-syntax-comment,\n.sas-syntax-macro-comment {\n color: #97c03f;\n}\n\n.sas-syntax-macro-ref,\n.sas-syntax-macro-sec-keyword {\n color: #dddfe4;\n font-weight: bold;\n}\n\n.sas-syntax-macro-keyword-param {\n color: #97c6fc;\n}\n\n.sas-syntax-cards-data {\n color: #faaa6b;\n}\n\n.sas-syntax-string {\n color: #f17e70;\n}\n\n.sas-syntax-date,\n.sas-syntax-time,\n.sas-syntax-dt,\n.sas-syntax-bitmask {\n color: #54b6a4;\n font-weight: bold;\n}\n\n.sas-syntax-namelit {\n color: #f17e70;\n font-weight: bold;\n}\n\n.sas-syntax-hex,\n.sas-syntax-numeric {\n color: #54b6a4;\n font-weight: bold;\n}\n\n.sas-syntax-format {\n color: #54b6a4;\n}\n\n.log-line.sas-log-error {\n color: #ff4d4d;\n}\n\n.log-line.sas-log-warning {\n color: #ffb829;\n}\n\n.log-line.sas-log-note {\n color: #2fa8fe;\n}\n"
},
{
"path": "client/src/components/notebook/exporters/templates/default.html",
"content": "\n\n \n \n \n \n \n \n ${content}\n \n\n"
},
{
"path": "client/src/components/notebook/exporters/templates/light.css",
"content": "body {\n color: #3b3b3b;\n --border-color: #b0b7bb;\n --highlight-background-color: #818b981f;\n}\n.hljs {\n background: #fff;\n}\n\n/* Syntax highlighting */\n.hljs-operator,\n.hljs-punctuation {\n color: #1b1d22;\n}\n\n.hljs-keyword {\n color: #3578c5;\n}\n\n.hljs-comment {\n color: #647f29;\n}\n\n.hljs-string {\n color: #8f4238;\n}\n\n.hljs-number {\n color: #3c8275;\n}\n\n.hljs-title {\n color: #224c7c;\n font-weight: bold;\n}\n\n.hljs-built_in {\n color: #aa0d91;\n}\n\n/* SAS Syntax */\n.sas-syntax-sep {\n color: #1b1d22;\n}\n\n.sas-syntax-keyword,\n.sas-syntax-macro-keyword {\n color: #3578c5;\n}\n\n.sas-syntax-sec-keyword,\n.sas-syntax-proc-name {\n color: #224c7c;\n font-weight: bold;\n}\n\n.sas-syntax-comment,\n.sas-syntax-macro-comment {\n color: #647f29;\n}\n\n.sas-syntax-macro-ref,\n.sas-syntax-macro-sec-keyword {\n color: #1b1d22;\n font-weight: bold;\n}\n\n.sas-syntax-macro-keyword-param {\n color: #054894;\n}\n\n.sas-syntax-cards-data {\n color: #ad6531;\n}\n\n.sas-syntax-string {\n color: #8f4238;\n}\n\n.sas-syntax-date,\n.sas-syntax-time,\n.sas-syntax-dt,\n.sas-syntax-bitmask {\n color: #3c8275;\n font-weight: bold;\n}\n\n.sas-syntax-namelit {\n color: #8f4238;\n font-weight: bold;\n}\n\n.sas-syntax-hex,\n.sas-syntax-numeric {\n color: #3c8275;\n font-weight: bold;\n}\n\n.sas-syntax-format {\n color: #3c8275;\n}\n\n.log-line.sas-log-error {\n color: #ff0000;\n}\n\n.log-line.sas-log-warning {\n color: #8f5f00;\n}\n\n.log-line.sas-log-note {\n color: #0000ff;\n}\n"
},
{
"path": "client/src/components/notebook/exporters/toHTML.ts",
"content": "// Copyright © 2025, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport {\n ColorThemeKind,\n NotebookCell,\n NotebookCellKind,\n NotebookCellOutput,\n NotebookDocument,\n TextDocument,\n window,\n} from \"vscode\";\nimport {\n LanguageClient,\n SemanticTokensRequest,\n} from \"vscode-languageclient/node\";\n\nimport { readFileSync } from \"fs\";\nimport hljs from \"highlight.js/lib/core\";\nimport python from \"highlight.js/lib/languages/python\";\nimport r from \"highlight.js/lib/languages/r\";\nimport sql from \"highlight.js/lib/languages/sql\";\nimport { marked } from \"marked\";\nimport markedKatex from \"marked-katex-extension\";\nimport path from \"path\";\n\nimport type { LogLine } from \"../../../connection\";\nimport { includeLogInNotebookExport } from \"../../utils/settings\";\n\nconst templatesDir = path.resolve(__dirname, \"../notebook/exporters/templates\");\n\nhljs.registerLanguage(\"python\", python);\nhljs.registerLanguage(\"r\", r);\nhljs.registerLanguage(\"sql\", sql);\n\n// Configure marked options\nmarked.setOptions({\n breaks: false,\n gfm: false,\n});\n\n// Enable KaTeX extension for marked\nmarked.use(markedKatex());\n\nexport const exportToHTML = async (\n notebook: NotebookDocument,\n client: LanguageClient,\n) => {\n const cells = notebook.getCells();\n\n let template = readFileSync(`${templatesDir}/default.html`).toString();\n\n const isDark =\n window.activeColorTheme.kind === ColorThemeKind.Dark ||\n window.activeColorTheme.kind === ColorThemeKind.HighContrast;\n const theme = readFileSync(\n `${templatesDir}/${isDark ? \"dark\" : \"light\"}.css`,\n ).toString();\n\n // Read KaTeX CSS from templates directory (copied from node_modules during build)\n const katexCss = readFileSync(`${templatesDir}/katex.css`).toString();\n\n template = template.replace(\"${katex}\", katexCss);\n template = template.replace(\"${theme}\", theme);\n template = template.replace(\"${content}\", await exportCells(cells, client));\n\n return template;\n};\n\nconst exportCells = async (cells: NotebookCell[], client: LanguageClient) => {\n let result = \"\";\n\n for (const cell of cells) {\n if (cell.kind === NotebookCellKind.Markup) {\n result += markdownToHTML(cell.document) + \"\\n\";\n } else {\n result += (await codeToHTML(cell.document, client)) + \"\\n\";\n if (cell.outputs.length > 0) {\n for (const output of cell.outputs) {\n if (includeLogInNotebookExport()) {\n result += logToHTML(output) + \"\\n\";\n }\n result += odsToHTML(output) + \"\\n\";\n }\n }\n }\n }\n\n return result;\n};\n\nconst markdownToHTML = (doc: TextDocument) => {\n let text = doc.getText();\n text = normalizeDisplayMathBlocks(text);\n\n return `\n${marked.parse(text)}\n`;\n};\n\n/**\n * Normalize display math blocks in markdown text.\n *\n * Ensures each $$...$$ display math block is on its own lines with blank\n * lines before and after, and trims leading/trailing whitespace inside the\n * delimiters while preserving internal newlines.\n *\n * @example\n * Inline: \"This is $$ x^2 $$ in text.\" -> \"This is\\n\\n$$\\n x^2\\n$$\\n\\n\"\n *\n * @example\n * Display: \"Text$$\\\\frac{1}{2}$$More\" -> \"Text\\n\\n$$\\n\\\\frac{1}{2}\\n$$\\n\\nMore\"\n */\nconst normalizeDisplayMathBlocks = (input: string) =>\n input.replace(/\\$\\$([\\s\\S]*?)\\$\\$/g, (_match, content) => {\n const trimmedContent = content.trim();\n return `\\n\\n$$\\n${trimmedContent}\\n$$\\n\\n`;\n });\n\nconst codeToHTML = async (doc: TextDocument, client: LanguageClient) => {\n let result = \"\";\n if (doc.languageId === \"sas\") {\n result = await SASToHTML(doc, client);\n } else {\n result = hljs.highlight(doc.getText(), {\n language: doc.languageId,\n }).value;\n }\n return `\n${result}
\n${doc.languageId}\n`;\n};\n\nconst SASToHTML = async (doc: TextDocument, client: LanguageClient) => {\n const result = [];\n const tokens = (\n await client.sendRequest(SemanticTokensRequest.type, {\n textDocument: {\n uri: doc.uri.toString(),\n },\n })\n ).data;\n const legend =\n client.initializeResult.capabilities.semanticTokensProvider.legend\n .tokenTypes;\n let tokenIndex = 0;\n let token =\n tokenIndex + 4 < tokens.length\n ? {\n line: tokens[tokenIndex],\n startChar: tokens[tokenIndex + 1],\n length: tokens[tokenIndex + 2],\n tokenType: tokens[tokenIndex + 3],\n }\n : null;\n\n for (let line = 0; line < doc.lineCount; line++) {\n const lineText = doc.lineAt(line).text;\n const parts = [];\n let end = 0;\n\n while (token && token.line === line) {\n parts.push(lineText.slice(end, token.startChar));\n end = token.startChar + token.length;\n parts.push(\n `${lineText.slice(\n token.startChar,\n end,\n )}`,\n );\n tokenIndex += 5;\n token =\n tokenIndex + 4 < tokens.length\n ? {\n line: tokens[tokenIndex] + token.line,\n startChar:\n tokens[tokenIndex + 1] +\n (tokens[tokenIndex] > 0 ? 0 : token.startChar),\n length: tokens[tokenIndex + 2],\n tokenType: tokens[tokenIndex + 3],\n }\n : null;\n }\n parts.push(lineText.slice(end));\n result.push(parts.join(\"\"));\n }\n return result.join(\"\\n\");\n};\n\nconst odsToHTML = (output: NotebookCellOutput) => {\n const ods = output.items.find(\n (item) => item.mime === \"application/vnd.sas.ods.html5\",\n );\n if (ods) {\n const html = ods.data.toString();\n const style = html.slice(\n html.indexOf(\"\") + 8,\n );\n const content = html.slice(\n html.indexOf(\"\"),\n );\n return `\n${style}\n\n`;\n }\n return \"\";\n};\n\nconst logToHTML = (output: NotebookCellOutput) => {\n const logItem = output.items.find(\n (item) => item.mime === \"application/vnd.sas.compute.log.lines\",\n );\n if (logItem) {\n const logs: LogLine[] = JSON.parse(logItem.data.toString());\n return `\n${logs\n .map(\n (line) => `${line.line}`,\n )\n .join(\"\\n\")}\n`;\n }\n return \"\";\n};\n"
},
{
"path": "client/src/components/notebook/exporters/toSAS.ts",
"content": "// Copyright © 2025, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { NotebookCell, NotebookDocument } from \"vscode\";\n\nexport const exportToSAS = (notebook: NotebookDocument) =>\n notebook\n .getCells()\n .map((cell) => exportCell(cell) + \"\\n\")\n .join(\"\\n\");\n\nconst exportCell = (cell: NotebookCell) => {\n const text = cell.document.getText();\n switch (cell.document.languageId) {\n case \"sas\":\n return text;\n case \"python\":\n return wrapPython(text);\n case \"r\":\n return wrapR(text);\n case \"sql\":\n return wrapSQL(text);\n case \"markdown\":\n return `/*\\n${text}\\n*/`;\n }\n};\n\nconst wrapSQL = (code: string) => {\n if (!code.trimEnd().endsWith(\";\")) {\n code = `${code};`;\n }\n return `proc sql;\n${code}\nquit;`;\n};\n\nconst wrapPython = (code: string) => `proc python;\nsubmit;\n${code}\nendsubmit;\nrun;`;\n\nconst wrapR = (code: string) => `proc r;\nsubmit;\n${code}\nendsubmit;\nrun;`;\n"
},
{
"path": "client/src/components/notebook/renderers/HTMLRenderer.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport type { ActivationFunction } from \"vscode-notebook-renderer\";\n\n/**\n * Replace the last occurrence of a substring\n */\nfunction replaceLast(\n base: string,\n searchValue: string,\n replaceValue: string,\n): string {\n const index = base.lastIndexOf(searchValue);\n if (index < 0) {\n return base;\n }\n return (\n base.slice(0, index) + replaceValue + base.slice(index + searchValue.length)\n );\n}\n\nexport const activate: ActivationFunction = () => ({\n renderOutputItem(data, element) {\n const html = data.text();\n let shadow = element.shadowRoot;\n if (!shadow) {\n shadow = element.attachShadow({ mode: \"open\" });\n }\n shadow.innerHTML = replaceLast(\n // it's not a whole webview, body not allowed\n html.replace(\"\",\n \"\",\n );\n },\n});\n"
},
{
"path": "client/src/components/notebook/renderers/LogRenderer.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport type { ActivationFunction } from \"vscode-notebook-renderer\";\n\nimport type { LogLine } from \"../../../connection\";\n\nconst colorMap = {\n error: \"var(--vscode-editorError-foreground)\",\n warning: \"var(--vscode-editorWarning-foreground)\",\n note: \"var(--vscode-editorInfo-foreground)\",\n};\n\nexport const activate: ActivationFunction = () => ({\n renderOutputItem(data, element) {\n const root = document.createElement(\"div\");\n root.style.whiteSpace = \"pre\";\n root.style.fontFamily = \"var(--vscode-editor-font-family)\";\n\n const logs: LogLine[] = data.json();\n for (const line of logs) {\n const color = colorMap[line.type];\n const div = document.createElement(\"div\");\n div.innerText = line.line;\n if (color) {\n div.style.color = color;\n }\n root.append(div);\n }\n element.replaceChildren(root);\n },\n});\n"
},
{
"path": "client/src/components/profile.ts",
"content": "// Copyright © 2022-2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport {\n ConfigurationTarget,\n QuickPickOptions,\n commands,\n l10n,\n window,\n workspace,\n} from \"vscode\";\n\nimport { readFileSync } from \"fs\";\n\nexport const EXTENSION_CONFIG_KEY = \"SAS\";\nexport const EXTENSION_DEFINE_PROFILES_CONFIG_KEY = \"connectionProfiles\";\nexport const EXTENSION_PROFILES_CONFIG_KEY = \"profiles\";\nexport const EXTENSION_ACTIVE_PROFILE_CONFIG_KEY = \"activeProfile\";\n\nenum ConnectionOptions {\n SAS9COM = \"SAS 9.4 (local)\",\n SAS9IOM = \"SAS 9.4 (remote - IOM)\",\n SAS9SSH = \"SAS 9.4 (remote - SSH)\",\n SASViya = \"SAS Viya\",\n}\n\nconst CONNECTION_PICK_OPTS: string[] = [\n ConnectionOptions.SASViya,\n ConnectionOptions.SAS9SSH,\n ConnectionOptions.SAS9IOM,\n ConnectionOptions.SAS9COM,\n];\n\n/**\n * The default compute context that will be used to create a SAS session.\n */\nexport const DEFAULT_COMPUTE_CONTEXT = \"SAS Job Execution compute context\";\nexport const DEFAULT_SSH_PORT = \"22\";\nexport const DEFAULT_IOM_PORT = \"8591\";\n\n/**\n * Dictionary is a type that maps a generic object with a string key.\n */\nexport type Dictionary = {\n [key: string]: T;\n};\n\n/**\n * Enum that represents the authentication type for a profile.\n */\nexport enum AuthType {\n AuthCode = \"authorization_code\",\n Error = \"error\",\n}\n\n/**\n * Enum that represents the connection type for a profile.\n */\nexport enum ConnectionType {\n COM = \"com\",\n IOM = \"iom\",\n Rest = \"rest\",\n SSH = \"ssh\",\n}\n\n/**\n * Profile is an interface that represents a users profile. Currently\n * supports two different authentication flows, token and password\n * flow with the clientId and clientSecret.\n *\n * Direct connect is also supported where a server is already started with\n * a static serverId. Setting serverId in the profile indicates that a connection\n * to that specific server with Id will be created. This overrides the context\n * value. Normally this option should not be set by the user since it is most likely\n * being set by an automated process.\n */\nexport interface ViyaProfile extends BaseProfile, ProfileWithFileRootOptions {\n connectionType: ConnectionType.Rest;\n endpoint: string;\n clientId?: string;\n clientSecret?: string;\n context?: string;\n serverId?: string;\n}\n\nexport interface SSHProfile extends BaseProfile {\n connectionType: ConnectionType.SSH;\n host: string;\n saspath: string;\n port: number;\n username: string;\n privateKeyFilePath?: string;\n}\n\nexport interface COMProfile extends BaseProfile, ProfileWithFileRootOptions {\n connectionType: ConnectionType.COM;\n host: string;\n}\n\nexport interface IOMProfile extends BaseProfile, ProfileWithFileRootOptions {\n connectionType: ConnectionType.IOM;\n host: string;\n username: string;\n port: number;\n}\n\nexport type Profile = ViyaProfile | SSHProfile | COMProfile | IOMProfile;\n\nexport enum AutoExecType {\n File = \"file\",\n Line = \"line\",\n}\n\nexport type AutoExec = AutoExecLine | AutoExecFile;\n\nexport interface AutoExecLine {\n type: AutoExecType.Line;\n line: string;\n}\n\nexport interface AutoExecFile {\n type: AutoExecType.File;\n filePath: string;\n}\n\nexport interface BaseProfile {\n sasOptions?: string[];\n autoExec?: AutoExec[];\n}\n\nexport interface ProfileWithFileRootOptions {\n fileNavigationCustomRootPath?: string;\n fileNavigationRoot?: \"CUSTOM\" | \"SYSTEM\" | \"USER\";\n}\n\nexport const toAutoExecLines = (autoExec: AutoExec[]): string[] => {\n const lines: string[] = [];\n\n for (const item of autoExec) {\n switch (item.type) {\n case AutoExecType.Line:\n lines.push(item.line);\n break;\n case AutoExecType.File:\n lines.push(...toAutoExecLinesFromPaths(item.filePath));\n break;\n default:\n break;\n }\n }\n return lines;\n};\n\n/**\n * Reads content from the given string paths.\n * Content is read sequentially from each path starting at the zeroth path,\n * appending each content line into the output array.\n *\n * If there is an error reading a file in the paths array, then\n * the file is skipped and content is not added.\n * @param paths string array of paths to read content from.\n * @returns string array of lines\n */\nconst toAutoExecLinesFromPaths = (filePath: string): string[] => {\n const lines: string[] = [];\n try {\n const content = readFileSync(filePath, \"utf8\").split(/\\n|\\r\\n/);\n lines.push(...content);\n } catch (e) {\n const err: Error = e;\n console.warn(\n `Error reading file: ${filePath}, error: ${err.message}, skipping...`,\n );\n }\n return lines;\n};\n\n/**\n * Profile detail is an interface that encapsulates the name of the profile\n * with the {@link Profile}.\n */\nexport interface ProfileDetail {\n name: string;\n profile: Profile;\n}\n\n/**\n * Profile validation is an interface that represents the validation\n * information from a profile needed when making a SAS connection.\n */\nexport interface ProfileValidation {\n type: AuthType;\n error: string;\n data?: string;\n profile: Profile;\n}\n\n/**\n * ProfileConfig manages a list of {@link Profile}s that are located in vscode settings.\n * Connection Profiles are designed to keep track of multiple\n * configurations of SAS Connections.\n */\nexport class ProfileConfig {\n /**\n * Helper function to migrate legacy profiles without a connection type.\n */\n async migrateLegacyProfiles() {\n const profiles = this.getAllProfiles();\n\n if (profiles) {\n for (const key in profiles) {\n const profile = profiles[key];\n if (profile.connectionType === undefined) {\n profile.connectionType = ConnectionType.Rest;\n await this.upsertProfile(key, profile);\n }\n if (\n profile.connectionType === ConnectionType.Rest &&\n /\\/$/.test(profile.endpoint)\n ) {\n profile.endpoint = profile.endpoint.replace(/\\/$/, \"\");\n await this.upsertProfile(key, profile);\n }\n }\n }\n }\n\n /**\n * Validates settings.json to confirm that SAS.connectionProfiles exists\n * as a key, and updates it, if the setting does not exists\n *\n * @returns Boolean for pass or fail\n */\n validateSettings(): boolean {\n const profileList: Dictionary = workspace\n .getConfiguration(EXTENSION_CONFIG_KEY)\n .get(EXTENSION_DEFINE_PROFILES_CONFIG_KEY)[EXTENSION_PROFILES_CONFIG_KEY];\n\n if (!profileList) {\n workspace.getConfiguration(EXTENSION_CONFIG_KEY).update(\n EXTENSION_DEFINE_PROFILES_CONFIG_KEY,\n {\n activeProfile: \"\",\n profiles: {},\n },\n ConfigurationTarget.Global,\n );\n return false;\n }\n return true;\n }\n\n /**\n * Get the active profile from the vscode settings.\n *\n * @returns String name to the active profile\n */\n getActiveProfile(): string {\n if (!this.validateSettings()) {\n return \"\";\n }\n const activeProfile: string = workspace\n .getConfiguration(EXTENSION_CONFIG_KEY)\n .get(EXTENSION_DEFINE_PROFILES_CONFIG_KEY)[\n EXTENSION_ACTIVE_PROFILE_CONFIG_KEY\n ];\n return activeProfile;\n }\n\n /**\n * Gets all profiles from the vscode settings.\n *\n * @returns Dictionary of profiles\n */\n getAllProfiles(): Dictionary {\n if (!this.validateSettings()) {\n return {};\n }\n const profileList: Dictionary = workspace\n .getConfiguration(EXTENSION_CONFIG_KEY)\n .get(EXTENSION_DEFINE_PROFILES_CONFIG_KEY)[EXTENSION_PROFILES_CONFIG_KEY];\n\n return profileList;\n }\n\n /**\n * Update VSCode settings with profile dictionary\n *\n * @param profileDict {@link Dictionary} the value for the key\n */\n async updateProfileSetting(profileDict: Dictionary): Promise {\n const currentActiveProfile = this.getActiveProfile();\n const profiles = {\n activeProfile: currentActiveProfile,\n profiles: profileDict,\n };\n await workspace\n .getConfiguration(EXTENSION_CONFIG_KEY)\n .update(\n EXTENSION_DEFINE_PROFILES_CONFIG_KEY,\n profiles,\n ConfigurationTarget.Global,\n );\n }\n\n /**\n * Update VSCode settings with active profile\n *\n * @param activeProfileParam {@link String} the value for the key\n */\n async updateActiveProfileSetting(activeProfileParam: string): Promise {\n const profileList = this.getAllProfiles();\n const profiles = {\n activeProfile: activeProfileParam,\n profiles: profileList,\n };\n if (activeProfileParam in profileList) {\n commands.executeCommand(\"SAS.close\", true);\n } else {\n profiles.activeProfile = \"\";\n }\n await workspace\n .getConfiguration(EXTENSION_CONFIG_KEY)\n .update(\n EXTENSION_DEFINE_PROFILES_CONFIG_KEY,\n profiles,\n ConfigurationTarget.Global,\n );\n }\n\n /**\n * Determines the number of profiles found in settings\n *\n * @returns number of profiles found in vscode settings\n */\n length(): number {\n return Object.keys(this.getAllProfiles()).length;\n }\n\n /**\n * Retreives the list of profile names.\n *\n * @returns List of profile names\n */\n listProfile(): string[] {\n return Object.keys(this.getAllProfiles());\n }\n\n /**\n * Retrieves the {@link Profile} by name from the profile configuration. If the profile\n * is not found by name, a default {@link Profile} will be generated and returned.\n *\n * @param name {@link String} of the profile name\n * @returns Profile object\n */\n getProfileByName(name: string): T {\n const profileList = this.getAllProfiles();\n if (name in profileList) {\n /* eslint-disable @typescript-eslint/consistent-type-assertions*/\n return profileList[name] as T;\n }\n return undefined;\n }\n\n /**\n * Retrieves the {@link ProfileDetail} of the active profile set in the profile\n * configurations.\n *\n * @returns Optional ProfileDetail\n */\n getActiveProfileDetail(): ProfileDetail | undefined {\n const activeProfileName = this.getActiveProfile();\n\n const profileList = this.getAllProfiles();\n if (activeProfileName in profileList) {\n const profile = { ...profileList[activeProfileName] };\n if (\n profile.connectionType === ConnectionType.Rest &&\n /\\/$/.test(profile.endpoint)\n ) {\n profile.endpoint = profile.endpoint.replace(/\\/$/, \"\");\n }\n const detail: ProfileDetail = {\n name: activeProfileName,\n profile,\n };\n return detail;\n } else {\n return undefined;\n }\n }\n\n /**\n * Upsert allows for add or update the new {@link Profile} into vscode settings.\n *\n * @param name {@link String} of the name of the profile\n * @param profile {@link Profile} object\n */\n async upsertProfile(name: string, profile: Profile): Promise {\n const profileList = this.getAllProfiles();\n // Cannot mutate VSCode Config Object, create a clone and add that to settings.json\n const newProfileList = JSON.parse(JSON.stringify(profileList));\n newProfileList[name] = profile;\n await this.updateProfileSetting(newProfileList);\n }\n\n /**\n * Deletes a profile from the vscode settings.\n *\n * @param name {@link String} of the name of the profile\n */\n async deleteProfile(name: string): Promise {\n const profileList = this.getAllProfiles();\n if (name in profileList) {\n // Cannot mutate VSCode Config Object, create a clone and add that to settings.json\n const newProfileList = JSON.parse(JSON.stringify(profileList));\n delete newProfileList[name];\n await this.updateProfileSetting(newProfileList);\n if (name === this.getActiveProfile()) {\n await this.updateActiveProfileSetting(\"\");\n }\n }\n }\n\n /**\n * Validates if the {@link ProfileDetail} meets the requirements needed for authentication\n * and returns back the authentication type.\n *\n * The validation process calculates the authentication flow by what is detailed in the\n * {@link ProfileDetail}. If the conditions to calculate the authentication flow are not\n * meet, then an error is provided in the {@link ProfileValidation}.\n *\n * @param profileDetail\n * @returns ProfileValidation object\n */\n validateProfile(profileDetail?: ProfileDetail): ProfileValidation {\n const pv: ProfileValidation = {\n type: AuthType.Error,\n error: \"\",\n profile: undefined,\n };\n\n //Validate active profile, return early if not valid\n if (!profileDetail?.profile) {\n pv.error = l10n.t(\"No Active Profile\");\n return pv;\n }\n\n const profile: Profile = profileDetail.profile;\n if (profile.connectionType === undefined) {\n pv.error = l10n.t(\"Missing connectionType in active profile.\");\n return pv;\n }\n if (profile.connectionType === ConnectionType.Rest) {\n if (!profile.endpoint) {\n pv.error = l10n.t(\"Missing endpoint in active profile.\");\n return pv;\n }\n } else if (profile.connectionType === ConnectionType.SSH) {\n if (!profile.host) {\n pv.error = l10n.t(\"Missing host in active profile.\");\n return pv;\n }\n\n if (!profile.port) {\n pv.error = l10n.t(\"Missing port in active profile.\");\n return pv;\n }\n\n if (!profile.saspath) {\n pv.error = l10n.t(\"Missing sas path in active profile.\");\n return pv;\n }\n if (!profile.username) {\n pv.error = l10n.t(\"Missing username in active profile.\");\n return pv;\n }\n }\n\n pv.profile = profileDetail.profile;\n pv.type = AuthType.AuthCode;\n return pv;\n }\n\n /**\n * Requests users input on updating or adding a new profile.\n *\n * @param name the {@link String} representation of the name of the profile\n */\n async prompt(name: string): Promise {\n const profile: Profile = this.getProfileByName(name);\n // Cannot mutate VSCode Config Object, create a clone and upsert\n let profileClone = { ...profile };\n if (!profile) {\n profileClone = {\n connectionType: ConnectionType.Rest,\n endpoint: undefined,\n };\n }\n\n const inputConnectionType: string = await createInputQuickPick(\n CONNECTION_PICK_OPTS,\n ProfilePromptType.ConnectionType,\n );\n if (inputConnectionType === undefined) {\n return;\n }\n\n profileClone.connectionType = mapQuickPickToEnum(inputConnectionType);\n\n if (profileClone.connectionType === ConnectionType.Rest) {\n profileClone.endpoint = await createInputTextBox(\n ProfilePromptType.Endpoint,\n profileClone.endpoint,\n );\n\n if (!profileClone.endpoint) {\n return;\n }\n profileClone.endpoint = profileClone.endpoint.replace(/\\/$/, \"\");\n\n profileClone.context = await createInputTextBox(\n ProfilePromptType.ComputeContext,\n profileClone.context || DEFAULT_COMPUTE_CONTEXT,\n );\n if (profileClone.context === undefined) {\n return;\n }\n if (\n profileClone.context === \"\" ||\n profileClone.context === DEFAULT_COMPUTE_CONTEXT\n ) {\n delete profileClone.context;\n }\n\n profileClone.clientId = await createInputTextBox(\n ProfilePromptType.ClientId,\n profileClone.clientId,\n );\n if (profileClone.clientId === undefined) {\n return;\n }\n if (profileClone.clientId === \"\") {\n delete profileClone.clientId;\n }\n\n if (profileClone.clientId) {\n profileClone.clientSecret = await createInputTextBox(\n ProfilePromptType.ClientSecret,\n profileClone.clientSecret,\n );\n if (profileClone.clientSecret === undefined) {\n return;\n }\n }\n\n await this.upsertProfile(name, profileClone);\n } else if (profileClone.connectionType === ConnectionType.SSH) {\n profileClone.host = await createInputTextBox(\n ProfilePromptType.Host,\n profileClone.host,\n );\n if (!profileClone.host) {\n return;\n }\n\n profileClone.saspath = await createInputTextBox(\n ProfilePromptType.SASPath,\n profileClone.saspath,\n );\n if (profileClone.saspath === undefined) {\n return;\n }\n\n profileClone.username = await createInputTextBox(\n ProfilePromptType.Username,\n profileClone.username,\n );\n if (profileClone.username === undefined) {\n return;\n }\n\n profileClone.port = parseInt(\n await createInputTextBox(ProfilePromptType.Port, DEFAULT_SSH_PORT),\n );\n if (isNaN(profileClone.port)) {\n return;\n }\n\n const keyPath = await createInputTextBox(\n ProfilePromptType.PrivateKeyFilePath,\n profileClone.privateKeyFilePath,\n );\n\n if (keyPath) {\n profileClone.privateKeyFilePath = keyPath;\n }\n\n await this.upsertProfile(name, profileClone);\n } else if (profileClone.connectionType === ConnectionType.COM) {\n profileClone.sasOptions = [];\n profileClone.host = \"localhost\"; //once remote support rolls out this should be set via prompting\n await this.upsertProfile(name, profileClone);\n } else if (profileClone.connectionType === ConnectionType.IOM) {\n profileClone.sasOptions = [];\n profileClone.host = await createInputTextBox(\n ProfilePromptType.Host,\n profileClone.host,\n );\n if (!profileClone.host) {\n return;\n }\n\n profileClone.port = parseInt(\n await createInputTextBox(ProfilePromptType.Port, DEFAULT_IOM_PORT),\n );\n if (isNaN(profileClone.port)) {\n return;\n }\n\n profileClone.username = await createInputTextBox(\n ProfilePromptType.Username,\n profileClone.username,\n );\n if (profileClone.username === undefined) {\n return;\n }\n\n await this.upsertProfile(name, profileClone);\n }\n }\n\n /**\n * Retrieves the remote target associated with the active profile. For SSH profiles, the host\n * value is used. For Viya, the endpoint value is used.\n * @param profileName - a profile name to retrieve.\n * @returns\n */\n remoteTarget(profileName: string): string {\n const activeProfile = this.getProfileByName(profileName);\n switch (activeProfile.connectionType) {\n case ConnectionType.SSH:\n case ConnectionType.COM:\n case ConnectionType.IOM:\n return activeProfile.host;\n case ConnectionType.Rest:\n return activeProfile.endpoint;\n }\n }\n}\n\n/**\n * Define an object to represent the values needed for prompting a window.showInputBox\n */\nexport interface ProfilePrompt {\n title: string;\n placeholder: string;\n description: string;\n}\n\n/**\n * An enum representing the types of prompts that can be returned for window.showInputBox\n */\nexport enum ProfilePromptType {\n Profile = 0,\n NewProfile,\n ClientId,\n Endpoint,\n ComputeContext,\n ClientSecret,\n ConnectionType,\n Host,\n SASPath,\n Port,\n Username,\n PrivateKeyFilePath,\n}\n\n/**\n * An interface that will map an enum of {@link ProfilePromptType} to an interface of {@link ProfilePrompt}.\n */\nexport type ProfilePromptInput = {\n [key in ProfilePromptType]: ProfilePrompt;\n};\n\n/**\n * Retrieves the {@link ProfilePrompt} by the enum {@link ProfilePromptType}\n *\n * @param type {@link ProfilePromptType}\n * @returns ProfilePrompt object\n */\nexport function getProfilePrompt(type: ProfilePromptType): ProfilePrompt {\n return input[type];\n}\n\n/**\n * Helper method to generate a window.ShowInputBox with using a defined set of {@link ProfilePrompt}s.\n *\n * @param profilePromptType {@link ProfilePromptType}\n * @param defaultValue the {@link String} of the default value that will be represented in the input box. Defaults to null\n * @param maskValue the {@link boolean} if the input value will be masked\n * @param username the {@link String} of the SAS User ID\n * @returns Thenable<{@link String}> of the users input\n */\nexport async function createInputTextBox(\n profilePromptType: ProfilePromptType,\n defaultValue: string | undefined = null,\n maskValue = false,\n): Promise {\n const profilePrompt = getProfilePrompt(profilePromptType);\n\n const entered = await window.showInputBox({\n title: profilePrompt.title,\n placeHolder: profilePrompt.placeholder,\n prompt: profilePrompt.description,\n password: maskValue,\n value: defaultValue,\n ignoreFocusOut: true,\n });\n return entered;\n}\n\n/**\n * Helper method to generate a window.ShowInputQuickPick using a defined set of {@link ProfilePrompt}s.\n * @param items list of selectable options to bind to the quickpick.\n * @param profilePromptType {@link ProfilePromptType}\n * @returns Thenable<{@link String}> of the users input\n */\nexport async function createInputQuickPick(\n items: readonly string[] | Thenable = [],\n profilePromptType: ProfilePromptType,\n): Promise {\n const profilePrompt = getProfilePrompt(profilePromptType);\n\n const options: QuickPickOptions = {\n title: profilePrompt.title,\n placeHolder: profilePrompt.placeholder,\n ignoreFocusOut: true,\n canPickMany: false,\n };\n\n const entered = await window.showQuickPick(items, options);\n\n return entered;\n}\n\n/**\n * Mapped {@link ProfilePrompt} to an enum of {@link ProfilePromptType}.\n */\nconst input: ProfilePromptInput = {\n [ProfilePromptType.Profile]: {\n title: l10n.t(\"Switch Current SAS Profile\"),\n placeholder: l10n.t(\"Select a SAS connection profile\"),\n description: \"\",\n },\n [ProfilePromptType.NewProfile]: {\n title: l10n.t(\"New SAS Connection Profile Name\"),\n placeholder: l10n.t(\"Enter connection name\"),\n description: l10n.t(\n \"You can also specify connection profile using the settings.json file.\",\n ),\n },\n [ProfilePromptType.Endpoint]: {\n title: l10n.t(\"SAS Viya Server\"),\n placeholder: l10n.t(\"Enter the URL\"),\n description: l10n.t(\n \"Enter the URL for the SAS Viya server. An example is https://example.sas.com.\",\n ),\n },\n [ProfilePromptType.ComputeContext]: {\n title: l10n.t(\"SAS Compute Context\"),\n placeholder: l10n.t(\"Enter the SAS compute context\"),\n description: l10n.t(\"Enter the SAS compute context.\"),\n },\n [ProfilePromptType.ClientId]: {\n title: l10n.t(\"Client ID\"),\n placeholder: l10n.t(\"Enter a client ID\"),\n description: l10n.t(\n \"Enter the registered client ID. An example is myapp.client.\",\n ),\n },\n [ProfilePromptType.ClientSecret]: {\n title: l10n.t(\"Client Secret\"),\n placeholder: l10n.t(\"Enter a client secret\"),\n description: l10n.t(\n \"Enter secret for client ID. An example is myapp.secret.\",\n ),\n },\n [ProfilePromptType.ConnectionType]: {\n title: l10n.t(\"Connection Type\"),\n placeholder: l10n.t(\"Select a Connection Type\"),\n description: l10n.t(\"Select a Connection Type.\"),\n },\n [ProfilePromptType.Host]: {\n title: l10n.t(\"SAS 9 Server\"),\n placeholder: l10n.t(\"Enter the server name\"),\n description: l10n.t(\"Enter the name of the SAS 9 server.\"),\n },\n [ProfilePromptType.SASPath]: {\n title: l10n.t(\"Server Path\"),\n placeholder: l10n.t(\"Enter the server path\"),\n description: l10n.t(\"Enter the server path of the SAS Executable.\"),\n },\n [ProfilePromptType.Port]: {\n title: l10n.t(\"Port Number\"),\n placeholder: l10n.t(\"Enter a port number\"),\n description: l10n.t(\"Enter a port number.\"),\n },\n [ProfilePromptType.Username]: {\n title: l10n.t(\"SAS Server Username\"),\n placeholder: l10n.t(\"Enter your username\"),\n description: l10n.t(\"Enter your SAS server username.\"),\n },\n [ProfilePromptType.PrivateKeyFilePath]: {\n title: l10n.t(\"Private Key File Path (optional)\"),\n placeholder: l10n.t(\"Enter the local private key file path\"),\n description: l10n.t(\"To use the SSH Agent or a password, leave blank.\"),\n },\n};\n\n/**\n * Helper function to map the quick pick item selection to a well known {@link ConnectionType}.\n * @param connectionTypePickInput - string value of one of the quick pick option inputs\n * @returns {@link ConnectionType}\n */\nfunction mapQuickPickToEnum(connectionTypePickInput: string): ConnectionType {\n /*\n Having a translation layer here allows the profile types to potentially evolve separately from the\n underlying technology used to implement the connection. Down the road its quite possible to have\n more than one selectable quick pick input that uses the same underlying connection methods..\n */\n switch (connectionTypePickInput) {\n case ConnectionOptions.SASViya:\n return ConnectionType.Rest;\n case ConnectionOptions.SAS9SSH:\n return ConnectionType.SSH;\n case ConnectionOptions.SAS9COM:\n return ConnectionType.COM;\n case ConnectionOptions.SAS9IOM:\n return ConnectionType.IOM;\n default:\n return undefined;\n }\n}\n"
},
{
"path": "client/src/components/tasks/SasTaskProvider.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport {\n CustomExecution,\n EventEmitter,\n ProviderResult,\n Pseudoterminal,\n Task,\n TaskProvider,\n TaskScope,\n l10n,\n} from \"vscode\";\n\nimport { hasRunningTask } from \"../../commands/run\";\nimport {\n Execute,\n SAS_TASK_TYPE,\n SasTaskDefinition,\n SasTaskNames,\n TaskInfo,\n runSasFileTask,\n} from \"./SasTasks\";\n\nexport class SasTaskProvider implements TaskProvider {\n provideTasks(): ProviderResult {\n return [generateTask(SasTaskNames.RunSasFile, runSasFileTask)];\n }\n\n resolveTask(task: Task): ProviderResult {\n if (task.definition.task === SasTaskNames.RunSasFile) {\n return generateTask(task, runSasFileTask);\n }\n }\n}\n\nexport class SasPseudoterminal implements Pseudoterminal {\n private messageEmitter = new EventEmitter();\n private closeEmitter = new EventEmitter();\n\n constructor(\n private execute: Execute,\n private taskInfo: TaskInfo,\n ) {}\n\n public onDidWrite = this.messageEmitter.event;\n public onDidClose? = this.closeEmitter.event;\n\n public open(): void {\n if (hasRunningTask()) {\n this.messageEmitter.fire(\n \"There is running task, please try again after that is complete.\\n\",\n );\n this.closeEmitter.fire(1);\n } else {\n this.executeTask();\n }\n }\n\n public close(): void {\n this.closeEmitter.fire(0);\n }\n public handleInput?(data): void {\n // press ctrl + c to cancel executing task.\n if (data === \"\u0003\") {\n this.messageEmitter.fire(\"Task is cancelled.\");\n this.closeEmitter.fire(1);\n }\n }\n\n private async executeTask(): Promise {\n return new Promise(() => {\n this.execute(this.messageEmitter, this.taskInfo, this.closeEmitter)\n .then(() => {\n this.messageEmitter.fire(l10n.t(\"Task is complete.\") + \"\\r\\n\\r\\n\");\n this.closeEmitter.fire(0);\n })\n .catch((reason) => {\n this.messageEmitter.fire(reason.message + \"\\r\\n\\r\\n\");\n this.messageEmitter.fire(l10n.t(\"Task is cancelled.\") + \"\\r\\n\\r\\n\");\n this.closeEmitter.fire(1);\n });\n });\n }\n}\n\nfunction generateTask(task: string | Task, execute: Execute) {\n const definition =\n typeof task === \"object\"\n ? task.definition\n : {\n type: SAS_TASK_TYPE,\n task: task,\n };\n\n return new Task(\n definition,\n TaskScope.Workspace,\n definition.task,\n SAS_TASK_TYPE,\n new CustomExecution(async (taskDefinition: SasTaskDefinition) => {\n return new SasPseudoterminal(execute, {\n definition: taskDefinition,\n label: typeof task === \"object\" ? task.name : task,\n });\n }),\n );\n}\n"
},
{
"path": "client/src/components/tasks/SasTasks.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { EventEmitter, TaskDefinition, l10n, window, workspace } from \"vscode\";\n\nimport { isAbsolute } from \"path\";\n\nimport { runTask } from \"../../commands/run\";\nimport { SASCodeDocument } from \"../utils/SASCodeDocument\";\nimport { getCodeDocumentConstructionParameters } from \"../utils/SASCodeDocumentHelper\";\n\nexport const SAS_TASK_TYPE = \"sas\";\n\nexport enum SasTaskNames {\n // Run the sas file indicated in the \"file\" property. if preamble or postamble provided, wrapping the sas code in the file with them.\n // If this task is called as predefined task or custom task without file or blank file name provided,\n // the code to run will be the selected code in active editor or be the active file code if there is no code selected.\n RunSasFile = \"Run sas file\",\n}\n\nexport interface SasTaskDefinition extends TaskDefinition {\n task: string;\n file?: string;\n preamble?: string;\n postamble?: string;\n}\n\nexport interface TaskInfo {\n definition: SasTaskDefinition;\n label: string;\n}\n\nexport type Execute = (\n messageEmitter: EventEmitter,\n taskInfo: TaskInfo,\n closeEmitter: EventEmitter,\n) => Promise;\n\nexport async function runSasFileTask(\n messageEmitter: EventEmitter,\n taskInfo: TaskInfo,\n closeEmitter: EventEmitter,\n) {\n const {\n definition: { file, preamble, postamble },\n label,\n } = taskInfo;\n\n const textDocument = await getTextDocumentFromFile(file);\n const parameters = getCodeDocumentConstructionParameters(textDocument, {\n selections:\n file === undefined || file.trim() === \"\"\n ? window.activeTextEditor?.selections\n : undefined,\n preamble,\n postamble,\n });\n\n const codeDoc = new SASCodeDocument(parameters);\n\n return runTask(codeDoc, messageEmitter, closeEmitter, label);\n}\n\nasync function getTextDocumentFromFile(file: string | undefined) {\n if (file === undefined || file.trim() === \"\") {\n return window.activeTextEditor.document;\n } else if (isAbsolute(file)) {\n return await workspace.openTextDocument(file);\n } else {\n const uri = (await workspace.findFiles(file))[0];\n if (uri === undefined) {\n throw new Error(l10n.t(\"Cannot find file: {file}\", { file }));\n } else {\n return await workspace.openTextDocument(uri);\n }\n }\n}\n"
},
{
"path": "client/src/components/utils/SASCodeDocument.ts",
"content": "// Copyright © 2024, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport {\n ProblemLocation,\n decomposeCodeLogLine,\n} from \"../logViewer/ProblemProcessor\";\n\nexport interface SASCodeDocumentParameters {\n languageId: string;\n code: string;\n selectedCode: string;\n uri?: string;\n fileName?: string;\n selections?: ReadonlyArray<{\n start: { line: number; character: number };\n end: { line: number; character: number };\n }>;\n preamble?: string;\n postamble?: string;\n outputHtml?: boolean;\n htmlStyle?: string;\n uuid?: string;\n checkKeyword: (LineNumber: number, ...keywords: string[]) => Promise;\n}\n\ntype LineNumber = number;\ntype Offset = { lineOffset: number; columnOffset: number };\n\nexport class SASCodeDocument {\n // there may be many selected raw code and they can be not continuous in editor.\n // this field provides a offset map for selected raw code in wrapped code and in whole raw code.\n private offsetMap: Map;\n\n public constructor(private parameters: SASCodeDocumentParameters) {}\n\n public getWrappedCode(): string {\n const code = this.getRawCode();\n return this.codeIsEmpty(code) ? \"\" : this.wrapCode(code);\n }\n\n public getUri(): string {\n return this.parameters.uri;\n }\n\n public getFileName(): string {\n return this.parameters.fileName;\n }\n\n public wrappedCodeLineAt(lineNumber: number) {\n return this.getWrappedCode().split(\"\\n\")[lineNumber];\n }\n\n public async getLocationInRawCode(\n locationFromLog: ProblemLocation,\n codeLinesInLog: string[],\n ): Promise {\n if (this.offsetMap === undefined) {\n await this.constructOffsetMap(codeLinesInLog);\n }\n\n const {\n lineNumber: lineNumberInLog,\n startColumn: startColumnInLog,\n endColumn: endColumnInLog,\n } = locationFromLog;\n\n const offset = this.offsetMap.get(lineNumberInLog);\n if (offset) {\n const { lineOffset: lineNumberInRaw, columnOffset } = offset;\n return {\n lineNumber: lineNumberInRaw,\n startColumn: startColumnInLog + columnOffset,\n endColumn: endColumnInLog + columnOffset,\n };\n }\n\n const firstLineNumber = this.offsetMap.keys().next().value;\n const lastLineNumber = Array.from(this.offsetMap.keys()).pop() ?? 0;\n\n // if the problem occurs before the first raw code line,\n // then re-locate it at the first character in the first raw code line.\n if (lineNumberInLog < firstLineNumber) {\n return {\n lineNumber: 0,\n startColumn: 0,\n endColumn: 1,\n };\n }\n\n // if the problem occurs after the last raw code line,\n // then re-located it at the last character in the last raw code line.\n if (lineNumberInLog > lastLineNumber) {\n const codeLinesInRaw = this.getRawCode().split(\"\\n\");\n const count = codeLinesInRaw[codeLinesInRaw.length - 1].length;\n let lastCharacterIndex = count === 0 ? 0 : count - 1;\n\n if (this.offsetMap.size === 1) {\n lastCharacterIndex = this.parameters.selections[0].end.character - 1;\n }\n\n const lineNumberInRaw =\n this.offsetMap.get(lastLineNumber)?.lineOffset ?? 0;\n\n return {\n lineNumber: lineNumberInRaw,\n startColumn: lastCharacterIndex,\n endColumn: lastCharacterIndex + 1,\n };\n }\n\n // the problem occurs in imported source code,\n // re-locate it at the nearest previous raw code line.\n const nearestPreviousLineNumberInLog = Array.from(this.offsetMap.keys())\n .reverse()\n .find((lineNumber) => lineNumberInLog > lineNumber);\n\n const nearestOffset = this.offsetMap.get(nearestPreviousLineNumberInLog);\n\n return {\n lineNumber: nearestOffset.lineOffset,\n startColumn: 0,\n endColumn: 1,\n };\n }\n\n private codeIsEmpty(code: string): boolean {\n return code.trim() === \"\";\n }\n\n private wrapCodeWithSASProgramFileName(code: string): string {\n let fileName = this.parameters.fileName;\n if (fileName === undefined) {\n return code;\n } else {\n fileName = fileName.replace(/[('\")]/g, \"%$&\");\n const wrapped =\n \"%let _SASPROGRAMFILE = %nrquote(%nrstr(\" + fileName + \"));\\n\" + code;\n return wrapped;\n }\n }\n\n private wrapCodeWithPreambleAndPostamble(code: string): string {\n return (\n (this.parameters?.preamble ? this.parameters?.preamble + \"\\n\" : \"\") +\n code +\n (this.parameters?.postamble ? \"\\n\" + this.parameters?.postamble : \"\")\n );\n }\n\n private wrapCodeWithOutputHtml(code: string): string {\n if (this.parameters.outputHtml) {\n const htmlStyle = this.parameters.htmlStyle.trim();\n const htmlStyleOption = htmlStyle !== \"\" ? ` style=${htmlStyle}` : \"\";\n const outputDestination = this.parameters?.uuid\n ? ` body=\"${this.parameters.uuid}.htm\"`\n : \"\";\n\n return `title;footnote;ods _all_ close;\nods graphics on;\nods html5(id=vscode)${htmlStyleOption} options(bitmap_mode='inline' svg_mode='inline')${outputDestination};\n${code}\n;*';*\";*/;run;quit;ods html5(id=vscode) close;\n`;\n } else {\n return code;\n }\n }\n\n private wrapSQL(code: string) {\n return `proc sql;\n${code}\n;quit;`;\n }\n\n private wrapPython(code: string) {\n return `proc python;\nsubmit;\n${code}\nendsubmit;\nrun;`;\n }\n\n private wrapR(code: string) {\n return `proc r;\nsubmit;\n${code}\nendsubmit;\nrun;`;\n }\n\n private insertLogStartIndicator(code: string): string {\n // add a comment line at the top of code,\n // this comment line will be used as indicator to the beginning of log related with this code\n return `/** LOG_START_INDICATOR **/\n${code}`;\n }\n\n private wrapCode(code: string): string {\n let wrapped = code;\n\n if (this.parameters.languageId === \"sql\") {\n wrapped = this.wrapSQL(wrapped);\n }\n\n if (this.parameters.languageId === \"python\") {\n wrapped = this.wrapPython(wrapped);\n }\n\n if (this.parameters.languageId === \"r\") {\n wrapped = this.wrapR(wrapped);\n }\n\n wrapped = this.wrapCodeWithSASProgramFileName(wrapped);\n\n wrapped = this.wrapCodeWithPreambleAndPostamble(wrapped);\n\n wrapped = this.wrapCodeWithOutputHtml(wrapped);\n\n wrapped = this.insertLogStartIndicator(wrapped);\n\n return wrapped;\n }\n\n // getWrappedCode() returns more code than raw code in editor, and addition code may be added at the beginning or end of raw code.\n // this method return the position at which the raw code begins in wrapped code.\n private getRawCodeBeginLineNumberInWrappedCode(): number {\n const FRONT_LOCATOR = \"LOCATOR-TO-MARK-THE-BEGIN-OF-USER-CODE\";\n const codeWithLocator = FRONT_LOCATOR + this.getRawCode();\n const wrapped = this.wrapCode(codeWithLocator);\n return wrapped\n .split(\"\\n\")\n .findIndex((line) => line.includes(FRONT_LOCATOR));\n }\n\n // return selected code line array, which is in {lineNumber, column, code} format.\n private constructCodeLinesInRaw(): {\n lineNumber: LineNumber;\n column: number;\n code: string;\n }[] {\n const codeLines = this.getRawCode().split(\"\\n\");\n let index = -1;\n const codeLinesInRaw = [];\n\n this.parameters.selections.forEach((selection) => {\n const { start, end } = selection;\n\n for (let lineNumber = start.line; lineNumber <= end.line; lineNumber++) {\n index++;\n codeLinesInRaw[index] = {\n lineNumber,\n column: lineNumber === start.line ? start.character : 0,\n code: codeLines[index],\n };\n }\n });\n\n return codeLinesInRaw;\n }\n\n private getNextValidCodeLineInLog(\n codeLines: string[],\n start: number,\n ): { code: string; lineNumber: number; index: number } {\n let index = start;\n let { code, lineNumber } = decomposeCodeLogLine(codeLines[index]);\n\n while (\n index < codeLines.length &&\n // code not included in the source file starts with \"+\" in the log.\n code.trim().startsWith(\"+\")\n ) {\n ({ code, lineNumber } = decomposeCodeLogLine(codeLines[++index]));\n }\n return { code, lineNumber, index };\n }\n\n private async constructOffsetMap(codeLinesInLog: string[]): Promise {\n const codeLinesInRaw = this.constructCodeLinesInRaw();\n let indexInRaw = 0;\n let codeLineInRaw: string;\n let lineNumberInRaw: number;\n let columnInRaw: number;\n\n let indexInLog = this.getRawCodeBeginLineNumberInWrappedCode();\n let codeLineInLog: string;\n let lineNumberInLog: number;\n let lastValidLineNumberInLog: number;\n\n let inInteractiveBlock = false;\n\n this.offsetMap = new Map();\n\n while (\n indexInRaw < codeLinesInRaw.length &&\n indexInLog < codeLinesInLog.length\n ) {\n ({\n code: codeLineInRaw,\n lineNumber: lineNumberInRaw,\n column: columnInRaw,\n } = codeLinesInRaw[indexInRaw]);\n\n if (inInteractiveBlock) {\n let index = indexInRaw;\n let lineInfo = codeLinesInRaw[++index];\n\n while (\n !(await this.parameters.checkKeyword(\n // this.parameters.uri,\n // lineInfo.code,\n lineInfo.lineNumber,\n \"endinteractive\",\n )) &&\n index < codeLinesInRaw.length\n ) {\n lineInfo = codeLinesInRaw[++index];\n }\n\n if (index < codeLinesInRaw.length) {\n ({\n code: codeLineInRaw,\n lineNumber: lineNumberInRaw,\n column: columnInRaw,\n } = codeLinesInRaw[++index]);\n indexInRaw = index;\n inInteractiveBlock = false;\n }\n }\n\n ({\n code: codeLineInLog,\n lineNumber: lineNumberInLog,\n index: indexInLog,\n } = this.getNextValidCodeLineInLog(codeLinesInLog, indexInLog));\n\n // The line numbers in the source code within the log should be continuous.\n // but if encountering datasets following a datalines statement or %INC statement,\n // the line numbers will not be continuous.\n // for datalines-like statements, it will skip the number of dataset lines.\n // for %INC-like statements, it will continue without skip\n const delta =\n lastValidLineNumberInLog === undefined\n ? 1\n : lineNumberInLog - lastValidLineNumberInLog;\n if (\n delta > 1 &&\n // if the code line in log can be found in raw,\n // think of the discontinuous line number is caused by %INC-like statements,\n // otherwise it is from datalines-like statements and need to skip lines in raw.\n !isSameOrStartsWith(codeLineInRaw.trim(), codeLineInLog.trim())\n ) {\n indexInRaw += delta - 1;\n ({\n code: codeLineInRaw,\n lineNumber: lineNumberInRaw,\n column: columnInRaw,\n } = codeLinesInRaw[indexInRaw]);\n }\n\n if (!isSameOrStartsWith(codeLineInRaw.trim(), codeLineInLog.trim())) {\n const match = this.getMatchedCodeLineInLog(\n codeLineInRaw,\n codeLinesInLog,\n indexInLog,\n );\n lineNumberInLog = match.lineNumber;\n indexInLog = match.index;\n }\n\n const offset = { lineOffset: lineNumberInRaw, columnOffset: columnInRaw };\n\n this.offsetMap.set(lineNumberInLog, offset);\n lastValidLineNumberInLog = lineNumberInLog;\n\n inInteractiveBlock = await this.parameters.checkKeyword(\n // this.parameters.uri,\n // codeLineInRaw,\n lineNumberInRaw,\n \"interactive\",\n \"i\",\n );\n\n indexInRaw++;\n indexInLog++;\n }\n }\n\n private getMatchedCodeLineInLog(\n codeLineInRaw: string,\n codeLinesInLog: string[],\n start: number,\n ): { code: string; lineNumber: number; index: number } | null {\n let validCodeLine = { code: \"\", lineNumber: -1, index: start };\n let indexInLog = start;\n do {\n validCodeLine = this.getNextValidCodeLineInLog(\n codeLinesInLog,\n indexInLog++,\n );\n } while (\n !isSameOrStartsWith(codeLineInRaw.trim(), validCodeLine.code.trim()) &&\n indexInLog < codeLinesInLog.length\n );\n\n return validCodeLine.index >= codeLinesInLog.length ? null : validCodeLine;\n }\n\n // priority return selected code, otherwise, return whole code.\n private getRawCode(): string {\n return this.parameters.selectedCode.trim() === \"\"\n ? this.parameters.code\n : this.parameters.selectedCode;\n }\n}\n\nfunction isSameOrStartsWith(base: string, target: string): boolean {\n return target === \"\" ? base === target : base.startsWith(target);\n}\n"
},
{
"path": "client/src/components/utils/SASCodeDocumentHelper.ts",
"content": "// Copyright © 2024, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport {\n ColorThemeKind,\n Hover,\n Position,\n Selection,\n TextDocument,\n commands,\n window,\n workspace,\n} from \"vscode\";\n\nimport { v4 } from \"uuid\";\n\nimport { profileConfig } from \"../../commands/profile\";\nimport { ConnectionType } from \"../profile\";\nimport { SASCodeDocumentParameters } from \"./SASCodeDocument\";\nimport { getHtmlStyle, isOutputHtmlEnabled } from \"./settings\";\n\nexport function getCodeDocumentConstructionParameters(\n textDocument: TextDocument,\n addition?: {\n selections?: ReadonlyArray;\n preamble?: string;\n postamble?: string;\n },\n): SASCodeDocumentParameters {\n // TODO #810 This is a temporary solution to prevent creating an excessive\n // number of result files for viya connections.\n // This todo will be cleaned up with remaining work in #810.\n const uuid = connectionTypeIsNotRest() ? v4() : undefined;\n\n return {\n languageId: textDocument.languageId,\n code: textDocument.getText(),\n selectedCode: getSelectedCode(textDocument, addition?.selections),\n uri: textDocument.uri.toString(),\n fileName: getFileName(textDocument),\n selections: getCodeSelections(addition?.selections, textDocument),\n preamble: addition?.preamble,\n postamble: addition?.postamble,\n htmlStyle: getHtmlStyleValue(),\n outputHtml: isOutputHtmlEnabled(),\n uuid,\n checkKeyword: async (lineNumber: number, ...keywords: string[]) => {\n const codeLines = textDocument.getText().split(\"\\n\");\n const codeLine = codeLines[lineNumber];\n\n const regExp = new RegExp(`\\\\b(${keywords.join(\"|\")})\\\\b`, \"gi\");\n const matches = Array.from(codeLine.matchAll(regExp));\n\n if (matches.length === 0) {\n return false;\n }\n\n for (const match of matches) {\n const [actualHover]: Hover[] = await commands.executeCommand(\n \"vscode.executeHoverProvider\",\n textDocument.uri,\n new Position(lineNumber, match.index),\n );\n if (actualHover !== undefined) {\n return true;\n }\n }\n return false;\n },\n };\n}\n\nfunction getSelectedCode(\n textDocument: TextDocument,\n selections?: ReadonlyArray,\n): string {\n if (selectionsAreNotEmpty(selections)) {\n return selections\n .map((selection) => {\n return textDocument.getText(selection);\n })\n .join(\"\\n\");\n } else {\n return \"\";\n }\n}\n\nfunction connectionTypeIsNotRest(): boolean {\n const activeProfile = profileConfig.getActiveProfileDetail();\n return (\n activeProfile &&\n activeProfile.profile.connectionType !== ConnectionType.Rest\n );\n}\n\nfunction selectionsAreNotEmpty(\n selections: ReadonlyArray | undefined,\n): boolean {\n return (\n selections?.length > 1 ||\n // the single cursor (if it is not in a selection) is always treated as a selection in Monaco Editor\n (selections?.length === 1 && !selections[0].isEmpty)\n );\n}\n\nfunction getHtmlStyleValue(): string {\n const htmlStyleSetting = getHtmlStyle();\n if (htmlStyleSetting === \"(auto)\") {\n // get the results.html.custom.style object from user settings\n const customStyle =\n workspace.getConfiguration(\"SAS\").get<{\n light?: string;\n dark?: string;\n highContrast?: string;\n highContrastLight?: string;\n }>(\"results.html.custom.style\") || {};\n\n switch (window.activeColorTheme.kind) {\n case ColorThemeKind.Light:\n return customStyle.light || \"Illuminate\";\n case ColorThemeKind.Dark:\n return customStyle.dark || \"Ignite\";\n case ColorThemeKind.HighContrast:\n return customStyle.highContrast || \"HighContrast\";\n case ColorThemeKind.HighContrastLight:\n return customStyle.highContrastLight || \"Illuminate\";\n default:\n return \"\";\n }\n } else if (htmlStyleSetting === \"(server default)\") {\n return \"\";\n } else {\n return htmlStyleSetting;\n }\n}\n\n// if no valid selection, return whole text as only selection\nfunction getCodeSelections(\n selections: ReadonlyArray,\n textDocument: TextDocument,\n): ReadonlyArray | undefined {\n if (selectionsAreNotEmpty(selections)) {\n const codeSelections: Selection[] = selections.filter(\n (selection) => !selection.isEmpty,\n );\n return codeSelections;\n } else {\n const lastLine = textDocument.lineCount - 1;\n const lastCharacter = textDocument.lineAt(lastLine).text.length;\n return [\n new Selection(new Position(0, 0), new Position(lastLine, lastCharacter)),\n ];\n }\n}\n\nfunction getFileName(textDocument: TextDocument): string {\n // Extract the query parameters\n const params = new URL(decodeURIComponent(textDocument.uri.toString()))\n .searchParams;\n\n let pathName: string;\n\n // Massage file path value\n const scheme = textDocument.uri?.scheme;\n if (scheme === \"sasServer\" && params.has(\"id\")) {\n const id = params.get(\"id\");\n // Viya - server file\n // id = /compute/sessions//files/~fs~studiodev~fs~myprogram.sas\n // result = /studiodev/myprogram.sas\n if (/^\\/compute\\/sessions\\/\\w+(-\\w+)+\\/files\\/~fs~[^/]+$/.test(id)) {\n pathName = `${id}`.split(\"/\").pop().replace(/~fs~/g, \"/\");\n }\n if (!pathName) {\n // IOM - server file\n // id = C:\\\\Users\\\\sasdemo\\\\Documents\\\\My SAS Files\\\\9.4\\\\myfolder\\\\myprogram.sas\n pathName = id;\n }\n }\n\n // Local files will default to utilizing the fileName.\n // fileName = c:\\\\Development\\\\VSCODE\\\\files\\\\myprogram.sas\n\n // We could consider utilizing the \"sasContent\" scheme id value but this\n // is the internal uri in the form of /files/files/ which would need to\n // potentially get translated to a readable path name. Also it doesn't work well\n // with the _SASPROGRAMDIR variable as it assumes folder structure.\n\n return pathName ?? textDocument.fileName ?? textDocument.uri?.fsPath;\n}\n"
},
{
"path": "client/src/components/utils/deferred.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\n\nexport interface Deferred {\n promise: Promise;\n resolve: (value: T | PromiseLike) => void;\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n reject: (reason?: any) => void;\n}\n\nexport function deferred() {\n // eslint-disable-next-line @typescript-eslint/consistent-type-assertions\n const deferred = {} as Deferred;\n deferred.promise = new Promise((resolve, reject) => {\n deferred.resolve = resolve;\n deferred.reject = reject;\n });\n return deferred;\n}\n"
},
{
"path": "client/src/components/utils/settings.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { workspace } from \"vscode\";\n\nexport function isOutputHtmlEnabled(): boolean {\n return !!workspace.getConfiguration(\"SAS\").get(\"results.html.enabled\");\n}\n\nexport function getHtmlStyle(): string {\n return workspace.getConfiguration(\"SAS\").get(\"results.html.style\");\n}\n\nexport function isSideResultEnabled(): string {\n return workspace.getConfiguration(\"SAS\").get(\"results.sideBySide\");\n}\n\nexport function isSinglePanelEnabled(): string {\n return workspace.getConfiguration(\"SAS\").get(\"results.singlePanel\");\n}\n\nexport function showLogOnExecutionStart(): boolean {\n return workspace.getConfiguration(\"SAS\").get(\"log.showOnExecutionStart\");\n}\n\nexport function showLogOnExecutionFinish(): boolean {\n return workspace.getConfiguration(\"SAS\").get(\"log.showOnExecutionFinish\");\n}\n\nexport function clearLogOnExecutionStart(): boolean {\n return workspace.getConfiguration(\"SAS\").get(\"log.clearOnExecutionStart\");\n}\n\nexport function isShowProblemsFromSASLogEnabled(): boolean {\n return workspace.getConfiguration(\"SAS\").get(\"problems.log.enabled\");\n}\n\nexport function includeLogInNotebookExport(): boolean {\n return workspace.getConfiguration(\"SAS\").get(\"notebook.export.includeLog\");\n}\n"
},
{
"path": "client/src/components/utils/throttle.ts",
"content": "// Copyright © 2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\n\n/**\n * Run tasks with limited concurrency\n * @param tasks array of tasks to run\n * @param limit limit of tasks that can run in parallel\n * @returns a promise like `Promise.all`\n */\nexport function throttle(tasks: Array<() => Promise>, limit: number) {\n const total = tasks.length;\n const results: T[] = Array(total);\n let count = 0;\n return new Promise((resolve, reject) => {\n function run() {\n const index = total - tasks.length;\n if (index === total) {\n if (count === total) {\n resolve(results);\n }\n return;\n }\n const task = tasks.shift();\n task().then((result) => {\n results[index] = result;\n ++count;\n run();\n }, reject);\n }\n Array(limit).fill(0).forEach(run);\n });\n}\n"
},
{
"path": "client/src/components/utils/treeViewSelections.ts",
"content": "// Copyright © 2024, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { TreeItem, TreeView } from \"vscode\";\n\n/**\n * Gets the selected items from a tree view based on the current selection state.\n *\n * If an item is present and is not part of selections, return the item. Otherwise,\n * return the selections.\n *\n * @param treeView - The VS Code TreeView instance\n * @param item - The item that was clicked/activated\n * @returns An array of selected items\n */\nexport function treeViewSelections(\n treeView: TreeView,\n item: T | undefined,\n): T[] {\n if (item) {\n const itemIsInSelection = treeView.selection.some(\n ({ id }) => id === item.id,\n );\n if (itemIsInSelection) {\n return [...treeView.selection];\n }\n\n return [item];\n }\n\n return [...treeView.selection];\n}\n"
},
{
"path": "client/src/connection/index.ts",
"content": "// Copyright © 2022-2023, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { l10n } from \"vscode\";\n\nimport {\n AuthType,\n ConnectionType,\n ProfileConfig,\n ViyaProfile,\n toAutoExecLines,\n} from \"../components/profile\";\nimport { getSession as getITCSession } from \"./itc\";\nimport { ITCProtocol } from \"./itc/types\";\nimport { Config as RestConfig, getSession as getRestSession } from \"./rest\";\nimport {\n Error2 as ComputeError,\n LogLine as ComputeLogLine,\n LogLineTypeEnum as ComputeLogLineTypeEnum,\n} from \"./rest/api/compute\";\nimport { Session } from \"./session\";\nimport { getSession as getSSHSession } from \"./ssh\";\n\nlet profileConfig: ProfileConfig;\n\nexport type ErrorRepresentation = ComputeError;\nexport type LogLine = ComputeLogLine;\nexport type LogLineTypeEnum = ComputeLogLineTypeEnum;\nexport type OnLogFn = (logs: LogLine[]) => void;\n\nexport interface RunResult {\n html5?: string;\n title?: string;\n}\n\nexport interface BaseConfig {\n sasOptions?: string[];\n autoExecLines?: string[];\n}\n\nexport function getSession(): Session {\n if (!profileConfig) {\n profileConfig = new ProfileConfig();\n }\n // retrieve active & valid profile\n const activeProfile = profileConfig.getActiveProfileDetail();\n const validProfile = profileConfig.validateProfile(activeProfile);\n\n if (validProfile.type === AuthType.Error) {\n throw new Error(validProfile.error);\n }\n\n switch (validProfile.profile?.connectionType) {\n case ConnectionType.Rest:\n return getRestSession(toRestConfig(validProfile.profile));\n case ConnectionType.SSH:\n return getSSHSession(validProfile.profile);\n case ConnectionType.COM:\n return getITCSession(validProfile.profile, ITCProtocol.COM);\n case ConnectionType.IOM:\n return getITCSession(validProfile.profile, ITCProtocol.IOMBridge);\n default:\n throw new Error(\n l10n.t(\"Invalid connectionType. Check Profile settings.\"),\n );\n }\n}\n\n/**\n * Translates a {@link ViyaProfile} interface to a {@link RestConfig} interface.\n * @param profile an input {@link ViyaProfile} to translate.\n * @returns RestConfig instance derived from the input profile.\n */\nfunction toRestConfig(profile: ViyaProfile): RestConfig {\n const mapped: RestConfig = profile;\n if (profile.autoExec) {\n mapped.autoExecLines = toAutoExecLines(profile.autoExec);\n }\n return mapped;\n}\n"
},
{
"path": "client/src/connection/itc/CodeRunner.ts",
"content": "// Copyright © 2024, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { commands } from \"vscode\";\n\nimport { v4 } from \"uuid\";\n\nimport { ITCSession } from \".\";\nimport { LogLine, getSession } from \"..\";\nimport { useRunStore } from \"../../store\";\nimport { Session } from \"../session\";\nimport { extractTextBetweenTags } from \"../util\";\n\nlet wait: Promise | undefined;\n\nexport async function executeRawCode(code: string): Promise {\n const randomId = v4();\n const startTag = `<${randomId}>`;\n const endTag = ``;\n const task = () =>\n _runCode(\n async (session) => {\n // eslint-disable-next-line @typescript-eslint/consistent-type-assertions\n await (session as ITCSession).execute(\n `Write-Host \"${startTag}\"\\n${code}\\nWrite-Host \"${endTag}\"\\n`,\n );\n },\n startTag,\n endTag,\n );\n\n wait = wait ? wait.then(task) : task();\n return wait;\n}\n\nexport async function runCode(\n code: string,\n startTag: string = \"\",\n endTag: string = \"\",\n): Promise {\n const task = () =>\n _runCode(\n // eslint-disable-next-line @typescript-eslint/consistent-type-assertions\n async (session) => await (session as ITCSession).run(code, true),\n startTag,\n endTag,\n );\n\n wait = wait ? wait.then(task) : task();\n return wait;\n}\n\nasync function _runCode(\n runCallback: (session: Session) => void,\n startTag: string = \"\",\n endTag: string = \"\",\n): Promise {\n // If we're already executing code, lets wait for it\n // to finish up.\n let unsubscribe;\n if (useRunStore.getState().isExecutingCode) {\n await new Promise((resolve) => {\n unsubscribe = useRunStore.subscribe(\n (state) => state.isExecutingCode,\n (isExecutingCode) => !isExecutingCode && resolve(true),\n );\n });\n }\n\n const { setIsExecutingCode } = useRunStore.getState();\n setIsExecutingCode(true, false);\n commands.executeCommand(\"setContext\", \"SAS.running\", true);\n const session = getSession();\n\n let logText = \"\";\n const onExecutionLogFn = session.onExecutionLogFn;\n const outputLines = [];\n\n const addLine = (logLines: LogLine[]) => {\n outputLines.push(...logLines.map(({ line }) => line));\n };\n\n try {\n await session.setup(true);\n\n // Lets capture output to use it on\n session.onExecutionLogFn = addLine;\n\n await runCallback(session);\n\n const logOutput = outputLines.filter((line) => line.trim()).join(\"\");\n logText = extractTextBetweenTags(logOutput, startTag, endTag);\n } finally {\n unsubscribe && unsubscribe();\n // Lets update our session to write to the log\n session.onExecutionLogFn = onExecutionLogFn;\n\n setIsExecutingCode(false);\n commands.executeCommand(\"setContext\", \"SAS.running\", false);\n }\n\n return logText;\n}\n"
},
{
"path": "client/src/connection/itc/ItcLibraryAdapter.ts",
"content": "// Copyright © 2024, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { l10n } from \"vscode\";\n\nimport type { SortModelItem } from \"ag-grid-community\";\nimport { ChildProcessWithoutNullStreams } from \"child_process\";\n\nimport { onRunError } from \"../../commands/run\";\nimport {\n LibraryAdapter,\n LibraryItem,\n TableData,\n TableQuery,\n TableRow,\n} from \"../../components/LibraryNavigator/types\";\nimport { ColumnCollection, TableInfo } from \"../rest/api/compute\";\nimport { getColumnIconType } from \"../util\";\nimport { executeRawCode, runCode } from \"./CodeRunner\";\nimport { Config } from \"./types\";\n\nclass ItcLibraryAdapter implements LibraryAdapter {\n protected hasEstablishedConnection: boolean = false;\n protected shellProcess: ChildProcessWithoutNullStreams;\n protected pollingForLogResults: boolean = false;\n protected log: string[] = [];\n protected endTag: string = \"\";\n protected outputFinished: boolean = false;\n protected config: Config;\n\n public async connect(): Promise {\n this.hasEstablishedConnection = true;\n }\n\n public async setup(): Promise {\n if (this.hasEstablishedConnection) {\n return;\n }\n\n await this.connect();\n }\n\n public async deleteTable(item: LibraryItem): Promise {\n const code = `\n proc datasets library=${item.library} nolist nodetails; delete ${item.name}; run;\n `;\n\n await this.runCode(code);\n }\n\n public async getColumns(item: LibraryItem): Promise {\n const code = `\n $runner.GetColumns(\"${item.library}\", \"${item.name}\")\n `;\n const output = await executeRawCode(code);\n const rawColumns = JSON.parse(output);\n const columns = rawColumns.map((column) => ({\n ...column,\n type: getColumnIconType(column),\n }));\n return {\n items: columns,\n count: -1,\n };\n }\n public async getLibraries(): Promise<{\n items: LibraryItem[];\n count: number;\n }> {\n const code = `\n $runner.GetLibraries()\n `;\n\n const output = await executeRawCode(code);\n const rawLibraries = JSON.parse(output).libraries;\n\n const libraries = rawLibraries.map((row: string[]) => {\n const [libName, readOnlyValue] = row;\n return {\n type: \"library\",\n uid: libName,\n id: libName,\n name: libName,\n readOnly: readOnlyValue === \"yes\",\n };\n });\n\n return {\n items: libraries,\n count: -1,\n };\n }\n\n public async getRows(\n item: LibraryItem,\n start: number,\n limit: number,\n sortModel: SortModelItem[],\n query: TableQuery | undefined,\n ): Promise {\n const { rows: rawRowValues, count } = await this.getDatasetInformation(\n item,\n start,\n limit,\n sortModel,\n query,\n );\n\n const rows = rawRowValues.map((line, idx: number): TableRow => {\n const rowData = [`${start + idx + 1}`].concat(line);\n return { cells: rowData };\n });\n\n return {\n rows,\n count,\n };\n }\n\n public async getRowsAsCSV(\n item: LibraryItem,\n start: number,\n limit: number,\n ): Promise {\n // We only need the columns for the first page of results\n const columns =\n start === 0\n ? {\n columns: [\"INDEX\"].concat(\n (await this.getColumns(item)).items.map((column) => column.name),\n ),\n }\n : {};\n\n const { rows } = await this.getRows(item, start, limit, [], undefined);\n\n rows.unshift(columns);\n // Fetching csv doesn't rely on count. Instead, we get the count\n // upfront via getTableRowCount\n return { rows, count: -1 };\n }\n\n public async getTableRowCount(\n item: LibraryItem,\n ): Promise<{ rowCount: number; maxNumberOfRowsToRead: number }> {\n const code = `\n proc sql;\n SELECT COUNT(1) into: COUNT FROM ${item.library}.${item.name};\n quit;\n %put &COUNT ;\n `;\n\n const output = await this.runCode(code, \"\", \" \");\n const rowCount = parseInt(output.replace(/[^0-9]/g, \"\"), 10);\n\n return { rowCount, maxNumberOfRowsToRead: 100 };\n }\n\n public async getTables(item: LibraryItem): Promise<{\n items: LibraryItem[];\n count: number;\n }> {\n const code = `\n $runner.GetTables(\"${item.name}\")\n `;\n\n const output = await executeRawCode(code);\n const rawTables = JSON.parse(output).tables;\n const tables = rawTables.map((table: string): LibraryItem => {\n return {\n type: \"table\",\n uid: `${item.name!}.${table}`,\n id: table,\n name: table,\n library: item.name,\n readOnly: item.readOnly,\n };\n });\n\n return { items: tables, count: -1 };\n }\n\n protected async getDatasetInformation(\n item: LibraryItem,\n start: number,\n limit: number,\n sortModel: SortModelItem[],\n query: TableQuery | undefined,\n ): Promise<{ rows: Array; count: number }> {\n const sortString = sortModel\n .map((col) => `${col.colId} ${col.sort}`)\n .join(\",\");\n const code = `\n $runner.GetDatasetRecords(\"${item.library}\",\"${item.name}\", ${start}, ${limit}, \"${sortString}\", '${query ? JSON.stringify(query) : \"\"}')\n `;\n const output = await executeRawCode(code);\n try {\n return JSON.parse(output);\n } catch (e) {\n console.warn(\"Failed to load table data with error\", e);\n console.warn(\"Raw output\", output);\n throw new Error(\n l10n.t(\n \"An error was encountered when loading table data. This usually happens when a table is too large or the data couldn't be processed. See console for more details.\",\n ),\n );\n }\n }\n\n public async getTableInfo(item: LibraryItem): Promise {\n const basicInfo: TableInfo = {\n bookmarkLength: 0, // Not available in vtable\n columnCount: 0,\n compressionRoutine: \"\",\n creationTimeStamp: \"\",\n encoding: \"\",\n engine: \"\",\n extendedType: \"\",\n label: \"\",\n libref: item.library,\n logicalRecordCount: 0,\n modifiedTimeStamp: \"\",\n name: item.name,\n physicalRecordCount: 0,\n recordLength: 0,\n rowCount: 0,\n type: \"DATA\",\n };\n\n try {\n // Use the PowerShell GetTableInfo function which queries sashelp.vtable\n const code = `\n $runner.GetTableInfo(\"${item.library}\", \"${item.name}\")\n `;\n const output = await executeRawCode(code);\n const tableInfo = JSON.parse(output);\n\n return {\n ...basicInfo,\n columnCount: tableInfo.columnCount || basicInfo.columnCount,\n compressionRoutine:\n tableInfo.compressionRoutine || basicInfo.compressionRoutine,\n creationTimeStamp:\n tableInfo.creationTimeStamp || basicInfo.creationTimeStamp,\n encoding: tableInfo.encoding || basicInfo.encoding,\n engine: tableInfo.engine || basicInfo.engine,\n extendedType: tableInfo.extendedType || basicInfo.extendedType,\n label: tableInfo.label || basicInfo.label,\n libref: tableInfo.libref || basicInfo.libref,\n logicalRecordCount: tableInfo.rowCount || basicInfo.logicalRecordCount,\n modifiedTimeStamp:\n tableInfo.modifiedTimeStamp || basicInfo.modifiedTimeStamp,\n name: tableInfo.name || basicInfo.name,\n physicalRecordCount:\n tableInfo.rowCount || basicInfo.physicalRecordCount,\n recordLength: tableInfo.recordLength || basicInfo.recordLength,\n rowCount: tableInfo.rowCount || basicInfo.rowCount,\n type: tableInfo.type || basicInfo.type,\n };\n } catch (error) {\n console.warn(\"Failed to get table info:\", error);\n // If anything fails, return basic info\n return basicInfo;\n }\n }\n\n protected async executionHandler(\n callback: () => Promise,\n ): Promise {\n try {\n return await callback();\n } catch (e) {\n onRunError(e);\n return \"\";\n }\n }\n\n protected async runCode(\n code: string,\n startTag: string = \"\",\n endTag: string = \"\",\n ): Promise {\n return this.executionHandler(() => runCode(code, startTag, endTag));\n }\n\n protected async executeRawCode(code: string): Promise {\n return this.executionHandler(() => executeRawCode(code));\n }\n}\n\nexport default ItcLibraryAdapter;\n"
},
{
"path": "client/src/connection/itc/ItcServerAdapter.ts",
"content": "// Copyright © 2025, SAS Institute Inc., Cary, NC, USA. All Rights Reserved.\n// SPDX-License-Identifier: Apache-2.0\nimport { FileType, Uri, workspace } from \"vscode\";\n\nimport { v4 } from \"uuid\";\n\nimport { onRunError } from \"../../commands/run\";\nimport {\n Messages,\n SAS_SERVER_ROOT_FOLDER,\n SAS_SERVER_ROOT_FOLDERS,\n SERVER_FOLDER_ID,\n} from \"../../components/ContentNavigator/const\";\nimport {\n ContentAdapter,\n ContentItem,\n RootFolderMap,\n} from \"../../components/ContentNavigator/types\";\nimport {\n ContextMenuAction,\n ContextMenuProvider,\n convertStaticFolderToContentItem,\n createStaticFolder,\n homeDirectoryNameAndType,\n sortedContentItems,\n} from \"../../components/ContentNavigator/utils\";\nimport { getGlobalStorageUri } from \"../../components/ExtensionContext\";\nimport { ProfileWithFileRootOptions } from \"../../components/profile\";\nimport { getLink, getResourceId, getSasServerUri } from \"../rest/util\";\nimport { executeRawCode } from \"./CodeRunner\";\nimport { PowershellResponse, ScriptActions } from \"./types\";\nimport { getDirectorySeparator } from \"./util\";\n\nclass ItcServerAdapter implements ContentAdapter {\n protected sessionId: string;\n private rootFolders: RootFolderMap;\n private contextMenuProvider: ContextMenuProvider;\n\n public constructor(\n protected readonly fileNavigationCustomRootPath: ProfileWithFileRootOptions[\"fileNavigationCustomRootPath\"],\n protected readonly fileNavigationRoot: ProfileWithFileRootOptions[\"fileNavigationRoot\"],\n ) {\n this.rootFolders = {};\n this.contextMenuProvider = new ContextMenuProvider(\n [\n ContextMenuAction.CreateChild,\n ContextMenuAction.Delete,\n ContextMenuAction.Update,\n ContextMenuAction.CopyPath,\n ContextMenuAction.AllowDownload,\n ],\n {\n [ContextMenuAction.CopyPath]: (item) => item.id !== SERVER_FOLDER_ID,\n },\n );\n }\n\n /* The following methods are needed for favorites, which are not applicable to sas server */\n public async addChildItem(): Promise {\n throw new Error(\"Method not implemented\");\n }\n public async addItemToFavorites(): Promise {\n throw new Error(\"Method not implemented\");\n }\n public removeItemFromFavorites(): Promise {\n throw new Error(\"Method not implemented\");\n }\n public getRootFolder(): ContentItem | undefined {\n return undefined;\n }\n\n /* The following is needed for creating a flow, which isn't supported on sas server */\n public async getParentOfItem(\n item: ContentItem,\n ): Promise {\n const parent = await this.getItemAtPath(item.parentFolderUri);\n if (!parent) {\n return undefined;\n }\n\n return parent;\n }\n\n public async getFolderPathForItem(): Promise {\n return \"\";\n }\n\n public async connect(): Promise