swc-plugin-component-annotate 1.2.1 → 1.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "swc-plugin-component-annotate",
-  "version": "1.2.1",
+  "version": "1.3.0",
   "description": "Use SWC to automatically annotate React components with data attributes for component tracking",
   "author": "scttcper <scttcper@gmail.com>",
   "license": "MIT",

package/src/lib.rs

@@ -13,7 +13,10 @@ use swc_core::{
         ast::*,
         visit::{noop_visit_mut_type, VisitMut, VisitMutWith},
     },
-    plugin::{plugin_transform, proxies::TransformPluginProgramMetadata},
+    plugin::{
+        metadata::TransformPluginMetadataContextKind, plugin_transform,
+        proxies::TransformPluginProgramMetadata,
+    },
 };
 
 pub struct ReactComponentAnnotateVisitor {
@@ -42,7 +45,7 @@ impl ReactComponentAnnotateVisitor {
     }
 
     #[inline]
-    fn should_ignore_component(&self, component_name: &str) -> bool {
+    pub fn should_ignore_component(&self, component_name: &str) -> bool {
         self.ignored_components_set.contains(component_name)
     }
 
@@ -310,8 +313,16 @@ pub fn process_transform(
         PluginConfig::default()
     };
 
-    let mut visitor =
-        ReactComponentAnnotateVisitor::new(config, &FileName::Custom("unknown".to_string()));
+    // Try to get the actual filename from the metadata context
+    let filename = if let Some(filename_str) =
+        metadata.get_context(&TransformPluginMetadataContextKind::Filename)
+    {
+        FileName::Custom(filename_str)
+    } else {
+        FileName::Custom("unknown".to_string())
+    };
+
+    let mut visitor = ReactComponentAnnotateVisitor::new(config, &filename);
     program.visit_mut_with(&mut visitor);
     program
 }