feat: provide getLibraries and getColumns API for Rest (#1450)

Author: scnwwu

client/src/components/APIProvider.ts

@@ -0,0 +1,32 @@
+// Copyright © 2025, SAS Institute Inc., Cary, NC, USA.  All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+import { authentication } from "vscode";
+
+import { profileConfig } from "../commands/profile";
+import { SASAuthProvider } from "./AuthProvider";
+import { ConnectionType } from "./profile";
+
+/* only Rest APIs for now */
+
+const apis = {};
+
+export const registerAPI = (name: string, fn) => {
+  apis[name] = fn;
+};
+
+export const getRestAPIs = async (accessToken: string) => {
+  const activeProfile = profileConfig.getProfileByName(
+    profileConfig.getActiveProfile(),
+  );
+  if (!activeProfile || activeProfile.connectionType !== ConnectionType.Rest) {
+    return;
+  }
+  const session = await authentication.getSession(SASAuthProvider.id, [], {
+    silent: true,
+  });
+  if (session.accessToken !== accessToken) {
+    return;
+  }
+
+  return apis;
+};

client/src/components/LibraryNavigator/LibraryDataProvider.ts

@@ -24,6 +24,7 @@ import {
 
 import { Writable } from "stream";
 
+import { registerAPI } from "../APIProvider";
 import { SubscriptionProvider } from "../SubscriptionProvider";
 import LibraryModel from "./LibraryModel";
 import { Icons, Messages, WorkLibraryId } from "./const";
@@ -62,6 +63,9 @@ class LibraryDataProvider
       this.selector(),
       this,
     );
+    registerAPI("getColumns", model.fetchColumns.bind(model));
+    registerAPI("getTables", model.getTables.bind(model));
+    registerAPI("getLibraries", model.getLibraries.bind(model));
   }
 
   public getSubscriptions(): Disposable[] {

client/src/components/LibraryNavigator/LibraryModel.ts

@@ -137,7 +137,7 @@ class LibraryModel {
     return await this.getTables(item);
   }
 
-  private async getLibraries(): Promise<LibraryItem[]> {
+  public async getLibraries(): Promise<LibraryItem[]> {
     await this.libraryAdapter.setup();
 
     let offset = 0;
@@ -159,7 +159,7 @@ class LibraryModel {
     return this.processItems(items, "library", undefined);
   }
 
-  private async getTables(item: LibraryItem): Promise<LibraryItem[]> {
+  public async getTables(item: LibraryItem): Promise<LibraryItem[]> {
     await this.libraryAdapter.setup();
 
     let offset = 0;

client/src/node/extension.ts

@@ -32,6 +32,7 @@ import {
   updateProfile,
 } from "../commands/profile";
 import { run, runRegion, runSelected } from "../commands/run";
+import { getRestAPIs } from "../components/APIProvider";
 import { SASAuthProvider } from "../components/AuthProvider";
 import { installCAs } from "../components/CAHelper";
 import ContentNavigator from "../components/ContentNavigator";
@@ -61,7 +62,7 @@ let client: LanguageClient;
 
 export let extensionContext: ExtensionContext | undefined;
 
-export function activate(context: ExtensionContext): void {
+export function activate(context: ExtensionContext) {
   // The server is implemented in node
   extensionContext = context;
   const serverModule = context.asAbsolutePath(
@@ -212,6 +213,10 @@ export function activate(context: ExtensionContext): void {
   profileConfig.migrateLegacyProfiles();
   triggerProfileUpdate();
   updateViewSettings();
+
+  return {
+    getRestAPIs,
+  };
 }
 
 function updateViewSettings(): void {