From 5b8ec11b6c97e14b9024e41123dab06b8f70473f Mon Sep 17 00:00:00 2001
From: httpjamesm <github@httpjames.space>
Date: Tue, 30 Jan 2024 11:21:33 -0500
Subject: [PATCH] feat: set client package based on redirect URL

---
 apps/accounts/src/pages/passkeys/flow/index.tsx | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/apps/accounts/src/pages/passkeys/flow/index.tsx b/apps/accounts/src/pages/passkeys/flow/index.tsx
index c3baf15e3..e934ca183 100644
--- a/apps/accounts/src/pages/passkeys/flow/index.tsx
+++ b/apps/accounts/src/pages/passkeys/flow/index.tsx
@@ -1,3 +1,4 @@
+import { APPS, CLIENT_PACKAGE_NAMES } from '@ente/shared/apps/constants';
 import {
     CenteredFlex,
     VerticallyCentered,
@@ -6,6 +7,7 @@ import EnteButton from '@ente/shared/components/EnteButton';
 import EnteSpinner from '@ente/shared/components/EnteSpinner';
 import FormPaper from '@ente/shared/components/Form/FormPaper';
 import { logError } from '@ente/shared/sentry';
+import { LS_KEYS, setData } from '@ente/shared/storage/localStorage';
 import InfoIcon from '@mui/icons-material/Info';
 import { Box, Typography } from '@mui/material';
 import { t } from 'i18next';
@@ -45,6 +47,15 @@ const PasskeysFlow = () => {
             }
         }
 
+        let pkg = CLIENT_PACKAGE_NAMES.get(APPS.PHOTOS);
+        if (redirectURL.protocol === 'enteauth:') {
+            pkg = CLIENT_PACKAGE_NAMES.get(APPS.AUTH);
+        } else if (redirectURL.hostname.startsWith('accounts')) {
+            pkg = CLIENT_PACKAGE_NAMES.get(APPS.ACCOUNTS);
+        }
+
+        setData(LS_KEYS.CLIENT_PACKAGE, { name: pkg });
+
         // get passkeySessionID from the query params
         const passkeySessionID = searchParams.get('passkeySessionID') as string;