path: root/Sora/Views/FavoritesView.swift

// swiftlint:disable file_length

import SwiftUI
import WaterfallGrid

struct FavoritesView: View {  // swiftlint:disable:this type_body_length
  @EnvironmentObject var settings: SettingsManager
  @EnvironmentObject var manager: BooruManager
  @Binding var selectedTab: Int
  @State private var searchText: String = ""
  @State private var isShowingRemoveAllConfirmation = false
  @Binding var isPresented: Bool
  @State private var isNewCollectionAlertPresented = false
  @State private var newCollectionName = ""
  @State private var itemPendingCollectionAssignment: UUID?
  @State private var isCollectionErrorAlertPresented = false
  @State private var selectedCollectionOption: CollectionPickerOption = .all
  @State private var sort: SettingsBookmarkSort = .dateAdded
  @State private var isCollectionPickerPresented = false
  @State private var isProviderPickerPresented = false
  @State private var selectedProvider: BooruProvider?
  @State private var navigationPath = NavigationPath()
  @State private var isDetailsSheetPresented = false
  @State private var selectedFavoriteForDetails: SettingsFavoritePost?

  var filteredFavorites: [SettingsFavoritePost] {
    settings.favorites.filter { favorite in
      let matchesFolder: Bool = {
        switch selectedCollectionOption {
        case .all:
          return true

        case .uncategorized:
          return favorite.folder == nil

        case .folder(let folderId):
          return favorite.folder == folderId

        case .topLevelFolder(let topLevelName):
          guard let favoriteFolderId = favorite.folder,
            let favoriteFolder = settings.folders.first(where: { $0.id == favoriteFolderId })
          else {
            return false
          }

          return favoriteFolder.topLevelName == topLevelName
        }
      }()
      let matchesSearch =
        searchText.isEmpty
        || favorite.tags
          .joined(separator: " ")
          .lowercased()
          .contains(searchText.lowercased())
      let matchesProvider =
        selectedProvider == nil
        || favorite.provider == selectedProvider

      return matchesFolder && matchesSearch && matchesProvider
    }
  }

  var sortedFilteredFavorites: [SettingsFavoritePost] {
    filteredFavorites.sorted { (lhs: SettingsFavoritePost, rhs: SettingsFavoritePost) in
      switch sort {
      case .dateAdded:
        return lhs.date > rhs.date

      case .lastVisited:
        return lhs.lastVisit > rhs.lastVisit

      case .visitCount:
        return lhs.visitedCount > rhs.visitedCount

      case .shuffle:
        return Bool.random()
      }
    }
  }

  private func isProviderFavorited(_ provider: BooruProvider) -> Bool {
    settings.favorites.contains { $0.provider == provider }
  }

  private func isCollectionPopulated(_ folder: UUID?) -> Bool {
    settings.favorites.contains { $0.folder == folder }
  }

