swc-plugin-component-annotate 1.11.0 → 1.13.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "swc-plugin-component-annotate",
-  "version": "1.11.0",
+  "version": "1.13.0",
   "description": "Use SWC to automatically annotate React components with data attributes for component tracking",
   "author": "scttcper <scttcper@gmail.com>",
   "license": "MIT",
@@ -9,7 +9,7 @@
     "swc"
   ],
   "scripts": {
-    "build": "cargo build --target wasm32-unknown-unknown",
+    "build": "cargo build --release --target wasm32-unknown-unknown",
     "test": "cargo test",
     "prepack": "cp -rf target/wasm32-unknown-unknown/release/swc_plugin_component_annotate.wasm ."
   },

package/src/jsx_utils.rs

@@ -15,7 +15,9 @@ pub fn is_react_fragment(element: &JSXElementName) -> bool {
             }
             false
         }
-        _ => false,
+        JSXElementName::JSXNamespacedName(_) => false,
+        #[cfg(swc_ast_unknown)]
+        _ => panic!("unknown jsx element name"),
     }
 }
 
@@ -30,6 +32,8 @@ pub fn get_element_name(element: &JSXElementName) -> Cow<str> {
         JSXElementName::JSXNamespacedName(namespaced) => {
             Cow::Owned(format!("{}:{}", namespaced.ns.sym, namespaced.name.sym))
         }
+        #[cfg(swc_ast_unknown)]
+        _ => panic!("unknown jsx element name"),
     }
 }
 
@@ -44,6 +48,8 @@ fn get_member_expression_name(member_expr: &JSXMemberExpr) -> String {
                 member_expr.prop.sym
             );
         }
+        #[cfg(swc_ast_unknown)]
+        _ => panic!("unknown jsx object"),
     };
 
     format!("{}.{}", obj_name, member_expr.prop.sym)
@@ -65,10 +71,10 @@ pub fn create_jsx_attr(name: &str, value: &str) -> JSXAttrOrSpread {
     JSXAttrOrSpread::JSXAttr(JSXAttr {
         span: Default::default(),
         name: JSXAttrName::Ident(IdentName::new(name.into(), Default::default())),
-        value: Some(JSXAttrValue::Lit(Lit::Str(Str {
+        value: Some(JSXAttrValue::Str(Str {
             span: Default::default(),
             value: value.into(),
             raw: None,
-        }))),
+        })),
     })
 }

package/src/lib.rs

@@ -86,6 +86,8 @@ impl ReactComponentAnnotateVisitor {
                     // Fragments are always transparent containers
                     jsx_fragment.visit_mut_with(self);
                 }
+                #[cfg(swc_ast_unknown)]
+                JSXElementChild::Unknown(..) => panic!("unknown jsx element child"),
                 _ => {}
             }
         }
@@ -101,6 +103,8 @@ impl ReactComponentAnnotateVisitor {
                 JSXElementChild::JSXFragment(jsx_fragment) => {
                     jsx_fragment.visit_mut_with(self);
                 }
+                #[cfg(swc_ast_unknown)]
+                JSXElementChild::Unknown(..) => panic!("unknown jsx element child"),
                 _ => {}
             }
         }
@@ -210,6 +214,8 @@ impl ReactComponentAnnotateVisitor {
             Expr::Paren(paren_expr) => {
                 self.process_return_expression(&mut paren_expr.expr);
             }
+            #[cfg(swc_ast_unknown)]
+            Expr::Unknown(..) => panic!("unknown expr"),
             _ => {}
         }
     }
