@wf-financing/ui 1.4.5 → 2.0.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wf-financing/ui",
-  "version": "1.4.5",
+  "version": "2.0.0",
   "exports": {
     ".": {
       "import": "./dist/index.es.js",
@@ -37,7 +37,7 @@
     "react-aria": "^3.41.1",
     "react-intl": "^6.2.5",
     "styled-components": "^6.1.19",
-    "@wf-financing/embedded-types": "0.3.0"
+    "@wf-financing/embedded-types": "0.4.0"
   },
   "peerDependencies": {
     "@wayflyer/flyui": "204.4.1"

package/sdk/index.ts

@@ -1,36 +1,29 @@
-import type { Themes } from '@wayflyer/flyui';
-import { PartnerCallbackType, MockedModeType } from '@wf-financing/embedded-types';
+import { PartnerCallbackType, SdkOptionsType } from '@wf-financing/embedded-types';
 
+import { PartnerTheme } from '../src/config';
 import { mountToTarget } from '../src/main';
+import { getPartnerIdFromToken, getPartnerThemeById } from '../src/utils';
 
 export class WayflyerUiCtaSdk {
-  private targetId: string;
-  private partnerDesignId: Themes;
-  private partnerCallback: PartnerCallbackType;
-  private companyToken: string;
-  private mockedMode?: MockedModeType;
+  private readonly partnerTheme: PartnerTheme;
 
   constructor(
-    targetId: string,
-    partnerDesignId: Themes,
-    partnerCallback: PartnerCallbackType,
-    companyToken: string,
-    mockedMode?: MockedModeType,
+    private readonly targetId: string,
+    private readonly partnerCallback: PartnerCallbackType,
+    private readonly companyToken: string,
+    private readonly options?: SdkOptionsType,
   ) {
-    this.targetId = targetId;
-    this.partnerDesignId = partnerDesignId;
-    this.partnerCallback = partnerCallback;
-    this.companyToken = companyToken;
-    this.mockedMode = mockedMode;
+    const partnerId = getPartnerIdFromToken(companyToken);
+    this.partnerTheme = getPartnerThemeById(partnerId);
   }
 
   mountCta() {
     if (document.readyState === 'loading') {
       document.addEventListener('DOMContentLoaded', () =>
-        mountToTarget(this.targetId, this.partnerDesignId, this.partnerCallback, this.companyToken, this.mockedMode),
+        mountToTarget(this.targetId, this.partnerTheme, this.partnerCallback, this.companyToken, this.options),
       );
     } else {
-      mountToTarget(this.targetId, this.partnerDesignId, this.partnerCallback, this.companyToken, this.mockedMode);
+      mountToTarget(this.targetId, this.partnerTheme, this.partnerCallback, this.companyToken, this.options);
     }
   }
 }

package/src/App.tsx

@@ -1,16 +1,17 @@
 import { QueryClient, QueryClientProvider } from '@tanstack/react-query';
-import { FlyUIProvider, Themes } from '@wayflyer/flyui';
-import { PartnerCallbackType, MockedModeType } from '@wf-financing/embedded-types';
+import { FlyUIProvider } from '@wayflyer/flyui';
+import { PartnerCallbackType, SdkOptionsType } from '@wf-financing/embedded-types';
 import { createIntl, createIntlCache, IntlShape } from 'react-intl';
 
+import { PartnerTheme } from './config';
 import { CtaWidget } from './CtaWidget';
 import enMessages from './locales/en.json';
 import { PartnerContext } from './utils';
 
 type AppPropsType = {
-  partnerDesignId: Themes;
+  theme: PartnerTheme;
   companyToken: string;
-  mockedMode?: MockedModeType;
+  options?: SdkOptionsType;
   partnerCallback: PartnerCallbackType;
   onWidgetClose: () => void;
   portalContainer: HTMLElement;
@@ -22,9 +23,9 @@ const messages = {
 };
 
 export const App = ({
-  partnerDesignId,
+  theme,
   companyToken,
-  mockedMode,
+  options,
   partnerCallback,
   onWidgetClose,
   portalContainer,
@@ -41,8 +42,8 @@ export const App = ({
 
   return (
     <QueryClientProvider client={queryClient}>
-      <PartnerContext.Provider value={{ companyToken, mockedMode, partnerCallback, onWidgetClose, portalContainer }}>
-        <FlyUIProvider theme={partnerDesignId} intl={intl}>
+      <PartnerContext.Provider value={{ companyToken, options, partnerCallback, onWidgetClose, portalContainer }}>
+        <FlyUIProvider theme={theme} intl={intl}>
           <CtaWidget />
         </FlyUIProvider>
       </PartnerContext.Provider>

package/src/api/continueHostedApplication.ts

@@ -1,12 +1,12 @@
-import { ContinueHostedApplicationResponseType, MockedModeType } from '@wf-financing/embedded-types';
+import { ContinueHostedApplicationResponseType, SdkOptionsType } from '@wf-financing/embedded-types';
 
 import { getHeadlessSdkInstance } from './getHeadlessSdkInstance';
 
 export const continueHostedApplication = async (
   companyToken: string,
-  mockedMode?: MockedModeType,
+  options?: SdkOptionsType,
 ): Promise<ContinueHostedApplicationResponseType> => {
-  const sdk = await getHeadlessSdkInstance(companyToken, mockedMode);
+  const sdk = await getHeadlessSdkInstance(companyToken, options);
 
   return await sdk.continueHostedApplication();
 };

package/src/api/ctaBanner.ts

@@ -1,9 +1,9 @@
-import { MockedModeType } from '@wf-financing/embedded-types';
+import { SdkOptionsType } from '@wf-financing/embedded-types';
 
 import { getHeadlessSdkInstance } from './getHeadlessSdkInstance';
 
-export const fetchCtaBanner = async (companyToken: string, mockedMode?: MockedModeType) => {
-  const sdk = await getHeadlessSdkInstance(companyToken, mockedMode);
+export const fetchCtaBanner = async (companyToken: string, options?: SdkOptionsType) => {
+  const sdk = await getHeadlessSdkInstance(companyToken, options);
   const cta = await sdk.getCta();
 
   return cta;

package/src/api/dismissCta.ts

@@ -1,11 +1,9 @@
-import { MockedModeType } from '@wf-financing/embedded-types';
+import { SdkOptionsType } from '@wf-financing/embedded-types';
 
 import { getHeadlessSdkInstance } from './getHeadlessSdkInstance';
 
-export const dismissCta = async (companyToken: string, mockedMode?: MockedModeType) => {
-  const sdk = await getHeadlessSdkInstance(companyToken, mockedMode);
+export const dismissCta = async (companyToken: string, options?: SdkOptionsType) => {
+  const sdk = await getHeadlessSdkInstance(companyToken, options);
 
-  if (!mockedMode) {
-    await sdk.dismissCta();
-  }
+  await sdk.dismissCta();
 };

package/src/api/fetchCtaBanner.test.ts

@@ -1,4 +1,4 @@
-import { SdkScenarios, MockedModeType } from '@wf-financing/embedded-types';
+import { SdkOptionsType } from '@wf-financing/embedded-types';
 import { afterEach, describe, expect, test, vi } from 'vitest';
 
 import { fetchCtaBanner } from './ctaBanner';
@@ -22,14 +22,13 @@ describe('fetchCtaBanner', () => {
     const mockCta = { banner: 'hello' };
     (getHeadlessSdkInstance as any).mockResolvedValueOnce(fakeSdk);
     fakeSdk.getCta.mockResolvedValueOnce(mockCta);
-    const mockedMode: MockedModeType = {
-      isMockedMode: true,
-      sdkScenario: SdkScenarios.NEW_APPLICATION,
+    const options: SdkOptionsType = {
+      isSandbox: true,
     };
 
-    const result = await fetchCtaBanner('TOKEN', mockedMode);
+    const result = await fetchCtaBanner('TOKEN', options);
 
-    expect(getHeadlessSdkInstance).toHaveBeenCalledWith('TOKEN', mockedMode);
+    expect(getHeadlessSdkInstance).toHaveBeenCalledWith('TOKEN', options);
     expect(fakeSdk.getCta).toHaveBeenCalled();
     expect(result).toBe(mockCta);
   });

package/src/api/getHeadlessSdkInstance.test.ts

@@ -1,4 +1,4 @@
-import { MockedModeType, SdkScenarios } from '@wf-financing/embedded-types';
+import { SdkOptionsType } from '@wf-financing/embedded-types';
 import { afterEach, describe, expect, Mock, test, vi } from 'vitest';
 
 vi.mock('./getHeadlessSdkInstance', () => ({
@@ -13,9 +13,8 @@ const fakeSdk = {
   startHostedApplication: vi.fn(),
 };
 
-const mockedMode: MockedModeType = {
-  isMockedMode: true,
-  sdkScenario: SdkScenarios.NEW_APPLICATION,
+const options: SdkOptionsType = {
+  isSandbox: true,
 };
 
 describe('startHostedApplication', () => {
@@ -31,9 +30,9 @@ describe('startHostedApplication', () => {
     (getHeadlessSdkInstance as unknown as Mock).mockResolvedValueOnce(fakeSdk);
     fakeSdk.startHostedApplication.mockResolvedValueOnce(fakeNextUrl);
 
-    const result = await startHostedApplication('TOKEN', partnerCallback, mockedMode);
+    const result = await startHostedApplication('TOKEN', partnerCallback, options);
 
-    expect(getHeadlessSdkInstance).toHaveBeenCalledWith('TOKEN', mockedMode);
+    expect(getHeadlessSdkInstance).toHaveBeenCalledWith('TOKEN', options);
     expect(partnerCallback).toHaveBeenCalled();
     expect(fakeSdk.startHostedApplication).toHaveBeenCalledWith(fakePartnerData);
     expect(result).toBe(fakeNextUrl);
@@ -46,7 +45,7 @@ describe('startHostedApplication', () => {
     (getHeadlessSdkInstance as unknown as Mock).mockResolvedValueOnce(fakeSdk);
     fakeSdk.startHostedApplication.mockResolvedValueOnce(fakeNextUrl);
 
-    const result = await startHostedApplication('TOKEN', partnerCallback, mockedMode);
+    const result = await startHostedApplication('TOKEN', partnerCallback);
 
     expect(fakeSdk.startHostedApplication).toHaveBeenCalledWith({});
     expect(result).toBe(fakeNextUrl);

package/src/api/getHeadlessSdkInstance.ts

@@ -1,16 +1,14 @@
-import { IHeadlessWayflyerSdk, MockedModeType } from '@wf-financing/embedded-types';
+import { IHeadlessWayflyerSdk, SdkOptionsType } from '@wf-financing/embedded-types';
 
 import { HEADLESS_SDK_URL, WAYFLYER_HEADLESS_SDK_ID } from '../config';
 import { initializeHeadlessSdk, loadScriptAndInitializeSdk } from '../utils';
 
 let cachedSdkInstance: IHeadlessWayflyerSdk | null = null;
 
-export const getHeadlessSdkInstance = async (companyToken: string, mockedMode?: MockedModeType) => {
+export const getHeadlessSdkInstance = async (companyToken: string, options?: SdkOptionsType) => {
   try {
     if (cachedSdkInstance) {
-      if (mockedMode) {
-        cachedSdkInstance.setSdkScenario(mockedMode.sdkScenario);
-      }
+      cachedSdkInstance.setCompanyToken(companyToken);
 
       return cachedSdkInstance;
     }
@@ -18,11 +16,11 @@ export const getHeadlessSdkInstance = async (companyToken: string, mockedMode?:
     const existingScript = document.getElementById(WAYFLYER_HEADLESS_SDK_ID) as HTMLScriptElement;
 
     if (window.WayflyerHeadlessSdk) {
-      return initializeHeadlessSdk(companyToken, mockedMode);
+      return initializeHeadlessSdk(companyToken, options);
     }
 
     if (existingScript) {
-      return loadScriptAndInitializeSdk(existingScript, companyToken, mockedMode);
+      return loadScriptAndInitializeSdk(existingScript, companyToken, options);
     }
 
     const script = document.createElement('script');
@@ -32,8 +30,7 @@ export const getHeadlessSdkInstance = async (companyToken: string, mockedMode?:
     script.async = true;
 
     document.head.appendChild(script);
-
-    const headlessSdk: IHeadlessWayflyerSdk = await loadScriptAndInitializeSdk(script, companyToken, mockedMode);
+    const headlessSdk: IHeadlessWayflyerSdk = await loadScriptAndInitializeSdk(script, companyToken, options);
     cachedSdkInstance = headlessSdk;
 
     return headlessSdk;

package/src/api/startHostedApplication.test.ts

@@ -1,4 +1,4 @@
-import { MockedModeType, SdkScenarios } from '@wf-financing/embedded-types';
+import { SdkOptionsType } from '@wf-financing/embedded-types';
 import { afterEach, describe, expect, test, vi } from 'vitest';
 
 vi.mock('./getHeadlessSdkInstance', () => ({
@@ -25,14 +25,13 @@ describe('startHostedApplication', () => {
 
     (getHeadlessSdkInstance as unknown as ReturnType<typeof vi.fn>).mockResolvedValueOnce(fakeSdk);
     fakeSdk.startHostedApplication.mockResolvedValueOnce(fakeNextUrl);
-    const mockedMode: MockedModeType = {
-      isMockedMode: true,
-      sdkScenario: SdkScenarios.NEW_APPLICATION,
+    const options: SdkOptionsType = {
+      isSandbox: true,
     };
 
-    const result = await startHostedApplication('TOKEN', partnerCallback, mockedMode);
+    const result = await startHostedApplication('TOKEN', partnerCallback, options);
 
-    expect(getHeadlessSdkInstance).toHaveBeenCalledWith('TOKEN', mockedMode);
+    expect(getHeadlessSdkInstance).toHaveBeenCalledWith('TOKEN', options);
     expect(partnerCallback).toHaveBeenCalled();
     expect(fakeSdk.startHostedApplication).toHaveBeenCalledWith(fakePartnerData);
     expect(result).toBe(fakeNextUrl);

package/src/api/startHostedApplication.ts

@@ -1,14 +1,14 @@
-import { PartnerCallbackType, MockedModeType } from '@wf-financing/embedded-types';
+import { PartnerCallbackType, SdkOptionsType } from '@wf-financing/embedded-types';
 
 import { getHeadlessSdkInstance } from './getHeadlessSdkInstance';
 
 export const startHostedApplication = async (
   companyToken: string,
   partnerCallback: PartnerCallbackType,
-  mockedMode?: MockedModeType,
+  options?: SdkOptionsType,
 ) => {
   let partnerData = await partnerCallback();
-  const sdk = await getHeadlessSdkInstance(companyToken, mockedMode);
+  const sdk = await getHeadlessSdkInstance(companyToken, options);
 
   if (!partnerData) {
     partnerData = {};

package/src/components/banner/CtaBanner.snapshot.stories.tsx

@@ -1,6 +1,5 @@
 import { Meta, StoryObj } from '@storybook/react';
-import { Themes } from '@wayflyer/flyui';
-import { SdkScenarios, MockedModeType } from '@wf-financing/embedded-types';
+import { PartnerTheme } from '../../config';
 import { App } from '../../App';
 import { CtaBanner } from './CtaBanner';
 
@@ -8,14 +7,17 @@ const portalContainer = document.createElement('div');
 document.body.append(portalContainer);
 
 const defaultArgs = {
-  partnerDesignId: 'whiteLabel' as Themes,
+  theme: 'whiteLabel' as PartnerTheme,
   companyToken: 'demo-token',
   partnerCallback: () => {},
   onWidgetClose: () => {},
   portalContainer,
+  options: {
+    isMockedMode: true,
+  },
 };
 
-type CtaBannerStoryArgs = typeof defaultArgs & { mockedMode: MockedModeType };
+type CtaBannerStoryArgs = typeof defaultArgs;
 
 const Template = (args: CtaBannerStoryArgs) => (
   <App {...args}>
@@ -26,17 +28,8 @@ const Template = (args: CtaBannerStoryArgs) => (
 const meta: Meta<typeof CtaBanner> = {
   title: 'CtaBanner',
   component: CtaBanner,
-  argTypes: {
-    mockedMode: {
-      control: { type: 'object' },
-    },
-  },
   args: {
     ...defaultArgs,
-    mockedMode: {
-      isMockedMode: true,
-      sdkScenario: SdkScenarios.GENERIC_NEW_APPLICATION,
-    },
   },
   decorators: [(_, context) => <Template {...(context.args as CtaBannerStoryArgs)} />],
 };
@@ -45,31 +38,5 @@ export default meta;
 type Story = StoryObj<typeof CtaBanner>;
 
 export const GenericOffer: Story = {
-  name: 'Generic Offer',
-  args: {
-    mockedMode: {
-      isMockedMode: true,
-      sdkScenario: SdkScenarios.GENERIC_NEW_APPLICATION,
-    },
-  },
-};
-
-export const IndicativeOffer: Story = {
-  name: 'Indicative Offer',
-  args: {
-    mockedMode: {
-      isMockedMode: true,
-      sdkScenario: SdkScenarios.INDICATIVE_NEW_APPLICATION,
-    },
-  },
-};
-
-export const ContinueApplication: Story = {
-  name: 'Continue Application',
-  args: {
-    mockedMode: {
-      isMockedMode: true,
-      sdkScenario: SdkScenarios.CONTINUE_APPLICATION,
-    },
-  },
+  name: 'CTA',
 };

package/src/components/modal/ConsentModal.snapshot.stories.tsx

@@ -1,14 +1,12 @@
 import type { Meta, StoryObj } from '@storybook/react';
 
-import { ConsentModal } from './ConsentModal';
 import { PartnerContext } from '../../utils';
-import { Themes } from '@wayflyer/flyui';
+import { ConsentModal } from './ConsentModal';
 
 const portalContainer = document.createElement('div');
 document.body.append(portalContainer);
 
 const appArgs = {
-  partnerDesignId: 'whiteLabel' as Themes,
   companyToken: 'demo-token',
   partnerCallback: () => {},
   onWidgetClose: () => {},

package/src/config/index.ts

@@ -3,3 +3,5 @@ export type { FontParamsType } from './fontParameters';
 export { ROOTS_CONFIG } from './rootsParameters';
 export { WAYFLYER_HEADLESS_SDK_ID } from './scriptId';
 export { HEADLESS_SDK_URL } from './url';
+export { WHITELISTED_PARTNER_IDS } from './whitelistedPartnerIds';
+export type { PartnerId, PartnerTheme } from './whitelistedPartnerIds';

package/src/config/url.ts

@@ -1 +1 @@
-export const HEADLESS_SDK_URL = 'https://unpkg.com/@wf-financing/headless@1/dist/index.es.js';
+export const HEADLESS_SDK_URL = 'https://embedded-finance-frontend.vercel.app/npm/@wf-financing/headless@2';

package/src/config/whitelistedPartnerIds.ts

@@ -0,0 +1,14 @@
+import { Themes } from '@wayflyer/flyui';
+
+export type PartnerTheme = Exclude<Themes, 'staff'>;
+export type PartnerId = (typeof WHITELISTED_PARTNER_IDS)[PartnerTheme];
+
+/**
+ * List of partner IDs that are allowed to use custom CTA themes
+ * when embedded into their platform.
+ */
+export const WHITELISTED_PARTNER_IDS = {
+  whiteLabel: '2416e2af-c9f7-40ac-a2f0-f37dcfe59674',
+  wayflyer: '1f309265-5324-4f8b-88d7-9b0d5c77a27c',
+  defaultTheme: 'defaultTheme',
+} as const satisfies Record<PartnerTheme, string>;

package/src/hooks/useContinueHostedApplication.ts

@@ -4,9 +4,9 @@ import { continueHostedApplication } from '../api';
 import { usePartnerContext } from './usePartnerContext';
 
 export const useContinueHostedApplication = () => {
-  const { companyToken, mockedMode } = usePartnerContext();
+  const { companyToken, options } = usePartnerContext();
 
   return useMutation({
-    mutationFn: () => continueHostedApplication(companyToken, mockedMode),
+    mutationFn: () => continueHostedApplication(companyToken, options),
   });
 };

package/src/hooks/useCtaBanner.ts

@@ -4,11 +4,11 @@ import { fetchCtaBanner } from '../api';
 import { usePartnerContext } from './usePartnerContext';
 
 export const useCtaBanner = () => {
-  const { companyToken, mockedMode } = usePartnerContext();
+  const { companyToken, options } = usePartnerContext();
 
   return useQuery({
-    queryKey: ['cta', companyToken, mockedMode],
-    queryFn: () => fetchCtaBanner(companyToken, mockedMode),
+    queryKey: ['cta', companyToken, options],
+    queryFn: () => fetchCtaBanner(companyToken, options),
     staleTime: Infinity,
   });
 };

package/src/hooks/useDismissCta.ts

@@ -4,9 +4,9 @@ import { dismissCta } from '../api';
 import { usePartnerContext } from './usePartnerContext';
 
 export const useDismissCta = () => {
-  const { companyToken, mockedMode } = usePartnerContext();
+  const { companyToken, options } = usePartnerContext();
 
   return useMutation({
-    mutationFn: () => dismissCta(companyToken, mockedMode),
+    mutationFn: () => dismissCta(companyToken, options),
   });
 };

package/src/hooks/useStartHostedApplication.ts

@@ -4,9 +4,9 @@ import { startHostedApplication } from '../api';
 import { usePartnerContext } from './usePartnerContext';
 
 export const useStartHostedApplication = () => {
-  const { companyToken, mockedMode, partnerCallback } = usePartnerContext();
+  const { companyToken, options, partnerCallback } = usePartnerContext();
 
   return useMutation({
-    mutationFn: () => startHostedApplication(companyToken, partnerCallback, mockedMode),
+    mutationFn: () => startHostedApplication(companyToken, partnerCallback, options),
   });
 };

package/src/main.tsx

@@ -1,19 +1,19 @@
-import type { Themes } from '@wayflyer/flyui';
-import { PartnerCallbackType, MockedModeType } from '@wf-financing/embedded-types';
+import { PartnerCallbackType, SdkOptionsType } from '@wf-financing/embedded-types';
 import ReactDOM from 'react-dom/client';
 import { StyleSheetManager } from 'styled-components';
 
 import { App } from './App';
-import { createRoots, applyFont } from './utils';
+import { PartnerTheme } from './config';
+import { applyFont, createRoots } from './utils';
 
 let root: ReactDOM.Root | undefined = undefined;
 
 export const mountToTarget = async (
   targetId: string,
-  partnerDesignId: Themes,
+  partnerTheme: PartnerTheme,
   partnerCallback: PartnerCallbackType,
   companyToken: string,
-  mockedMode?: MockedModeType,
+  options?: SdkOptionsType,
 ): Promise<void> => {
   const hostEl = document.getElementById(targetId);
   if (!hostEl) throw new Error(`Target element with id "${targetId}" not found.`);
@@ -21,7 +21,7 @@ export const mountToTarget = async (
   const shadow = hostEl.shadowRoot ?? hostEl.attachShadow({ mode: 'open' });
 
   try {
-    await applyFont(shadow, partnerDesignId);
+    await applyFont(shadow, partnerTheme);
   } catch (error) {
     console.error(error);
   }
@@ -39,9 +39,9 @@ export const mountToTarget = async (
   root.render(
     <StyleSheetManager target={styledComponentsStyleRoot}>
       <App
-        partnerDesignId={partnerDesignId}
+        theme={partnerTheme}
         companyToken={companyToken}
-        mockedMode={mockedMode}
+        options={options}
         partnerCallback={partnerCallback}
         onWidgetClose={handleCloseWidget}
         portalContainer={portalRoot}

package/src/utils/applyFont.ts

@@ -1,19 +1,16 @@
-import type { Themes } from '@wayflyer/flyui';
-
-import { FONT_PARAMS } from '../config';
+import { FONT_PARAMS, PartnerTheme } from '../config';
 import { loadFont } from './loadFont';
 
-type ApplyFontsType = (shadow: ShadowRoot, partnerDesignId: Themes) => Promise<void>;
+type ApplyFontsType = (shadow: ShadowRoot, partnerTheme: PartnerTheme) => Promise<void>;
 
-export const applyFont: ApplyFontsType = async (shadow, partnerDesignId) => {
+export const applyFont: ApplyFontsType = async (shadow, partnerTheme) => {
   const { dmSansParams, merrionSansParams } = FONT_PARAMS;
 
-  switch (partnerDesignId) {
+  switch (partnerTheme) {
     case 'whiteLabel':
       await loadFont(shadow, dmSansParams);
       break;
     case 'wayflyer':
-    case 'staff':
     case 'defaultTheme':
       await loadFont(shadow, merrionSansParams);
       break;

package/src/utils/getPartnerIdFromToken.ts

@@ -0,0 +1,9 @@
+import { PartnerId } from '../config';
+import { parseJwt } from './parseJwt';
+
+export const getPartnerIdFromToken = (token: string): PartnerId => {
+  const company = parseJwt(token);
+  const partnerId = company?.sub.split('|')[0] as PartnerId;
+
+  return partnerId;
+};

package/src/utils/getPartnerThemeById.ts

@@ -0,0 +1,9 @@
+import { PartnerId, PartnerTheme, WHITELISTED_PARTNER_IDS } from '../config';
+
+const PARTNER_BY_VALUE: Record<PartnerId, PartnerTheme> = Object.fromEntries(
+  Object.entries(WHITELISTED_PARTNER_IDS).map(([k, v]) => [v, k]),
+) as Record<PartnerId, PartnerTheme>;
+
+export const getPartnerThemeById = (value: PartnerId): PartnerTheme => {
+  return PARTNER_BY_VALUE[value] || 'whiteLabel';
+};

package/src/utils/index.ts

@@ -1,5 +1,7 @@
+export { applyFont } from './applyFont';
+export { createRoots } from './createRoots';
+export { getPartnerIdFromToken } from './getPartnerIdFromToken';
+export { getPartnerThemeById } from './getPartnerThemeById';
 export { initializeHeadlessSdk } from './initializeHeadlessSdk';
 export { loadScriptAndInitializeSdk } from './loadScriptAndInitializeSdk';
 export { PartnerContext, type PartnerContextType } from './partnerContext';
-export { createRoots } from './createRoots';
-export { applyFont } from './applyFont';

package/src/utils/initializeHeadlessSdk.ts

@@ -1,18 +1,14 @@
-import { IHeadlessWayflyerSdk, MockedModeType } from '@wf-financing/embedded-types';
+import { IHeadlessWayflyerSdk, SdkOptionsType } from '@wf-financing/embedded-types';
 
-export const initializeHeadlessSdk = (companyToken: string, mockedMode?: MockedModeType): IHeadlessWayflyerSdk => {
+export const initializeHeadlessSdk = (companyToken: string, options?: SdkOptionsType): IHeadlessWayflyerSdk => {
   if (!window.WayflyerHeadlessSdk) {
     throw new Error('Failed to load WayflyerHeadlessSdk from the script.');
   }
 
   const WayflyerHeadlessSdk = window.WayflyerHeadlessSdk;
 
-  if (mockedMode) {
-    mockedMode.sdkScenario;
-    const { isMockedMode, sdkScenario } = mockedMode;
-
-    const wayflyerCtaSdk = new WayflyerHeadlessSdk(companyToken, isMockedMode);
-    wayflyerCtaSdk.setSdkScenario(sdkScenario);
+  if (options) {
+    const wayflyerCtaSdk = new WayflyerHeadlessSdk(companyToken, options);
 
     return wayflyerCtaSdk;
   }

package/src/utils/loadScriptAndInitializeSdk.ts

@@ -1,16 +1,16 @@
-import { IHeadlessWayflyerSdk, MockedModeType } from '@wf-financing/embedded-types';
+import { IHeadlessWayflyerSdk, SdkOptionsType } from '@wf-financing/embedded-types';
 
 import { initializeHeadlessSdk } from './initializeHeadlessSdk';
 
 export const loadScriptAndInitializeSdk = (
   script: HTMLScriptElement,
   companyToken: string,
-  mockedMode?: MockedModeType,
+  options?: SdkOptionsType,
 ): Promise<IHeadlessWayflyerSdk> => {
   return new Promise<IHeadlessWayflyerSdk>((resolve, reject) => {
     script.onload = () => {
       try {
-        resolve(initializeHeadlessSdk(companyToken, mockedMode));
+        resolve(initializeHeadlessSdk(companyToken, options));
       } catch (error) {
         reject(error);
       }

package/src/utils/parseJwt.ts

@@ -0,0 +1,12 @@
+type CompanyToken = {
+  sub: string;
+};
+
+export const parseJwt = (token: string): CompanyToken | null => {
+  try {
+    return JSON.parse(atob(token.split('.')[1]));
+  } catch (error) {
+    console.error('Error parsing a company token:', error);
+    return null;
+  }
+};