//
/*
 * 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 2021 Aiden Vigue & Jellyfin Contributors
 */

import Combine
import Foundation
import JellyfinAPI

final class SeriesItemViewModel: DetailItemViewModel {

    @Published var seasons = [BaseItemDto]()
    @Published var nextUpItem: BaseItemDto?

    override init(item: BaseItemDto) {
        super.init(item: item)
        self.item = item

        requestSeasons()
        getNextUp()
    }

    func getNextUp() {
        LogManager.shared.log.debug("Getting next up for show \(self.item.id!) (\(self.item.name!))")
        TvShowsAPI.getNextUp(userId: SessionManager.current.user.user_id!, fields: [.primaryImageAspectRatio, .seriesPrimaryImage, .seasonUserData, .overview, .genres, .people], seriesId: self.item.id!, enableUserData: true)
            .trackActivity(loading)
            .sink(receiveCompletion: { [weak self] completion in
                self?.handleAPIRequestError(completion: completion)
            }, receiveValue: { [weak self] response in
                self?.nextUpItem = response.items?.first ?? nil
            })
            .store(in: &cancellables)
    }

    func getRunYears() -> String {
        let dateFormatter = DateFormatter()
        dateFormatter.dateFormat = "yyyy"

        var startYear: String?
        var endYear: String?

        if item.premiereDate != nil {
            startYear = dateFormatter.string(from: item.premiereDate!)
        }

        if item.endDate != nil {
            endYear = dateFormatter.string(from: item.endDate!)
        }

        return "\(startYear ?? "Unknown") - \(endYear ?? "Present")"
    }

    func requestSeasons() {
        LogManager.shared.log.debug("Getting seasons of show \(self.item.id!) (\(self.item.name!))")
        TvShowsAPI.getSeasons(seriesId: item.id ?? "", userId: SessionManager.current.user.user_id!, fields: [.primaryImageAspectRatio, .seriesPrimaryImage, .seasonUserData, .overview, .genres, .people], enableUserData: true)
            .trackActivity(loading)
            .sink(receiveCompletion: { [weak self] completion in
                self?.handleAPIRequestError(completion: completion)
            }, receiveValue: { [weak self] response in
                self?.seasons = response.items ?? []
                LogManager.shared.log.debug("Retrieved \(String(self?.seasons.count ?? 0)) seasons")
            })
            .store(in: &cancellables)
    }
}