feat!: align AssetGeneratorDataUrlFunction with webpack (#8614)

feat: align AssetGeneratorDataUrlFunction with webpack

crates/node_binding/binding.d.ts

@@ -1113,9 +1113,9 @@ export interface RawAliasOptionItem {
   redirect: Array<string | false>
 }
 
-export interface RawAssetGeneratorDataUrlFnArgs {
+export interface RawAssetGeneratorDataUrlFnCtx {
   filename: string
-  content: string
+  module: JsModule
 }
 
 export interface RawAssetGeneratorDataUrlOptions {
@@ -1127,11 +1127,11 @@ export interface RawAssetGeneratorOptions {
   emit?: boolean
   filename?: JsFilename
   publicPath?: "auto" | JsFilename
-  dataUrl?: RawAssetGeneratorDataUrlOptions | ((arg: RawAssetGeneratorDataUrlFnArgs) => string)
+  dataUrl?: RawAssetGeneratorDataUrlOptions | ((source: Buffer, context: RawAssetGeneratorDataUrlFnCtx) => string)
 }
 
 export interface RawAssetInlineGeneratorOptions {
-  dataUrl?: RawAssetGeneratorDataUrlOptions | ((arg: RawAssetGeneratorDataUrlFnArgs) => string)
+  dataUrl?: RawAssetGeneratorDataUrlOptions | ((source: Buffer, context: RawAssetGeneratorDataUrlFnCtx) => string)
 }
 
 export interface RawAssetParserDataUrl {

crates/rspack_binding_options/src/options/raw_module/mod.rs

@@ -2,12 +2,12 @@ use std::fmt::Formatter;
 use std::{collections::HashMap, sync::Arc};
 
 use derive_more::Debug;
-use napi::bindgen_prelude::Either3;
+use napi::bindgen_prelude::{Buffer, Either3};
 use napi::Either;
 use napi_derive::napi;
-use rspack_binding_values::JsFilename;
+use rspack_binding_values::{JsFilename, JsModuleWrapper};
 use rspack_core::{
-  AssetGeneratorDataUrl, AssetGeneratorDataUrlFnArgs, AssetGeneratorDataUrlOptions,
+  AssetGeneratorDataUrl, AssetGeneratorDataUrlFnCtx, AssetGeneratorDataUrlOptions,
   AssetGeneratorOptions, AssetInlineGeneratorOptions, AssetParserDataUrl,
   AssetParserDataUrlOptions, AssetParserOptions, AssetResourceGeneratorOptions,
   CssAutoGeneratorOptions, CssAutoParserOptions, CssGeneratorOptions, CssModuleGeneratorOptions,
@@ -500,7 +500,7 @@ pub struct RawAssetGeneratorOptions {
   pub public_path: Option<JsFilename>,
   #[debug(skip)]
   #[napi(
-    ts_type = "RawAssetGeneratorDataUrlOptions | ((arg: RawAssetGeneratorDataUrlFnArgs) => string)"
+    ts_type = "RawAssetGeneratorDataUrlOptions | ((source: Buffer, context: RawAssetGeneratorDataUrlFnCtx) => string)"
   )]
   pub data_url: Option<RawAssetGeneratorDataUrl>,
 }
@@ -523,7 +523,7 @@ impl From<RawAssetGeneratorOptions> for AssetGeneratorOptions {
 pub struct RawAssetInlineGeneratorOptions {
   #[debug(skip)]
   #[napi(
-    ts_type = "RawAssetGeneratorDataUrlOptions | ((arg: RawAssetGeneratorDataUrlFnArgs) => string)"
+    ts_type = "RawAssetGeneratorDataUrlOptions | ((source: Buffer, context: RawAssetGeneratorDataUrlFnCtx) => string)"
   )]
   pub data_url: Option<RawAssetGeneratorDataUrl>,
 }
@@ -559,22 +559,26 @@ impl From<RawAssetResourceGeneratorOptions> for AssetResourceGeneratorOptions {
 
 type RawAssetGeneratorDataUrl = Either<
   RawAssetGeneratorDataUrlOptions,
-  ThreadsafeFunction<RawAssetGeneratorDataUrlFnArgs, String>,
+  ThreadsafeFunction<(Buffer, RawAssetGeneratorDataUrlFnCtx), String>,
 >;
 struct RawAssetGeneratorDataUrlWrapper(RawAssetGeneratorDataUrl);
 
-#[derive(Debug, Clone)]
 #[napi(object)]
-pub struct RawAssetGeneratorDataUrlFnArgs {
+pub struct RawAssetGeneratorDataUrlFnCtx {
   pub filename: String,
-  pub content: String,
+  #[napi(ts_type = "JsModule")]
+  pub module: JsModuleWrapper,
 }
 
-impl From<AssetGeneratorDataUrlFnArgs> for RawAssetGeneratorDataUrlFnArgs {
-  fn from(value: AssetGeneratorDataUrlFnArgs) -> Self {
+impl From<AssetGeneratorDataUrlFnCtx<'_>> for RawAssetGeneratorDataUrlFnCtx {
+  fn from(value: AssetGeneratorDataUrlFnCtx) -> Self {
     Self {
       filename: value.filename,
-      content: value.content,
+      module: JsModuleWrapper::new(
+        value.module,
+        value.compilation.id(),
+        Some(value.compilation),
+      ),
     }
   }
 }
@@ -584,7 +588,9 @@ impl From<RawAssetGeneratorDataUrlWrapper> for AssetGeneratorDataUrl {
     use pollster::block_on;
     match value.0 {
       Either::A(a) => Self::Options(a.into()),
-      Either::B(b) => Self::Func(Arc::new(move |ctx| block_on(b.call(ctx.into())))),
+      Either::B(b) => Self::Func(Arc::new(move |source, ctx| {
+        block_on(b.call((source.into(), ctx.into())))
+      })),
     }
   }
 }

crates/rspack_core/src/options/module.rs

@@ -13,7 +13,7 @@ use rspack_regex::RspackRegex;
 use rspack_util::{try_all, try_any, MergeFrom};
 use rustc_hash::FxHashMap as HashMap;
 
-use crate::{Filename, Module, ModuleType, PublicPath, Resolve};
+use crate::{Compilation, Filename, Module, ModuleType, PublicPath, Resolve};
 
 #[derive(Debug)]
 pub struct ParserOptionsMap(HashMap<String, ParserOptions>);
@@ -382,13 +382,14 @@ pub struct AssetGeneratorOptions {
   pub data_url: Option<AssetGeneratorDataUrl>,
 }
 
-pub struct AssetGeneratorDataUrlFnArgs {
+pub struct AssetGeneratorDataUrlFnCtx<'a> {
   pub filename: String,
-  pub content: String,
+  pub module: &'a dyn Module,
+  pub compilation: &'a Compilation,
 }
 
 pub type AssetGeneratorDataUrlFn =
-  Arc<dyn Fn(AssetGeneratorDataUrlFnArgs) -> Result<String> + Sync + Send>;
+  Arc<dyn Fn(Vec<u8>, AssetGeneratorDataUrlFnCtx) -> Result<String> + Sync + Send>;
 
 #[cacheable]
 pub enum AssetGeneratorDataUrl {

crates/rspack_plugin_asset/src/lib.rs

@@ -7,7 +7,7 @@ use rayon::prelude::*;
 use rspack_cacheable::{cacheable, cacheable_dyn};
 use rspack_core::{
   rspack_sources::{BoxSource, RawBufferSource, RawStringSource, SourceExt},
-  AssetGeneratorDataUrl, AssetGeneratorDataUrlFnArgs, AssetInfo, AssetParserDataUrl,
+  AssetGeneratorDataUrl, AssetGeneratorDataUrlFnCtx, AssetInfo, AssetParserDataUrl,
   BuildMetaDefaultObject, BuildMetaExportsType, ChunkGraph, ChunkUkey, CodeGenerationDataAssetInfo,
   CodeGenerationDataFilename, CodeGenerationDataUrl, Compilation, CompilationRenderManifest,
   CompilerOptions, Filename, GenerateContext, GeneratorOptions, LocalFilenameFn, Module,
@@ -135,18 +135,19 @@ impl AssetParserAndGenerator {
     resource_data: &ResourceData,
     data_url: Option<&AssetGeneratorDataUrl>,
     source: &BoxSource,
+    module: &dyn Module,
+    compilation: &Compilation,
   ) -> Option<String> {
-    let func_args = AssetGeneratorDataUrlFnArgs {
-      filename: resource_data
-        .resource_path
-        .as_deref()
-        .map(|p| p.as_str().to_string())
-        .unwrap_or_default(),
-      content: source.source().into_owned(),
+    let func_ctx = AssetGeneratorDataUrlFnCtx {
+      filename: resource_data.resource.clone(),
+      module,
+      compilation,
     };
 
     if let Some(AssetGeneratorDataUrl::Func(data_url)) = data_url {
-      return Some(data_url(func_args).expect("call data_url function failed"));
+      return Some(
+        data_url(source.buffer().to_vec(), func_ctx).expect("call data_url function failed"),
+      );
     }
     None
   }
@@ -397,7 +398,7 @@ impl ParserAndGenerator for AssetParserAndGenerator {
   fn generate(
     &self,
     source: &BoxSource,
-    module: &dyn rspack_core::Module,
+    module: &dyn Module,
     generate_context: &mut GenerateContext,
   ) -> Result<BoxSource> {
     let compilation = generate_context.compilation;
@@ -418,7 +419,9 @@ impl ParserAndGenerator for AssetParserAndGenerator {
 
           let encoded_source: String;
 
-          if let Some(custom_data_url) = self.get_data_url(resource_data, data_url, source) {
+          if let Some(custom_data_url) =
+            self.get_data_url(resource_data, data_url, source, module, compilation)
+          {
             encoded_source = custom_data_url;
           } else {
             let mimetype = self.get_mimetype(resource_data, data_url)?;