jellyflood/JellyfinPlayer/LibraryView.swift

/*
 * JellyfinPlayer/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 SwiftUI
import NukeUI
import JellyfinAPI
import Combine

struct LibraryView: View {

    @StateObject
    var tempViewModel = ViewModel()
    @State private var items: [BaseItemDto] = []
    @State private var isLoading: Bool = false

    private var usingParentID: String = ""
    private var title: String = ""
    private var filters: LibraryFilters = LibraryFilters(filters: [], sortOrder: [.ascending], withGenres: [], sortBy: ["SortName"])
    private var personId: String = ""
    private var genre: String = ""
    private var studio: String = ""

    @State private var totalPages: Int = 0
    @State private var currentPage: Int = 0
    @State private var isSearching: String? = ""
    @State private var viewDidLoad: Bool = false

    init(usingParentID: String, title: String) {
        self.usingParentID = usingParentID
        self.title = title
    }

    init(usingParentID: String, title: String, usingFilters: LibraryFilters) {
        self.usingParentID = usingParentID
        self.title = title
        self.filters = usingFilters
    }

    init(withPerson: BaseItemPerson) {
        self.usingParentID = ""
        self.title = withPerson.name ?? ""
        self.personId = withPerson.id!
    }

    init(withGenre: NameGuidPair) {
        self.usingParentID = ""
        self.title = withGenre.name ?? ""
        self.genre = withGenre.id!
    }

    init(withStudio: NameGuidPair) {
        self.usingParentID = ""
        self.title = withStudio.name ?? ""
        self.studio = withStudio.id!
    }

    func onAppear() {
        recalcTracks()

        if viewDidLoad {
            return
        }

        isLoading = true
        items = []

        DispatchQueue.global(qos: .userInitiated).async {
            ItemsAPI.getItemsByUserId(userId: SessionManager.current.userID!, startIndex: currentPage * 100, limit: 100, recursive: true, searchTerm: nil, sortOrder: filters.sortOrder, parentId: (usingParentID != "" ? usingParentID : nil), fields: [.primaryImageAspectRatio, .seriesPrimaryImage, .seasonUserData, .overview, .genres, .people], includeItemTypes: ["Movie", "Series"], filters: filters.filters, sortBy: filters.sortBy, enableUserData: true, personIds: (personId == "" ? nil : [personId]), studioIds: (studio == "" ? nil : [studio]), genreIds: (genre == "" ? nil : [genre]), enableImages: true)
                .sink(receiveCompletion: { completion in
                    print(completion)
                    isLoading = false
                }, receiveValue: { response in
                    let x = ceil(Double(response.totalRecordCount!) / 100.0)
                    totalPages = Int(x)
                    items = response.items ?? []
                    isLoading = false
                    viewDidLoad = true
                })
                .store(in: &tempViewModel.cancellables)
        }
    }

    // MARK: tracks for grid
    @State private var tracks: [GridItem] = []
    func recalcTracks() {
        let trkCnt = Int(floor(UIScreen.main.bounds.size.width / 125))
        _tracks.wrappedValue = []
        for _ in 0 ..< trkCnt {
            _tracks.wrappedValue.append(GridItem(.flexible()))
        }
    }

    var body: some View {
        ZStack {
            if isLoading == true {
                ProgressView()
            } else {
                if !items.isEmpty {
                    VStack {
                        ScrollView(.vertical) {
                            Spacer().frame(height: 16)
                            LazyVGrid(columns: tracks) {
                                ForEach(items, id: \.id) { item in
                                    NavigationLink(destination: ItemView(item: item)) {
                                        VStack(alignment: .leading) {
                                            ImageView(src: item.getPrimaryImage(baseURL: ServerEnvironment.current.server.baseURI!, maxWidth: 100), bh: item.getPrimaryImageBlurHash())
                                                .frame(width: 100, height: 150)
                                                .cornerRadius(10)
                                            Text(item.name ?? "")
                                                .font(.caption)
                                                .fontWeight(.semibold)
                                                .foregroundColor(.primary)
                                                .lineLimit(1)
                                            if item.productionYear != nil {
                                                Text(String(item.productionYear!))
                                                    .foregroundColor(.secondary)
                                                    .font(.caption)
                                                    .fontWeight(.medium)
                                            } else {
                                                Text(item.type ?? "")
                                            }
                                        }.frame(width: 100)
                                    }
                                }
                            }.onRotate { _ in
                                recalcTracks()
                            }
                            if totalPages > 1 {
                                HStack {
                                    Spacer()
                                    HStack {
                                        Button {
                                            currentPage = currentPage - 1
                                            onAppear()
                                        } label: {
                                            Image(systemName: "chevron.left")
                                                .font(.system(size: 25))
                                        }.disabled(currentPage == 0)
                                        Text("Page \(String(currentPage+1)) of \(String(totalPages))")
                                            .font(.headline)
                                            .fontWeight(.semibold)
                                        Button {
                                            currentPage = currentPage + 1
                                            onAppear()
                                        } label: {
                                            Image(systemName: "chevron.right")
                                                .font(.system(size: 25))
                                        }.disabled(currentPage > totalPages - 1)
                                    }
                                    Spacer()
                                }
                            }
                            Spacer().frame(height: 16)
                        }
                    }
                } else {
                    Text("No results.")
                }
            }
        }
        .onAppear(perform: onAppear)
        .navigationBarTitle(title, displayMode: .inline)
        .toolbar {
            ToolbarItemGroup(placement: .navigationBarTrailing) {
                if currentPage > 0 {
                    Button {
                        currentPage = currentPage - 1
                        onAppear()
                    } label: {
                        Image(systemName: "chevron.left")
                    }
                }
                if currentPage < totalPages - 1 {
                    Button {
                        currentPage = currentPage + 1
                        onAppear()
                    } label: {
                        Image(systemName: "chevron.right")
                    }
                }
                if usingParentID != "" {
                    NavigationLink(destination: LibrarySearchView(usingParentID: usingParentID)) {
                        Image(systemName: "magnifyingglass")
                    }
                }
            }
        }
    }
}

// stream BM^S by nicki!
//