//
// 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) 2022 Jellyfin & Jellyfin Contributors
//

import Defaults
import Foundation
import JellyfinAPI

final class MediaViewModel: ViewModel {

    @Published
    private var libraries: [LibraryItem] = []
    @Published
    var libraryImages: [String: [ImageSource]] = [:]

    @Default(.Experimental.liveTVAlphaEnabled)
    private var liveTVEnabled

    var libraryItems: [LibraryItem] {
        [.init(library: .init(name: L10n.favorites, collectionType: "favorites"), viewModel: self)]
            .appending(.init(library: .init(name: "LiveTV", collectionType: "liveTV"), viewModel: self), if: liveTVEnabled)
            .appending(libraries)
    }

    private static let supportedCollectionTypes: [String] = ["boxsets", "folders", "movies", "tvshows", "unknown"]

    override init() {
        super.init()

        requestLibraries()
        getRandomItemImageSource(with: [.isFavorite], id: nil, key: "favorites")
    }

    func requestLibraries() {
        UserViewsAPI.getUserViews(userId: SessionManager.main.currentLogin.user.id)
            .sink(receiveCompletion: { completion in
                self.handleAPIRequestError(completion: completion)
            }, receiveValue: { response in
                guard let items = response.items else { return }
                let filteredLibraries = items.filter { Self.supportedCollectionTypes.contains($0.collectionType ?? "unknown") }
                filteredLibraries.forEach {
                    self.getRandomItemImageSource(with: nil, id: $0.id, key: $0.id ?? "")
                }

                self.libraries = filteredLibraries.map { .init(library: $0, viewModel: self) }
            })
            .store(in: &cancellables)
    }

    private func getRandomItemImageSource(with filters: [ItemFilter]?, id: String?, key: String) {
        ItemsAPI.getItemsByUserId(
            userId: SessionManager.main.currentLogin.user.id,
            limit: 3,
            recursive: true,
            parentId: id,
            includeItemTypes: [.movie, .series],
            filters: filters,
            sortBy: ["Random"]
        )
        .sink(receiveCompletion: { [weak self] completion in
            self?.handleAPIRequestError(completion: completion)
        }, receiveValue: { [weak self] response in
            guard let items = response.items else { return }
            let imageSources = items.map { $0.imageSource(.backdrop, maxWidth: 500) }
            self?.libraryImages[key] = imageSources
        })
        .store(in: &cancellables)
    }
}