101 lines
2.0 KiB
Swift
101 lines
2.0 KiB
Swift
//
|
|
// Folder.swift
|
|
// DataModel
|
|
//
|
|
// Created by Brent Simmons on 7/1/17.
|
|
// Copyright © 2017 Ranchero Software, LLC. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
public final class Folder: UnreadCountProvider {
|
|
|
|
public let account: Account
|
|
public var nameForDisplay: String
|
|
var childObjects = [AnyObject]()
|
|
|
|
init(account: Account, nameForDisplay: String) {
|
|
|
|
self.account = account
|
|
self.nameForDisplay = nameForDisplay
|
|
|
|
NotificationCenter.default.addObserver(self, selector: #selector(unreadCountDidChange(_:)), name: .UnreadCountDidChange, object: nil)
|
|
}
|
|
|
|
// MARK: Notifications
|
|
|
|
@objc dynamic public func unreadCountDidChange(_ note: Notification) {
|
|
|
|
guard let obj = note.object else {
|
|
return
|
|
}
|
|
let potentialChild = obj as AnyObject
|
|
if isChild(potentialChild) {
|
|
updateUnreadCount()
|
|
}
|
|
}
|
|
|
|
// MARK: UnreadCountProvider
|
|
|
|
public var unreadCount = 0 {
|
|
didSet {
|
|
if unreadCount != oldValue {
|
|
postUnreadCountDidChangeNotification()
|
|
}
|
|
}
|
|
}
|
|
|
|
public func updateUnreadCount() {
|
|
|
|
unreadCount = calculateUnreadCount(childObjects)
|
|
}
|
|
}
|
|
|
|
extension Folder: Container {
|
|
|
|
public func flattenedFeeds() -> Set<Feed> {
|
|
|
|
var feeds = Set<Feed>()
|
|
for oneChild in childObjects {
|
|
if let oneFeed = oneChild as? Feed {
|
|
feeds.insert(oneFeed)
|
|
}
|
|
else if let oneContainer = oneChild as? Container {
|
|
feeds.formUnion(oneContainer.flattenedFeeds())
|
|
}
|
|
}
|
|
return feeds
|
|
}
|
|
|
|
public func isChild(_ obj: AnyObject) -> Bool {
|
|
|
|
return childObjects.contains(where: { (oneObject) -> Bool in
|
|
return oneObject === obj
|
|
})
|
|
}
|
|
|
|
public func visitObjects(_ recurse: Bool, _ visitBlock: VisitBlock) -> Bool {
|
|
|
|
for oneObject in childObjects {
|
|
|
|
if let oneContainer = oneObject as? Container {
|
|
if visitBlock(oneObject) {
|
|
return true
|
|
}
|
|
if recurse && oneContainer.visitObjects(recurse, visitBlock) {
|
|
return true
|
|
}
|
|
}
|
|
else {
|
|
if visitBlock(oneObject) {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
}
|
|
|
|
|