@@ -223,6 +229,8 @@ impl ReactComponentAnnotateVisitor {
         let callee_name = match call_expr.callee.as_expr() {
             Some(expr) => match expr.as_ref() {
                 Expr::Ident(ident) => ident.sym.as_ref(),
+                #[cfg(swc_ast_unknown)]
+                Expr::Unknown(..) => panic!("unknown expr"),
                 _ => return None,
             },
             _ => return None,
@@ -340,7 +348,7 @@ impl VisitMut for ReactComponentAnnotateVisitor {
     fn visit_mut_import_decl(&mut self, import_decl: &mut ImportDecl) {
         // Track imports from @emotion/styled (only if enabled)
         if self.config.experimental_rewrite_emotion_styled
-            && import_decl.src.value.as_ref() == "@emotion/styled"
+            && import_decl.src.value == "@emotion/styled"
         {
             for specifier in &import_decl.specifiers {
                 match specifier {
@@ -352,16 +360,22 @@ impl VisitMut for ReactComponentAnnotateVisitor {
                     ImportSpecifier::Named(named_import) => {
                         // Check if the imported name is 'default' or 'styled'
                         let imported_name = match &named_import.imported {
-                            Some(ModuleExportName::Ident(ident)) => ident.sym.as_ref(),
-                            None => named_import.local.sym.as_ref(),
-                            _ => continue,
+                            Some(ModuleExportName::Ident(ident)) => Some(ident.sym.as_ref()),
+                            Some(ModuleExportName::Str(str)) => str.value.as_str(),
+                            None => Some(named_import.local.sym.as_ref()),
+                            #[cfg(swc_ast_unknown)]
+                            Some(_) => panic!("unknown module export name"),
                         };
 
-                        if imported_name == "default" || imported_name == "styled" {
-                            self.styled_import = Some(named_import.local.sym.to_string());
+                        if let Some(imported_name) = imported_name {
+                            if imported_name == "default" || imported_name == "styled" {
+                                self.styled_import = Some(named_import.local.sym.to_string());
+                            }
                         }
                     }
-                    _ => {}
+                    ImportSpecifier::Namespace(_) => {}
+                    #[cfg(swc_ast_unknown)]
+                    _ => panic!("unknown import specifier"),
                 }
             }
         }
@@ -416,6 +430,8 @@ impl VisitMut for ReactComponentAnnotateVisitor {
                                 // Direct expression return
                                 self.process_return_expression(expr);
                             }
+                            #[cfg(swc_ast_unknown)]
+                            _ => panic!("unknown block stmt or expr"),
                         }
 
                         self.current_component_name = None;
@@ -423,6 +439,8 @@ impl VisitMut for ReactComponentAnnotateVisitor {
                     Expr::Fn(func_expr) => {
                         self.find_jsx_in_function_body(&mut func_expr.function, component_name);
                     }
+                    #[cfg(swc_ast_unknown)]
+                    Expr::Unknown(..) => panic!("unknown expr"),
                     _ => {}
                 }
             }
@@ -436,25 +454,33 @@ impl VisitMut for ReactComponentAnnotateVisitor {
 
         // Look for render method
         for member in &mut class_decl.class.body {
-            if let ClassMember::Method(method) = member {
-                if let PropName::Ident(ident) = &method.key {
-                    if ident.sym.as_ref() == "render" {
-                        if let Some(body) = &mut method.function.body {
-                            self.current_component_name = Some(component_name.clone());
-
-                            // Look for return statements
-                            for stmt in &mut body.stmts {
-                                if let Stmt::Return(return_stmt) = stmt {
-                                    if let Some(arg) = &mut return_stmt.arg {
-                                        self.process_return_expression(arg);
+            match member {
+                ClassMember::Method(method) => match &method.key {
+                    PropName::Ident(ident) => {
+                        if ident.sym.as_ref() == "render" {
+                            if let Some(body) = &mut method.function.body {
+                                self.current_component_name = Some(component_name.clone());
+
+                                // Look for return statements
+                                for stmt in &mut body.stmts {
+                                    if let Stmt::Return(return_stmt) = stmt {
+                                        if let Some(arg) = &mut return_stmt.arg {
+                                            self.process_return_expression(arg);
+                                        }
                                     }
                                 }
-                            }
 
-                            self.current_component_name = None;
+                                self.current_component_name = None;
+                            }
                         }
                     }
-                }
+                    #[cfg(swc_ast_unknown)]
+                    PropName::Unknown(..) => panic!("unknown prop name"),
+                    _ => {}
+                },
+                #[cfg(swc_ast_unknown)]
+                ClassMember::Unknown(..) => panic!("unknown class member"),
+                _ => {}
             }
         }