mermaid-builder 0.1.2

//! Submodule providing structs for style class definitions in Mermaid diagrams.

mod builder;
mod color;
mod error;
mod font_style;
mod font_weight;
mod style_properties;
mod units;
use alloc::{string::String, vec::Vec};
use core::fmt::Display;

pub use builder::StyleClassBuilder;
pub use color::Color;
pub use error::StyleClassError;
pub use font_weight::FontWeight;
pub use style_properties::StyleProperty;
pub use units::Unit;

#[derive(Clone, Debug, PartialEq, Eq, Hash, PartialOrd, Ord)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
/// Represents a style class in Mermaid diagrams, which can be used to define
/// styles for nodes, edges, and other elements. It includes a name and a set of
/// properties that define the style.
pub struct StyleClass {
    /// The name of the style class.
    name: String,
    /// The properties associated with the style class.
    properties: Vec<StyleProperty>,
}

impl StyleClass {
    #[must_use]
    /// Returns the name of the style class.
    pub fn name(&self) -> &str {
        &self.name
    }
    #[must_use]
    /// Returns the properties of the style class.
    pub fn properties(&self) -> &[StyleProperty] {
        &self.properties
    }
}

impl Display for StyleClass {
    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
        use crate::traits::TabbedDisplay;
        self.fmt_tabbed(f, 0)
    }
}

impl crate::traits::TabbedDisplay for StyleClass {
    fn fmt_tabbed(&self, f: &mut core::fmt::Formatter<'_>, tab_count: usize) -> core::fmt::Result {
        write!(f, "{:indent$}classDef {} ", "", self.name, indent = tab_count * 2)?;
        for (property_number, property) in self.properties.iter().enumerate() {
            if property_number > 0 {
                write!(f, ",")?;
            }
            write!(f, "{property}")?;
        }
        writeln!(f)
    }
}

#[cfg(test)]
mod tests {
    use alloc::{format, string::ToString, vec};

    use super::*;
    use crate::shared::style_class::color::Color;

    #[test]
    fn test_style_class_display() {
        let style_class = StyleClass {
            name: "myClass".to_string(),
            properties: vec![
                StyleProperty::Fill(Color::from((255, 0, 0))),
                StyleProperty::Stroke(Color::from((0, 0, 255))),
            ],
        };

        assert_eq!(format!("{style_class}"), "classDef myClass fill: #ff0000,stroke: #0000ff\n");
    }

    #[test]
    fn test_style_class_getters() {
        let style_class = StyleClass {
            name: "myClass".to_string(),
            properties: vec![StyleProperty::Fill(Color::from((255, 0, 0)))],
        };

        assert_eq!(style_class.name(), "myClass");
        assert_eq!(style_class.properties().len(), 1);
    }
}