feat(es/module): Support more import.meta properties (#10179)

**Related issue:** - Closes #10177

Author: magic-akari

.changeset/fluffy-phones-fetch.md

@@ -0,0 +1,6 @@
+---
+swc_core: minor
+swc_ecma_transforms_module: minor
+---
+
+feat(es/module): Support more `import.meta` properties

crates/swc_ecma_transforms_module/src/amd.rs

@@ -6,7 +6,7 @@ use swc_common::{
  comments::{CommentKind, Comments},
  source_map::PURE_SP,
  util::take::Take,
- Mark, Span, SyntaxContext, DUMMY_SP,
+ Mark, Span, Spanned, SyntaxContext, DUMMY_SP,
 };
 use swc_ecma_ast::*;
 use swc_ecma_transforms_base::{feature::FeatureFlag, helper_expr};
@@ -283,17 +283,75 @@ where
  );
  }
  Expr::Member(MemberExpr { span, obj, prop })
- if prop.is_ident_with("url")
- && !self.config.preserve_import_meta
+ if !self.config.preserve_import_meta
  && obj
  .as_meta_prop()
  .map(|p| p.kind == MetaPropKind::ImportMeta)
  .unwrap_or_default() =>
  {
- obj.visit_mut_with(self);
-
- *n = amd_import_meta_url(*span, self.module());
+ let p = match prop {
+ MemberProp::Ident(IdentName { sym, .. }) => &**sym,
+ MemberProp::Computed(ComputedPropName { expr, .. }) => match &**expr {
+ Expr::Lit(Lit::Str(s)) => &s.value,
+ _ => return,
+ },
+ MemberProp::PrivateName(..) => return,
+ };
  self.found_import_meta = true;
+
+ match p {
+ // new URL(module.uri, document.baseURI).href
+ "url" => {
+ *n = amd_import_meta_url(*span, self.module());
+ }
+ // require.toUrl()
+ "resolve" => {
+ let mut require = self.require.clone();
+ require.span = obj.span();
+ *obj = require.into();
+
+ match prop {
+ MemberProp::Ident(IdentName { sym, .. }) => *sym = "toUrl".into(),
+ MemberProp::Computed(ComputedPropName { expr, .. }) => {
+ match &mut **expr {
+ Expr::Lit(Lit::Str(s)) => {
+ s.value = "toUrl".into();
+ s.raw = None;
+ }
+ _ => unreachable!(),
+ }
+ }
+ MemberProp::PrivateName(..) => unreachable!(),
+ }
+ }
+ // module.uri.split("/").pop()
+ "filename" => {
+ *n = amd_import_meta_filename(*span, self.module());
+ }
+ // require.toUrl(".")
+ "dirname" => {
+ let mut require = self.require.clone();
+ require.span = obj.span();
+ *obj = require.into();
+
+ match prop {
+ MemberProp::Ident(IdentName { sym, .. }) => *sym = "toUrl".into(),
+ MemberProp::Computed(ComputedPropName { expr, .. }) => {
+ match &mut **expr {
+ Expr::Lit(Lit::Str(s)) => {
+ s.value = "toUrl".into();
+ s.raw = None;
+ }
+ _ => unreachable!(),
+ }
+ }
+ MemberProp::PrivateName(..) => unreachable!(),
+ }
+
+ *n = n.take().as_call(n.span(), vec![quote_str!(".").as_arg()]);
+ }
+ _ => {}
+ }
  }
  _ => n.visit_mut_children_with(self),
  }
@@ -513,3 +571,13 @@ fn amd_import_meta_url(span: Span, module: Ident) -> Expr {
  }
  .into()
 }
+
+// module.uri.split("/").pop()
+fn amd_import_meta_filename(span: Span, module: Ident) -> Expr {
+ module
+ .make_member(quote_ident!("uri"))
+ .make_member(quote_ident!("split"))
+ .as_call(DUMMY_SP, vec![quote_str!("/").as_arg()])
+ .make_member(quote_ident!("pop"))
+ .as_call(span, vec![])
+}

crates/swc_ecma_transforms_module/src/common_js.rs

@@ -1,5 +1,7 @@
 use rustc_hash::FxHashSet;
