jellyflood/Shared/ViewModels/ItemViewModel/CollectionItemViewModel.swift

62 lines
1.7 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) 2025 Jellyfin & Jellyfin Contributors
//
import Combine
import Foundation
import JellyfinAPI
import OrderedCollections
final class CollectionItemViewModel: ItemViewModel {
// MARK: - Published Collection Items
@Published
private(set) var collectionItems: OrderedDictionary<BaseItemKind, [BaseItemDto]> = [:]
override var presentPlayButton: Bool {
false
}
// MARK: - On Refresh
override func onRefresh() async throws {
let collectionItems = try await self.getCollectionItems()
await MainActor.run {
self.collectionItems = collectionItems
}
}
// MARK: - Get Collection Items
private func getCollectionItems() async throws -> OrderedDictionary<BaseItemKind, [BaseItemDto]> {
var parameters = Paths.GetItemsByUserIDParameters()
parameters.fields = .MinimumFields
parameters.includeItemTypes = BaseItemKind.supportedCases
.appending(.episode)
parameters.parentID = item.id
let request = Paths.getItemsByUserID(
userID: userSession.user.id,
parameters: parameters
)
let response = try await userSession.client.send(request)
let items = response.value.items ?? []
let result = OrderedDictionary<BaseItemKind?, [BaseItemDto]>(
grouping: items,
by: \.type
)
.compactKeys()
.sortedKeys { $0.rawValue < $1.rawValue }
return result
}
}