From c775239f51337f4cb7238ecc60ee0577ce5debfd Mon Sep 17 00:00:00 2001
From: Jakub Niechaj <quban123@gmail.com>
Date: Thu, 19 Oct 2023 18:02:14 +0200
Subject: [PATCH] Add conditional Keycloak provider

---
 src/App.tsx                                   |  2 +-
 .../Dialog/RejectKeycloakUserDialog.tsx       |  4 +-
 src/ThreeEditor/js/Storage.js                 | 53 ++++++++++++++-----
 .../components/Panels/LoginPanel.tsx          |  4 +-
 src/services/AuthService.tsx                  | 13 ++---
 src/services/GenericContext.ts                | 22 ++++++++
 src/services/KeycloakAuthService.tsx          | 33 ++++++++++--
 7 files changed, 98 insertions(+), 33 deletions(-)

diff --git a/src/App.tsx b/src/App.tsx
index a5c6f69c8..213d272de 100644
--- a/src/App.tsx
+++ b/src/App.tsx
@@ -86,11 +86,11 @@ function App() {
 				<ThemeProvider theme={theme} />,
 				<SnackbarProvider maxSnack={3} />,
 				<KeycloakAuth />,
+				<Store />,
 				<DialogProvider />,
 				<Auth />,
 				<ShSimulation />,
 				<PythonConverterService />,
-				<Store />,
 				<Loader />
 			]}>
 			<WrapperApp />
diff --git a/src/ThreeEditor/components/Dialog/RejectKeycloakUserDialog.tsx b/src/ThreeEditor/components/Dialog/RejectKeycloakUserDialog.tsx
index 81001014c..162b39e1a 100644
--- a/src/ThreeEditor/components/Dialog/RejectKeycloakUserDialog.tsx
+++ b/src/ThreeEditor/components/Dialog/RejectKeycloakUserDialog.tsx
@@ -1,13 +1,13 @@
 import { Button } from '@mui/material';
-import { useKeycloak } from 'keycloak-react-web';
 
