feat(LibraryNavigator) wip for adding properties and column information

kpatl1 · kpatl1 · commit 468e3c4e4e8b · 2025-08-19T17:04:28.000-04:00
Signed-off-by: Kishan Patel &lt;kpatel7927@gmail.com&gt;
diff --git a/client/src/components/LibraryNavigator/LibraryDataProvider.ts b/client/src/components/LibraryNavigator/LibraryDataProvider.ts
@@ -66,6 +66,7 @@ class LibraryDataProvider
     registerAPI("getColumns", model.fetchColumns.bind(model));
     registerAPI("getTables", model.getTables.bind(model));
     registerAPI("getLibraries", model.getLibraries.bind(model));
+    registerAPI("getTableInfo", model.getTableInfo.bind(model));
   }
 
   public getSubscriptions(): Disposable[] {
@@ -178,6 +179,14 @@ class LibraryDataProvider
     this.model.useAdapter(libraryAdapter);
     this._onDidChangeTreeData.fire(undefined);
   }
+
+  public async getTableInfo(item: LibraryItem) {
+    return await this.model.getTableInfo(item);
+  }
+
+  public async fetchColumns(item: LibraryItem) {
+    return await this.model.fetchColumns(item);
+  }
 }
 
 export default LibraryDataProvider;
diff --git a/client/src/components/LibraryNavigator/LibraryModel.ts b/client/src/components/LibraryNavigator/LibraryModel.ts
@@ -126,6 +126,14 @@ class LibraryModel {
     }
   }
 
+  public async getTableInfo(item: LibraryItem) {
+    await this.libraryAdapter.setup();
+    if (this.libraryAdapter.getTableInfo) {
+      return await this.libraryAdapter.getTableInfo(item);
+    }
+    throw new Error("Table properties not supported for this connection type");
+  }
+
   public async getChildren(item?: LibraryItem): Promise<LibraryItem[]> {
     if (!this.libraryAdapter) {
       return [];
diff --git a/client/src/components/LibraryNavigator/index.ts b/client/src/components/LibraryNavigator/index.ts
@@ -17,6 +17,7 @@ import * as path from "path";
 import { profileConfig } from "../../commands/profile";
 import { Column } from "../../connection/rest/api/compute";
 import DataViewer from "../../panels/DataViewer";
+import TablePropertiesViewer from "../../panels/TablePropertiesViewer";
 import { WebViewManager } from "../../panels/WebviewManager";
 import { SubscriptionProvider } from "../SubscriptionProvider";
 import LibraryAdapterFactory from "./LibraryAdapterFactory";
@@ -101,6 +102,54 @@ class LibraryNavigator implements SubscriptionProvider {
           );
         },
       ),
