@stack-spot/portal-network 1.0.0-betaadp.3 → 1.0.0-dev.1768423147036

package/src/index.ts

@@ -1,7 +1,7 @@
 export { CancelledError } from '@tanstack/react-query'
-export { apiAddresses, getApiAddresses, getBaseUrlWithOverride, setApisOverride } from './api-addresses'
+export { apiAddresses, getApiAddresses, getApisBaseUrlConfig, getBaseUrlByTenantWithOverride } from './api-addresses'
 export { accountClient } from './client/account'
-export * from './client/discover'
+export { accountAssetManagerClient } from './client/account-asset-manager'
 export { agentClient } from './client/agent'
 export { agentToolsClient } from './client/agent-tools'
 export { aiClient } from './client/ai'
@@ -14,7 +14,9 @@ export { cloudServicesClient } from './client/cloud-services'
 export { codeShiftClient } from './client/code-shift'
 export { contentClient } from './client/content'
 export { dataIntegrationClient } from './client/data-integration'
+export * from './client/discover'
 export { discoverClient } from './client/discover'
+export { edpBfaClient } from './client/edp-bfa'
 export { eventBusClient } from './client/event-bus'
 export { genAiInferenceClient } from './client/gen-ai-inference'
 export { insightsClient } from './client/insights'
@@ -35,9 +37,10 @@ export { StreamError } from './error/StreamError'
 export { StreamJsonError } from './error/StreamJsonError'
 export { NetworkClient } from './network/NetworkClient'
 export { queryClient } from './network/react-query-client'
-export { OperationResult, OperationVariables, UseQueryObjectOptions } from './network/types'
+export { Env, OperationResult, OperationVariables, Tenant, UseQueryObjectOptions } from './network/types'
 export { StreamingStatus } from './types'
 export { StreamedArray } from './utils/StreamedArray'
 export { StreamedJson } from './utils/StreamedJson'
 export { useExtendedList } from './utils/use-extended-list'
 export { useStreamedArray } from './utils/use-streamed-array'
+

package/src/network/NetworkClient.ts

@@ -1,9 +1,10 @@
-import { AuthenticationError, SessionExpiredError } from '@stack-spot/auth'
-import { requestPermission } from '@stack-spot/opa'
+import { AuthenticationError, Session, SessionExpiredError } from '@stack-spot/auth'
+import { requestPermission, setup as setupPermissions } from '@stack-spot/opa'
 import { events } from 'fetch-event-stream'
+import { getApisBaseUrlConfig, getBaseUrlByTenantWithOverride, getPermissionsAPIMap } from '../api-addresses'
 import { StackspotAPIError } from '../error/StackspotAPIError'
-import { Env, FetchEventStream, HTTPMethod, SessionManager } from './types'
-import { getBaseUrlWithOverride } from '../api-addresses'
+import { queryClient } from './react-query-client'
+import { Env, FetchEventStream, HTTPMethod, SessionManager, Tenant } from './types'
 
 /**
  * A set of methods for performing network requests to an API.
@@ -21,6 +22,7 @@ export abstract class NetworkClient {
   private baseURL: Record<Env, string>
   static readonly sessionManager?: SessionManager
   private static env?: Env
+  private static tenant?: Tenant 
 
   /**
    * @param baseURL An object with the keys "dev", "stg" and "prd". The values must be the url for each of these environments.
@@ -33,11 +35,21 @@ export abstract class NetworkClient {
    * Sets up all network clients. Must be called before attempting to make any request.
    * @param sessionManager An object with functions capable of checking, retrieving and ending the current session.
    * @param env The environment to send the requests to.
+   * @param tenant - {@link Tenant} tenant identifier used to scope requests.
    */
