refactor: code

snitin315 · snitin315 · commit 480bf1dc9e05 · 2021-09-02T16:10:59.000+05:30
diff --git a/lib/Server.js b/lib/Server.js
@@ -1956,21 +1956,17 @@ class Server {
             splitKey.length - 1 === splitOutputFilename.length &&
             splitKey[0] === outputFilename.split(".")[0]
           ) {
-            extension = splitKey[splitKey.length - 1];
+            extension = path.extname(key);
             isOutputModule = value.javascriptModule;
           }
         });
 
         if (isOutputModule) {
-          filename = this.middleware.getFilenameFromUrl(
-            `${_path}.${extension}`
-          );
+          filename = this.middleware.getFilenameFromUrl(`${_path}${extension}`);
           scriptType = "module";
         } else {
-          extension = "js";
-          filename = this.middleware.getFilenameFromUrl(
-            `${_path}.${extension}`
-          );
+          extension = ".js";
+          filename = this.middleware.getFilenameFromUrl(`${_path}${extension}`);
           scriptType = "text/javascript";
         }
 
@@ -1984,7 +1980,7 @@ class Server {
 
         // Serve a page that executes the javascript
         const queries = req._parsedUrl.search || "";
-        const responsePage = `<!DOCTYPE html><html><head><meta charset="utf-8"/></head><body><script type="${scriptType}" charset="utf-8" src="${_path}.${extension}${queries}"></script></body></html>`;
+        const responsePage = `<!DOCTYPE html><html><head><meta charset="utf-8"/></head><body><script type="${scriptType}" charset="utf-8" src="${_path}${extension}${queries}"></script></body></html>`;
 
         res.send(responsePage);
       } catch (error) {
