From bab6e24a0d85e14ec337e1649d6f525003b0d547 Mon Sep 17 00:00:00 2001
From: Kenneth <kenneth@nym.sh>
Date: Mon, 15 Dec 2025 22:59:35 +0000
Subject: [PATCH] fix: update dir content query data on create dir

---
 apps/drive-web/src/vfs/api.ts | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/apps/drive-web/src/vfs/api.ts b/apps/drive-web/src/vfs/api.ts
index eba5f0c..67ae674 100644
--- a/apps/drive-web/src/vfs/api.ts
+++ b/apps/drive-web/src/vfs/api.ts
@@ -81,8 +81,6 @@ export const directoryContentQueryAtom = atomFamily((directoryId: string) =>
 	}),
 )
 
-// Directory Mutations
-
 export const createDirectoryMutationAtom = atom((get) => {
 	const account = get(currentAccountAtom)
 	return mutationOptions({
@@ -105,6 +103,13 @@ export const createDirectoryMutationAtom = atom((get) => {
 				get(directoryInfoQueryAtom(data.id)).queryKey,
 				data,
 			)
+			const parent = data.path.at(-2)
+			if (parent) {
+				client.setQueryData(
+					get(directoryContentQueryAtom(parent.id)).queryKey,
+					(prev) => (prev ? [...prev, data] : [data]),
+				)
+			}
 		},
 	})
 })