NetNewsWire/Shared/Extensions/Node-Extensions.swift

68 lines
1.5 KiB
Swift
Raw Normal View History

2017-05-27 19:43:27 +02:00
//
// Node-Extensions.swift
// Local
//
// Created by Brent Simmons on 8/10/16.
// Copyright © 2016 Ranchero Software, LLC. All rights reserved.
//
import Foundation
2024-03-22 01:21:50 +01:00
import Tree
import Articles
import Core
2017-05-27 19:43:27 +02:00
2017-11-19 01:56:36 +01:00
extension Array where Element == Node {
2024-03-22 01:21:50 +01:00
@MainActor func sortedAlphabetically() -> [Node] {
2017-11-19 01:56:36 +01:00
return Node.nodesSortedAlphabetically(self)
}
2024-03-22 01:21:50 +01:00
@MainActor func sortedAlphabeticallyWithFoldersAtEnd() -> [Node] {
2017-11-19 01:56:36 +01:00
return Node.nodesSortedAlphabeticallyWithFoldersAtEnd(self)
}
}
private extension Node {
2017-05-27 19:43:27 +02:00
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
}
}
}
2017-11-19 01:56:36 +01:00