fix: 🐛 fixing deleted workspaces being cached

apps/web/src/components/common/sidebar/sections/workspaces/components/workspace-menu.tsx

@@ -2,7 +2,9 @@ import * as DropdownMenu from "@radix-ui/react-dropdown-menu";
 
 import useDeleteWorkspace from "@/hooks/mutations/workspace/use-delete-workspace";
 import { cn } from "@/lib/utils";
-import queryClient from "@/query-client";
+import useProjectStore from "@/store/project";
+import useWorkspaceStore from "@/store/workspace";
+import { useQueryClient } from "@tanstack/react-query";
 import { MoreHorizontal, Pencil, Plus, Trash2 } from "lucide-react";
 import { useState } from "react";
 import AddProjectModal from "../../projects/add-project-modal";
@@ -13,14 +15,19 @@ type WorkspaceMenuProps = {
 
 function WorkspaceMenu({ id }: WorkspaceMenuProps) {
   const [isCreateProjectModalOpen, setCreateProjectModalOpen] = useState(false);
+  const { setWorkspace } = useWorkspaceStore();
+  const { setProject } = useProjectStore();
+  const queryClient = useQueryClient();
   const { mutateAsync: deleteWorkspace } = useDeleteWorkspace({
     id,
   });
 
   async function handleDeleteWorkspace() {
     await deleteWorkspace();
-    queryClient.invalidateQueries({
-      queryKey: ["workspaces", "projects"],
+    setWorkspace(undefined);
+    setProject(undefined);
+    queryClient.removeQueries({
+      queryKey: ["workspaces"],
     });
   }
 

apps/web/src/store/workspace.ts

@@ -2,11 +2,11 @@ import type { Workspace } from "@/types/workspace";
 import { create } from "zustand";
 
 const useWorkspaceStore = create<{
-  workspace: Workspace | null;
-  setWorkspace: (updatedWorkspace: Workspace) => void;
+  workspace: Workspace | undefined;
+  setWorkspace: (updatedWorkspace: Workspace | undefined) => void;
 }>((set) => ({
-  workspace: null,
-  setWorkspace: (updatedWorkspace: Workspace) =>
+  workspace: undefined,
+  setWorkspace: (updatedWorkspace) =>
     set(() => ({ workspace: updatedWorkspace })),
 }));