From 389dfd08f774acd93402489ed44ce2d8463d6bec Mon Sep 17 00:00:00 2001
From: jeffvli <jeffvictorli@gmail.com>
Date: Tue, 9 Dec 2025 19:58:24 -0800
Subject: [PATCH] render login page on SERVER_LOCK (#1172)

---
 .../components/action-required-container.tsx       |  2 --
 .../action-required/components/server-required.tsx | 12 +++---------
 .../routes/action-required-route.tsx               | 13 +++++++++----
 src/renderer/features/login/routes/login-route.tsx | 12 +++++-------
 .../features/search/components/server-commands.tsx |  7 ++-----
 .../servers/components/add-server-form.tsx         | 14 ++++----------
 .../sidebar/components/server-selector-items.tsx   |  7 ++-----
 .../features/titlebar/components/app-menu.tsx      |  4 +---
 src/renderer/router/app-outlet.tsx                 | 12 +-----------
 9 files changed, 27 insertions(+), 56 deletions(-)

diff --git a/src/renderer/features/action-required/components/action-required-container.tsx b/src/renderer/features/action-required/components/action-required-container.tsx
index ac5d78f82..f9448e5c2 100644
--- a/src/renderer/features/action-required/components/action-required-container.tsx
+++ b/src/renderer/features/action-required/components/action-required-container.tsx
@@ -1,7 +1,6 @@
 import { ReactNode } from 'react';
 
 import { Group } from '/@/shared/components/group/group';
-import { Icon } from '/@/shared/components/icon/icon';
 import { Stack } from '/@/shared/components/stack/stack';
 import { Text } from '/@/shared/components/text/text';
 
@@ -13,7 +12,6 @@ interface ActionRequiredContainerProps {
 export const ActionRequiredContainer = ({ children, title }: ActionRequiredContainerProps) => (
     <Stack style={{ cursor: 'default', maxWidth: '700px' }}>
         <Group>
-            <Icon fill="warn" icon="warn" size="lg" />
             <Text size="xl" style={{ textTransform: 'uppercase' }}>
                 {title}
             </Text>
diff --git a/src/renderer/features/action-required/components/server-required.tsx b/src/renderer/features/action-required/components/server-required.tsx
index dc4489fef..58c3fda98 100644
--- a/src/renderer/features/action-required/components/server-required.tsx
+++ b/src/renderer/features/action-required/components/server-required.tsx
@@ -28,20 +28,14 @@ const localSettings = isElectron() ? window.api.localSettings : null;
 export const ServerRequired = () => {
     const serverList = useServerList();
 
-    const serverLock =
-        (localSettings
-            ? !!localSettings.env.SERVER_LOCK
-            : !!window.SERVER_LOCK &&
-              window.SERVER_TYPE &&
-              window.SERVER_NAME &&
-              window.SERVER_URL) || false;
+    const isServerLock = Boolean(window.SERVER_LOCK) || false;
 
-    if (Object.keys(serverList).length > 0) {
+    if (Object.keys(serverList).length > 1) {
         return (
             <ScrollArea>
                 <Stack miw="300px">
                     <ServerSelector />
-                    {!serverLock && (
+                    {!isServerLock && (
                         <>
                             <Divider my="lg" />
                             <AddServerForm onCancel={null} />
diff --git a/src/renderer/features/action-required/routes/action-required-route.tsx b/src/renderer/features/action-required/routes/action-required-route.tsx
index 0e511d7fa..721c0e49c 100644
--- a/src/renderer/features/action-required/routes/action-required-route.tsx
+++ b/src/renderer/features/action-required/routes/action-required-route.tsx
@@ -1,5 +1,4 @@
 import { openModal } from '@mantine/modals';
-import isElectron from 'is-electron';
 import { useTranslation } from 'react-i18next';
 import { Navigate } from 'react-router';
 
@@ -7,6 +6,7 @@ import { PageHeader } from '/@/renderer/components/page-header/page-header';
 import { ActionRequiredContainer } from '/@/renderer/features/action-required/components/action-required-container';
 import { ServerCredentialRequired } from '/@/renderer/features/action-required/components/server-credential-required';
 import { ServerRequired } from '/@/renderer/features/action-required/components/server-required';
+import LoginRoute from '/@/renderer/features/login/routes/login-route';
 import { ServerList } from '/@/renderer/features/servers/components/server-list';
 import { AnimatedPage } from '/@/renderer/features/shared/components/animated-page';
 import { PageErrorBoundary } from '/@/renderer/features/shared/components/page-error-boundary';
@@ -18,14 +18,15 @@ import { Group } from '/@/shared/components/group/group';
 import { Icon } from '/@/shared/components/icon/icon';
 import { Stack } from '/@/shared/components/stack/stack';
 
-const localSettings = isElectron() ? window.api.localSettings : null;
-
 const ActionRequiredRoute = () => {
     const { t } = useTranslation();
     const currentServer = useCurrentServerWithCredential();
     const isServerRequired = !currentServer;
     const isCredentialRequired = currentServer && !currentServer.credential;
 
+    const isServerLock = Boolean(window.SERVER_LOCK) || false;
+    const isLoginRequired = isServerLock && !currentServer;
+
     const checks = [
         {
             component: <ServerCredentialRequired />,
@@ -49,6 +50,10 @@ const ActionRequiredRoute = () => {
         });
     };
 
+    if (isLoginRequired) {
+        return <LoginRoute />;
+    }
+
     return (
         <AnimatedPage>
             <PageHeader />
@@ -64,7 +69,7 @@ const ActionRequiredRoute = () => {
                     <Stack mt="2rem">
                         {canReturnHome && <Navigate to={AppRoute.HOME} />}
                         {/* This should be displayed if a credential is required */}
-                        {isCredentialRequired && !localSettings?.env.SERVER_LOCK && (
+                        {isCredentialRequired && !isServerLock && (
                             <Group justify="center" wrap="nowrap">
                                 <Button
                                     fullWidth
diff --git a/src/renderer/features/login/routes/login-route.tsx b/src/renderer/features/login/routes/login-route.tsx
index 4f133f70b..359a5e46b 100644
--- a/src/renderer/features/login/routes/login-route.tsx
+++ b/src/renderer/features/login/routes/login-route.tsx
@@ -48,18 +48,16 @@ const LoginRoute = () => {
     const currentServer = useCurrentServer();
 
     // Check if server lock is configured
-    const serverLock = localSettings?.env.SERVER_LOCK || false;
-    const serverType = localSettings?.env.SERVER_TYPE
-        ? toServerType(localSettings.env.SERVER_TYPE)
-        : null;
-    const serverName = localSettings?.env.SERVER_NAME || '';
-    const serverUrl = localSettings?.env.SERVER_URL || '';
+    const isServerLock = Boolean(window.SERVER_LOCK) || false;
+    const serverType = window.SERVER_TYPE ? toServerType(window.SERVER_TYPE) : null;
+    const serverName = window.SERVER_NAME || '';
+    const serverUrl = window.SERVER_URL || '';
 
     const config = [
         {
             isValid: true,
             key: 'SERVER_LOCK',
-            value: serverLock,
+            value: isServerLock,
         },
         {
             isValid: serverType !== null,
diff --git a/src/renderer/features/search/components/server-commands.tsx b/src/renderer/features/search/components/server-commands.tsx
index 90b4b03ff..6fbd18a54 100644
--- a/src/renderer/features/search/components/server-commands.tsx
+++ b/src/renderer/features/search/components/server-commands.tsx
@@ -1,5 +1,4 @@
 import { openModal } from '@mantine/modals';
-import isElectron from 'is-electron';
 import { Dispatch, useCallback } from 'react';
 import { useTranslation } from 'react-i18next';
 import { useNavigate } from 'react-router';
@@ -10,8 +9,6 @@ import { AppRoute } from '/@/renderer/router/routes';
 import { useAuthStoreActions, useServerList } from '/@/renderer/store';
 import { ServerListItemWithCredential } from '/@/shared/types/domain-types';
 
-const localSettings = isElectron() ? window.api.localSettings : null;
-
 interface ServerCommandsProps {
     handleClose: () => void;
     setPages: (pages: CommandPalettePages[]) => void;
@@ -45,7 +42,7 @@ export const ServerCommands = ({ handleClose, setPages, setQuery }: ServerComman
         [handleClose, navigate, setCurrentServer, setPages, setQuery],
     );
 
-    const serverLock = localSettings?.env.SERVER_LOCK || false;
+    const isServerLock = Boolean(window.SERVER_LOCK) || false;
 
     return (
         <>
@@ -59,7 +56,7 @@ export const ServerCommands = ({ handleClose, setPages, setQuery }: ServerComman
                     >{`${serverList[key].name}...`}</Command.Item>
                 ))}
             </Command.Group>
-            {!serverLock && (
+            {!isServerLock && (
                 <Command.Group heading={t('common.manage', { postProcess: 'sentenceCase' })}>
                     <Command.Item onSelect={handleManageServersModal}>
                         {t('page.appMenu.manageServers', { postProcess: 'sentenceCase' })}...
diff --git a/src/renderer/features/servers/components/add-server-form.tsx b/src/renderer/features/servers/components/add-server-form.tsx
index 2e7579d5b..24e9e8bf5 100644
--- a/src/renderer/features/servers/components/add-server-form.tsx
+++ b/src/renderer/features/servers/components/add-server-form.tsx
@@ -112,13 +112,7 @@ export const AddServerForm = ({ onCancel }: AddServerFormProps) => {
     });
 
     // server lock for web is only true if lock is true *and* all other properties are set
-    const serverLock =
-        (localSettings
-            ? !!localSettings.env.SERVER_LOCK
-            : !!window.SERVER_LOCK &&
-              window.SERVER_TYPE &&
-              window.SERVER_NAME &&
-              window.SERVER_URL) || false;
+    const isServerLock = Boolean(window.SERVER_LOCK) || false;
 
     const isSubmitDisabled = !form.values.name || !form.values.url || !form.values.username;
 
@@ -227,7 +221,7 @@ export const AddServerForm = ({ onCancel }: AddServerFormProps) => {
                 <Stack m={5} ref={focusTrapRef}>
                     <SegmentedControl
                         data={ALL_SERVERS}
-                        disabled={Boolean(serverLock)}
+                        disabled={isServerLock}
                         p="md"
                         withItemsBorders={false}
                         {...form.getInputProps('type')}
@@ -235,7 +229,7 @@ export const AddServerForm = ({ onCancel }: AddServerFormProps) => {
                     <Group grow>
                         <TextInput
                             data-autofocus
-                            disabled={Boolean(serverLock)}
+                            disabled={isServerLock}
                             label={t('form.addServer.input', {
                                 context: 'name',
                                 postProcess: 'titleCase',
@@ -244,7 +238,7 @@ export const AddServerForm = ({ onCancel }: AddServerFormProps) => {
                             {...form.getInputProps('name')}
                         />
                         <TextInput
-                            disabled={Boolean(serverLock)}
+                            disabled={isServerLock}
                             label={t('form.addServer.input', {
                                 context: 'url',
                                 postProcess: 'titleCase',
diff --git a/src/renderer/features/sidebar/components/server-selector-items.tsx b/src/renderer/features/sidebar/components/server-selector-items.tsx
index cfa3e9709..cc7e8756f 100644
--- a/src/renderer/features/sidebar/components/server-selector-items.tsx
+++ b/src/renderer/features/sidebar/components/server-selector-items.tsx
@@ -1,6 +1,5 @@
 import { openModal } from '@mantine/modals';
 import { useQuery, useQueryClient } from '@tanstack/react-query';
-import isElectron from 'is-electron';
 import { useTranslation } from 'react-i18next';
 import { useNavigate } from 'react-router';
 
@@ -17,8 +16,6 @@ import { Icon } from '/@/shared/components/icon/icon';
 import { ServerListItemWithCredential, ServerType } from '/@/shared/types/domain-types';
 import { ServerFeature } from '/@/shared/types/features-types';
 
-const localSettings = isElectron() ? window.api.localSettings : null;
-
 export const ServerSelectorItems = () => {
     const { t } = useTranslation();
     const navigate = useNavigate();
@@ -91,7 +88,7 @@ export const ServerSelectorItems = () => {
         });
     };
 
-    const serverLock = localSettings?.env.SERVER_LOCK || false;
+    const isServerLock = Boolean(window.SERVER_LOCK) || false;
 
     return (
         <>
@@ -126,7 +123,7 @@ export const ServerSelectorItems = () => {
                     </DropdownMenu.Item>
                 );
             })}
-            {!serverLock && (
+            {!isServerLock && (
                 <DropdownMenu.Item
                     leftSection={<Icon icon="edit" />}
                     onClick={handleManageServersModal}
diff --git a/src/renderer/features/titlebar/components/app-menu.tsx b/src/renderer/features/titlebar/components/app-menu.tsx
index 40b07640f..9f1491ad9 100644
--- a/src/renderer/features/titlebar/components/app-menu.tsx
+++ b/src/renderer/features/titlebar/components/app-menu.tsx
@@ -13,8 +13,6 @@ import { DropdownMenu, MenuItemProps } from '/@/shared/components/dropdown-menu/
 import { Icon } from '/@/shared/components/icon/icon';
 import { toast } from '/@/shared/components/toast/toast';
 
-const localSettings = isElectron() ? window.api.localSettings : null;
-
 const browser = isElectron() ? window.api.browser : null;
 
 interface BaseMenuItem {
@@ -177,7 +175,7 @@ export const AppMenu = () => {
             type: 'divider',
         },
         {
-            condition: !localSettings?.env.SERVER_LOCK,
+            condition: !window.SERVER_LOCK,
             id: 'manage-servers',
             item: {
                 label: t('page.appMenu.manageServers', { postProcess: 'sentenceCase' }),
diff --git a/src/renderer/router/app-outlet.tsx b/src/renderer/router/app-outlet.tsx
index efe28b51e..af7d7aa49 100644
--- a/src/renderer/router/app-outlet.tsx
+++ b/src/renderer/router/app-outlet.tsx
@@ -1,4 +1,3 @@
-import isElectron from 'is-electron';
 import { useMemo } from 'react';
 import { Navigate, Outlet } from 'react-router';
 
@@ -9,14 +8,10 @@ import { Center } from '/@/shared/components/center/center';
 import { Spinner } from '/@/shared/components/spinner/spinner';
 import { AuthState } from '/@/shared/types/types';
 
-const localSettings = isElectron() ? window.api.localSettings : null;
-
 export const AppOutlet = () => {
     const currentServer = useCurrentServer();
     const authState = useServerAuthenticated();
 
-    const serverLock = localSettings?.env.SERVER_LOCK || false;
-
     const isActionsRequired = useMemo(() => {
         const isServerRequired = !currentServer;
 
@@ -34,12 +29,7 @@ export const AppOutlet = () => {
         );
     }
 
-    // When server lock is enabled always redirect to login
-    if (serverLock && !currentServer) {
-        return <Navigate replace to={AppRoute.LOGIN} />;
-    }
-
-    if (!serverLock && (isActionsRequired || authState === AuthState.INVALID)) {
+    if (isActionsRequired || authState === AuthState.INVALID) {
         return <Navigate replace to={AppRoute.ACTION_REQUIRED} />;
     }