/workspaces refactor

src/apps/officer-web/state/useServerSettings.ts

@@ -12,7 +12,6 @@ type ServerSettings = {
   accountMode?: 'organization' | 'single';
   aiHarnesses?: AIHarnesses;
   plugins?: Record<string, boolean>;
-  terminalSandboxed?: boolean;
 };
 
 const SETTINGS_KEY = ['SERVER_SETTINGS'];
@@ -30,8 +29,6 @@ export const useServerSettings = () => {
   const accountMode = settings?.accountMode;
   const aiHarnesses = settings?.aiHarnesses;
   const plugins = settings?.plugins;
-  const terminalSandboxed = settings?.terminalSandboxed;
-
   const saveSettings = useCallback(
     async (update: Partial<ServerSettings>) => {
       const result = await client.put<ServerSettings>('/server-settings', update);
@@ -40,5 +37,5 @@ export const useServerSettings = () => {
     [client, queryClient],
   );
 
-  return { onboardingComplete, accountMode, aiHarnesses, plugins, terminalSandboxed, isLoading, saveSettings };
+  return { onboardingComplete, accountMode, aiHarnesses, plugins, isLoading, saveSettings };
 };

src/apps/officer-web/state/useWorkspacesState.ts

@@ -0,0 +1,39 @@
+import { useCallback, useRef } from 'react';
+import { useQuery, useQueryClient } from '@tanstack/react-query';
+import { useClient } from 'hooks/useClient';
+import { useAuth } from 'hooks/useAuth';
+import type { UserState } from './types/user-settings';
+
+const QUERY_KEY = ['WORKSPACES_STATE'];
+
+export function useWorkspacesState<T>(key: string, defaultValue: T): [T, (value: T | ((prev: T) => T)) => void, boolean] {
+  const client = useClient();
+  const { isAuthenticated } = useAuth();
+  const queryClient = useQueryClient();
+  const clientRef = useRef(client);
+  clientRef.current = client;
+
+  const { data: state = {}, isSuccess } = useQuery<UserState>({
+    queryKey: QUERY_KEY,
+    enabled: isAuthenticated,
+    queryFn: () => client.get<UserState>('/user/workspaces-state'),
+    staleTime: Infinity,
+  });
+
+  const value = key in state ? (state[key] as T) : defaultValue;
+
+  const setValue = useCallback(
+    (update: T | ((prev: T) => T)) => {
+      const currentState = queryClient.getQueryData<UserState>(QUERY_KEY) ?? {};
+      const currentValue = key in currentState ? (currentState[key] as T) : defaultValue;
+      const newValue = typeof update === 'function' ? (update as (prev: T) => T)(currentValue) : update;
+
+      queryClient.setQueryData(QUERY_KEY, { ...currentState, [key]: newValue });
+
+      clientRef.current.patch('/user/workspaces-state', { [key]: newValue }).catch(() => {});
+    },
+    [key, defaultValue, queryClient],
+  );
+
+  return [value, setValue, isSuccess];
+}