49 lines
1.2 KiB
Swift
49 lines
1.2 KiB
Swift
//
|
||
// 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 DataModel
|
||
|
||
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 can’t have subfolders.
|
||
// This will have to be revised later.
|
||
|
||
var folderNodes = [Node]()
|
||
|
||
let _ = AccountManager.sharedInstance.localAccount.visitChildren { (oneRepresentedObject) in
|
||
|
||
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
|
||
}
|
||
}
|