ARK-OS/arkos/arkrt/ark.ui.basic/Sources/Views/Modifiers/Layout/OverlayModifier.swift

extension View {
    /// Overlays another view on top of this view.
    ///
    /// - Parameter alignment: The alignment that the modifier uses to position
    ///   the overlay relative to the underlying content.
    /// - Parameter content: The view to overlay this view with.
    public func overlay(
        alignment: Alignment = .center,
        @ViewBuilder content: () -> some View
    ) -> some View {
        OverlayModifier(content: self, overlay: content(), alignment: alignment)
    }
}

struct OverlayModifier<Content: View, Overlay: View>: TypeSafeView {
    typealias Children = TupleView2<Content, Overlay>.Children

    var body: TupleView2<Content, Overlay>
    var alignment: Alignment

    init(content: Content, overlay: Overlay, alignment: Alignment) {
        body = TupleView2(content, overlay)
        self.alignment = alignment
    }

    func children<Backend: BaseAppBackend>(
        backend: Backend,
        snapshots: [ViewGraphSnapshotter.NodeSnapshot]?,
        environment: EnvironmentValues
    ) -> TupleView2<Content, Overlay>.Children {
        body.children(
            backend: backend,
            snapshots: snapshots,
            environment: environment
        )
    }

    func layoutableChildren<Backend: BaseAppBackend>(
        backend: Backend,
        children: TupleView2<Content, Overlay>.Children
    ) -> [LayoutSystem.LayoutableChild] {
        []
    }

    func asWidget<Backend: BaseAppBackend>(
        _ children: TupleView2<Content, Overlay>.Children,
        backend: Backend
    ) -> Backend.Widget {
        body.asWidget(children, backend: backend)
    }

    func computeLayout<Backend: BaseAppBackend>(
        _ widget: Backend.Widget,
        children: TupleView2<Content, Overlay>.Children,
        proposedSize: ProposedViewSize,
        environment: EnvironmentValues,
        backend: Backend
    ) -> ViewLayoutResult {
        let contentResult = children.child0.computeLayout(
            with: body.view0,
            proposedSize: proposedSize,
            environment: environment
        )
        let contentSize = contentResult.size
        let overlayResult = children.child1.computeLayout(
            with: body.view1,
            proposedSize: ProposedViewSize(contentSize),
            environment: environment
        )
        let overlaySize = overlayResult.size

        let size = ViewSize(
            max(contentSize.width, overlaySize.width),
            max(contentSize.height, overlaySize.height)
        )

        return ViewLayoutResult(
            size: size,
            childResults: [contentResult, overlayResult]
        )
    }

    func commit<Backend: BaseAppBackend>(
        _ widget: Backend.Widget,
        children: TupleView2<Content, Overlay>.Children,
        layout: ViewLayoutResult,
        environment: EnvironmentValues,
        backend: Backend
    ) {
        let frameSize = layout.size.vector
        let contentSize = children.child0.commit().size.vector
        let overlaySize = children.child1.commit().size.vector

        let contentPosition = alignment.position(ofChild: contentSize, in: frameSize)
        let overlayPosition = alignment.position(ofChild: overlaySize, in: frameSize)

        backend.setPosition(ofChildAt: 0, in: widget, to: contentPosition)
        backend.setPosition(ofChildAt: 1, in: widget, to: overlayPosition)

        backend.setSize(of: widget, to: frameSize)
    }
}