use alloc::boxed::Box;
use rlvgl_core::{
event::Event,
renderer::Renderer,
widget::{Color, Rect, Widget},
};
use rlvgl_widgets::switch::Switch as BaseSwitch;
use crate::theme::{ColorScheme, ComponentSize, Theme, Variant};
pub struct Switch {
inner: BaseSwitch,
on_change: Option<Box<dyn FnMut(bool)>>,
}
impl Switch {
pub fn new(bounds: Rect) -> Self {
Self {
inner: BaseSwitch::new(bounds),
on_change: None,
}
}
pub fn on_change<F: FnMut(bool) + 'static>(mut self, handler: F) -> Self {
self.on_change = Some(Box::new(handler));
self
}
pub fn on(mut self, value: bool) -> Self {
self.inner.set_on(value);
self
}
pub fn is_on(&self) -> bool {
self.inner.is_on()
}
pub fn set_on(&mut self, value: bool) {
self.inner.set_on(value);
}
pub fn knob_color(mut self, color: Color) -> Self {
self.inner.knob_color = color;
self
}
pub fn knob_color_value(&self) -> Color {
self.inner.knob_color
}
pub fn set_knob_color(&mut self, color: Color) {
self.inner.knob_color = color;
}
pub fn themed(
mut self,
theme: &Theme,
scheme: ColorScheme,
variant: Variant,
size: ComponentSize,
) -> Self {
let resolved = theme.component_style(scheme, variant, size);
self.inner.style = resolved.style;
self.inner.knob_color = resolved.accent_color;
self
}
pub fn style(&self) -> &rlvgl_core::style::Style {
&self.inner.style
}
pub fn style_mut(&mut self) -> &mut rlvgl_core::style::Style {
&mut self.inner.style
}
}
impl Widget for Switch {
fn bounds(&self) -> Rect {
self.inner.bounds()
}
fn draw(&self, renderer: &mut dyn Renderer) {
self.inner.draw(renderer);
}
fn handle_event(&mut self, event: &Event) -> bool {
let before = self.inner.is_on();
let handled = self.inner.handle_event(event);
let after = self.inner.is_on();
if !handled || before == after {
return handled;
}
if let Some(cb) = self.on_change.as_mut() {
cb(after);
}
handled
}
}
#[cfg(test)]
mod tests {
use super::*;
use alloc::rc::Rc;
use core::cell::Cell;
#[test]
fn switch_on_change_triggers() {
let rect = Rect {
x: 0,
y: 0,
width: 40,
height: 20,
};
let state = Rc::new(Cell::new(false));
let s = state.clone();
let mut sw = Switch::new(rect).on_change(move |v| s.set(v));
let event = Event::PressRelease { x: 5, y: 5 };
sw.handle_event(&event);
assert!(state.get());
}
#[test]
fn switch_themed_sets_style_and_knob_color() {
let rect = Rect {
x: 0,
y: 0,
width: 40,
height: 20,
};
let theme = Theme::material_light();
let sw = Switch::new(rect).themed(
&theme,
ColorScheme::Primary,
Variant::Subtle,
ComponentSize::Lg,
);
assert_eq!(
sw.knob_color_value(),
theme.scheme(ColorScheme::Primary).solid
);
assert_eq!(sw.style().radius, theme.tokens.radii.lg);
}
}