trezor-firmware/core/embed/rust/src/ui/model_tt/component/dialog.rs

use crate::ui::{
    component::{base::ComponentExt, Child, Component, Event, EventCtx},
    geometry::{Grid, Rect},
};

pub enum DialogMsg<T, L, R> {
    Content(T),
    Left(L),
    Right(R),
}

pub struct Dialog<T, L, R> {
    pub content: Child<T>,
    pub left: Child<L>,
    pub right: Child<R>,
}

impl<T, L, R> Dialog<T, L, R>
where
    T: Component,
    L: Component,
    R: Component,
{
    pub fn new(
        area: Rect,
        content: impl FnOnce(Rect) -> T,
        left: impl FnOnce(Rect) -> L,
        right: impl FnOnce(Rect) -> R,
    ) -> Self {
        let layout = DialogLayout::middle(area);
        Self {
            content: content(layout.content).into_child(),
            left: left(layout.left).into_child(),
            right: right(layout.right).into_child(),
        }
    }
}

impl<T, L, R> Component for Dialog<T, L, R>
where
    T: Component,
    L: Component,
    R: Component,
{
    type Msg = DialogMsg<T::Msg, L::Msg, R::Msg>;

    fn event(&mut self, ctx: &mut EventCtx, event: Event) -> Option<Self::Msg> {
        self.content
            .event(ctx, event)
            .map(Self::Msg::Content)
            .or_else(|| self.left.event(ctx, event).map(Self::Msg::Left))
            .or_else(|| self.right.event(ctx, event).map(Self::Msg::Right))
    }

    fn paint(&mut self) {
        self.content.paint();
        self.left.paint();
        self.right.paint();
    }
}

pub struct DialogLayout {
    pub content: Rect,
    pub left: Rect,
    pub right: Rect,
}

impl DialogLayout {
    pub fn middle(area: Rect) -> Self {
        let grid = Grid::new(area, 5, 2);
        Self {
            content: Rect::new(
                grid.row_col(0, 0).top_left(),
                grid.row_col(3, 1).bottom_right(),
            ),
            left: grid.row_col(4, 0),
            right: grid.row_col(4, 1),
        }
    }
}

#[cfg(feature = "ui_debug")]
impl<T, L, R> crate::trace::Trace for Dialog<T, L, R>
where
    T: crate::trace::Trace,
    L: crate::trace::Trace,
    R: crate::trace::Trace,
{
    fn trace(&self, t: &mut dyn crate::trace::Tracer) {
        t.open("Dialog");
        t.field("content", &self.content);
        t.field("left", &self.left);
        t.field("right", &self.right);
        t.close();
    }
}