+      commands.registerCommand(
+        "SAS.showTableProperties",
+        async (item: LibraryItem) => {
+          try {
+            const tableInfo = await this.libraryDataProvider.getTableInfo(item);
+            const columns = await this.libraryDataProvider.fetchColumns(item);
+
+            this.webviewManager.render(
+              new TablePropertiesViewer(
+                this.extensionUri,
+                item.uid,
+                tableInfo,
+                columns,
+                false, // Show properties tab
+              ),
+              `properties-${item.uid}`,
+            );
+          } catch (error) {
+            window.showErrorMessage(
+              `Failed to load table properties: ${error.message}`,
+            );
+          }
+        },
+      ),
+      commands.registerCommand(
+        "SAS.showTableColumns",
+        async (item: LibraryItem) => {
+          try {
+            const tableInfo = await this.libraryDataProvider.getTableInfo(item);
+            const columns = await this.libraryDataProvider.fetchColumns(item);
+
+            this.webviewManager.render(
+              new TablePropertiesViewer(
+                this.extensionUri,
+                item.uid,
+                tableInfo,
+                columns,
+                true, // Show columns tab
+              ),
+              `columns-${item.uid}`,
+            );
+          } catch (error) {
+            window.showErrorMessage(
+              `Failed to load table columns: ${error.message}`,
+            );
+          }
+        },
+      ),
       commands.registerCommand("SAS.collapseAllLibraries", () => {
         commands.executeCommand(
           "workbench.actions.treeView.librarydataprovider.collapseAll",
diff --git a/client/src/components/LibraryNavigator/types.ts b/client/src/components/LibraryNavigator/types.ts
@@ -1,6 +1,6 @@
 // Copyright © 2023, SAS Institute Inc., Cary, NC, USA.  All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
-import { ColumnCollection } from "../../connection/rest/api/compute";
+import { ColumnCollection, TableInfo } from "../../connection/rest/api/compute";
 
 export const LibraryType = "library";
 export const TableType = "table";
@@ -56,5 +56,6 @@ export interface LibraryAdapter {
     items: LibraryItem[];
     count: number;
   }>;
+  getTableInfo?(item: LibraryItem): Promise<TableInfo>;
   setup(): Promise<void>;
 }
diff --git a/client/src/connection/itc/ItcLibraryAdapter.ts b/client/src/connection/itc/ItcLibraryAdapter.ts
@@ -11,7 +11,7 @@ import {
   TableData,
   TableRow,
 } from "../../components/LibraryNavigator/types";
-import { ColumnCollection } from "../rest/api/compute";
+import { ColumnCollection, TableInfo } from "../rest/api/compute";
 import { getColumnIconType } from "../util";
 import { executeRawCode, runCode } from "./CodeRunner";
 import { Config } from "./types";
@@ -196,6 +196,28 @@ class ItcLibraryAdapter implements LibraryAdapter {
     }
   }
 
+  public async getTableInfo(item: LibraryItem): Promise<TableInfo> {
+    const code = `
+      $runner.GetTableInfo("${item.library}", "${item.name}")
+    `;
+    const output = await executeRawCode(code);
+    const tableInfo = JSON.parse(output);
+
+    return {
+      name: tableInfo.name,
+      type: tableInfo.type,
+      creationTimeStamp: tableInfo.creationTimeStamp,
+      modifiedTimeStamp: tableInfo.modifiedTimeStamp,
+      rowCount: tableInfo.rowCount,
+      columnCount: tableInfo.columnCount,
+      compressionRoutine: tableInfo.compressionRoutine,
+      label: tableInfo.label,
+      engine: tableInfo.engine,
+      extendedType: tableInfo.extendedType,
+      libref: tableInfo.libref,
+    };
+  }
+
   protected async executionHandler(
     callback: () => Promise<string>,
   ): Promise<string> {
diff --git a/client/src/connection/itc/script.ts b/client/src/connection/itc/script.ts
@@ -580,6 +580,44 @@ class SASRunner{
     Write-Host $(ConvertTo-Json -Depth 10 -InputObject $result -Compress)
   }
 
+  [void]GetTableInfo([string]$libname, [string]$memname) {
+    $objRecordSet = New-Object -comobject ADODB.Recordset
+    $objRecordSet.ActiveConnection = $this.dataConnection
+    $query = @"
+      select memname, memtype, crdate, modate, nobs, nvar, compress, 
+             memlabel, engine, typemem, filesize, delobs
+      from sashelp.vtable
+      where libname='$libname' and memname='$memname';
+"@
+    $objRecordSet.Open(
+      $query,
+      [System.Reflection.Missing]::Value, # Use the active connection
+      2, # adOpenDynamic
+      1, # adLockReadOnly
+      1  # adCmdText
+    )
+
+    $result = New-Object psobject
+    if (-not $objRecordSet.EOF) {
+      $result | Add-Member -MemberType NoteProperty -Name "name" -Value $objRecordSet.Fields.Item(0).Value
+      $result | Add-Member -MemberType NoteProperty -Name "type" -Value $objRecordSet.Fields.Item(1).Value
+      $result | Add-Member -MemberType NoteProperty -Name "creationTimeStamp" -Value $objRecordSet.Fields.Item(2).Value
+      $result | Add-Member -MemberType NoteProperty -Name "modifiedTimeStamp" -Value $objRecordSet.Fields.Item(3).Value
+      $result | Add-Member -MemberType NoteProperty -Name "rowCount" -Value $objRecordSet.Fields.Item(4).Value
+      $result | Add-Member -MemberType NoteProperty -Name "columnCount" -Value $objRecordSet.Fields.Item(5).Value
+      $result | Add-Member -MemberType NoteProperty -Name "compressionRoutine" -Value $objRecordSet.Fields.Item(6).Value
+      $result | Add-Member -MemberType NoteProperty -Name "label" -Value $objRecordSet.Fields.Item(7).Value
+      $result | Add-Member -MemberType NoteProperty -Name "engine" -Value $objRecordSet.Fields.Item(8).Value
+      $result | Add-Member -MemberType NoteProperty -Name "extendedType" -Value $objRecordSet.Fields.Item(9).Value
+      $result | Add-Member -MemberType NoteProperty -Name "fileSize" -Value $objRecordSet.Fields.Item(10).Value
+      $result | Add-Member -MemberType NoteProperty -Name "deletedObs" -Value $objRecordSet.Fields.Item(11).Value
+      $result | Add-Member -MemberType NoteProperty -Name "libref" -Value $libname
+    }
+    $objRecordSet.Close()
+
+    Write-Host $(ConvertTo-Json -Depth 10 -InputObject $result -Compress)
+  }
+
   [void]GetTables([string]$libname) {
     $objRecordSet = New-Object -comobject ADODB.Recordset
     $objRecordSet.ActiveConnection = $this.dataConnection
diff --git a/client/src/connection/rest/RestLibraryAdapter.ts b/client/src/connection/rest/RestLibraryAdapter.ts
@@ -9,7 +9,12 @@ import {
   TableData,
 } from "../../components/LibraryNavigator/types";
 import { appendSessionLogFn } from "../../components/logViewer";
-import { ColumnCollection, DataAccessApi, RowCollection } from "./api/compute";
+import {
+  ColumnCollection,
+  DataAccessApi,
+  RowCollection,
+  TableInfo,
+} from "./api/compute";
 import { getApiConfig } from "./common";
 
 const requestOptions = {
@@ -137,6 +142,23 @@ class RestLibraryAdapter implements LibraryAdapter {
     return { rowCount: response.data.rowCount, maxNumberOfRowsToRead: 1000 };
   }
 
+  public async getTableInfo(item: LibraryItem): Promise<TableInfo> {
+    await this.setup();
+    const response = await this.retryOnFail(
+      async () =>
+        await this.dataAccessApi.getTable(
+          {
+            sessionId: this.sessionId,
+            libref: item.library || "",
+            tableName: item.name,
+          },
+          { headers: { Accept: "application/json" } },
+        ),
+    );
+
+    return response.data;
+  }
+
   private async retryOnFail<T>(
     callbackFn: () => Promise<AxiosResponse<T>>,
   ): Promise<AxiosResponse<T>> {
diff --git a/client/src/panels/TablePropertiesViewer.ts b/client/src/panels/TablePropertiesViewer.ts
diff --git a/package.json b/package.json
diff --git a/package.nls.json b/package.nls.json

Original file line number	Diff line number	Diff line change
`@@ -126,6 +126,14 @@ class LibraryModel {`
`126`	`126`	`}`
`127`	`127`	`}`
`128`	`128`
	`129`	`+ public async getTableInfo(item: LibraryItem) {`
	`130`	`+ await this.libraryAdapter.setup();`
	`131`	`+ if (this.libraryAdapter.getTableInfo) {`
	`132`	`+ return await this.libraryAdapter.getTableInfo(item);`
	`133`	`+ }`
	`134`	`+ throw new Error("Table properties not supported for this connection type");`
	`135`	`+ }`
	`136`	`+`
`129`	`137`	`public async getChildren(item?: LibraryItem): Promise<LibraryItem[]> {`
`130`	`138`	`if (!this.libraryAdapter) {`
`131`	`139`	`return [];`