fix(core): allow upgrade-success route in browser for electron

toeverything · Apr 22, 2024 · f70c827 · f70c827
1 parent b2242bb
commit f70c827
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/packages/frontend/electron/renderer/app.tsx b/packages/frontend/electron/renderer/app.tsx
@@ -29,11 +29,16 @@ import type { PropsWithChildren, ReactElement } from 'react';
 import { lazy, Suspense } from 'react';
 import { RouterProvider } from 'react-router-dom';
 
+const desktopWhiteList = [
+  '/desktop-signin',
+  '/open-app/signin-redirect',
+  '/upgrade-success',
+  '/ai-upgrade-success',
+];
 if (
   !environment.isDesktop &&
   environment.isDebug &&
-  !location.pathname.includes('/desktop-signin') &&
-  !location.pathname.includes('/open-app/signin-redirect')
+  desktopWhiteList.every(path => !location.pathname.startsWith(path))
 ) {
   document.body.innerHTML = `<h1 style="color:red;font-size:5rem;text-align:center;">Don't run electron entry in browser.</h1>`;
   throw new Error('Wrong distribution');