mirror of
https://github.com/yattee/yattee.git
synced 2024-12-13 13:50:32 +05:30
183 lines
5.8 KiB
Swift
183 lines
5.8 KiB
Swift
import Cache
|
|
import Foundation
|
|
import Logging
|
|
import Siesta
|
|
import SwiftUI
|
|
import SwiftyJSON
|
|
|
|
final class SubscribedChannelsModel: ObservableObject, CacheModel {
|
|
static var shared = SubscribedChannelsModel()
|
|
let logger = Logger(label: "stream.yattee.cache.channels")
|
|
|
|
static let diskConfig = DiskConfig(name: "channels")
|
|
static let memoryConfig = MemoryConfig()
|
|
|
|
let storage = try? Storage<String, JSON>(
|
|
diskConfig: SubscribedChannelsModel.diskConfig,
|
|
memoryConfig: SubscribedChannelsModel.memoryConfig,
|
|
transformer: BaseCacheModel.jsonTransformer
|
|
)
|
|
|
|
@Published var isLoading = false
|
|
@Published var channels = [Channel]()
|
|
@Published var error: RequestError?
|
|
|
|
var accounts: AccountsModel { .shared }
|
|
var unwatchedFeedCount: UnwatchedFeedCountModel { .shared }
|
|
|
|
var resource: Resource? {
|
|
accounts.api.subscriptions
|
|
}
|
|
|
|
var all: [Channel] {
|
|
channels.sorted { $0.name.lowercased() < $1.name.lowercased() }
|
|
}
|
|
|
|
var allByUnwatchedCount: [Channel] {
|
|
if let account = accounts.current {
|
|
return all.sorted { c1, c2 in
|
|
let c1HasUnwatched = (unwatchedFeedCount.unwatchedByChannel[account]?[c1.id] ?? -1) > 0
|
|
let c2HasUnwatched = (unwatchedFeedCount.unwatchedByChannel[account]?[c2.id] ?? -1) > 0
|
|
let nameIncreasing = c1.name.lowercased() < c2.name.lowercased()
|
|
|
|
return c1HasUnwatched ? (c2HasUnwatched ? nameIncreasing : true) : (c2HasUnwatched ? false : nameIncreasing)
|
|
}
|
|
}
|
|
return all
|
|
}
|
|
|
|
func subscribe(_ channelID: String, onSuccess: @escaping () -> Void = {}) {
|
|
accounts.api.subscribe(channelID) {
|
|
self.scheduleLoad(onSuccess: onSuccess)
|
|
}
|
|
}
|
|
|
|
func unsubscribe(_ channelID: String, onSuccess: @escaping () -> Void = {}) {
|
|
accounts.api.unsubscribe(channelID) {
|
|
self.scheduleLoad(onSuccess: onSuccess)
|
|
}
|
|
}
|
|
|
|
func isSubscribing(_ channelID: String) -> Bool {
|
|
channels.contains { $0.id == channelID }
|
|
}
|
|
|
|
func load(force: Bool = false, onSuccess: @escaping () -> Void = {}) {
|
|
guard accounts.app.supportsSubscriptions, !isLoading, accounts.signedIn, let account = accounts.current else {
|
|
channels = []
|
|
return
|
|
}
|
|
|
|
DispatchQueue.main.async { [weak self] in
|
|
guard let self else { return }
|
|
let request = force ? self.resource?.load() : self.resource?.loadIfNeeded()
|
|
guard request != nil else { return }
|
|
|
|
self.loadCachedChannels(account)
|
|
|
|
self.isLoading = true
|
|
|
|
request?
|
|
.onCompletion { [weak self] _ in
|
|
self?.isLoading = false
|
|
}
|
|
.onSuccess { resource in
|
|
self.error = nil
|
|
if let channels: [Channel] = resource.typedContent() {
|
|
self.channels = channels
|
|
channels.forEach { ChannelsCacheModel.shared.storeIfMissing($0) }
|
|
self.storeChannels(account: account, channels: channels)
|
|
FeedModel.shared.calculateUnwatchedFeed()
|
|
onSuccess()
|
|
}
|
|
}
|
|
.onFailure { self.error = $0 }
|
|
}
|
|
}
|
|
|
|
func loadCachedChannels(_ account: Account) {
|
|
let cache = getChannels(account: account)
|
|
if !cache.isEmpty {
|
|
DispatchQueue.main.async {
|
|
self.channels = cache
|
|
}
|
|
}
|
|
}
|
|
|
|
func storeChannels(account: Account, channels: [Channel]) {
|
|
let date = iso8601DateFormatter.string(from: Date())
|
|
logger.info("caching channels \(channelsDateCacheKey(account)) -- \(date)")
|
|
|
|
channels.forEach { ChannelsCacheModel.shared.storeIfMissing($0) }
|
|
|
|
let dateObject: JSON = ["date": date]
|
|
let channelsObject: JSON = ["channels": channels.map(\.json).map(\.object)]
|
|
|
|
try? storage?.setObject(dateObject, forKey: channelsDateCacheKey(account))
|
|
try? storage?.setObject(channelsObject, forKey: channelsCacheKey(account))
|
|
}
|
|
|
|
func getChannels(account: Account) -> [Channel] {
|
|
logger.info("getting channels \(channelsDateCacheKey(account))")
|
|
|
|
if let json = try? storage?.object(forKey: channelsCacheKey(account)),
|
|
let channels = json.dictionaryValue["channels"]
|
|
{
|
|
return channels.arrayValue.compactMap { json in
|
|
let channel = Channel.from(json)
|
|
if !channel.hasExtendedDetails,
|
|
let cache = ChannelsCacheModel.shared.retrieve(channel.cacheKey)
|
|
{
|
|
return cache.channel
|
|
}
|
|
|
|
return channel
|
|
}
|
|
}
|
|
|
|
return []
|
|
}
|
|
|
|
private func scheduleLoad(onSuccess: @escaping () -> Void) {
|
|
DispatchQueue.main.asyncAfter(deadline: .now() + 2) {
|
|
self.load(force: true, onSuccess: onSuccess)
|
|
}
|
|
}
|
|
|
|
private func channelsCacheKey(_ account: Account) -> String {
|
|
"channels-\(account.id)"
|
|
}
|
|
|
|
private func channelsDateCacheKey(_ account: Account) -> String {
|
|
"channels-\(account.id)-date"
|
|
}
|
|
|
|
func getChannelsTime(account: Account) -> Date? {
|
|
if let json = try? storage?.object(forKey: channelsDateCacheKey(account)),
|
|
let string = json.dictionaryValue["date"]?.string,
|
|
let date = iso8601DateFormatter.date(from: string)
|
|
{
|
|
return date
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
var channelsTime: Date? {
|
|
if let account = accounts.current {
|
|
return getChannelsTime(account: account)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
var formattedCacheTime: String {
|
|
getFormattedDate(channelsTime)
|
|
}
|
|
|
|
func onAccountChange() {
|
|
channels = []
|
|
load(force: true)
|
|
}
|
|
}
|