grass/src/css.rs

//! # Convert from SCSS AST to CSS
use std::fmt;
use std::io::Write;
use std::sync::atomic::Ordering;

use crate::atrule::AtRule;
use crate::error::SassResult;
use crate::lexer::IS_UTF8;
use crate::{RuleSet, Selector, Stmt, Style, StyleSheet};

#[derive(Debug, Clone)]
enum Toplevel {
    RuleSet(Selector, Vec<BlockEntry>),
    MultilineComment(String),
    AtRule(AtRule),
    Newline,
}

#[derive(Debug, Clone)]
enum BlockEntry {
    Style(Box<Style>),
    MultilineComment(String),
}

impl fmt::Display for BlockEntry {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            BlockEntry::Style(s) => writeln!(f, "{}", s),
            BlockEntry::MultilineComment(s) => writeln!(f, "/*{}*/", s),
        }
    }
}

impl Toplevel {
    const fn new_rule(selector: Selector) -> Self {
        Toplevel::RuleSet(selector, Vec::new())
    }

    fn push_style(&mut self, mut s: Style) -> SassResult<()> {
        s.value = s.value.eval()?;
        if s.value.is_null() {
            return Ok(());
        }
        if let Toplevel::RuleSet(_, entries) = self {
            entries.push(BlockEntry::Style(Box::new(s)));
        }
        Ok(())
    }

    fn push_comment(&mut self, s: String) {
        if let Toplevel::RuleSet(_, entries) = self {
            entries.push(BlockEntry::MultilineComment(s));
        }
    }
}

#[derive(Debug, Clone)]
pub struct Css {
    blocks: Vec<Toplevel>,
}

impl Css {
    pub const fn new() -> Self {
        Css { blocks: Vec::new() }
    }

    pub fn from_stylesheet(s: StyleSheet) -> SassResult<Self> {
        Css::new().parse_stylesheet(s)
    }

    fn parse_stmt(&mut self, stmt: Stmt) -> SassResult<Vec<Toplevel>> {
        Ok(match stmt {
            Stmt::RuleSet(RuleSet {
                selector,
                super_selector,
                rules,
            }) => {
                let selector = super_selector.zip(&selector).remove_placeholders();
                if selector.is_empty() {
                    return Ok(Vec::new());
                }
                let mut vals = vec![Toplevel::new_rule(selector)];
                for rule in rules {
                    match rule {
                        Stmt::RuleSet(_) => vals.extend(self.parse_stmt(rule)?),
                        Stmt::Style(s) => vals
                            .get_mut(0)
                            .expect("expected block to exist")
                            .push_style(*s)?,
                        Stmt::MultilineComment(s) => vals
                            .get_mut(0)
                            .expect("expected block to exist")
                            .push_comment(s),
                        Stmt::AtRule(r) => vals.push(Toplevel::AtRule(r)),
                    };
                }
                vals
            }
            Stmt::MultilineComment(s) => vec![Toplevel::MultilineComment(s)],
            Stmt::Style(_) => panic!("expected toplevel element, found style"),
            Stmt::AtRule(r) => vec![Toplevel::AtRule(r)],
        })
    }

    fn parse_stylesheet(mut self, s: StyleSheet) -> SassResult<Css> {
        let mut is_first = true;
        for stmt in s.0 {
            let v = self.parse_stmt(stmt)?;
            // this is how we print newlines between unrelated styles
            // it could probably be refactored
            if !v.is_empty() {
                if let Toplevel::MultilineComment(..) = v[0] {
                } else if is_first {
                    is_first = false;
                } else {
                    self.blocks.push(Toplevel::Newline);
                }
                self.blocks.extend(v);
            }
        }
        Ok(self)
    }

    pub fn pretty_print<W: Write>(self, buf: &mut W, nesting: usize) -> SassResult<()> {
        let mut has_written = false;
        let padding = vec![' '; nesting * 2].iter().collect::<String>();
        if IS_UTF8.swap(false, Ordering::Relaxed) {
            writeln!(buf, "@charset \"UTF-8\";")?;
        }
        for block in self.blocks {
            match block {
                Toplevel::RuleSet(selector, styles) => {
                    if styles.is_empty() {
                        continue;
                    }
                    has_written = true;
                    writeln!(buf, "{}{} {{", padding, selector)?;
                    for style in styles {
                        write!(buf, "{}  {}", padding, style)?;
                    }
                    writeln!(buf, "{}}}", padding)?;
                }
                Toplevel::MultilineComment(s) => {
                    has_written = true;
                    writeln!(buf, "{}/*{}*/", padding, s)?;
                }
                Toplevel::AtRule(r) => match r {
                    AtRule::Unknown(u) => {
                        if u.params.is_empty() {
                            writeln!(buf, "{}@{} {{", padding, u.name)?;
                        } else {
                            writeln!(buf, "{}@{} {} {{", padding, u.name, u.params)?;
                        }
                        Css::from_stylesheet(StyleSheet::from_stmts(u.body))?
                            .pretty_print(buf, nesting + 1)?;
                        writeln!(buf, "{}}}", padding)?;
                    }
                    _ => todo!("at-rule other than unknown at toplevel"),
                },
                Toplevel::Newline => {
                    if has_written {
                        writeln!(buf)?
                    }
                }
            }
        }
        Ok(())
    }
}