feat(deepLinks): make deeplinking work with multi org (#5344)

Author: blegesse-w

src/identity/apis/auth.ts

@@ -1,6 +1,7 @@
 // Functions calling API
 import {
   getAccount,
+  getAccounts,
   getAccountsOrgs,
   getIdentity,
   getMe as getMeQuartz,
@@ -282,3 +283,31 @@ export const updateDefaultOrgByAccountID = async ({
 
   // success status code is 204; no response body expected.
 }
+
+// fetch user default account's default org
+export const getDefaultAccountDefaultOrg = async (): Promise<OrganizationSummaries[number]> => {
+  const response = await getAccounts({})
+
+  if (response.status === 401) {
+    throw new UnauthorizedError(response.data.message)
+  }
+
+  if (response.status === 500) {
+    throw new ServerError(response.data.message)
+  }
+  const {data} = response
+
+  if (Array.isArray(data) && data.length) {
+    const defaultAccount = data.find(account => account.isDefault)
+
+    // fetch default org
+    if (defaultAccount) {
+      const quartzOrg = await fetchOrgsByAccountID(defaultAccount.id)
+      const defaultQuartzOrg =
+        quartzOrg.find(org => org.isDefault) || quartzOrg[0]
+
+      return defaultQuartzOrg
+    }
+    throw new GenericError('No default account found')
+  }
+}

src/shared/components/NotFound.tsx

@@ -21,6 +21,7 @@ import {getOrg as fetchOrg} from 'src/organizations/apis'
 // Utils
 import {buildDeepLinkingMap} from 'src/utils/deepLinks'
 import {event} from 'src/cloud/utils/reporting'
+import {shouldUseQuartzIdentity} from 'src/identity/utils/shouldUseQuartzIdentity'
 
 // Components
 import LogoWithCubo from 'src/checkout/LogoWithCubo'
@@ -30,6 +31,9 @@ import {Organization} from 'src/types'
 // Constants
 import {CLOUD} from 'src/shared/constants'
 
+// API
+import {getDefaultAccountDefaultOrg} from 'src/identity/apis/auth'
+
 const NotFoundNew: FC = () => (
   <AppWrapper type="funnel" className="page-not-found" testID="not-found">
     <FunnelPage enableGraphic={true} className="page-not-found-funnel">
@@ -136,10 +140,20 @@ const NotFound: FC = () => {
 
   const handleDeepLink = useCallback(async () => {
     if (!org.current) {
-      setIsFetchingOrg(true)
-      org.current = await fetchOrg()
+      if (shouldUseQuartzIdentity()) {
+        try {
+          setIsFetchingOrg(true)
+          const defaultQuartzOrg = await getDefaultAccountDefaultOrg()
+          org.current = defaultQuartzOrg
+        } catch {
+          history.push(`/no-orgs`)
+          return
+        }
+      } else {
+        setIsFetchingOrg(true)
+        org.current = await fetchOrg()
+      }
     }
-
     const deepLinkingMap = buildDeepLinkingMap(org.current)
 
     if (deepLinkingMap.hasOwnProperty(location.pathname)) {

src/utils/deepLinks.ts

@@ -1,7 +1,10 @@
 import {Organization} from 'src/client'
+import {OrganizationSummaries} from 'src/client/unityRoutes'
 import {PROJECT_NAME_PLURAL} from 'src/flows'
 
-export const buildDeepLinkingMap = (org: Organization) => ({
+export const buildDeepLinkingMap = (
+  org: Organization | OrganizationSummaries[number]
+) => ({
   '/me/about': `/orgs/${org.id}/about`,
   '/me/alerts': `/orgs/${org.id}/alerting`,
   '/me/billing': `/orgs/${org.id}/billing`,