jellyflood/Shared/ViewModels/LiveTVProgramsViewModel.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 LiveTVProgramsViewModel: ViewModel {

	@Published
	var recommendedItems = [BaseItemDto]()
	@Published
	var seriesItems = [BaseItemDto]()
	@Published
	var movieItems = [BaseItemDto]()
	@Published
	var sportsItems = [BaseItemDto]()
	@Published
	var kidsItems = [BaseItemDto]()
	@Published
	var newsItems = [BaseItemDto]()

	private var channels = [String: BaseItemDto]()

	override init() {
		super.init()

		getChannels()
	}

	func findChannel(id: String) -> BaseItemDto? {
		channels[id]
	}

	private func getChannels() {
		LiveTvAPI.getLiveTvChannels(userId: SessionManager.main.currentLogin.user.id,
		                            startIndex: 0,
		                            limit: 1000,
		                            enableImageTypes: [.primary],
		                            enableUserData: false,
		                            enableFavoriteSorting: true)
			.trackActivity(loading)
			.sink(receiveCompletion: { [weak self] completion in
				self?.handleAPIRequestError(completion: completion)
			}, receiveValue: { [weak self] response in
				LogManager.log.debug("Received \(response.items?.count ?? 0) Channels")
				guard let self = self else { return }
				if let chans = response.items {
					for chan in chans {
						if let chanId = chan.id {
							self.channels[chanId] = chan
						}
					}
					self.getRecommendedPrograms()
					self.getSeries()
					self.getMovies()
					self.getSports()
					self.getKids()
					self.getNews()
				}
			})
			.store(in: &cancellables)
	}

	private func getRecommendedPrograms() {
		LiveTvAPI.getRecommendedPrograms(userId: SessionManager.main.currentLogin.user.id,
		                                 limit: 9,
		                                 isAiring: true,
		                                 imageTypeLimit: 1,
		                                 enableImageTypes: [.primary, .thumb],
		                                 fields: [.channelInfo, .primaryImageAspectRatio],
		                                 enableTotalRecordCount: false)
			.trackActivity(loading)
			.sink(receiveCompletion: { [weak self] completion in
				self?.handleAPIRequestError(completion: completion)
			}, receiveValue: { [weak self] response in
				LogManager.log.debug("Received \(String(response.items?.count ?? 0)) Recommended Programs")
				guard let self = self else { return }
				self.recommendedItems = response.items ?? []
			})
			.store(in: &cancellables)
	}

	private func getSeries() {
		let getProgramsRequest = GetProgramsRequest(userId: SessionManager.main.currentLogin.user.id,
		                                            hasAired: false,
		                                            isMovie: false,
		                                            isSeries: true,
		                                            isNews: false,
		                                            isKids: false,
		                                            isSports: false,
		                                            limit: 9,
		                                            enableTotalRecordCount: false,
		                                            enableImageTypes: [.primary, .thumb],
		                                            fields: [.channelInfo, .primaryImageAspectRatio])

		LiveTvAPI.getPrograms(getProgramsRequest: getProgramsRequest)
			.trackActivity(loading)
			.sink(receiveCompletion: { [weak self] completion in
				self?.handleAPIRequestError(completion: completion)
			}, receiveValue: { [weak self] response in
				LogManager.log.debug("Received \(String(response.items?.count ?? 0)) Series Items")
				guard let self = self else { return }
				self.seriesItems = response.items ?? []
			})
			.store(in: &cancellables)
	}

	private func getMovies() {
		let getProgramsRequest = GetProgramsRequest(userId: SessionManager.main.currentLogin.user.id,
		                                            hasAired: false,
		                                            isMovie: true,
		                                            isSeries: false,
		                                            isNews: false,
		                                            isKids: false,
		                                            isSports: false,
		                                            limit: 9,
		                                            enableTotalRecordCount: false,
		                                            enableImageTypes: [.primary, .thumb],
		                                            fields: [.channelInfo, .primaryImageAspectRatio])

		LiveTvAPI.getPrograms(getProgramsRequest: getProgramsRequest)
			.trackActivity(loading)
			.sink(receiveCompletion: { [weak self] completion in
				self?.handleAPIRequestError(completion: completion)
			}, receiveValue: { [weak self] response in
				LogManager.log.debug("Received \(String(response.items?.count ?? 0)) Movie Items")
				guard let self = self else { return }
				self.movieItems = response.items ?? []
			})
			.store(in: &cancellables)
	}

	private func getSports() {
		let getProgramsRequest = GetProgramsRequest(userId: SessionManager.main.currentLogin.user.id,
		                                            hasAired: false,
		                                            isSports: true,
		                                            limit: 9,
		                                            enableTotalRecordCount: false,
		                                            enableImageTypes: [.primary, .thumb],
		                                            fields: [.channelInfo, .primaryImageAspectRatio])

		LiveTvAPI.getPrograms(getProgramsRequest: getProgramsRequest)
			.trackActivity(loading)
			.sink(receiveCompletion: { [weak self] completion in
				self?.handleAPIRequestError(completion: completion)
			}, receiveValue: { [weak self] response in
				LogManager.log.debug("Received \(String(response.items?.count ?? 0)) Sports Items")
				guard let self = self else { return }
				self.sportsItems = response.items ?? []
			})
			.store(in: &cancellables)
	}

	private func getKids() {
		let getProgramsRequest = GetProgramsRequest(userId: SessionManager.main.currentLogin.user.id,
		                                            hasAired: false,
		                                            isKids: true,
		                                            limit: 9,
		                                            enableTotalRecordCount: false,
		                                            enableImageTypes: [.primary, .thumb],
		                                            fields: [.channelInfo, .primaryImageAspectRatio])

		LiveTvAPI.getPrograms(getProgramsRequest: getProgramsRequest)
			.trackActivity(loading)
			.sink(receiveCompletion: { [weak self] completion in
				self?.handleAPIRequestError(completion: completion)
			}, receiveValue: { [weak self] response in
				LogManager.log.debug("Received \(String(response.items?.count ?? 0)) Kids Items")
				guard let self = self else { return }
				self.kidsItems = response.items ?? []
			})
			.store(in: &cancellables)
	}

	private func getNews() {
		let getProgramsRequest = GetProgramsRequest(userId: SessionManager.main.currentLogin.user.id,
		                                            hasAired: false,
		                                            isNews: true,
		                                            limit: 9,
		                                            enableTotalRecordCount: false,
		                                            enableImageTypes: [.primary, .thumb],
		                                            fields: [.channelInfo, .primaryImageAspectRatio])

		LiveTvAPI.getPrograms(getProgramsRequest: getProgramsRequest)
			.trackActivity(loading)
			.sink(receiveCompletion: { [weak self] completion in
				self?.handleAPIRequestError(completion: completion)
			}, receiveValue: { [weak self] response in
				LogManager.log.debug("Received \(String(response.items?.count ?? 0)) News Items")
				guard let self = self else { return }
				self.newsItems = response.items ?? []
			})
			.store(in: &cancellables)
	}

	func fetchVideoPlayerViewModel(item: BaseItemDto, completion: @escaping (VideoPlayerViewModel) -> Void) {
		item.createLiveTVVideoPlayerViewModel()
			.sink { completion in
				self.handleAPIRequestError(completion: completion)
			} receiveValue: { videoPlayerViewModels in
				if let viewModel = videoPlayerViewModels.first {
					completion(viewModel)
				}
			}
			.store(in: &self.cancellables)
	}
}