guise-ui 0.2.3

//! `RadioGroup` — a controlled set of mutually-exclusive [`Radio`]s.
//!
//! The parent owns the selected index; the group wires exclusivity and reports
//! the new index through `on_change`. This is the ergonomic layer over the bare
//! `Radio`, which leaves grouping to the caller.

use std::rc::Rc;

use gpui::prelude::*;
use gpui::{div, px, App, IntoElement, SharedString, Window};

use super::Radio;
use crate::reactive::Binding;
use crate::theme::{theme, ColorName, Size};

type GroupHandler = Rc<dyn Fn(usize, &mut Window, &mut App) + 'static>;

/// A vertical group of radios with a single selected value.
#[derive(IntoElement)]
pub struct RadioGroup {
    options: Vec<SharedString>,
    value: Option<usize>,
    color: ColorName,
    size: Size,
    label: Option<SharedString>,
    binding: Option<Binding<usize>>,
    on_change: Option<GroupHandler>,
}

impl RadioGroup {
    pub fn new() -> Self {
        RadioGroup {
            options: Vec::new(),
            value: None,
            color: ColorName::Blue,
            size: Size::Sm,
            label: None,
            binding: None,
            on_change: None,
        }
    }

    pub fn options<I, S>(mut self, options: I) -> Self
    where
        I: IntoIterator<Item = S>,
        S: Into<SharedString>,
    {
        self.options = options.into_iter().map(Into::into).collect();
        self
    }

    /// The currently selected index.
    pub fn value(mut self, value: usize) -> Self {
        self.value = Some(value);
        self
    }

    pub fn color(mut self, color: ColorName) -> Self {
        self.color = color;
        self
    }

    pub fn size(mut self, size: Size) -> Self {
        self.size = size;
        self
    }

    pub fn label(mut self, label: impl Into<SharedString>) -> Self {
        self.label = Some(label.into());
        self
    }

    /// Two-way bind the selected index. Overrides `value`; clicking a radio
    /// writes its index back through the binding, then runs any `on_change`.
    pub fn bind(mut self, binding: Binding<usize>) -> Self {
        self.binding = Some(binding);
        self
    }

    /// Called with the newly selected index when a radio is clicked.
    pub fn on_change(mut self, handler: impl Fn(usize, &mut Window, &mut App) + 'static) -> Self {
        self.on_change = Some(Rc::new(handler));
        self
    }
}

impl Default for RadioGroup {
    fn default() -> Self {
        RadioGroup::new()
    }
}

impl RenderOnce for RadioGroup {
    fn render(self, _window: &mut Window, cx: &mut App) -> impl IntoElement {
        let t = theme(cx);
        let gap = t.spacing(Size::Xs);
        let text = t.text().hsla();
        let font = t.font_size(Size::Sm);
        let value = self
            .binding
            .as_ref()
            .map_or(self.value, |b| Some(b.get(cx)));

        let mut column = div().flex().flex_col().gap(px(gap));
        if let Some(label) = self.label.clone() {
            column = column.child(div().text_size(px(font)).text_color(text).child(label));
        }

        for (i, option) in self.options.iter().enumerate() {
            let mut radio = Radio::new(("guise-radiogroup", i))
                .label(option.clone())
                .checked(value == Some(i))
                .color(self.color)
                .size(self.size);
            if self.binding.is_some() || self.on_change.is_some() {
                let binding = self.binding.clone();
                let handler = self.on_change.clone();
                radio = radio.on_change(move |_ev, window, cx| {
                    if let Some(binding) = &binding {
                        binding.set(cx, i);
                    }
                    if let Some(handler) = &handler {
                        handler(i, window, cx);
                    }
                });
            }
            column = column.child(radio);
        }
        column
    }
}