+import { useKeycloakAuth } from '../../../services/KeycloakAuthService';
 import { ConcreteDialogProps, CustomDialog } from './CustomDialog';
 
 export function RejectKeycloakUserDialog({
 	onClose,
 	reason
 }: ConcreteDialogProps & { reason: string }) {
-	const { keycloak, initialized } = useKeycloak();
+	const { keycloak, initialized } = useKeycloakAuth();
 
 	return (
 		<CustomDialog
diff --git a/src/ThreeEditor/js/Storage.js b/src/ThreeEditor/js/Storage.js
index 22485481f..698d682a9 100644
--- a/src/ThreeEditor/js/Storage.js
+++ b/src/ThreeEditor/js/Storage.js
@@ -39,27 +39,52 @@ function Storage() {
 
 			request.onerror = function (event) {
 				console.error('IndexedDB', event);
+				// delete database
+				const req = indexedDB.deleteDatabase(name);
+				req.onsuccess = function () {
+					console.log('Deleted database successfully');
+				};
+
+				req.onerror = function () {
+					console.log('Could not delete database');
+				};
+
+				req.onblocked = function () {
+					console.log('Could not delete database due to the operation being blocked');
+				};
 			};
 		},
 
 		get: function (callback) {
-			const transaction = database.transaction(['states'], 'readwrite');
-			const objectStore = transaction.objectStore('states');
-			const request = objectStore.get(0);
-			request.onsuccess = function (event) {
-				callback(event.target.result);
-			};
+			try {
+				const transaction = database.transaction(['states'], 'readwrite');
+				const objectStore = transaction.objectStore('states');
+				const request = objectStore.get(0);
+				request.onsuccess = function (event) {
+					callback(event.target.result);
+				};
+			} catch (error) {
+				console.error(error);
+				callback();
+			}
 		},
 
 		set: function (data) {
-			const start = performance.now();
-
-			const transaction = database.transaction(['states'], 'readwrite');
-			const objectStore = transaction.objectStore('states');
-			const request = objectStore.put(data, 0);
-			request.onsuccess = function () {
-				devLog('Saved state to IndexedDB.', `${(performance.now() - start).toFixed(2)}ms`);
-			};
+			try {
+				const start = performance.now();
+
+				const transaction = database.transaction(['states'], 'readwrite');
+				const objectStore = transaction.objectStore('states');
+				const request = objectStore.put(data, 0);
+				request.onsuccess = function () {
+					devLog(
+						'Saved state to IndexedDB.',
+						`${(performance.now() - start).toFixed(2)}ms`
+					);
+				};
+			} catch (error) {
+				console.error(error);
+			}
 		},
 
 		clear: function () {
diff --git a/src/WrapperApp/components/Panels/LoginPanel.tsx b/src/WrapperApp/components/Panels/LoginPanel.tsx
index 973fbdf68..d246660dd 100644
--- a/src/WrapperApp/components/Panels/LoginPanel.tsx
+++ b/src/WrapperApp/components/Panels/LoginPanel.tsx
@@ -1,14 +1,14 @@
 import { Box, Button, Card, CardContent, TextField, Typography, useTheme } from '@mui/material';
-import { useKeycloak } from 'keycloak-react-web';
 import { ChangeEvent, useCallback, useEffect, useState } from 'react';
 
 import { useConfig } from '../../../config/ConfigService';
 import { useAuth } from '../../../services/AuthService';
+import { useKeycloakAuth } from '../../../services/KeycloakAuthService';
 
 export default function LoginPanel() {
 	const { altAuth } = useConfig();
 	const { login } = useAuth();
-	const { keycloak, initialized } = useKeycloak();
+	const { keycloak, initialized } = useKeycloakAuth();
 	const theme = useTheme();
 
 	const [username, setUsername] = useState('');
diff --git a/src/services/AuthService.tsx b/src/services/AuthService.tsx
index cd0e5ff7f..c74e75da0 100644
--- a/src/services/AuthService.tsx
+++ b/src/services/AuthService.tsx
@@ -1,5 +1,4 @@
 import { Backdrop, CircularProgress, Theme, Typography } from '@mui/material';
-import { useKeycloak } from 'keycloak-react-web';
 import ky, { HTTPError } from 'ky';
 import { KyInstance } from 'ky/distribution/types/ky';
 import { useSnackbar } from 'notistack';
@@ -18,6 +17,7 @@ import useIntervalAsync from '../util/hooks/useIntervalAsync';
 import { snakeToCamelCase } from '../util/Notation/Notation';
 import { useDialog } from './DialogService';
 import { createGenericContext, GenericContextProviderProps } from './GenericContext';
+import { useKeycloakAuth } from './KeycloakAuthService';
 
 type AuthUser = Pick<ResponseAuthStatus, 'username' | 'source'>;
 
@@ -80,7 +80,7 @@ const Auth = ({ children }: GenericContextProviderProps) => {
 	const [user, setUser] = useState<AuthUser | null>(load(StorageKey.USER, isAuthUser));
 	const [reachInterval, setReachInterval] = useState<number>();
 	const [refreshInterval, setRefreshInterval] = useState<number | undefined>(180000); // 3 minutes in ms default interval for refresh token
-	const { keycloak, initialized } = useKeycloak();
+	const { keycloak, initialized } = useKeycloakAuth();
 	const [keyCloakInterval, setKeyCloakInterval] = useState<number>();
 	const [isServerReachable, setIsServerReachable] = useState<boolean | null>(null);
 	const { enqueueSnackbar } = useSnackbar();
@@ -215,14 +215,7 @@ const Auth = ({ children }: GenericContextProviderProps) => {
 								: err.message
 					});
 				});
-	}, [
-		initialized,
-		keycloak.authenticated,
-		keycloak.token,
-		keycloak.tokenParsed?.preferred_username,
-		kyRef,
-		open
-	]);
+	}, [initialized, keycloak, kyRef, open]);
 
 	useEffect(() => {
 		if (initialized && keycloak.authenticated)
diff --git a/src/services/GenericContext.ts b/src/services/GenericContext.ts
index 4190277a0..7821375ce 100644
--- a/src/services/GenericContext.ts
+++ b/src/services/GenericContext.ts
@@ -21,3 +21,25 @@ export const createGenericContext = <T extends unknown>() => {
 
 	return [useGenericContext, genericContext.Provider] as const;
 };
+
+export const createSubstituteContext = <T extends unknown>(useSubstituted: () => T) => {
+	// Create a context with a generic parameter or undefined
+	const genericContext = createContext<T | undefined>(undefined);
+
+	// Check if the value provided to the context is defined or call the substitute
+	const useSubstituteOrGenericContext = () => {
+		const contextIsDefined = useContext(genericContext);
+
+		try {
+			return useSubstituted();
+		} catch (e) {
+			if (!contextIsDefined) {
+				throw new Error('useSubstituteOrGenericContext must be used within a Provider');
+			}
+
+			return contextIsDefined;
+		}
+	};
+
+	return [useSubstituteOrGenericContext, genericContext.Provider] as const;
+};
diff --git a/src/services/KeycloakAuthService.tsx b/src/services/KeycloakAuthService.tsx
index cfb229531..d37b3d162 100644
--- a/src/services/KeycloakAuthService.tsx
+++ b/src/services/KeycloakAuthService.tsx
@@ -1,7 +1,8 @@
 import Keycloak, { KeycloakInitOptions } from 'keycloak-js';
-import { KeycloakProvider } from 'keycloak-react-web';
+import { KeycloakProvider, useKeycloak } from 'keycloak-react-web';
 
-import { GenericContextProviderProps } from './GenericContext';
+import { useConfig } from '../config/ConfigService';
+import { createSubstituteContext, GenericContextProviderProps } from './GenericContext';
 
 const authInstance = new Keycloak({
 	url: `${
@@ -20,12 +21,36 @@ const initOptions = {
 	enableLogging: false
 } as const satisfies KeycloakInitOptions;
 
-export const KeycloakAuth = ({ children }: GenericContextProviderProps) => {
-	return (
+export type KeycloakAuthContext =
+	| {
+			initialized: false;
+			keycloak?: Keycloak;
+	  }
+	| {
+			initialized: true;
+			keycloak: Keycloak;
+	  };
+
+const [useKeycloakAuth, KeycloakAuthContextProvider] =
+	createSubstituteContext<KeycloakAuthContext>(useKeycloak);
+
+const KeycloakAuth = ({ children }: GenericContextProviderProps) => {
+	const { altAuth } = useConfig();
+
+	return altAuth ? (
 		<KeycloakProvider
 			client={authInstance}
 			initOptions={initOptions}>
 			{children}
 		</KeycloakProvider>
+	) : (
+		<KeycloakAuthContextProvider
+			value={{
+				initialized: false
+			}}>
+			{children}
+		</KeycloakAuthContextProvider>
 	);
 };
+
+export { KeycloakAuth, useKeycloakAuth };