81 lines
2.9 KiB
Swift
81 lines
2.9 KiB
Swift
//
|
|
/*
|
|
* 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: ItemViewModel {
|
|
|
|
@Published var seasons: [BaseItemDto] = []
|
|
|
|
override init(item: BaseItemDto) {
|
|
super.init(item: item)
|
|
|
|
requestSeasons()
|
|
getNextUp()
|
|
}
|
|
|
|
override func playButtonText() -> String {
|
|
guard let playButtonItem = playButtonItem else { return L10n.play }
|
|
guard let episodeLocator = playButtonItem.getEpisodeLocator() else { return L10n.play }
|
|
return episodeLocator
|
|
}
|
|
|
|
override func shouldDisplayRuntime() -> Bool {
|
|
return false
|
|
}
|
|
|
|
private func getNextUp() {
|
|
|
|
LogManager.shared.log.debug("Getting next up for show \(self.item.id!) (\(self.item.name!))")
|
|
TvShowsAPI.getNextUp(userId: SessionManager.main.currentLogin.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
|
|
if let nextUpItem = response.items?.first {
|
|
self?.playButtonItem = nextUpItem
|
|
}
|
|
})
|
|
.store(in: &cancellables)
|
|
}
|
|
|
|
private 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")"
|
|
}
|
|
|
|
private func requestSeasons() {
|
|
LogManager.shared.log.debug("Getting seasons of show \(self.item.id!) (\(self.item.name!))")
|
|
TvShowsAPI.getSeasons(seriesId: item.id ?? "", userId: SessionManager.main.currentLogin.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)
|
|
}
|
|
}
|