Merge pull request #54 from chenjiahan/enhance-writeToDisk

feat: support filter writeToDisk by compilation name

Author: 9aoy

src/index.js

@@ -100,7 +100,7 @@ const noop = () => {};
  * @template {IncomingMessage} [RequestInternal = IncomingMessage]
  * @template {ServerResponse} [ResponseInternal = ServerResponse]
  * @typedef {Object} Options
- * @property {boolean | ((targetPath: string) => boolean)} [writeToDisk]
+ * @property {boolean | ((targetPath: string, compilationName?: string) => boolean)} [writeToDisk]
  * @property {NonNullable<Configuration["output"]>["publicPath"]} [publicPath]
  * @property {boolean | string} [index]
  * @property {boolean} [lastModified]

src/utils/setupWriteToDisk.js

@@ -29,11 +29,13 @@ function setupWriteToDisk(context) {
 
       compiler.hooks.assetEmitted.tapAsync(
         "DevMiddleware",
-        (file, info, callback) => {
-          const { targetPath, content } = info;
+        (_file, info, callback) => {
+          const { targetPath, content, compilation } = info;
           const { writeToDisk: filter } = context.options;
           const allowWrite =
-            filter && typeof filter === "function" ? filter(targetPath) : true;
+            filter && typeof filter === "function"
+              ? filter(targetPath, compilation.name)
+              : true;
 
           if (!allowWrite) {
             return callback();

types/index.d.ts

@@ -75,7 +75,7 @@ export = wdm;
  * @template {IncomingMessage} [RequestInternal = IncomingMessage]
  * @template {ServerResponse} [ResponseInternal = ServerResponse]
  * @typedef {Object} Options
- * @property {boolean | ((targetPath: string) => boolean)} [writeToDisk]
+ * @property {boolean | ((targetPath: string, compilationName?: string) => boolean)} [writeToDisk]
  * @property {NonNullable<Configuration["output"]>["publicPath"]} [publicPath]
  * @property {boolean | string} [index]
  * @property {boolean} [lastModified]
@@ -256,7 +256,10 @@ type Options<
   RequestInternal extends IncomingMessage = import("http").IncomingMessage,
   ResponseInternal extends ServerResponse = ServerResponse,
 > = {
-  writeToDisk?: boolean | ((targetPath: string) => boolean) | undefined;
+  writeToDisk?:
+    | boolean
+    | ((targetPath: string, compilationName?: string) => boolean)
+    | undefined;
   publicPath?: NonNullable<Configuration["output"]>["publicPath"];
   index?: string | boolean | undefined;
   lastModified?: boolean | undefined;