NetNewsWire/Evergreen/MainWindow/AddFeed/FolderTreeControllerDelegat...

50 lines
1.2 KiB
Swift
Raw Normal View History

2017-05-27 19:43:27 +02:00
//
// FolderTreeControllerDelegate.swift
// Evergreen
//
// Created by Brent Simmons on 8/10/16.
// Copyright © 2016 Ranchero Software, LLC. All rights reserved.
//
import Foundation
import RSCore
import RSTree
import Data
2017-09-24 21:24:44 +02:00
import Account
2017-05-27 19:43:27 +02:00
final class FolderTreeControllerDelegate: TreeControllerDelegate {
func treeController(treeController: TreeController, childNodesFor node: Node) -> [Node]? {
return node.isRoot ? childNodesForRootNode(node) : nil
}
}
private extension FolderTreeControllerDelegate {
func childNodesForRootNode(_ node: Node) -> [Node]? {
// Root node is Top Level and children are folders. Folders cant have subfolders.
// This will have to be revised later.
var folderNodes = [Node]()
2017-09-24 21:24:44 +02:00
let _ = AccountManager.shared.localAccount.visitChildren { (oneRepresentedObject) in
2017-05-27 19:43:27 +02:00
if let folder = oneRepresentedObject as? Folder {
folderNodes += [createNode(folder, parent: node)]
}
return false
}
return Node.nodesSortedAlphabetically(folderNodes)
}
func createNode(_ folder: Folder, parent: Node) -> Node {
let node = Node(representedObject: folder as AnyObject, parent: parent)
node.canHaveChildNodes = false
return node
}
}