Store workspace in session storage (#6973)

frontend/src/lib/stores.ts

@@ -36,11 +36,11 @@ export interface UserWorkspace {
 	disabled: boolean
 }
 
-const persistedWorkspace = BROWSER && getWorkspace()
+const persistedWorkspace = BROWSER && getWorkspaceFromStorage()
 
-function getWorkspace(): string | undefined {
+export function getWorkspaceFromStorage(): string | undefined {
 	try {
-		return localStorage.getItem('workspace') ?? undefined
+		return sessionStorage.getItem('workspace') ?? localStorage.getItem('workspace') ?? undefined
 	} catch (e) {
 		console.error('error interacting with local storage', e)
 	}
@@ -116,7 +116,6 @@ export const codeCompletionSessionEnabled = writable<boolean>(
 	getLocalSetting(CODE_COMPLETION_SETTING_NAME) != 'false'
 )
 
-
 export const usedTriggerKinds = writable<string[]>([])
 
 type SQLBaseSchema = {

frontend/src/routes/(root)/(logged)/+layout.svelte

@@ -97,6 +97,7 @@
 	async function updateUserStore(workspace: string | undefined) {
 		if (workspace) {
 			try {
+				sessionStorage.setItem('workspace', String(workspace))
 				localStorage.setItem('workspace', String(workspace))
 			} catch (e) {
 				console.error('Could not persist workspace to local storage', e)

frontend/src/routes/(root)/(logged)/user/(user)/workspaces/+page.svelte

@@ -10,7 +10,8 @@
 		usersWorkspaceStore,
 		userWorkspaces,
 		workspaceStore,
-		userStore
+		userStore,
+		getWorkspaceFromStorage
 	} from '$lib/stores'
 	import { Button, Skeleton } from '$lib/components/common'
 	import Toggle from '$lib/components/Toggle.svelte'
@@ -58,7 +59,7 @@
 		}
 		if ($usersWorkspaceStore) {
 			if (!$workspaceStore) {
-				const local = localStorage.getItem('workspace')?.toString()
+				const local = getWorkspaceFromStorage()?.toString()
 				switchWorkspace(local)
 			}
 		} else {