SCAL-233969: Optimise info call (WIP)

src/auth.ts

@@ -42,7 +42,7 @@
 }
 
 /**
  * Enum for auth status emitted by the emitter returned from {@link init}.
  * @group Authentication / Init
  */
 export enum AuthStatus {
@@ -54,6 +54,10 @@
      * Emits when the SDK authenticates successfully
      */
     SDK_SUCCESS = 'SDK_SUCCESS',
+    /**
+     * Emits when the SDK authenticates successfully
+     */
+    SESSION_INFO_SUCCESS = 'SESSION_INFO_SUCCESS',
     /**
      * Emits when the app sends an authentication success message
      */
@@ -66,13 +70,13 @@
      * Emitted when inPopup is true in the SAMLRedirect flow and the
      * popup is waiting to be triggered either programmatically
      * or by the trigger button.
      * @version SDK: 1.19.0
      */
     WAITING_FOR_POPUP = 'WAITING_FOR_POPUP',
 }
 
 /**
  * Event emitter returned from {@link init}.
  * @group Authentication / Init
  */
 export interface AuthEventEmitter {
@@ -100,7 +104,7 @@
     once(event: AuthStatus.SUCCESS, listener: (sessionInfo: any) => void): this;
     /**
      * Trigger an event on the emitter returned from init.
      * @param {@link AuthEvent}
      */
     emit(event: AuthEvent, ...args: any[]): boolean;
     /**
@@ -119,7 +123,7 @@
 }
 
 /**
  * Events which can be triggered on the emitter returned from {@link init}.
  * @group Authentication / Init
  */
 export enum AuthEvent {
@@ -132,7 +136,7 @@
 
 let authEE: EventEmitter<AuthStatus | AuthEvent>;
 
 /**
  *
  */
 export function getAuthEE(): EventEmitter<AuthStatus | AuthEvent> {
@@ -141,7 +145,7 @@
 
 /**
  *
  * @param eventEmitter
  */
 export function setAuthEE(eventEmitter: EventEmitter<AuthStatus | AuthEvent>): void {
     authEE = eventEmitter;
@@ -176,7 +180,7 @@
 
 /**
  *
  * @param failureType
  */
 export function notifyAuthFailure(failureType: AuthFailureType): void {
     if (!authEE) {

src/embed/ts-embed.ts

@@ -62,6 +62,7 @@ import {
 import { AuthFailureType } from '../auth';
 import { getEmbedConfig } from './embedConfig';
 import { ERROR_MESSAGE } from '../errors';
+import { getSessionInfo } from '../utils/sessionInfoService';
 
 const { version } = pkgInfo;
 
@@ -654,6 +655,11 @@ export class TsEmbed {
                             elHeight: this.iFrame.clientHeight,
                             timeTookToLoad: loadTimestamp - initTimestamp,
                         });
+                        getSessionInfo().then((data) => {
+                            if (data?.info) {
+                                this.trigger(HostEvent.InfoSuccess, data);
+                            }
+                        });
                     });
                     this.iFrame.addEventListener('error', () => {
                         nextInQueue();

src/types.ts

@@ -3180,7 +3180,17 @@ export enum HostEvent {
      * @version SDK: 1.29.0 | Thoughtspot: 10.1.0.cl
      */
     GetParameters = 'GetParameters',
-}
+    /**
+     * Notify when info call is completed successfully
+     * ```js
+     * liveboardEmbed.trigger(HostEvent.GetParameters).then((parameter) => {
+     *  console.log('parameters', parameter);
+     * });
+     *```
+     * @version SDK: 1.29.0 | Thoughtspot: 10.1.0.cl
+     */
+     InfoSuccess = 'InfoSuccess',
+    }
 
 /**
  * The different visual modes that the data sources panel within

src/utils/authService/authService.spec.ts

@@ -29,7 +29,7 @@ describe('Unit test for authService', () => {
         const response = await fetchSessionInfoService(thoughtSpotHost);
         expect(response.success).toBe(true);
         expect(fetch).toHaveBeenCalledTimes(1);
-        expect(fetch).toBeCalledWith(`${thoughtSpotHost}${EndPoints.SESSION_INFO}`, {
+        expect(fetch).toBeCalledWith(`${thoughtSpotHost}${EndPoints.PREAUTH_INFO}`, {
             credentials: 'include',
         });
     });
@@ -116,7 +116,7 @@ describe('Unit test for authService', () => {
         } catch (e) {
             //
         }
-        expect(logger.error).toHaveBeenCalledWith('Failed to fetch http://localhost:3000/callosum/v1/session/info', 'error');
+        expect(logger.error).toHaveBeenCalledWith('Failed to fetch http://localhost:3000/prism/preauth/info', 'error');
 
         try {
             await fetchBasicAuthService(thoughtSpotHost, username, password);

src/utils/authService/authService.ts

@@ -3,6 +3,7 @@ import { logger } from '../logger';
 export const EndPoints = {
     AUTH_VERIFICATION: '/callosum/v1/session/info',
     SESSION_INFO: '/callosum/v1/session/info',
+    PREAUTH_INFO: '/prism/preauth/info',
     SAML_LOGIN_TEMPLATE: (targetUrl: string) => `/callosum/v1/saml/login?targetURLPath=${targetUrl}`,
     OIDC_LOGIN_TEMPLATE: (targetUrl: string) => `/callosum/v1/oidc/login?targetURLPath=${targetUrl}`,
     TOKEN_LOGIN: '/callosum/v1/session/login/token',

src/utils/authService/tokenizedAuthService.ts

@@ -25,14 +25,70 @@ function tokenizedFailureLoggedFetch(url: string, options: RequestInit = {}): Pr
  *  const response = await sessionInfoService();
  * ```
  */
-export async function fetchSessionInfoService(thoughtspotHost: string): Promise<any> {
+export async function fetchPreauthInfoService(thoughtspotHost: string): Promise<any> {
+    const sessionInfoPath = `${thoughtspotHost}${EndPoints.PREAUTH_INFO}`;
+    const response = await tokenizedFailureLoggedFetch(sessionInfoPath);
+    if (!response.ok) {
+        throw new Error(`Failed to fetch auth info: ${response.statusText}`);
+    }
+
+    return response;
+}
+
+/**
+ * Fetches the session info from the ThoughtSpot server.
+ * @param thoughtspotHost
+ * @returns {Promise<any>}
+ * @example
+ * ```js
+ *  const response = await sessionInfoService();
+ * ```
+ */
+export async function fetchV1InfoService(thoughtspotHost: string): Promise<any> {
     const sessionInfoPath = `${thoughtspotHost}${EndPoints.SESSION_INFO}`;
     const response = await tokenizedFailureLoggedFetch(sessionInfoPath);
     if (!response.ok) {
         throw new Error(`Failed to fetch session info: ${response.statusText}`);
     }
-    const data = await response.json();
-    return data;
+
+    return response;
+}
+
+/**
+ * Fetches the session info from the ThoughtSpot server.
+ * @param thoughtspotHost
+ * @returns {Promise<any>}
+ * @example
+ * ```js
+ *  const response = await sessionInfoService();
+ * ```
+ */
+export async function fetchSessionInfoService(thoughtspotHost: string): Promise<any> {
+    try {
+        const response = await fetchPreauthInfoService(thoughtspotHost);
+
+        // Convert Headers to a plain object
+        const headers: Record<string, string> = {};
+        response?.headers?.forEach((value: string, key: string) => {
+            headers[key] = value;
+        });
+        const data = await response.json();
+
+        return {
+            ...data,
+            status: 200,
+            headers,
+        };
+    } catch (error) {
+        if (error.status === 404) {
+            const response = await fetchV1InfoService(thoughtspotHost);
+            const data = await response.json();
+
+            return data;
+        }
+
+        return {};
+    }
 }
 
 /**

src/utils/sessionInfoService.ts

@@ -66,18 +66,25 @@ export function getCachedSessionInfo(): SessionInfo | null {
  * @version SDK: 1.28.3 | ThoughtSpot: *
  */
 export const getSessionDetails = (sessionInfoResp: any): SessionInfo => {
-    const devMixpanelToken = sessionInfoResp.configInfo.mixpanelConfig.devSdkKey;
-    const prodMixpanelToken = sessionInfoResp.configInfo.mixpanelConfig.prodSdkKey;
-    const mixpanelToken = sessionInfoResp.configInfo.mixpanelConfig.production
+    const infoResp = sessionInfoResp?.info ?? sessionInfoResp;
+    let configInfo = sessionInfoResp?.info
+        ? sessionInfoResp.info?.configInfo
+        : sessionInfoResp.configInfo;
+
+    configInfo = configInfo || {};
+
+    const devMixpanelToken = configInfo.mixpanelConfig?.devSdkKey;
+    const prodMixpanelToken = configInfo.mixpanelConfig?.prodSdkKey;
+    const mixpanelToken = configInfo.mixpanelConfig?.production
         ? prodMixpanelToken
         : devMixpanelToken;
     return {
-        userGUID: sessionInfoResp.userGUID,
+        userGUID: infoResp.userGUID,
         mixpanelToken,
-        isPublicUser: sessionInfoResp.configInfo.isPublicUser,
+        isPublicUser: configInfo.isPublicUser,
         releaseVersion: sessionInfoResp.releaseVersion,
-        clusterId: sessionInfoResp.configInfo.selfClusterId,
-        clusterName: sessionInfoResp.configInfo.selfClusterName,
+        clusterId: configInfo.selfClusterId,
+        clusterName: configInfo.selfClusterName,
         ...sessionInfoResp,
     };
 };