-use swc_common::{source_map::PURE_SP, util::take::Take, Mark, Span, SyntaxContext, DUMMY_SP};
+use swc_common::{
+ source_map::PURE_SP, util::take::Take, Mark, Span, Spanned, SyntaxContext, DUMMY_SP,
+};
 use swc_ecma_ast::*;
 use swc_ecma_transforms_base::{feature::FeatureFlag, helper_expr};
 use swc_ecma_utils::{
@@ -200,20 +202,56 @@ impl VisitMut for Cjs {
  );
  }
  Expr::Member(MemberExpr { span, obj, prop })
- if prop.is_ident_with("url")
- && !self.config.preserve_import_meta
+ if !self.config.preserve_import_meta
  && obj
  .as_meta_prop()
  .map(|p| p.kind == MetaPropKind::ImportMeta)
  .unwrap_or_default() =>
  {
- obj.visit_mut_with(self);
+ let p = match prop {
+ MemberProp::Ident(IdentName { sym, .. }) => &**sym,
+ MemberProp::Computed(ComputedPropName { expr, .. }) => match &**expr {
+ Expr::Lit(Lit::Str(s)) => &s.value,
+ _ => return,
+ },
+ MemberProp::PrivateName(..) => return,
+ };
 
- let require = quote_ident!(
- SyntaxContext::empty().apply_mark(self.unresolved_mark),
- "require"
- );
- *n = cjs_import_meta_url(*span, require, self.unresolved_mark);
+ match p {
+ "url" => {
+ let require = quote_ident!(
+ SyntaxContext::empty().apply_mark(self.unresolved_mark),
+ "require"
+ );
+ *n = cjs_import_meta_url(*span, require, self.unresolved_mark);
+ }
+ "resolve" => {
+ let require = quote_ident!(
+ SyntaxContext::empty().apply_mark(self.unresolved_mark),
+ obj.span(),
+ "require"
+ );
+
+ *obj = Box::new(require.into());
+ }
+ "filename" => {
+ *n = quote_ident!(
+ SyntaxContext::empty().apply_mark(self.unresolved_mark),
+ *span,
+ "__filename"
+ )
+ .into();
+ }
+ "dirname" => {
+ *n = quote_ident!(
+ SyntaxContext::empty().apply_mark(self.unresolved_mark),
+ *span,
+ "__dirname"
+ )
+ .into();
+ }
+ _ => {}
+ }
  }
  _ => n.visit_mut_children_with(self),
  }

crates/swc_ecma_transforms_module/tests/fixture/common/import-meta/input.js

@@ -0,0 +1,6 @@
+const react = import.meta.resolve("react");
+const url = import.meta.url;
+const filename = import.meta.filename;
+const dirname = import.meta.dirname;
+
+console.log(react, url, filename, dirname);

crates/swc_ecma_transforms_module/tests/fixture/common/import-meta/output.amd.js

@@ -0,0 +1,11 @@
+define([
+ "require",
+ "module"
+], function(require, module) {
+ "use strict";
+ const react = require.toUrl("react");
+ const url = new URL(module.uri, document.baseURI).href;
+ const filename = module.uri.split("/").pop();
+ const dirname = require.toUrl(".");
+ console.log(react, url, filename, dirname);
+});

crates/swc_ecma_transforms_module/tests/fixture/common/import-meta/output.cjs

@@ -0,0 +1,6 @@
+"use strict";
+const react = require.resolve("react");
+const url = require("url").pathToFileURL(__filename).toString();
+const filename = __filename;
+const dirname = __dirname;
+console.log(react, url, filename, dirname);

crates/swc_ecma_transforms_module/tests/fixture/common/import-meta/output.umd.js

@@ -0,0 +1,12 @@
+(function(global, factory) {
+ if (typeof module === "object" && typeof module.exports === "object") factory();
+ else if (typeof define === "function" && define.amd) define([], factory);
+ else if (global = typeof globalThis !== "undefined" ? globalThis : global || self) factory();
+})(this, function() {
+ "use strict";
+ const react = import.meta.resolve("react");
+ const url = import.meta.url;
+ const filename = import.meta.filename;
+ const dirname = import.meta.dirname;
+ console.log(react, url, filename, dirname);
+});