re_ui 0.31.1

use egui::{InnerResponse, Response, Ui, Vec2};

use crate::design_tokens::AlertVisuals;
use crate::{Icon, UiExt as _, icons};

enum AlertKind {
    Info,
    Success,
    Warning,
    Error,
}

impl AlertKind {
    fn colors(&self, ui: &Ui) -> &AlertVisuals {
        match self {
            Self::Info => &ui.tokens().alert_info,
            Self::Success => &ui.tokens().alert_success,
            Self::Warning => &ui.tokens().alert_warning,
            Self::Error => &ui.tokens().alert_error,
        }
    }

    fn icon(&self) -> Icon {
        match self {
            Self::Info => icons::INFO,
            Self::Success => icons::SUCCESS,
            Self::Warning => icons::WARNING,
            Self::Error => icons::ERROR,
        }
    }
}

pub struct AlertText {
    alert: Alert,
    visible_text: String,
    full_text: Option<String>,
}

impl egui::Widget for AlertText {
    fn ui(self, ui: &mut Ui) -> Response {
        self.alert.show_text(ui, self.visible_text, self.full_text)
    }
}

/// Show a pretty info / error message
pub struct Alert {
    kind: AlertKind,
}

impl Alert {
    pub fn success() -> Self {
        Self::new(AlertKind::Success)
    }

    pub fn info() -> Self {
        Self::new(AlertKind::Info)
    }

    pub fn warning() -> Self {
        Self::new(AlertKind::Warning)
    }

    pub fn error() -> Self {
        Self::new(AlertKind::Error)
    }

    fn new(kind: AlertKind) -> Self {
        Self { kind }
    }

    pub fn frame(&self, ui: &Ui) -> egui::Frame {
        let colors = self.kind.colors(ui);

        egui::Frame::new()
            .fill(colors.fill)
            .corner_radius(6)
            .inner_margin(6.0)
            .outer_margin(1.0) // Needed because we set clip_rect_margin. TODO(emilk/egui#5605): remove clip_rect_margin
    }

    pub fn show<T>(self, ui: &mut Ui, content: impl FnOnce(&mut Ui) -> T) -> InnerResponse<T> {
        self.frame(ui).show(ui, |ui| {
            ui.horizontal(|ui| {
                ui.spacing_mut().item_spacing = Vec2::splat(4.0);
                ui.small_icon(&self.kind.icon(), Some(self.kind.colors(ui).icon));
                content(ui)
            })
            .inner
        })
    }

    pub fn show_text(
        self,
        ui: &mut Ui,
        visible_text: impl Into<String>,
        full_text: Option<String>,
    ) -> Response {
        let visible_text = visible_text.into();
        let text_color = self.kind.colors(ui).text;
        self.show(ui, |ui| {
            ui.spacing_mut().item_spacing.x = 4.0;
            ui.style_mut().wrap_mode = Some(egui::TextWrapMode::Wrap);
            let response = ui
                .label(
                    egui::RichText::new(&visible_text)
                        .strong()
                        .color(text_color),
                )
                .on_hover_ui(|ui| {
                    if let Some(full_text) = &full_text {
                        ui.label(full_text);
                        ui.add_space(8.0);
                    }
                    ui.label("Click to copy text.");
                });
            if response.clicked() {
                ui.copy_text(full_text.unwrap_or(visible_text));
            }
        })
        .response
    }

    pub fn text_widget(
        self,
        visible_text: impl Into<String>,
        full_text: Option<String>,
    ) -> AlertText {
        AlertText {
            alert: self,
            visible_text: visible_text.into(),
            full_text,
        }
    }
}

#[cfg(test)]
mod tests {
    use egui::Theme;
    use egui_kittest::{Harness, SnapshotResults};

    use crate::UiExt as _;

    #[test]
    fn test_alert() {
        let mut results = SnapshotResults::new();
        for theme in [egui::Theme::Dark, egui::Theme::Light] {
            let mut harness = Harness::builder().with_theme(theme).build_ui(|ui| {
                crate::apply_style_and_install_loaders(ui.ctx());

                ui.info_label("This is an info alert.");
                ui.success_label("This is a success alert.");
                ui.warning_label("This is a warning alert.");
                ui.error_label("This is an error alert.");
            });

            harness.run();
            harness.fit_contents();
            harness.snapshot(format!(
                "alert_{}",
                match theme {
                    Theme::Dark => "dark",
                    Theme::Light => "light",
                }
            ));
            results.extend_harness(&mut harness);
        }
    }
}