NetNewsWire/Evergreen/MainWindow/AddFeed/FolderTreeControllerDelegate.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 Data
import Account

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.shared.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
	}
}