chore: finish plugin

Author: ErKeLost

package.json

@@ -1,6 +1,6 @@
 {
   "name": "farm-plugin-replace-dirname",
-  "version": "0.1.7",
+  "version": "0.2.0",
   "main": "scripts/index.js",
   "types": "scripts/index.d.ts",
   "type": "module",
@@ -30,7 +30,8 @@
     "./package.json": "./package.json"
   },
   "scripts": {
-    "dev": "cargo watch -w src -s 'scripts/watch.sh'",
+    "dev:mac": "cargo watch -w src -s 'scripts/watch.sh'",
+    "dev:win": "cargo watch -w src -s 'cmd /C scripts\\watch.bat'",
     "bump": "npx changeset version",
     "play": "cd playground && farm",
     "build:play": "cd playground && farm build",

playground/index.js

@@ -1,3 +1,5 @@
 const bbbbbb = 123132132132
 console.log('Hello World', bbbbbb);
 console.log(__dirname)
+console.log(__filename);
+console.log(import.meta.url);

scripts/watch.bat

@@ -0,0 +1,3 @@
+@echo off
+echo Running watch script...
+npm run build

src/base.rs

@@ -12,7 +12,7 @@ use farmfe_core::{
   error::CompilationError,
   plugin::{Plugin, PluginProcessModuleHookParam, PluginTransformHookResult},
   swc_common::{comments::NoopComments, BytePos, Mark, SourceMap, DUMMY_SP},
-  swc_ecma_ast::{self, Ident, Module},
+  swc_ecma_ast::{self, Expr, Ident, Lit, MemberExpr, MemberProp, Module, Str},
   swc_ecma_parser::{EsConfig, Parser, StringInput, Syntax},
 };
 
@@ -45,6 +45,7 @@ impl Default for ReplaceDirnameOptions {
     }
   }
 }
+
 impl FarmPluginReplaceDirname {
   fn new(config: &Config, options: String) -> Self {
     let options: ReplaceDirnameOptions = serde_json::from_str(&options).unwrap_or_default();
@@ -66,54 +67,76 @@ impl Plugin for FarmPluginReplaceDirname {
     if !filter.execute(&param.module_id.relative_path()) {
       return Ok(None);
     }
-    let absolute_path = env::current_dir()
+
+    let file_path = env::current_dir()
       .unwrap()
       .join(param.module_id.relative_path());
 
-    let dir_path: &str = Path::new(&absolute_path)
+    let dir_path: &str = Path::new(&file_path)
       .parent()
       .map_or("", |p| p.to_str().unwrap_or(""));
 
     let ast = &mut param.meta.as_script_mut().ast;
-    replace_dirname_with_ast(param);
+    replace_dirname_with_ast(ast, dir_path, file_path.to_str().unwrap());
     Ok(Some(()))
   }
 }
 
-pub fn replace_dirname_with_ast(param: &mut PluginProcessModuleHookParam) {
-  let absolute_path = env::current_dir()
-    .unwrap()
-    .join(param.module_id.relative_path());
-
-  let dir_path: &str = Path::new(&absolute_path)
-    .parent()
-    .map_or("", |p| p.to_str().unwrap_or(""));
-
-  let ast = &mut param.meta.as_script_mut().ast;
-
+pub fn replace_dirname_with_ast(ast: &mut Module, dir_path: &str, file_path: &str) {
   struct ReplaceLibVisitor<'a> {
     dir_path: &'a str,
-    absolute_path: &'a str,
+    file_path: &'a str,
   }
 
   impl<'a> VisitMut for ReplaceLibVisitor<'a> {
-    fn visit_mut_ident(&mut self, ident: &mut Ident) {
-      match &*ident.sym {
-        "__dirname" => {
-          *ident = Ident::new(format!("\"{}\"", self.dir_path).into(), DUMMY_SP);
+    fn visit_mut_expr(&mut self, expr: &mut Expr) {
+      match expr {
+        Expr::Ident(ident) => {
+          match &*ident.sym {
+            "__dirname" => {
+              *expr = Expr::Lit(Lit::Str(Str {
+                value: self.dir_path.into(),
+                span: DUMMY_SP,
+                raw: None,
+              }));
+            }
+            "__filename" => {
+              *expr = Expr::Lit(Lit::Str(Str {
+                value: self.file_path.into(),
+                span: DUMMY_SP,
+                raw: None,
+              }));
+            }
+            _ => {}
+          }
+        }
+        Expr::Member(MemberExpr { obj, prop, .. }) => {
+          if let Expr::MetaProp(meta_prop) = &**obj {
+            if meta_prop.kind == swc_ecma_ast::MetaPropKind::ImportMeta {
+              if let MemberProp::Ident(ident) = &prop {
+                if ident.sym == "url" {
+                  *expr = Expr::Lit(Lit::Str(Str {
+                    value: self.file_path.into(),
+                    span: DUMMY_SP,
+                    raw: None,
+                  }));
+                }
+              }
+            }
+          }
         }
-        "__filename" => {
-          *ident = Ident::new(format!("\"{}\"", self.absolute_path).into(), DUMMY_SP);
+        _ => {
+          // 递归访问子节点
+          expr.visit_mut_children_with(self);
         }
-        _ => {}
       }
     }
   }
 
   let mut visitor = ReplaceLibVisitor {
     dir_path,
-    absolute_path: absolute_path.to_str().unwrap(),
+    file_path,
   };
 
   ast.visit_mut_with(&mut visitor);
-}
+}