feat: Development commit

1 files changed, 72 insertions, 19 deletions

diff --git a/Sora/Views/Generic/GenericListView.swift b/Sora/Views/Generic/GenericListView.swift
index b0ee422..0c8da79 100644
--- a/Sora/Views/Generic/GenericListView.swift
+++ b/Sora/Views/Generic/GenericListView.swift
@@ -1,6 +1,9 @@
+// swiftlint:disable file_length
+
 import SwiftUI
 
-struct GenericListView<T: Identifiable & Hashable & GenericItem>: View {  // swiftlint:disable:this type_body_length
+// swiftlint:disable:next type_body_length
+struct GenericListView<T: Identifiable & Hashable & GenericItem>: View {
   @EnvironmentObject private var settings: SettingsManager
   @EnvironmentObject private var manager: BooruManager
   @Binding var selectedTab: Int
@@ -11,7 +14,7 @@ struct GenericListView<T: Identifiable & Hashable & GenericItem>: View {  // swi
   @State private var newCollectionName = ""
   @State private var itemPendingCollectionAssignment: UUID?
   @State private var isCollectionErrorAlertPresented = false
-  @State private var selectedFolder: UUID?
+  @State private var selectedCollectionOption: CollectionPickerOption = .all
   @State private var sort: SettingsBookmarkSort = .dateAdded
   @State private var isCollectionPickerPresented = false
   @State private var isProviderPickerPresented = false
@@ -31,15 +34,25 @@ struct GenericListView<T: Identifiable & Hashable & GenericItem>: View {  // swi
   var filteredItems: [T] {
     items.filter { item in
       let matchesFolder: Bool = {
-        if selectedFolder == nil {
+        switch selectedCollectionOption {
+        case .all:
           return true
-        }
 
-        if selectedFolder == UUID.nilUUID() {
+        case .uncategorized:
           return item.folder == nil
-        }
 
-        return item.folder == selectedFolder
+        case .topLevel(let topLevelName):
+          if let folderId = item.folder,
+            let folder = settings.folders.first(where: { $0.id == folderId }),
+            let itemTopLevelName = folder.topLevelName
+          {
+            return itemTopLevelName == topLevelName
+          }
+          return false
+
+        case .folder(let folderId):
+          return item.folder == folderId
+        }
       }()
       let matchesSearch =
         searchText.isEmpty
@@ -76,7 +89,7 @@ struct GenericListView<T: Identifiable & Hashable & GenericItem>: View {  // swi
 
   @ViewBuilder private var listContent: some View {
     List {
-      if filteredItems.isEmpty && (!searchText.isEmpty || !(selectedFolder == nil)) {
+      if filteredItems.isEmpty && (!searchText.isEmpty || !(selectedCollectionOption == .all)) {
         Text("No matching items found")
           .foregroundColor(.secondary)
       }
@@ -116,16 +129,37 @@ struct GenericListView<T: Identifiable & Hashable & GenericItem>: View {  // swi
               .padding()
               .pickerStyle(.menu)
 
-              Picker("Collection", selection: $selectedFolder) {
-                Text("All").tag(nil as UUID?)
+              Picker("Collection", selection: $selectedCollectionOption) {
+                Text(CollectionPickerOption.all.name(settings)).tag(CollectionPickerOption.all)
 
                 if items.contains(where: { $0.folder == nil }) {
-                  Text("Uncategorised").tag(UUID.nilUUID() as UUID?)
+                  Text(CollectionPickerOption.uncategorized.name(settings)).tag(
+                    CollectionPickerOption.uncategorized
+                  )
+                }
+
+                ForEach(
+                  settings.folders
+                    .filter { $0.topLevelName != nil }
+                    .map { $0.topLevelName! }
+                    .removingDuplicates(), id: \.self
+                ) { topLevelName in
+                  Text(CollectionPickerOption.topLevel(topLevelName).name(settings)).tag(
+                    CollectionPickerOption.topLevel(topLevelName)
+                  )
                 }
 
-                ForEach(settings.folders, id: \.id) { folder in
+                ForEach(
+                  settings.folders.filter { folder in
+                    folder.topLevelName == nil
+                      || (selectedCollectionOption == .topLevel(folder.topLevelName!)
+                        && folder.topLevelName != nil)
+                  }, id: \.id
+                ) { folder in
                   if isCollectionPopulated(folder.id) {
-                    Text(folder.name).tag(folder.id)
+                    Text(CollectionPickerOption.folder(folder.id).name(settings)).tag(
+                      CollectionPickerOption.folder(folder.id)
+                    )
                   }
                 }
               }
@@ -199,16 +233,35 @@ struct GenericListView<T: Identifiable & Hashable & GenericItem>: View {  // swi
             }
 
             Menu {
-              Picker("Collection", selection: $selectedFolder) {
-                Text("All").tag(nil as UUID?)
+              Picker("Collection", selection: $selectedCollectionOption) {
+                Text(CollectionPickerOption.all.name(settings)).tag(CollectionPickerOption.all)
 
                 if items.contains(where: { $0.folder == nil }) {
-                  Text("Uncategorised").tag(UUID.nilUUID())
+                  Text(CollectionPickerOption.uncategorized.name(settings)).tag(
+                    CollectionPickerOption.uncategorized
+                  )
+                }
+
+                ForEach(
+                  settings.folders
+                    .filter { $0.topLevelName != nil }
+                    .map { $0.topLevelName! }
+                    .removingDuplicates(), id: \.self
+                ) { topLevelName in
+                  Text(CollectionPickerOption.topLevel(topLevelName).name(settings)).tag(
+                    CollectionPickerOption.topLevel(topLevelName)
+                  )
                 }
 
-                ForEach(settings.folders, id: \.id) { folder in
-                  Text(folder.name)
-                    .tag(folder.id)
+                ForEach(
+                  settings.folders.filter { folder in
+                    folder.topLevelName == nil
+                      || (selectedCollectionOption == .topLevel(folder.topLevelName!)
+                        && folder.topLevelName != nil)
+                  }, id: \.id
+                ) { folder in
+                  Text(CollectionPickerOption.folder(folder.id).name(settings))
+                    .tag(CollectionPickerOption.folder(folder.id))
                     .selectionDisabled(!isCollectionPopulated(folder.id))
                 }
               }