swc-plugin-component-annotate 1.0.0 → 1.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "swc-plugin-component-annotate",
-  "version": "1.0.0",
+  "version": "1.2.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

@@ -1,9 +1,11 @@
 pub mod config;
 mod constants;
 mod jsx_utils;
+pub mod path_utils;
 
 use config::PluginConfig;
 use jsx_utils::*;
+use path_utils::extract_filename;
 use rustc_hash::FxHashSet;
 use swc_core::{
     common::FileName,
@@ -292,23 +294,9 @@ impl VisitMut for ReactComponentAnnotateVisitor {
     }
 }
 
-fn extract_filename(filename: &FileName) -> Option<String> {
-    match filename {
-        FileName::Real(path) => path
-            .file_name()
-            .and_then(|name| name.to_str())
-            .map(|s| s.to_string()),
-        FileName::Custom(custom) => {
-            if custom.contains('/') {
-                custom.split('/').last().map(|s| s.to_string())
-            } else if custom.contains('\\') {
-                custom.split('\\').last().map(|s| s.to_string())
-            } else {
-                Some(custom.clone())
-            }
-        }
-        _ => None,
-    }
+// Export for testing
+pub fn extract_filename_for_test(filename: &FileName) -> Option<String> {
+    extract_filename(filename)
 }
 
 #[plugin_transform]

package/src/path_utils.rs

@@ -0,0 +1,75 @@
+use swc_core::common::FileName;
+
+// Platform-specific path parsing functions
+fn parse_unix_path(path: &str) -> Vec<&str> {
+    path.split('/').collect()
+}
+
+fn parse_windows_path(path: &str) -> Vec<&str> {
+    path.split('\\').collect()
+}
+
+// Fallback for mixed or unknown paths - checks the string content
+fn parse_path_with_detection(path: &str) -> Vec<&str> {
+    if path.contains('\\') {
+        parse_windows_path(path)
+    } else {
+        parse_unix_path(path)
+    }
+}
+
+pub fn extract_filename(filename: &FileName) -> Option<String> {
+    match filename {
+        FileName::Real(path) => {
+            if let Some(file_name) = path.file_name().and_then(|name| name.to_str()) {
+                // Check if it's an index file
+                if file_name.starts_with("index.") {
+                    // Get parent directory name and combine with filename
+                    if let Some(parent) = path
+                        .parent()
+                        .and_then(|p| p.file_name())
+                        .and_then(|n| n.to_str())
+                    {
+                        Some(format!("{}/{}", parent, file_name))
+                    } else {
+                        Some(file_name.to_string())
+                    }
+                } else {
+                    Some(file_name.to_string())
+                }
+            } else {
+                None
+            }
+        }
+        FileName::Custom(custom) => {
+            // Always use detection for Custom filenames since they can come from any platform
+            let parts = parse_path_with_detection(custom);
+
+            let file_part = if parts.len() > 1 {
+                parts.last().copied().unwrap_or(custom)
+            } else {
+                custom
+            };
+
+            // Check if it's an index file
+            if file_part.starts_with("index.") {
+                // Extract parent directory from the full path
+                let parent = if parts.len() >= 2 {
+                    Some(parts[parts.len() - 2])
+                } else {
+                    None
+                };
+
+                if let Some(parent_name) = parent {
+                    // Always use forward slash in output for consistency
+                    Some(format!("{}/{}", parent_name, file_part))
+                } else {
+                    Some(file_part.to_string())
+                }
+            } else {
+                Some(file_part.to_string())
+            }
+        }
+        _ => None,
+    }
+}