[#35] Build CSS code outside html cleaning block (#36)

* Build CSS code outside html cleaning block

* 2.0.2

Author: marco-prontera

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-css-injected-by-js",
-  "version": "2.0.1",
+  "version": "2.0.2",
   "description": "A Vite plugin that takes the CSS and adds it to the page through the JS. For those who want a single JS file.",
   "main": "dist/cjs/index.js",
   "module": "dist/esm/index.js",

package.json

@@ -30,22 +30,26 @@ export default function cssInjectedByJsPlugin(
                     !bundle[i].fileName.includes('polyfill')
             );
 
+            const allCssCode = cssAssets.reduce(function extractCssCodeAndDeleteFromBundle(previousValue, cssName) {
+                const cssAsset = bundle[cssName] as OutputAsset;
+                const result = previousValue + cssAsset.source;
+                delete bundle[cssName];
+
+                return result;
+            }, '');
+
+            if (allCssCode.length > 0) {
+                cssToInject = allCssCode;
+            }
+
             for (const name of htmlFiles) {
                 const htmlChunk = bundle[name] as OutputAsset;
                 let replacedHtml = htmlChunk.source as string;
 
-                const allCssCode = cssAssets.reduce(function extractCssCodeAndDeleteFromBundle(previousValue, cssName) {
-                    const cssAsset = bundle[cssName] as OutputAsset;
-                    const result = previousValue + cssAsset.source;
-                    delete bundle[cssName];
+                cssAssets.forEach((cssName) => {
                     replacedHtml = removeLinkStyleSheets(replacedHtml, cssName);
                     htmlChunk.source = replacedHtml;
-                    return result;
-                }, '');
-
-                if (allCssCode.length > 0) {
-                    cssToInject = allCssCode;
-                }
+                });
             }
 
             const jsAsset = bundle[jsAssets[0]] as OutputChunk;