jellyflood/Shared/ViewModels/ItemViewModel/EpisodeItemViewModel.swift

72 lines
2.0 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 (c) 2022 Jellyfin & Jellyfin Contributors
//
import Combine
import Foundation
import JellyfinAPI
import Stinsen
final class EpisodeItemViewModel: ItemViewModel {
@Published
var seriesItem: BaseItemDto?
override init(item: BaseItemDto) {
super.init(item: item)
getSeriesItem()
}
override func updateItem() {
ItemsAPI.getItems(
userId: SessionManager.main.currentLogin.user.id,
limit: 1,
fields: [
.primaryImageAspectRatio,
.seriesPrimaryImage,
.seasonUserData,
.overview,
.genres,
.people,
.chapters,
],
enableUserData: true,
ids: [item.id ?? ""]
)
.sink { completion in
self.handleAPIRequestError(completion: completion)
} receiveValue: { response in
if let item = response.items?.first {
self.item = item
self.playButtonItem = item
}
}
.store(in: &cancellables)
}
private func getSeriesItem() {
guard let seriesID = item.seriesId else { return }
ItemsAPI.getItems(
userId: SessionManager.main.currentLogin.user.id,
limit: 1,
fields: ItemFields.allCases,
enableUserData: true,
ids: [seriesID]
)
.trackActivity(loading)
.sink(receiveCompletion: { [weak self] completion in
self?.handleAPIRequestError(completion: completion)
}, receiveValue: { [weak self] response in
guard let firstItem = response.items?.first else { return }
self?.seriesItem = firstItem
})
.store(in: &cancellables)
}
}