//
// Swiftfin is subject to the terms of the Mozilla Public
// License, v2.0. If a copy of the MPL was not distributed with this
// file, you can obtain one at https://mozilla.org/MPL/2.0/.
//
// Copyright (c) 2023 Jellyfin & Jellyfin Contributors
//

import Combine
import Foundation
import Get
import JellyfinAPI

final class ItemTypeLibraryViewModel: PagingLibraryViewModel {

    let itemTypes: [BaseItemKind]
    let filterViewModel: FilterViewModel

    init(itemTypes: [BaseItemKind], filters: ItemFilters) {
        self.itemTypes = itemTypes
        self.filterViewModel = .init(parent: nil, currentFilters: filters)
        super.init()

        filterViewModel.$currentFilters
            .sink { newFilters in
                self.requestItems(with: newFilters, replaceCurrentItems: true)
            }
            .store(in: &cancellables)
    }

    private func requestItems(with filters: ItemFilters, replaceCurrentItems: Bool = false) {

        if replaceCurrentItems {
            items = []
            currentPage = 0
            hasNextPage = true
        }

        Task {
            var parameters = self._getDefaultParams()
            let request = Paths.getItems(parameters: parameters)
            let response = try await userSession.client.send(request)

            guard let items = response.value.items, !items.isEmpty else {
                hasNextPage = false
                return
            }

            await MainActor.run {
                self.items.append(contentsOf: items)
            }
        }
    }

    override func _getDefaultParams() -> Paths.GetItemsParameters? {
        let filters = filterViewModel.currentFilters
        let genreIDs = filters.genres.compactMap(\.id)
        let sortBy: [String] = filters.sortBy.map(\.filterName).appending("IsFolder")
        let sortOrder = filters.sortOrder.map { SortOrder(rawValue: $0.filterName) ?? .ascending }
        let itemFilters: [ItemFilter] = filters.filters.compactMap { .init(rawValue: $0.filterName) }

        let parameters = Paths.GetItemsParameters(
            userID: userSession.user.id,
            startIndex: currentPage * pageItemSize,
            limit: pageItemSize,
            isRecursive: true,
            sortOrder: sortOrder,
            fields: ItemFields.allCases,
            includeItemTypes: itemTypes,
            filters: itemFilters,
            sortBy: sortBy,
            enableUserData: true,
            genreIDs: genreIDs
        )

        return parameters
    }

    override func _requestNextPage() {
        requestItems(with: filterViewModel.currentFilters)
    }
}