-  static setup(sessionManager: SessionManager, env: Env) {
+  static setup(sessionManager: SessionManager, env: Env, tenant?: Tenant) {
     //@ts-ignore This is the only place we would link session manager to be set.
     NetworkClient.sessionManager = sessionManager
     NetworkClient.env = env
+    NetworkClient.tenant = tenant
+    const url = getApisBaseUrlConfig(env, tenant).permissionValidation
+    const apiMap = getPermissionsAPIMap(env, tenant)
+    const onChange = (session?: Session) => {
+      queryClient.invalidateQueries()
+      session && setupPermissions({ url, session, apiMap })
+    }
+    sessionManager.onChange?.(onChange)
+    if (sessionManager.hasSession()) onChange(sessionManager.getSession())
   }
 
   /**
@@ -62,7 +74,7 @@ export abstract class NetworkClient {
    * @returns the final baseURL for the current environment, considering overrides if available.
    */
   protected getBaseURL(): string {
-    return getBaseUrlWithOverride(this.baseURL, NetworkClient.env);
+    return getBaseUrlByTenantWithOverride(this.baseURL, NetworkClient.env, NetworkClient.tenant)
   }
 
   /**

package/src/network/types.ts

@@ -8,8 +8,11 @@ export interface SessionManager {
   hasSession(): boolean,
   endSession(): void,
   getSession(): Session,
+  onChange?: (listener: (session: Session | undefined) => void) => (() => void),
 }
 
+export type Tenant = 'stackspot' | 'itau'
+
 export type Env = 'dev' | 'stg' | 'prd'
 
 export type HTTPMethod = 'post' | 'patch' | 'delete' | 'put' | 'get'

package/src/utils/StreamedJson.tsx

@@ -31,6 +31,7 @@ interface ConstructorParams<T> {
    * of merged.
    */
   ignoreKeys?: (keyof T)[],
+  textFromErrorEvent?: (data: Partial<T>) => string,
 }
 
 /**
@@ -44,16 +45,18 @@ export class StreamedJson<T> {
   private abortController: AbortController | undefined
   private transform?: (event: Partial<T>, data: Partial<T>) => void | Promise<void>
   private ignoreKeys?: (keyof T)[]
+  private textFromErrorEvent: (data: Partial<T>) => string
 
   /**
    * @param response the fetch response.
    * @param minChangeIntervalMS a stream can be too fast. This sets a minimum interval between running the listeners. The default is 50ms.
    */
-  constructor({ eventsPromise, abortController, minChangeIntervalMS = 50, transform, ignoreKeys }: ConstructorParams<T>) {
+  constructor({ eventsPromise, abortController, minChangeIntervalMS = 50, transform, ignoreKeys, textFromErrorEvent }: ConstructorParams<T>) {
     this.abortController = abortController
     this.transform = transform
     this.ignoreKeys = ignoreKeys
     this.run(eventsPromise, minChangeIntervalMS)
+    this.textFromErrorEvent = textFromErrorEvent ?? ((data) => JSON.stringify(data))
   }
 
   private async run(eventsPromise: Promise<FetchEventStream>, minChangeIntervalMS: number) {
@@ -64,10 +67,17 @@ export class StreamedJson<T> {
       for await (const event of events) {
         if (this.error) return
         if (event.data) {
-          const json = JSON.parse(event.data)
+          let json
+          try {
+            json = JSON.parse(event.data)
+          } catch (e) {
+            // eslint-disable-next-line no-console
+            console.warn('Mal formed JSON in streaming:', event.data)
+            continue
+          }
           await this.transform?.(json, this.data)
           this.merge(json, this.data)
-          if (new Date().getTime() - lastChangeCall >= minChangeIntervalMS) {
+          if (event.event !== 'error' && new Date().getTime() - lastChangeCall >= minChangeIntervalMS) {
             this.onChangeListeners.forEach(l => l(this.data))
             lastChangeCall = new Date().getTime()
             flushed = true
@@ -75,6 +85,11 @@ export class StreamedJson<T> {
             flushed = false
           }
         }
+        if (event.event === 'error') {
+          const error = new Error(this.textFromErrorEvent(this.data))
+          this.data = {}
+          throw error
+        }
       }
       if (!flushed) this.onChangeListeners.forEach(l => l(this.data))
     } catch (error: any) {
@@ -138,7 +153,7 @@ export class StreamedJson<T> {
   onChange(listener: (value: Partial<T>) => void) {
     if (this.fullPromise.resolved) {
       listener(this.data)
-      return () => {}
+      return () => { }
     }
     this.onChangeListeners.push(listener)
     return () => {