101 lines
3.4 KiB
Swift
101 lines
3.4 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 LibraryViewModel: ViewModel {
|
|
var parentID: String?
|
|
var person: BaseItemPerson?
|
|
var genre: NameGuidPair?
|
|
var studio: NameGuidPair?
|
|
|
|
@Published
|
|
var items = [BaseItemDto]()
|
|
|
|
@Published
|
|
var totalPages = 0
|
|
@Published
|
|
var currentPage = 0
|
|
@Published
|
|
var hasNextPage = false
|
|
@Published
|
|
var hasPreviousPage = false
|
|
|
|
// temp
|
|
@Published
|
|
var filters: LibraryFilters
|
|
|
|
var enabledFilterType: [FilterType] {
|
|
if genre == nil {
|
|
return [.tag, .genre, .sortBy, .sortOrder, .filter]
|
|
} else {
|
|
return [.tag, .sortBy, .sortOrder, .filter]
|
|
}
|
|
}
|
|
|
|
init(parentID: String? = nil,
|
|
person: BaseItemPerson? = nil,
|
|
genre: NameGuidPair? = nil,
|
|
studio: NameGuidPair? = nil,
|
|
filters: LibraryFilters = LibraryFilters(filters: [], sortOrder: [.ascending], withGenres: [], sortBy: [.name])) {
|
|
self.parentID = parentID
|
|
self.person = person
|
|
self.genre = genre
|
|
self.studio = studio
|
|
self.filters = filters
|
|
super.init()
|
|
|
|
$filters
|
|
.sink(receiveValue: requestItems(with:))
|
|
.store(in: &cancellables)
|
|
}
|
|
|
|
func requestItems(with filters: LibraryFilters) {
|
|
let personIDs: [String] = [person].compactMap(\.?.id)
|
|
let studioIDs: [String] = [studio].compactMap(\.?.id)
|
|
let genreIDs: [String]
|
|
if filters.withGenres.isEmpty {
|
|
genreIDs = [genre].compactMap(\.?.id)
|
|
} else {
|
|
genreIDs = filters.withGenres.compactMap(\.id)
|
|
}
|
|
let sortBy = filters.sortBy.map(\.rawValue)
|
|
|
|
ItemsAPI.getItemsByUserId(userId: SessionManager.current.user.user_id!, startIndex: currentPage * 100, limit: 100, recursive: true,
|
|
searchTerm: nil, sortOrder: filters.sortOrder, parentId: parentID,
|
|
fields: [.primaryImageAspectRatio, .seriesPrimaryImage, .seasonUserData, .overview, .genres, .people],
|
|
includeItemTypes: ["Movie", "Series"], filters: filters.filters, sortBy: sortBy, tags: filters.tags,
|
|
enableUserData: true, personIds: personIDs, studioIds: studioIDs, genreIds: genreIDs, enableImages: true)
|
|
.trackActivity(loading)
|
|
.sink(receiveCompletion: { [weak self] completion in
|
|
self?.handleAPIRequestCompletion(completion: completion)
|
|
}, receiveValue: { [weak self] response in
|
|
guard let self = self else { return }
|
|
let totalPages = ceil(Double(response.totalRecordCount ?? 0) / 100.0)
|
|
self.totalPages = Int(totalPages)
|
|
self.hasPreviousPage = self.currentPage > 0
|
|
self.hasNextPage = self.currentPage < self.totalPages - 1
|
|
self.items = response.items ?? []
|
|
})
|
|
.store(in: &cancellables)
|
|
}
|
|
|
|
func requestNextPage() {
|
|
currentPage += 1
|
|
requestItems(with: filters)
|
|
}
|
|
|
|
func requestPreviousPage() {
|
|
currentPage -= 1
|
|
requestItems(with: filters)
|
|
}
|
|
}
|