feat: 支持将样式节点记录写入 ast 中

Author: luckyadam

Cargo.lock

@@ -14,7 +14,8 @@ once_cell = "1.18.0"
 selectors = "0.25.0"
 smallvec = "1.11.0"
 style = "0.1.0"
-swc_common = {version = "0.32.1", features = ["tty-emitter", "sourcemap"]}
-swc_ecma_ast = {version = "0.109.1"}
-swc_ecma_parser = "0.139.1"
-swc_ecma_visit = "0.95.1"
+swc_common = {version = "0.33.0", features = ["tty-emitter", "sourcemap"]}
+swc_ecma_ast = {version = "0.110.0"}
+swc_ecma_codegen = "0.146.0"
+swc_ecma_parser = "0.141.0"
+swc_ecma_visit = "0.96.0"

Cargo.toml

@@ -0,0 +1,4 @@
+.mod {
+	padding-top:29px;
+	padding-bottom:29px;
+}

asset/test.scss

@@ -1,18 +1,25 @@
+use std::{collections::HashMap, rc};
+
 use ego_tree::Tree;
-use swc_common::{sync::Lrc, SourceMap, errors::{Handler, ColorConfig}};
-use swc_ecma_ast::EsVersion;
+use swc_common::{sync::Lrc, SourceMap, errors::{Handler, ColorConfig}, comments::SingleThreadedComments};
+use swc_ecma_ast::{EsVersion, Module};
+use swc_ecma_codegen::{text_writer::JsWriter, Emitter};
 use swc_ecma_parser::{lexer::Lexer, Syntax, TsConfig, StringInput, Parser};
-use swc_ecma_visit::VisitWith;
+use swc_ecma_visit::{VisitWith, VisitMutWith};
 
-use crate::{scraper::{Node, Selector, ElementRef}, visitor::AstVisitor};
+use crate::{scraper::{Node, Selector, ElementRef}, visitor::{AstVisitor, JSXRecord, AstMutVisitor}};
 
 pub struct JSXDocument {
-  pub tree: Tree<Node>
+  pub tree: Tree<Node>,
+  pub module: Option<Module>,
 }
 
 impl JSXDocument {
   pub fn new() -> Self {
-    JSXDocument { tree: Tree::new(Node::Document) }
+    JSXDocument {
+      tree: Tree::new(Node::Document),
+      module: None,
+    }
   }
 
   pub fn parse(&mut self, jsx: String) {
@@ -50,15 +57,36 @@ impl JSXDocument {
       e.into_diagnostic(&handler).emit();
     }
 
-    let module = parser
+    let mut module = parser
       .parse_module()
       .map_err(|e| {
         e.into_diagnostic(&handler).emit()
       })
       .expect("failed to parser module");
-
-    let mut vistor = AstVisitor::new(&module, &mut self.tree);
+    let mut jsx_record: JSXRecord = HashMap::new();
+    let mut vistor = AstVisitor::new(&module, &mut self.tree, &mut jsx_record);
     module.visit_with(&mut vistor);
+    let mut mut_visitor = AstMutVisitor::new( &mut jsx_record);
+    module.visit_mut_with(&mut mut_visitor);
+    // // ast 转代码
+    // let cm = rc::Rc::new(SourceMap::default());
+    // let handler = Handler::with_tty_emitter(ColorConfig::Auto, true, false, Some(cm.clone()));
+    // let comments = SingleThreadedComments::default();
+
+    // let mut buf = Vec::new();
+    // {
+    //   let writer = Box::new(JsWriter::new(cm.clone(), "\n", &mut buf, None));
+    //   let mut emitter = Emitter {
+    //     cfg: Default::default(),
+    //     cm: cm.clone(),
+    //     wr: writer,
+    //     comments: Some(&comments),
+    //   };
+    //   emitter.emit_module(&module).unwrap();
+    // }
+    // let code = String::from_utf8(buf).unwrap();
+    // println!("{}", code);
+    self.module = Some(module);
   }
 
   pub fn select<'a>(&self, selector: &'a Selector) -> Vec<ElementRef> {

src/document.rs

@@ -1,4 +1,4 @@
-use std::collections::HashSet;
+use std::{collections::{HashSet, hash_map::DefaultHasher}, hash::{Hash, Hasher}};
 
 use html5ever::{QualName, ns, LocalName, namespace_url};
 use lightningcss::properties::PropertyId;
@@ -95,3 +95,9 @@ pub fn is_style_inheritable(style: PropertyId<'_>) -> bool {
 pub fn is_starts_with_uppercase(str: &str) -> bool {
   str.chars().next().unwrap().is_uppercase()
 }
+
+pub fn calculate_hash<T: Hash>(t: &T) -> u64 {
+  let mut hasher = DefaultHasher::new();
+  t.hash(&mut hasher);
+  hasher.finish()
+}