Update local model stores in all mutations (#129)

src-web/hooks/useDeleteWorkspace.tsx

@@ -1,16 +1,20 @@
 import { useMutation } from '@tanstack/react-query';
 import type { Workspace } from '@yaakapp-internal/models';
+import {useSetAtom} from "jotai";
 import { InlineCode } from '../components/core/InlineCode';
 import { trackEvent } from '../lib/analytics';
 import { invokeCmd } from '../lib/tauri';
 import { useActiveWorkspace } from './useActiveWorkspace';
 import { useAppRoutes } from './useAppRoutes';
 import { useConfirm } from './useConfirm';
+import {removeModelById} from "./useSyncModelStores";
+import {workspacesAtom} from "./useWorkspaces";
 
 export function useDeleteWorkspace(workspace: Workspace | null) {
   const activeWorkspace = useActiveWorkspace();
   const routes = useAppRoutes();
   const confirm = useConfirm();
+  const setWorkspaces = useSetAtom(workspacesAtom);
 
   return useMutation<Workspace | null, string>({
     mutationKey: ['delete_workspace', workspace?.id],
@@ -32,6 +36,9 @@ export function useDeleteWorkspace(workspace: Workspace | null) {
     onSuccess: async (workspace) => {
       if (workspace === null) return;
 
+      // Optimistic update
+      setWorkspaces(removeModelById(workspace));
+
       const { id: workspaceId } = workspace;
       if (workspaceId === activeWorkspace?.id) {
         routes.navigate('workspaces');