fix: include root folder items in top-level favorites filter

1 files changed, 16 insertions, 0 deletions

diff --git a/Sora/Views/Shared/FolderHierarchy.swift b/Sora/Views/Shared/FolderHierarchy.swift
index 99c4f8e..e7a2bc0 100644
--- a/Sora/Views/Shared/FolderHierarchy.swift
+++ b/Sora/Views/Shared/FolderHierarchy.swift
@@ -41,4 +41,20 @@ struct FolderHierarchy {
   func hasTopLevelUncategorized(forTopLevelName topLevelName: String) -> Bool {
     folderNames.contains(topLevelName)
   }
+
+  func isInTopLevelFolder(_ folderIdentifier: UUID?, named topLevelName: String) -> Bool {
+    guard let folderIdentifier, let folder = folder(for: folderIdentifier) else {
+      return false
+    }
+
+    return folder.name == topLevelName || folder.topLevelName == topLevelName
+  }
+
+  func isInTopLevelUncategorized(_ folderIdentifier: UUID?, named topLevelName: String) -> Bool {
+    guard let folderIdentifier, let folder = folder(for: folderIdentifier) else {
+      return false
+    }
+
+    return folder.name == topLevelName
+  }
 }