  var body: some View {
    NavigationStack(path: $navigationPath) {
      VStack(spacing: 0) {
        if settings.favorites.isEmpty {
          ContentUnavailableView(
            "No Favorites",
            systemImage: "heart",
            description: Text("Tap the heart button on a post to add it to favorites.")
          )
        } else {
          #if os(macOS)
            if !settings.folders.isEmpty {
              Picker("Sort", selection: $sort) {
                ForEach(SettingsBookmarkSort.allCases, id: \.rawValue) { sortMode in
                  Text(sortMode.rawValue).tag(sortMode)
                }
              }
              .padding()
              .pickerStyle(.menu)

              Picker("Collection", selection: $selectedCollectionOption) {
                Text(CollectionPickerOption.all.name(settings: settings))
                  .tag(CollectionPickerOption.all)

                if settings.favorites.contains(where: { $0.folder == nil }) {
                  Text(CollectionPickerOption.uncategorized.name(settings: settings))
                    .tag(CollectionPickerOption.uncategorized)
                }

                ForEach(settings.folders.filter { $0.topLevelName == nil }, id: \.id) { folder in
                  Text(folder.name).tag(CollectionPickerOption.folder(folder.id))
                }

                let topLevelFolders = settings.folders
                  .reduce(into: [String: [SettingsFolder]]()) { result, folder in
                    guard let topLevelName = folder.topLevelName else { return }

                    result[topLevelName, default: []].append(folder)
                  }

                ForEach(topLevelFolders.keys.sorted(), id: \.self) { topLevelName in
                  Menu {
                    ForEach(topLevelFolders[topLevelName] ?? [], id: \.id) { folder in
                      Button(action: {
                        selectedCollectionOption = .folder(folder.id)
                      }) {
                        Text(folder.shortName)
                      }
                    }
                  } label: {
                    Text(topLevelName)
                  }
                }
              }
              .padding(.bottom)
              .padding(.horizontal)
              .pickerStyle(.menu)

              Picker("Provider", selection: $selectedProvider) {
                Text("All").tag(nil as BooruProvider?)

                ForEach(BooruProvider.allCases, id: \.rawValue) { provider in
                  if isProviderFavorited(provider) {
                    Text(provider.rawValue).tag(provider)
                  }
                }
              }
              .padding(.bottom)
              .padding(.horizontal)
              .pickerStyle(.menu)
            }
          #endif

          ScrollView {
            if filteredFavorites.isEmpty
              && (!searchText.isEmpty || !(selectedCollectionOption == .all))
            {
              ContentUnavailableView(
                "No matching favorites found",
                systemImage: "heart.slash",
                description: Text("Try adjusting your search or collection filter.")
              )
            } else {
              gridView(columnCount: settings.thumbnailGridColumns)
            }
          }
          .searchable(text: $searchText)
        }
      }
    }
    .navigationTitle("Favorites")
    .refreshable { settings.syncFromCloud() }
    .toolbar {
      #if os(macOS)
        ToolbarItem {
          Button(action: {
            settings.syncFromCloud()
          }) {
            Label("Refresh", systemImage: "arrow.clockwise")
          }
        }
      #endif

      #if os(macOS)
        ToolbarItem {
          Button(
            role: .destructive,
            action: {
              isShowingRemoveAllConfirmation = true
            }
          ) {
            Label("Remove All", systemImage: "trash")
          }
        }
      #endif

      #if os(iOS)
        ToolbarItemGroup(placement: .secondaryAction) {
          Menu {
            Picker("Sort By", selection: $sort) {
              ForEach(SettingsBookmarkSort.allCases, id: \.rawValue) { sortMode in
                Text(sortMode.rawValue).tag(sortMode)
              }
            }
          } label: {
            Label("Sort By", systemImage: "arrow.up.arrow.down")
          }

          Menu {
            Button(action: {
              selectedCollectionOption = .all
            }) {
              Label("All", systemImage: "folder")
            }

            if settings.favorites.contains(where: { $0.folder == nil }) {
              Button(action: {
                selectedCollectionOption = .uncategorized
              }) {
                Label("Uncategorized", systemImage: "folder.badge.questionmark")
              }
            }

            ForEach(settings.folders.filter { $0.topLevelName == nil }, id: \.id) { folder in
              Button(action: {
                selectedCollectionOption = .folder(folder.id)
              }) {
                Label(folder.name, systemImage: "folder")
              }
            }

            let topLevelFolders = settings.folders
              .reduce(into: [String: [SettingsFolder]]()) { result, folder in
                guard let topLevelName = folder.topLevelName else { return }

                result[topLevelName, default: []].append(folder)
              }

            ForEach(topLevelFolders.keys.sorted(), id: \.self) { topLevelName in
              Menu {
                Button(action: {
                  selectedCollectionOption = .topLevelFolder(topLevelName)
                }) {
                  Label("All \(topLevelName)", systemImage: "folder.fill")
                }

                Divider()

                ForEach(topLevelFolders[topLevelName] ?? [], id: \.id) { folder in
                  Button(action: {
                    selectedCollectionOption = .folder(folder.id)
                  }) {
                    Label(folder.shortName, systemImage: "folder")
                  }
                }
              } label: {
                Label(topLevelName, systemImage: "folder.fill")
              }
            }
          } label: {
            Label("Collection", systemImage: "folder")
          }

          Menu {
            Picker("Provider", selection: $selectedProvider) {
              Text("All").tag(nil as BooruProvider?)

              ForEach(BooruProvider.allCases, id: \.rawValue) { provider in
                Text(provider.rawValue)
                  .tag(provider)
                  .selectionDisabled(!isProviderFavorited(provider))
              }
            }
          } label: {
            Label("Provider", systemImage: "globe")
          }

          Button(
            role: .destructive,
            action: {
              isShowingRemoveAllConfirmation = true
            }
          ) {
            Label("Delete All", systemImage: "trash")
          }
        }
      #endif
    }
    .alert(
      "Are you sure you want to remove all favorites? This action cannot be undone.",
      isPresented: $isShowingRemoveAllConfirmation
    ) {
      Button("Remove All Favorites") {
        settings.favorites.removeAll()
      }

      Button("Cancel", role: .cancel) { () }
    }
    .collectionAlerts(
      isNewCollectionAlertPresented: $isNewCollectionAlertPresented,
      newCollectionName: $newCollectionName,
      isCollectionErrorAlertPresented: $isCollectionErrorAlertPresented
    ) { newCollectionName in
      let newFolder = SettingsFolder(name: newCollectionName)

      settings.folders.append(newFolder)

      if let id = itemPendingCollectionAssignment {
        settings.updateFavoriteFolder(withID: id, folder: newFolder.id)
      }

      itemPendingCollectionAssignment = nil
    }
    .navigationDestination(for: PostWithContext.self) { context in
      PostDetailsView(
        post: context.post,
        navigationPath: $navigationPath,
        posts: context.posts,
        baseSearchText: context.baseSearchText
      )
    }
    .navigationDestination(for: String.self) { tag in
      PostGridView(
        selectedTab: $selectedTab, navigationPath: $navigationPath, initialTag: tag
      )
    }
    .sheet(isPresented: $isDetailsSheetPresented) {
      if let favorite = selectedFavoriteForDetails {
        NavigationView {
          VStack(alignment: .leading, spacing: 16) {
            Text("Favorite Details")
              .font(.title2)
              .fontWeight(.bold)

            VStack(alignment: .leading, spacing: 12) {
              VStack(alignment: .leading, spacing: 4) {
                Text("Tags")
                  .font(.headline)
                Text(favorite.tags.joined(separator: ", ").lowercased())
                  .font(.body)
                  .foregroundStyle(Color.secondary)
              }

              VStack(alignment: .leading, spacing: 4) {
                Text("Favorited")
                  .font(.headline)
                Text(favorite.createdAt.formatted())
                  .font(.body)
                  .foregroundStyle(Color.secondary)
              }

              VStack(alignment: .leading, spacing: 4) {
                Text("Provider")
                  .font(.headline)
                Text(favorite.provider.rawValue)
                  .font(.body)
                  .foregroundStyle(Color.secondary)
              }

              if let folder = favorite.folder, let folderName = settings.folderName(forID: folder) {
                VStack(alignment: .leading, spacing: 4) {
                  Text("Collection")
                    .font(.headline)
                  Text(folderName)
                    .font(.body)
                    .foregroundStyle(Color.secondary)
                }
              }

              VStack(alignment: .leading, spacing: 4) {
                Text("Visit Count")
                  .font(.headline)
                Text("\(favorite.visitedCount) times")
                  .font(.body)
                  .foregroundStyle(Color.secondary)
              }

              if favorite.lastVisit != favorite.createdAt {
                VStack(alignment: .leading, spacing: 4) {
                  Text("Last Visited")
                    .font(.headline)
                  Text(favorite.lastVisit.formatted())
                    .font(.body)
                    .foregroundStyle(Color.secondary)
                }
              }
            }

            Spacer()
          }
          .padding()
          .navigationTitle("Details")
          #if os(iOS)
            .navigationBarTitleDisplayMode(.inline)
          #endif
          .toolbar {
            #if os(iOS)
              ToolbarItem(placement: .navigationBarTrailing) {
                Button("Done") {
                  isDetailsSheetPresented = false
                }
              }
            #else
              ToolbarItem(placement: .cancellationAction) {
                Button("Done") {
                  isDetailsSheetPresented = false
                }
              }
            #endif
          }
        }
      }
    }
  }

