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

extension Scene {
    /// Adds menu commands to this scene.
    ///
    /// The commands will typically be displayed in the system's global menu bar
    /// if it has one, or in individual windows' menu bars otherwise.
    ///
    /// - Parameter commands: The commands to add.
    public func commands(@CommandsBuilder _ commands: () -> Commands) -> some Scene {
        CommandsModifier(content: self, commands: commands())
    }
}

struct CommandsModifier<Content: Scene>: Scene {
    typealias Node = CommandsModifierNode<Content>

    var content: Content
    var commands: Commands

    init(content: Content, commands: Commands) {
        self.content = content
        self.commands = commands
    }
}

final class CommandsModifierNode<Content: Scene>: SceneGraphNode {
    typealias NodeScene = CommandsModifier<Content>

    var commands: Commands
    var contentNode: Content.Node

    init<Backend: BaseAppBackend>(
        from scene: NodeScene,
        backend: Backend,
        environment: EnvironmentValues
    ) {
        self.commands = scene.commands
        self.contentNode = Content.Node(
            from: scene.content,
            backend: backend,
            environment: environment
        )
    }

    func updateNode(
        _ newScene: NodeScene?,
        environment: EnvironmentValues
    ) -> SceneNodeUpdateResult {
        if let newScene {
            self.commands = newScene.commands
        }

        var result = contentNode.updateNode(newScene?.content, environment: environment)
        result.preferences.commands = result.preferences.commands.overlayed(with: commands)
        return result
    }

    func update<Backend: BaseAppBackend>(
        backend: Backend,
        environment: EnvironmentValues
    ) {
        contentNode.update(
            backend: backend,
            environment: environment
        )
    }
}