NetNewsWire/Shared/SmartFeeds/SmartFeed.swift

118 lines
2.6 KiB
Swift
Raw Normal View History

2017-11-19 22:57:42 +01:00
//
// SmartFeed.swift
2018-08-29 07:18:24 +02:00
// NetNewsWire
2017-11-19 22:57:42 +01:00
//
// Created by Brent Simmons on 11/19/17.
// Copyright © 2017 Ranchero Software. All rights reserved.
//
import Foundation
import RSCore
import Articles
2017-11-19 22:57:42 +01:00
import Account
final class SmartFeed: PseudoFeed {
2019-11-15 13:19:14 +01:00
var feedID: FeedIdentifier? {
delegate.feedID
}
var nameForDisplay: String {
return delegate.nameForDisplay
}
2017-11-19 22:57:42 +01:00
var unreadCount = 0 {
didSet {
if unreadCount != oldValue {
postUnreadCountDidChangeNotification()
}
}
}
var smallIcon: IconImage? {
return delegate.smallIcon
}
#if os(macOS)
var pasteboardWriter: NSPasteboardWriting {
return SmartFeedPasteboardWriter(smartFeed: self)
}
#endif
private let delegate: SmartFeedDelegate
private var unreadCounts = [String: Int]()
2017-11-19 22:57:42 +01:00
init(delegate: SmartFeedDelegate) {
self.delegate = delegate
2017-11-19 22:57:42 +01:00
NotificationCenter.default.addObserver(self, selector: #selector(unreadCountDidChange(_:)), name: .UnreadCountDidChange, object: nil)
queueFetchUnreadCounts() // Fetch unread count at startup
2017-11-19 22:57:42 +01:00
}
@objc func unreadCountDidChange(_ note: Notification) {
if note.object is AppDelegate {
queueFetchUnreadCounts()
2017-11-19 22:57:42 +01:00
}
}
@objc func fetchUnreadCounts() {
let activeAccounts = AccountManager.shared.activeAccounts
// Remove any accounts that are no longer active or have been deleted
let activeAccountIDs = activeAccounts.map { $0.accountID }
unreadCounts.keys.forEach { accountID in
if !activeAccountIDs.contains(accountID) {
unreadCounts.removeValue(forKey: accountID)
}
}
if activeAccounts.isEmpty {
updateUnreadCount()
} else {
activeAccounts.forEach { self.fetchUnreadCount(for: $0) }
}
}
}
extension SmartFeed: ArticleFetcher {
func fetchArticles() -> Set<Article> {
return delegate.fetchArticles()
}
func fetchArticlesAsync(_ callback: @escaping ArticleSetBlock) {
delegate.fetchArticlesAsync(callback)
}
func fetchUnreadArticles() -> Set<Article> {
return delegate.fetchUnreadArticles()
}
func fetchUnreadArticlesAsync(_ callback: @escaping ArticleSetBlock) {
delegate.fetchUnreadArticlesAsync(callback)
}
2017-11-19 22:57:42 +01:00
}
private extension SmartFeed {
2017-11-19 22:57:42 +01:00
func queueFetchUnreadCounts() {
CoalescingQueue.standard.add(self, #selector(fetchUnreadCounts))
}
2017-11-19 22:57:42 +01:00
func fetchUnreadCount(for account: Account) {
delegate.fetchUnreadCount(for: account) { (accountUnreadCount) in
self.unreadCounts[account.accountID] = accountUnreadCount
2017-11-19 22:57:42 +01:00
self.updateUnreadCount()
}
}
func updateUnreadCount() {
unreadCount = AccountManager.shared.activeAccounts.reduce(0) { (result, account) -> Int in
if let oneUnreadCount = unreadCounts[account.accountID] {
2017-11-19 22:57:42 +01:00
return result + oneUnreadCount
}
return result
}
}
}