Allow access to rspack and webpack LoaderContext in the UnpluginBuild…

…Context that is available in the load and transform hooks
unjs · Aug 23, 2024 · ad57f4a · ad57f4a
1 parent 8f7e5ee
commit ad57f4a
Show file tree

Hide file tree

Showing 7 changed files with 24 additions and 2 deletions.
diff --git a/src/rspack/context.ts b/src/rspack/context.ts
@@ -7,6 +7,7 @@ import type { UnpluginBuildContext, UnpluginContext, UnpluginMessage } from '../
 interface ContextOptions {
   addWatchFile: (file: string) => void
   getWatchFiles: () => string[]
+  getNativeBuildContext?: () => any
 }
 
 export function contextOptionsFromCompilation(compilation: Compilation): ContextOptions {

diff --git a/src/rspack/loaders/load.ts b/src/rspack/loaders/load.ts
@@ -19,7 +19,12 @@ export default async function load(this: LoaderContext, source: string, map: any
   const res = await plugin.load.call(
     Object.assign(
       {},
-      this._compilation && createBuildContext(contextOptionsFromCompilation(this._compilation), this._compilation),
+      this._compilation && createBuildContext({
+        ...contextOptionsFromCompilation(this._compilation),
+        getNativeBuildContext: () => {
+          return this
+        },
+      }, this._compilation),
       context,
     ),
     normalizeAbsolutePath(id),

diff --git a/src/rspack/loaders/transform.ts b/src/rspack/loaders/transform.ts
@@ -27,7 +27,12 @@ export default async function transform(
   const res = await plugin.transform.call(
     Object.assign(
       {},
-      this._compilation && createBuildContext(contextOptionsFromCompilation(this._compilation), this._compilation),
+      this._compilation && createBuildContext({
+        ...contextOptionsFromCompilation(this._compilation),
+        getNativeBuildContext: () => {
+          return this
+        },
+      }, this._compilation),
       context,
     ),
     source,

diff --git a/src/types.ts b/src/types.ts
@@ -45,6 +45,7 @@ export interface UnpluginBuildContext {
   emitFile: (emittedFile: EmittedAsset) => void
   getWatchFiles: () => string[]
   parse: (input: string, options?: any) => AstNode
+  getNativeBuildContext?: () => any
 }
 
 export interface UnpluginOptions {

diff --git a/src/webpack/context.ts b/src/webpack/context.ts
@@ -9,6 +9,7 @@ import type { UnpluginBuildContext, UnpluginContext, UnpluginMessage } from '../
 interface ContextOptions {
   addWatchFile: (file: string) => void
   getWatchFiles: () => string[]
+  getNativeBuildContext?: () => any
 }
 
 export function contextOptionsFromCompilation(compilation: Compilation): ContextOptions {
@@ -59,6 +60,9 @@ export function createBuildContext(options: ContextOptions, compilation?: Compil
     getWatchFiles() {
       return options.getWatchFiles()
     },
+    getNativeBuildContext() {
+      return options.getNativeBuildContext?.()
+    },
   }
 }
 

diff --git a/src/webpack/loaders/load.ts b/src/webpack/loaders/load.ts
@@ -23,6 +23,9 @@ export default async function load(this: LoaderContext<any>, source: string, map
       getWatchFiles: () => {
         return this.getDependencies()
       },
+      getNativeBuildContext: () => {
+        return this
+      },
     }, this._compilation), ...context },
     normalizeAbsolutePath(id),
   )

diff --git a/src/webpack/loaders/transform.ts b/src/webpack/loaders/transform.ts
@@ -27,6 +27,9 @@ export default async function transform(this: LoaderContext<{ unpluginName: stri
       getWatchFiles: () => {
         return this.getDependencies()
       },
+      getNativeBuildContext: () => {
+        return this
+      },
     }, this._compilation), ...context },
     source,
     this.resource,