From 8eb0d4fd0d4ee2be6814dde030843c3a05c5b2cd Mon Sep 17 00:00:00 2001
From: Xu Quan <xuquan1225@hotmail.com>
Date: Tue, 24 Jun 2025 16:08:53 +0800
Subject: [PATCH] fix: use isRoot to check if folder is top level

---
 src/store/folder/structure.ts | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/store/folder/structure.ts b/src/store/folder/structure.ts
index 73553aa..fe5bbbd 100644
--- a/src/store/folder/structure.ts
+++ b/src/store/folder/structure.ts
@@ -11,6 +11,7 @@ export interface FolderStructureSlice {
 	rootFolder: TFolder | null;
 
 	getTopLevelFolders: () => TFolder[];
+	isTopLevelFolder: (folder: TFolder) => boolean;
 	getFilesCountInFolder: (
 		folder: TFolder,
 		includeSubfolderFiles: boolean
@@ -29,10 +30,14 @@ export const createFolderStructureSlice =
 		folders: plugin.app.vault.getAllFolders() || [],
 		rootFolder: plugin.app.vault.getRoot() || null,
 
+		isTopLevelFolder: (folder: TFolder): boolean => {
+			return Boolean(folder.parent?.isRoot());
+		},
 		getTopLevelFolders: () => {
+			const {isTopLevelFolder } = get();
 			return plugin.app.vault
 				.getAllFolders()
-				.filter((folder) => folder.parent?.parent === null);
+				.filter((folder) => isTopLevelFolder(folder));
 		},
 		hasFolderChildren: (folder: TFolder): boolean => {
 			return folder.children.some((child) => isFolder(child));