Vorrania/ios/Sources/Baum.swift

import Foundation

/// Eine Zeile einer flach ausgelegten Baumliste, mit ihrer Tiefe.
///
/// Lagerorte und Kategorien sind Baeume, kommen vom Server aber als flache
/// Liste nach Namen sortiert. Daraus wird eine Auswahl, in der "Garage → Fach A"
/// und "Keller → Fach A" irgendwo zwischen allem anderen stehen - Geschwister
/// und Kinder finden nie zusammen. Diese Reihenfolge stellt das her.
struct Baumzeile<T: TreeItem>: Identifiable {
    let item: T
    let tiefe: Int

    var id: T.ID { item.id }

    /// Name mit Einrueckung je Ebene. Macht die Hierarchie in einer flachen
    /// Auswahl sichtbar, ohne jede Zeile mit dem vollen Pfad zu ueberladen.
    var label: String { String(repeating: "   ", count: tiefe) + item.name }
}

/// Bringt eine flache Eltern-Kind-Liste in Baumreihenfolge, mit Tiefe.
///
/// Waisen - deren Elternteil weggefiltert oder geloescht ist - gelten als
/// oberste Ebene. Ohne diese Regel verschwaenden sie lautlos aus der Auswahl.
func baum<T: TreeItem>(_ liste: [T]) -> [Baumzeile<T>] {
    let ids = Set(liste.map(\.id))
    var ergebnis: [Baumzeile<T>] = []

    func gehe(_ knoten: T, _ tiefe: Int) {
        ergebnis.append(Baumzeile(item: knoten, tiefe: tiefe))
        for kind in liste.filter({ $0.parentId == knoten.id }) {
            gehe(kind, tiefe + 1)
        }
    }
    for wurzel in liste.filter({ $0.parentId == nil || !ids.contains($0.parentId!) }) {
        gehe(wurzel, 0)
    }
    return ergebnis
}