157 lines
3.4 KiB
Swift
157 lines
3.4 KiB
Swift
|
||
//
|
||
// Container.swift
|
||
// NetNewsWire
|
||
//
|
||
// Created by Brent Simmons on 4/17/16.
|
||
// Copyright © 2016 Ranchero Software, LLC. All rights reserved.
|
||
//
|
||
|
||
import Foundation
|
||
import RSCore
|
||
import Articles
|
||
|
||
extension Notification.Name {
|
||
|
||
public static let ChildrenDidChange = Notification.Name("ChildrenDidChange")
|
||
}
|
||
|
||
public protocol Container: class, ContainerIdentifiable {
|
||
|
||
var account: Account? { get }
|
||
var topLevelWebFeeds: Set<WebFeed> { get set }
|
||
var folders: Set<Folder>? { get set }
|
||
|
||
func hasAtLeastOneWebFeed() -> Bool
|
||
func objectIsChild(_ object: AnyObject) -> Bool
|
||
|
||
func hasChildFolder(with: String) -> Bool
|
||
func childFolder(with: String) -> Folder?
|
||
|
||
func removeWebFeed(_ webFeed: WebFeed)
|
||
func addWebFeed(_ webFeed: WebFeed)
|
||
|
||
//Recursive — checks subfolders
|
||
func flattenedWebFeeds() -> Set<WebFeed>
|
||
func has(_ webFeed: WebFeed) -> Bool
|
||
func hasWebFeed(with webFeedID: String) -> Bool
|
||
func hasWebFeed(withURL url: String) -> Bool
|
||
func existingWebFeed(withWebFeedID: String) -> WebFeed?
|
||
func existingWebFeed(withURL url: String) -> WebFeed?
|
||
func existingFolder(with name: String) -> Folder?
|
||
func existingFolder(withID: Int) -> Folder?
|
||
|
||
func postChildrenDidChangeNotification()
|
||
}
|
||
|
||
public extension Container {
|
||
|
||
func hasAtLeastOneWebFeed() -> Bool {
|
||
return topLevelWebFeeds.count > 0
|
||
}
|
||
|
||
func hasChildFolder(with name: String) -> Bool {
|
||
return childFolder(with: name) != nil
|
||
}
|
||
|
||
func childFolder(with name: String) -> Folder? {
|
||
guard let folders = folders else {
|
||
return nil
|
||
}
|
||
for folder in folders {
|
||
if folder.name == name {
|
||
return folder
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func objectIsChild(_ object: AnyObject) -> Bool {
|
||
if let feed = object as? WebFeed {
|
||
return topLevelWebFeeds.contains(feed)
|
||
}
|
||
if let folder = object as? Folder {
|
||
return folders?.contains(folder) ?? false
|
||
}
|
||
return false
|
||
}
|
||
|
||
func flattenedWebFeeds() -> Set<WebFeed> {
|
||
var feeds = Set<WebFeed>()
|
||
feeds.formUnion(topLevelWebFeeds)
|
||
if let folders = folders {
|
||
for folder in folders {
|
||
feeds.formUnion(folder.flattenedWebFeeds())
|
||
}
|
||
}
|
||
return feeds
|
||
}
|
||
|
||
func hasWebFeed(with webFeedID: String) -> Bool {
|
||
return existingWebFeed(withWebFeedID: webFeedID) != nil
|
||
}
|
||
|
||
func hasWebFeed(withURL url: String) -> Bool {
|
||
return existingWebFeed(withURL: url) != nil
|
||
}
|
||
|
||
func has(_ webFeed: WebFeed) -> Bool {
|
||
return flattenedWebFeeds().contains(webFeed)
|
||
}
|
||
|
||
func existingWebFeed(withWebFeedID webFeedID: String) -> WebFeed? {
|
||
for feed in flattenedWebFeeds() {
|
||
if feed.webFeedID == webFeedID {
|
||
return feed
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func existingWebFeed(withURL url: String) -> WebFeed? {
|
||
for feed in flattenedWebFeeds() {
|
||
if feed.url == url {
|
||
return feed
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func existingFolder(with name: String) -> Folder? {
|
||
guard let folders = folders else {
|
||
return nil
|
||
}
|
||
|
||
for folder in folders {
|
||
if folder.name == name {
|
||
return folder
|
||
}
|
||
if let subFolder = folder.existingFolder(with: name) {
|
||
return subFolder
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func existingFolder(withID folderID: Int) -> Folder? {
|
||
guard let folders = folders else {
|
||
return nil
|
||
}
|
||
|
||
for folder in folders {
|
||
if folder.folderID == folderID {
|
||
return folder
|
||
}
|
||
if let subFolder = folder.existingFolder(withID: folderID) {
|
||
return subFolder
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func postChildrenDidChangeNotification() {
|
||
NotificationCenter.default.post(name: .ChildrenDidChange, object: self)
|
||
}
|
||
}
|
||
|