postcss-rs · justjavac · Nov 23, 2021 · Nov 22, 2021 · Nov 22, 2021 · Nov 22, 2021
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/assets/simple.css b/assets/simple.css
diff --git a/assets/test.css b/assets/test.css
diff --git a/crates/recursive-parser/Cargo.toml b/crates/recursive-parser/Cargo.toml
@@ -10,6 +10,7 @@ edition = "2021"
 # cssparser = "0.29.0"
 tokenizer = { path = "../tokenizer" }
 mimalloc-rust = "0.1"
+anyhow = "1.0.47"
 
 [profile.release]
 lto = true
@@ -18,7 +19,8 @@ panic = "abort"
 
 [dev-dependencies]
 criterion = "0.3.5"
+similar-asserts = "1.1.0"
 
 [[bench]]
 name = "parse"
-harness = false
+harness = false
diff --git a/crates/recursive-parser/src/ast_util.rs b/crates/recursive-parser/src/ast_util.rs
@@ -0,0 +1,168 @@
+use anyhow::Result;
+use std::io::Write;
+
+use crate::{parser, visitor::Visit};
+#[derive(Default)]
+pub struct AstPrinter<W: Write> {
+  level: usize,
+  writer: W,
+}
+
+impl<W: Write> AstPrinter<W> {
+  pub fn new(level: usize, writer: W) -> Self {
+    Self { level, writer }
+  }
+
+  pub fn print<'a>(&mut self, root: &'a parser::Root<'a>) -> Result<()> {
+    self.visit_root(root)?;
+    Ok(())
+  }
+
+  pub fn result(self) -> W {
+    self.writer
+  }
+}
+
+impl<'a, W: Write> Visit<'a, Result<()>> for AstPrinter<W> {
+  fn visit_root(&mut self, root: &parser::Root) -> Result<()> {
+    writeln!(
+      self.writer,
+      "{}Root@{:?}",
+      " ".repeat(self.level * 2),
+      root.start..root.end
+    )?;
+    self.level += 1;
+    for child in &root.children {
+      match child {
+        parser::RuleOrAtRuleOrDecl::Rule(rule) => {
+          self.visit_rule(rule)?;
+        }
+        parser::RuleOrAtRuleOrDecl::AtRule(at_rule) => {
+          self.visit_at_rule(at_rule)?;
+        }
+        parser::RuleOrAtRuleOrDecl::Declaration(decl) => {
+          self.visit_declaration(decl)?;
+        }
+      }
+    }
+    self.level -= 1;
+    Ok(())
+  }
+
+  fn visit_rule(&mut self, rule: &parser::Rule) -> Result<()> {
+    writeln!(
+      self.writer,
+      "{}Rule@{:?}",
+      " ".repeat(self.level * 2),
+      rule.start..rule.end
+    )?;
+    self.level += 1;
+    writeln!(
+      self.writer,
+      "{}selector: `{}`",
+      " ".repeat(self.level * 2),
+      rule.selector.content,
+    )?;
+    for child in &rule.children {
+      match child {
+        parser::RuleOrAtRuleOrDecl::Rule(rule) => {
+          self.visit_rule(rule)?;
+        }
+        parser::RuleOrAtRuleOrDecl::AtRule(at_rule) => {
+          self.visit_at_rule(at_rule)?;
+        }
+        parser::RuleOrAtRuleOrDecl::Declaration(decl) => {
+          self.visit_declaration(decl)?;
+        }
+      }
+    }
+    self.level -= 1;
+    Ok(())
+  }
+
+  fn visit_at_rule(&mut self, at_rule: &parser::AtRule) -> Result<()> {
+    writeln!(
+      self.writer,
+      "{}AtRule@{:?}",
+      " ".repeat(self.level * 2),
+      at_rule.start..at_rule.end
+    )?;
+    self.level += 1;
+    writeln!(
+      self.writer,
+      "{}name: `{}`",
+      " ".repeat(self.level * 2),
+      at_rule.name,
+    )?;
+    writeln!(
+      self.writer,
+      "{}params: `{}`",
+      " ".repeat(self.level * 2),
+      at_rule.params,
+    )?;
+    for child in &at_rule.children {
+      match child {
+        parser::RuleOrAtRuleOrDecl::Rule(rule) => {
+          self.visit_rule(rule)?;
+        }
+        parser::RuleOrAtRuleOrDecl::AtRule(at_rule) => {
+          self.visit_at_rule(at_rule)?;
+        }
+        parser::RuleOrAtRuleOrDecl::Declaration(decl) => {
+          self.visit_declaration(decl)?;
+        }
+      }
+    }
+    self.level -= 1;
+    Ok(())
+  }
+
+  fn visit_declaration(&mut self, decl: &parser::Declaration) -> Result<()> {
+    writeln!(
+      self.writer,
+      "{}Declaration@{:?}",
+      " ".repeat(self.level * 2),
+      decl.start..decl.end
+    )?;
+    self.level += 1;
+    writeln!(
+      self.writer,
+      "{}prop: `{}`",
+      " ".repeat(self.level * 2),
+      decl.prop.content,
+    )?;
+    writeln!(
+      self.writer,
+      "{}value: `{}`",
+      " ".repeat(self.level * 2),
+      decl.value.content,
+    )?;
+    self.level -= 1;
+    Ok(())
+  }
+}
+
+#[derive(Debug, Default)]
+pub struct WrapString(pub String);
+impl WrapString {
+  pub fn inner_string(mut self) -> String {
+    self.0
+  }
+}
+
+impl From<String> for WrapString {
+  fn from(string: String) -> Self {
+    Self(string)
+  }
+}
+
+impl Write for WrapString {
+  fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
+    self.0 += std::str::from_utf8(buf).unwrap();
+    Ok(buf.len())
+  }
+
+  fn flush(&mut self) -> std::io::Result<()> {
+    Ok(())
+  }
+}