  @ViewBuilder
  private func gridView(columnCount: Int) -> some View {
    if settings.alternativeThumbnailGrid {
      let columnsData = getColumnsData(columnCount: columnCount)

      HStack(alignment: .top) {
        ForEach(0..<columnCount, id: \.self) { columnIndex in
          LazyVStack {
            ForEach(columnsData[columnIndex], id: \.id) { favorite in
              favoriteGridContent(favorite: favorite)
                .id(favorite.id)
            }
          }
          .transaction { $0.animation = nil }
        }
      }
      #if os(macOS)
        .padding(8)
      #else
        .padding(.horizontal)
      #endif
      .transition(.opacity)
    } else {
      WaterfallGrid(sortedFilteredFavorites, id: \.id) { favorite in
        favoriteGridContent(favorite: favorite)
          .id(favorite.id)
      }
      .gridStyle(columns: columnCount)
      .transaction { $0.animation = nil }
      #if os(macOS)
        .padding(8)
      #else
        .padding(.horizontal)
      #endif
      .transition(.opacity)
    }
  }

  private func getColumnsData(columnCount: Int) -> [[SettingsFavoritePost]] {
    (0..<columnCount).map { columnIndex in
      sortedFilteredFavorites.enumerated().compactMap { index, favorite in
        index % columnCount == columnIndex ? favorite : nil
      }
    }
  }

