jellyflood/Shared/ViewModels/LibraryListViewModel.swift

65 lines
1.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 (c) 2022 Jellyfin & Jellyfin Contributors
//
import Foundation
import JellyfinAPI
final class LibraryListViewModel: ViewModel {
@Published
var libraries: [BaseItemDto] = []
@Published
var libraryRandomItems: [BaseItemDto: BaseItemDto] = [:]
// temp
var withFavorites = LibraryFilters(filters: [.isFavorite], sortOrder: [], withGenres: [], sortBy: [])
override init() {
super.init()
requestLibraries()
}
func requestLibraries() {
UserViewsAPI.getUserViews(userId: SessionManager.main.currentLogin.user.id)
.trackActivity(loading)
.sink(receiveCompletion: { completion in
self.handleAPIRequestError(completion: completion)
}, receiveValue: { response in
if let libraries = response.items {
self.libraries = libraries
for library in libraries {
self.getRandomLibraryItem(for: library)
}
}
})
.store(in: &cancellables)
}
// MARK: Library random item
func getRandomLibraryItem(for library: BaseItemDto) {
guard library.itemType == .collectionFolder else { return }
ItemsAPI.getItems(userId: SessionManager.main.currentLogin.user.id,
limit: 1,
parentId: library.id)
.sink { completion in
self.handleAPIRequestError(completion: completion)
} receiveValue: { result in
if let item = result.items?.first {
self.libraryRandomItems[library] = item
} else {
self.libraryRandomItems[library] = library
}
}
.store(in: &cancellables)
}
}