refactor: code

Author: snitin315

lib/Server.js

@@ -1993,28 +1993,18 @@ class Server {
 
         let isOutputModule = false;
 
-        if (typeof this.compiler.compilers !== "undefined") {
-          this.stats.stats.forEach((stats) => {
-            stats.compilation.assetsInfo.forEach((value, key) => {
-              const assetExtension = path.extname(key);
-              const assetBasename = path.basename(key, assetExtension);
-
-              if (assetBasename === outputFilename) {
-                extension = assetExtension;
-                isOutputModule = value.javascriptModule;
-              }
-            });
-          });
-        } else {
-          this.stats.compilation.assetsInfo.forEach((value, key) => {
+        const stats = this.stats.stats ? this.stats.stats : [this.stats];
+
+        for (const item of stats) {
+          for (const [key, value] of item.compilation.assetsInfo) {
             const assetExtension = path.extname(key);
             const assetBasename = path.basename(key, assetExtension);
 
             if (assetBasename === outputFilename) {
               extension = assetExtension;
               isOutputModule = value.javascriptModule;
             }
-          });
+          }
         }
 
         if (isOutputModule) {