webpack-federation-stats-plugin 1.0.3-beta.0 → 1.1.0

package/FederationStatsPlugin.js

@@ -3,20 +3,19 @@ const PLUGIN_NAME = "FederationStatsPlugin"
 const EXTENSION_REGEX = /\.[^/.]+$/
 
 class FederationStatsPlugin {
-  constructor(options = {fileName: "federation-stats.json"}) {
-    if (!options.fileName) throw new Error("fileName option is required.")
-    this._options = options
+  constructor(options = {}) {
+    this._options = {fileName: "federation-stats.json", ...options}
   }
 
   apply(compiler) {
-    const federationPlugin = compiler.options.plugins.find((plugin) => plugin.constructor.name === "ModuleFederationPlugin")
+    const federationPlugin = compiler.options.plugins && compiler.options.plugins.find((plugin) => plugin.constructor.name === "ModuleFederationPlugin")
 
-    if (!federationPlugin) {
-      throw new Error("No ModuleFederationPlugin found.")
-    }
+    if (!federationPlugin) throw new Error("No ModuleFederationPlugin found.")
 
     const appName = federationPlugin._options.name
-    const exposedFiles = new Map(Object.entries(federationPlugin._options.exposes || {}).map(([k, v]) => [v, k]))
+
+    // get exposed modules from the ModuleFederationPlugin
+    const exposedFiles = new Map(Object.entries(federationPlugin._options.exposes || {}).map(([k, v]) => (typeof v === "object" ? [v.import, k] : [v, k])))
 
     compiler.hooks.thisCompilation.tap(PLUGIN_NAME, (compilation) => {
       compilation.hooks.processAssets.tapPromise(
@@ -25,51 +24,59 @@ class FederationStatsPlugin {
           stage: compilation.constructor.PROCESS_ASSETS_STAGE_REPORT,
         },
         async () => {
-          const stats = compilation.getStats().toJson()
-          const modules = stats.modules.filter((module) => {
-            const moduleName = module.name.replace(EXTENSION_REGEX, "")
-            return exposedFiles.has(moduleName)
-          })
+          const stats = compilation.getStats().toJson({})
+          // find mf modules
+          const mfModules = stats.modules.filter((module) => module.issuerName === "container entry" && exposedFiles.has(module.name.replace(EXTENSION_REGEX, "")))
 
           const chunksReducer = (chunksArr, current) => {
             current.siblings.forEach((s) => {
               const chunk = stats.chunks.find((c) => c.id === s)
-              if (chunk) {
-                chunk.files.forEach((f) => chunksArr.push(f))
-              }
+              chunk.files.forEach((f) => chunksArr.push(f))
             })
             current.files.forEach((f) => chunksArr.push(f))
             return chunksArr
           }
 
-          const chunks = modules.reduce((acc, module) => {
-            const moduleName = module.name.replace(EXTENSION_REGEX, "")
-            let exposedAs = exposedFiles.get(moduleName)
-            if (exposedAs.startsWith("./")) {
-              exposedAs = exposedAs.substring(2)
-            }
-
-            const moduleChunks = module.chunks
+          const chunks = mfModules.map((module) => {
+            const exposedAs = exposedFiles.get(module.name.replace(EXTENSION_REGEX, ""))
+            const chunks = module.chunks
               .map((chunkId) => stats.chunks.find((chunk) => chunk.id === chunkId))
-              .filter((chunk) => chunk && chunk.runtime.includes(appName))
+              .filter((chunk) => chunk.runtime.includes(appName))
               .reduce(chunksReducer, [])
+            return {
+              module: exposedAs,
+              chunks: chunks,
+              id: module.id,
+            }
+          })
 
-            acc[exposedAs] = [...new Set(moduleChunks)]
-            return acc
-          }, {})
+          const exposes = chunks.reduce((result, current) => Object.assign(result, {[current.module.replace("./", "")]: current.chunks}), {})
+          const name = (federationPlugin._options.library && federationPlugin._options.library.name) || federationPlugin._options.name
 
           const statsResult = {
-            name: appName,
-            exposes: chunks,
-            publicUrl: this._options.publicUrl || "",
+            name,
+            exposes,
+          }
+
+          const publicUrl = this._options.publicUrl
+          if (publicUrl) {
+            statsResult.publicUrl = publicUrl
           }
 
           const fileName = this._options.fileName
-          compilation.emitAsset(fileName, {
-            source: () => Buffer.from(JSON.stringify(statsResult, null, 2), "utf-8"),
-            size: () => Buffer.byteLength(JSON.stringify(statsResult, null, 2)),
-          })
-        },
+          const statsBuffer = Buffer.from(JSON.stringify(statsResult), "utf-8")
+          const mfStats = {
+            source: () => statsBuffer,
+            size: () => statsBuffer.length,
+          }
+
+          const asset = compilation.getAsset(fileName)
+          if (asset) {
+            compilation.updateAsset(fileName, mfStats)
+          } else {
+            compilation.emitAsset(fileName, mfStats)
+          }
+        }
       )
     })
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "webpack-federation-stats-plugin",
-  "version": "1.0.3-beta.0",
+  "version": "1.1.0",
   "types": "index.d.ts",
   "description": "export module federation stats",
   "main": "FederationStatsPlugin.js",