fix(platform): Replace localStorage with Jotai

apps/platform/src/components/ui/combobox.tsx

@@ -22,9 +22,9 @@ import { apiClient } from '@/lib/api-client'
 import type { Workspace } from '@/types'
 import { zWorkspace } from '@/types'
 import {
-  getCurrentWorkspace,
-  setCurrentWorkspace,
-  setWorkspace
+  GetCurrentWorkspace,
+  SetCurrentWorkspace,
+  SetWorkspace
 } from '@/lib/workspace-storage'
 import { Input } from './input'
 import { Label } from './label'
@@ -74,7 +74,7 @@ export function Combobox(): React.JSX.Element {
       const response = await apiClient.post<Workspace>('/workspace', {
         name
       })
-      setCurrentWorkspace(response)
+      SetCurrentWorkspace(response)
       setOpen(false)
     } catch (error) {
       // eslint-disable-next-line no-console -- we need to log the error
@@ -87,7 +87,7 @@ export function Combobox(): React.JSX.Element {
       .then((data) => {
         if (data) {
           setAllWorkspace(data)
-          setWorkspace(data)
+          SetWorkspace(data)
         }
       })
       .catch((error) => {
@@ -112,7 +112,7 @@ export function Combobox(): React.JSX.Element {
             </div>
             <div className="flex flex-col items-start">
               <div className="text-lg text-white">
-                {getCurrentWorkspace()?.name ?? 'No workspace'}
+                {GetCurrentWorkspace()?.name ?? 'No workspace'}
               </div>
               <span className="text-xs text-white/55">100+ projects</span>
             </div>
@@ -132,7 +132,7 @@ export function Combobox(): React.JSX.Element {
                   <CommandItem
                     key={workspace.id}
                     onSelect={() => {
-                      setCurrentWorkspace(workspace)
+                      SetCurrentWorkspace(workspace)
                       router.refresh()
                       setOpen(false)
                     }}
@@ -141,7 +141,7 @@ export function Combobox(): React.JSX.Element {
                     <Check
                       className={cn(
                         'mr-2 h-4 w-4',
-                        getCurrentWorkspace()?.name === workspace.name
+                        GetCurrentWorkspace()?.name === workspace.name
                           ? 'opacity-100'
                           : 'opacity-0'
                       )}

apps/platform/src/lib/workspace-storage.ts

@@ -1,29 +1,60 @@
+import { atom, useAtom } from 'jotai'
 import type { Workspace } from '@/types'
 
-export function setWorkspace(workspaceData: Workspace[]): void {
+const defaultWorkspaceAtom = atom<Workspace | null>(null)
+const currentWorkspaceAtom = atom<Workspace | null>(null)
+
+export const useDefaultWorkspace = (): {
+  defaultWorkspace: Workspace | null
+  setDefaultWorkspace: (workspace: Workspace | null) => void
+} => {
+  const [defaultWorkspace, setDefaultWorkspace] = useAtom(defaultWorkspaceAtom)
+  return { defaultWorkspace, setDefaultWorkspace }
+}
+
+export const useCurrentWorkspace = (): {
+  currentWorkspace: Workspace | null
+  setCurrentWorkspace: (workspace: Workspace | null) => void
+} => {
+  const [currentWorkspace, setCurrentWorkspace] = useAtom(currentWorkspaceAtom)
+  return { currentWorkspace, setCurrentWorkspace }
+}
+
+export const SetWorkspace = (workspaceData: Workspace[]): void => {
+  const { setDefaultWorkspace } = useDefaultWorkspace()
+  const { currentWorkspace, setCurrentWorkspace } = useCurrentWorkspace()
+
   const defaultWorkspace =
     workspaceData.find((workspace) => workspace.isDefault) || null
-  if (typeof localStorage !== 'undefined') {
-    localStorage.setItem('defaultWorkspace', JSON.stringify(defaultWorkspace))
 
-    if (getCurrentWorkspace() === null) {
-      localStorage.setItem('currentWorkspace', JSON.stringify(defaultWorkspace))
-    }
+  setDefaultWorkspace(defaultWorkspace)
+
+  if (currentWorkspace === null && defaultWorkspace) {
+    setCurrentWorkspace(defaultWorkspace)
   }
 }
 
-export function getCurrentWorkspace(): Workspace | null {
-  const currentWorkspace =
-    typeof localStorage !== 'undefined'
-      ? localStorage.getItem('currentWorkspace')
-      : `{}`
+export const GetCurrentWorkspace = (): Workspace | null => {
+  const { currentWorkspace } = useCurrentWorkspace()
+  return currentWorkspace
+}
 
-  if (currentWorkspace) {
-    return JSON.parse(currentWorkspace) as Workspace
-  }
-  return null
+export const SetCurrentWorkspace = (workspace: Workspace): void => {
+  const { setCurrentWorkspace } = useCurrentWorkspace()
+  setCurrentWorkspace(workspace)
+}
+
+//Utility functions
+export const UpdateDefaultWorkspaceAtom = (
+  workspace: Workspace | null
+): void => {
+  const { setDefaultWorkspace } = useDefaultWorkspace()
+  setDefaultWorkspace(workspace)
 }
 
-export function setCurrentWorkspace(workspace: Workspace): void {
-  localStorage.setItem('currentWorkspace', JSON.stringify(workspace))
+export const UpdateCurrentWorkspaceAtom = (
+  workspace: Workspace | null
+): void => {
+  const { setCurrentWorkspace } = useCurrentWorkspace()
+  setCurrentWorkspace(workspace)
 }