68 lines
1.5 KiB
Swift
68 lines
1.5 KiB
Swift
//
|
|
// Node-Extensions.swift
|
|
// Local
|
|
//
|
|
// Created by Brent Simmons on 8/10/16.
|
|
// Copyright © 2016 Ranchero Software, LLC. All rights reserved.
|
|
//
|
|
|
|
import Foundation
|
|
import RSTree
|
|
import Articles
|
|
import RSCore
|
|
|
|
extension Array where Element == Node {
|
|
|
|
func sortedAlphabetically() -> [Node] {
|
|
|
|
return Node.nodesSortedAlphabetically(self)
|
|
}
|
|
|
|
func sortedAlphabeticallyWithFoldersAtEnd() -> [Node] {
|
|
|
|
return Node.nodesSortedAlphabeticallyWithFoldersAtEnd(self)
|
|
}
|
|
}
|
|
|
|
private extension Node {
|
|
|
|
class func nodesSortedAlphabetically(_ nodes: [Node]) -> [Node] {
|
|
|
|
return nodes.sorted { (node1, node2) -> Bool in
|
|
|
|
guard let obj1 = node1.representedObject as? DisplayNameProvider, let obj2 = node2.representedObject as? DisplayNameProvider else {
|
|
return false
|
|
}
|
|
|
|
let name1 = obj1.nameForDisplay
|
|
let name2 = obj2.nameForDisplay
|
|
|
|
return name1.localizedStandardCompare(name2) == .orderedAscending
|
|
}
|
|
}
|
|
|
|
class func nodesSortedAlphabeticallyWithFoldersAtEnd(_ nodes: [Node]) -> [Node] {
|
|
|
|
return nodes.sorted { (node1, node2) -> Bool in
|
|
|
|
if node1.canHaveChildNodes != node2.canHaveChildNodes {
|
|
if node1.canHaveChildNodes {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
guard let obj1 = node1.representedObject as? DisplayNameProvider, let obj2 = node2.representedObject as? DisplayNameProvider else {
|
|
return false
|
|
}
|
|
|
|
let name1 = obj1.nameForDisplay
|
|
let name2 = obj2.nameForDisplay
|
|
|
|
return name1.localizedStandardCompare(name2) == .orderedAscending
|
|
}
|
|
}
|
|
}
|
|
|
|
|