  private func favoriteGridContent(favorite: SettingsFavoritePost) -> some View {
    NavigationLink(
      value: PostWithContext(
        post: favorite.toBooruPost(),
        posts: sortedFilteredFavorites.map { $0.toBooruPost() },
        baseSearchText: nil
      )
    ) {
      FavoritePostThumbnailView(
        favorite: favorite,
      ) {
        settings.removeFavorite(withID: favorite.id)
      }
    }
    .buttonStyle(PlainButtonStyle())
    .contextMenu {
      Button(action: {
        selectedFavoriteForDetails = favorite
        isDetailsSheetPresented = true
      }) {
        Label("Show Details", systemImage: "info.circle")
      }

      Button(action: {
        manager.searchText += " \(favorite.tags.joined(separator: " "))"
        manager.selectedPost = nil

        let localManager = manager
        let localSettings = settings

        isPresented.toggle()

        Task(priority: .userInitiated) {
          await localManager.performSearch(settings: localSettings)
        }
      }) {
        Label("Add to Search", systemImage: "plus")
      }

      Menu {
        ForEach(settings.folders.filter { $0.topLevelName == nil }, id: \.id) { folder in
          Button(action: {
            settings.updateFavoriteFolder(withID: favorite.id, folder: folder.id)
          }) {
            Label(folder.name, systemImage: "folder")
          }
          .disabled(favorite.folder == folder.id)
        }

        let topLevelFolders = settings.folders
          .reduce(into: [String: [SettingsFolder]]()) { result, folder in
            guard let topLevelName = folder.topLevelName else { return }

            result[topLevelName, default: []].append(folder)
          }

        ForEach(topLevelFolders.keys.sorted(), id: \.self) { topLevelName in
          Menu {
            ForEach(topLevelFolders[topLevelName] ?? [], id: \.id) { folder in
              Button(action: {
                settings.updateFavoriteFolder(withID: favorite.id, folder: folder.id)
              }) {
                Text(folder.shortName)
              }
              .disabled(favorite.folder == folder.id)
            }
          } label: {
            Text(topLevelName)
          }
        }

        Button(action: {
          itemPendingCollectionAssignment = favorite.id
          isNewCollectionAlertPresented = true
        }) {
          Label("New Collection", systemImage: "plus")
        }
      } label: {
        Label("\(favorite.folder != nil ? "Move" : "Add") to Collection", systemImage: "folder")
      }

      if favorite.folder != nil {
        Button(action: {
          settings.updateFavoriteFolder(withID: favorite.id, folder: nil)
        }) {
          Label("Remove from Collection", systemImage: "folder.badge.minus")
        }
      }

      Button {
        settings.removeFavorite(withID: favorite.id)
      } label: {
        Label("Delete", systemImage: "trash")
      }
    }
  }
}

extension SettingsFavoritePost {
  func toBooruPost() -> BooruPost {
    BooruPost(
      id: postId,
      height: height,
      score: "0",
      fileURL: fileUrl.flatMap(URL.init) ?? URL(string: "https://example.com")!,
      parentID: "0",
      sampleURL: previewUrl.flatMap(URL.init) ?? URL(string: "https://example.com")!,
      sampleWidth: width,
      sampleHeight: height,
      previewURL: thumbnailUrl.flatMap(URL.init) ?? URL(string: "https://example.com")!,
      rating: rating,
      tags: tags,
      width: width,
      change: nil,
      md5: "",
      creatorID: "0",
      authorID: nil,
      createdAt: createdAt,
      status: "active",
      source: "",
      previewWidth: width,
      previewHeight: height
    )
  }
}

#Preview {
  FavoritesView(selectedTab: .constant(1), isPresented: .constant(false))
    .environmentObject(SettingsManager())
    .environmentObject(BooruManager(.yandere))
}