ARK-OS/arkos/arkrt/ark.ui.basic/Sources/Scenes/Commands.swift

/// A set of menus to be displayed in a menu bar.
public struct Commands {
    /// Represents an empty menu bar.
    public static var empty: Commands {
        Commands(menus: [])
    }

    var menus: [CommandMenu]

    init(menus: [CommandMenu]) {
        self.menus = menus
    }

    /// Overlays `newCommands` onto `self`.
    ///
    /// If top-level menus in `newCommands` and `self` have conflicting names,
    /// the menus get merged, with the items from `self`'s menu first, followed
    /// by the items from `newCommands`'s menus.
    ///
    /// - Parameter newCommands: The commands to overlay.
    /// - Returns: The overlayed commands.
    public consuming func overlayed(with newCommands: Commands) -> Commands {
        var newMenusByName: [String: Int] = [:]
        for (i, menu) in newCommands.menus.enumerated() {
            newMenusByName[menu.name] = i
        }

        for (i, menu) in menus.enumerated() {
            guard let newMenuIndex = newMenusByName[menu.name] else {
                continue
            }
            menus[i] = CommandMenu(
                name: menu.name,
                content: menu.content + newCommands.menus[newMenuIndex].content
            )
        }

        let existingMenuNames = Set(menus.map(\.name))
        for newMenu in newCommands.menus {
            guard !existingMenuNames.contains(newMenu.name) else {
                continue
            }
            menus.append(newMenu)
        }

        return self
    }

    /// Resolves the menus to a representation used by backends.
    @MainActor
    func resolve() -> [ResolvedMenu.Submenu] {
        menus.map { menu in
            menu.resolve()
        }
    }
}