use crate::prelude::FluentBuilder as _;
use crate::{
ActiveTheme, App, Axis, Div, Hsla, IntoElement, ParentElement, RenderOnce, SharedString,
StyleRefinement, Styled, StyledExt as _, Window, div, px,
};
#[derive(Debug, Clone, Copy, PartialEq, Default)]
pub enum SeparatorStyle {
#[default]
Solid,
Dashed,
}
#[derive(IntoElement)]
pub struct Separator {
base: Div,
style: StyleRefinement,
label: Option<SharedString>,
axis: Axis,
color: Option<Hsla>,
line_style: SeparatorStyle,
}
impl Separator {
pub fn vertical() -> Self {
Self {
base: div().h_full(),
axis: Axis::Vertical,
label: None,
color: None,
style: StyleRefinement::default(),
line_style: SeparatorStyle::Solid,
}
}
pub fn horizontal() -> Self {
Self {
base: div(),
axis: Axis::Horizontal,
label: None,
color: None,
style: StyleRefinement::default(),
line_style: SeparatorStyle::Solid,
}
}
pub fn vertical_dashed() -> Self {
Self::vertical().dashed()
}
pub fn horizontal_dashed() -> Self {
Self::horizontal().dashed()
}
pub fn label(mut self, label: impl Into<SharedString>) -> Self {
self.label = Some(label.into());
self
}
pub fn color(mut self, color: impl Into<Hsla>) -> Self {
self.color = Some(color.into());
self
}
pub fn dashed(mut self) -> Self {
self.line_style = SeparatorStyle::Dashed;
self
}
fn render_base(axis: Axis) -> Div {
div().absolute().map(|this| match axis {
Axis::Vertical => this.w(px(1.)).h_full(),
Axis::Horizontal => this.h(px(1.)).w_full(),
})
}
fn render_solid(axis: Axis, color: Hsla) -> impl IntoElement {
Self::render_base(axis).bg(color)
}
fn render_dashed(axis: Axis, color: Hsla) -> impl IntoElement {
div()
.absolute()
.map(|this| match axis {
Axis::Horizontal => this.w_full().h(px(1.)).border_t_1(),
Axis::Vertical => this.h_full().w(px(1.)).border_l_1(),
})
.border_dashed()
.border_color(color)
}
}
impl Styled for Separator {
fn style(&mut self) -> &mut StyleRefinement {
&mut self.style
}
}
impl RenderOnce for Separator {
fn render(self, _: &mut Window, cx: &mut App) -> impl IntoElement {
let color = self.color.unwrap_or(cx.theme().border);
let axis = self.axis;
let line_style = self.line_style;
self.base
.flex()
.flex_shrink_0()
.items_center()
.justify_center()
.refine_style(&self.style)
.child(match line_style {
SeparatorStyle::Solid => Self::render_solid(axis, color).into_any_element(),
SeparatorStyle::Dashed => Self::render_dashed(axis, color).into_any_element(),
})
.when_some(self.label, |this, label| {
this.child(
div()
.px_2()
.py_1()
.mx_auto()
.text_xs()
.bg(cx.theme().tokens.background)
.text_color(cx.theme().muted_foreground)
.child(label),
)
})
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_separator_axis() {
let v = Separator::vertical();
assert_eq!(v.axis, Axis::Vertical);
let h = Separator::horizontal();
assert_eq!(h.axis, Axis::Horizontal);
}
#[test]
fn test_separator_dashed() {
let s = Separator::horizontal_dashed();
assert_eq!(s.line_style, SeparatorStyle::Dashed);
assert_eq!(s.axis, Axis::Horizontal);
}
#[test]
fn test_separator_label_color() {
let s = Separator::vertical().label("分隔").color(crate::red_500());
assert_eq!(s.label, Some("分隔".into()));
assert!(s.color.is_some());
}
}