jdrouet/mrml

use std::borrow::Cow;

#[derive(Default)]
pub struct Style {
    selectors: Vec<Cow<'static, str>>,
    content: Vec<Cow<'static, str>>,
}

impl Style {
    pub fn add_content<V: Into<Cow<'static, str>>>(mut self, value: V) -> Self {
        self.content.push(value.into());
        self
    }

    pub fn add_selector<V: Into<Cow<'static, str>>>(mut self, name: V) -> Self {
        self.selectors.push(name.into());
        self
    }
}

impl std::fmt::Display for Style {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "{} {{ {} }}",
            self.selectors.join(",\n"),
            self.content.join("\n")
        )
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn basic() {
        let result = Style::default()
            .add_selector("body".to_string())
            .add_selector("main")
            .add_content("background: red;".to_string())
            .add_content("color: blue;")
            .to_string();
        let expected = "body,\nmain { background: red;\ncolor: blue; }";
        assert_eq!(result, expected);
    }
}