@thoughtspot/visual-embed-sdk 1.25.0 → 1.26.0-token-cache.0

package/src/types.ts

@@ -463,7 +463,6 @@ export interface EmbedConfig {
      * @version SDK: 1.12.0 | ThoughtSpot: 8.4.0.cl, 8.4.1.sw*
      */
     suppressSearchEmbedBetaWarning?: boolean;
-
     /**
      * Hide beta alert warning message for SageEmbed.
      *
@@ -531,6 +530,11 @@ export interface EmbedConfig {
      * @version SDK: 1.27.0 | ThoughtSpot: 9.8.0.cl
      */
     pendoTrackingKey?: string;
+
+    /**
+     * If passed as true all alerts will be suppressed in the embedded app.
+     */
+    suppressErrorAlerts?: boolean;
 }
 
 // eslint-disable-next-line @typescript-eslint/no-empty-interface
@@ -1443,11 +1447,17 @@ export enum EmbedEvent {
      */
     SageWorksheetUpdated = 'sageWorksheetUpdated',
     /**
-     * Emitten when a user updates a connection in Data tab
+     * Emitted when a user updates a connection in Data tab
      *
      * @version SDK : 1.27.0 | Thoughtspot: 9.8.0.cl
      */
     UpdateConnection = 'updateConnection',
+    /**
+     * Emitted when a user updates a connection in Data tab
+     *
+     * @version SDK : 1.27.0 | Thoughtspot: 9.8.0.cl
+     */
+    CreateConnection = 'createConnection',
     /**
      * Emitted when name, status (private or public) or filter values of a
      * PersonalisedView is updated.

package/src/utils/authService.spec.ts

@@ -1,4 +1,3 @@
-import { embedConfig } from 'src/auth.spec';
 import {
     fetchSessionInfoService,
     fetchAuthTokenService,
@@ -6,8 +5,8 @@ import {
     fetchBasicAuthService,
     fetchAuthPostService,
     verifyTokenService,
+    EndPoints,
 } from './authService';
-import { EndPoints } from '../auth';
 
 const thoughtSpotHost = 'http://10.79.135.124:3000';
 

package/src/utils/authService.ts

@@ -1,5 +1,14 @@
-// eslint-disable-next-line import/no-cycle
-import { EndPoints } from '../auth';
+export const EndPoints = {
+    AUTH_VERIFICATION: '/callosum/v1/session/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',
+    BASIC_LOGIN: '/callosum/v1/session/login',
+    LOGOUT: '/callosum/v1/session/logout',
+    EXECUTE_TML: '/api/rest/2.0/metadata/tml/import',
+    EXPORT_TML: '/api/rest/2.0/metadata/tml/export',
+    IS_ACTIVE: '/callosum/v1/session/isactive',
+};
 
 /**
  *
@@ -31,15 +40,25 @@ export function fetchSessionInfoService(authVerificationUrl: string): Promise<an
  * @param thoughtSpotHost : ThoughtSpot host to verify the token against.
  * @param authToken : Auth token to verify.
  */
-export function verifyTokenService(thoughtSpotHost: string, authToken: string): Promise<Response> {
+export async function verifyTokenService(
+    thoughtSpotHost: string,
+    authToken: string,
+): Promise<boolean> {
     const authVerificationUrl = `${thoughtSpotHost}${EndPoints.IS_ACTIVE}`;
-    return fetch(authVerificationUrl, {
-        headers: {
-            Authorization: `Bearer ${authToken}`,
-            'x-requested-by': 'ThoughtSpot',
-        },
-        credentials: 'omit',
-    });
+    try {
+        const res = await fetch(authVerificationUrl, {
+            headers: {
+                Authorization: `Bearer ${authToken}`,
+                'x-requested-by': 'ThoughtSpot',
+            },
+            credentials: 'omit',
+        });
+        return res.ok;
+    } catch (e) {
+        console.error(`Token Verification Service failed : ${e.message}`);
+    }
+
+    return false;
 }
 
 /**

package/src/utils/graphql/answerService/answerService.spec.ts

@@ -1,7 +1,10 @@
 import 'jest-fetch-mock';
-import { VizPoint } from 'src/types';
+import { AuthType, VizPoint } from '../../../types';
 import { AnswerService } from './answerService';
 import { getAnswerData, removeColumns } from './answer-queries';
+import * as baseInstance from '../../../embed/base';
+import * as authTokenInstance from '../../../authToken';
+import * as tokenizedFetch from '../../../tokenizedFetch';
 
 const defaultSession = {
     sessionId: 'id',
@@ -108,14 +111,35 @@ describe('Answer service tests', () => {
         expect(data.data.foo).toBe(1);
     });
 
-    test('fetchCSVBlob should call the right API', async () => {
+    test('fetchCSVBlob should call the right API', () => {
         fetchMock.once('Bla');
         const answerService = createAnswerService();
+
+        const mockEmbedConfig = {
+            thougthspotHost: '/test',
+            authType: AuthType.TrustedAuthTokenCookieless,
+        };
+        jest.spyOn(baseInstance, 'getEmbedConfig').mockReturnValueOnce(mockEmbedConfig);
+        jest.spyOn(authTokenInstance, 'getAuthenticationToken').mockReturnValueOnce(Promise.resolve('token'));
+        const mockTokenizedFetch = jest.spyOn(tokenizedFetch, 'tokenizedFetch');
         answerService.fetchCSVBlob(undefined, true);
-        expect(fetchMock).toHaveBeenCalledWith(
+
+        expect(mockTokenizedFetch).toHaveBeenCalledWith(
             `https://tshost/prism/download/answer/csv?sessionId=${defaultSession.sessionId}&genNo=${defaultSession.genNo}&userLocale=en-us&exportFileName=data&hideCsvHeader=false`,
             expect.objectContaining({}),
         );
+
+        answerService.fetchCSVBlob('en-uk', true);
+        expect(mockTokenizedFetch).toHaveBeenCalledWith(
+            `https://tshost/prism/download/answer/csv?sessionId=${defaultSession.sessionId}&genNo=${defaultSession.genNo}&userLocale=en-uk&exportFileName=data&hideCsvHeader=false`,
+            expect.objectContaining({}),
+        );
+
+        answerService.fetchCSVBlob(undefined, false);
+        expect(mockTokenizedFetch).toHaveBeenCalledWith(
+            `https://tshost/prism/download/answer/csv?sessionId=${defaultSession.sessionId}&genNo=${defaultSession.genNo}&userLocale=en-us&exportFileName=data&hideCsvHeader=true`,
+            expect.objectContaining({}),
+        );
     });
 
     test('getUnderlyingDataForPoint should call the right APIs', async () => {

package/src/utils/graphql/answerService/answerService.ts

@@ -1,3 +1,4 @@
+import { tokenizedFetch } from '../../../tokenizedFetch';
 import type { ColumnValue, VizPoint } from '../../../types';
 import { deepMerge, removeTypename } from '../../../utils';
 import { graphqlQuery } from '../graphql-request';
@@ -102,12 +103,16 @@ export class AnswerService {
      * @returns Response
      */
     public async fetchCSVBlob(userLocale = 'en-us', includeInfo = false): Promise<Response> {
-        const fetchUrl = `${this.thoughtSpotHost}/prism/download/answer/csv?sessionId=${this.session.sessionId}&genNo=${this.session.genNo}&userLocale=${userLocale}&exportFileName=data&hideCsvHeader=${!includeInfo}`;
-        return fetch(fetchUrl, {
+        const fetchUrl = this.getFetchCSVBlobUrl(userLocale, includeInfo);
+        return tokenizedFetch(fetchUrl, {
             credentials: 'include',
         });
     }
 
+    public getFetchCSVBlobUrl(userLocale = 'en-us', includeInfo = false): string {
+        return `${this.thoughtSpotHost}/prism/download/answer/csv?sessionId=${this.session.sessionId}&genNo=${this.session.genNo}&userLocale=${userLocale}&exportFileName=data&hideCsvHeader=${!includeInfo}`;
+    }
+
     /**
      * Get underlying data given a point and the output column names.
      *

package/src/utils/processData.spec.ts

@@ -7,7 +7,7 @@ import { EmbedEvent, AuthType } from '../types';
 describe('Unit test for process data', () => {
     beforeAll(() => {
         base.init({
-            thoughtSpotHost: '',
+            thoughtSpotHost: 'https://tshost',
             authType: AuthType.None,
         });
     });
@@ -153,7 +153,7 @@ describe('Unit test for process data', () => {
         base.init({
             loginFailedMessage: 'Hello',
             autoLogin: true,
-            thoughtSpotHost: '',
+            thoughtSpotHost: 'https://tshost',
             authType: AuthType.None,
         });
         jest.spyOn(base, 'getEmbedConfig').mockRestore();

package/src/utils/processData.ts

@@ -56,9 +56,10 @@ function processNoCookieAccess(e: any, containerEl: Element) {
         loginFailedMessage,
         suppressNoCookieAccessAlert,
         ignoreNoCookieAccess,
+        suppressErrorAlerts,
     } = getEmbedConfig();
     if (!ignoreNoCookieAccess) {
-        if (!suppressNoCookieAccessAlert) {
+        if (!suppressNoCookieAccessAlert && !suppressErrorAlerts) {
             // eslint-disable-next-line no-alert
             alert(
                 'Third party cookie access is blocked on this browser, please allow third party cookies for this to work properly. \nYou can use `suppressNoCookieAccessAlert` to suppress this message.',