From 775e26dc5f3e38705b3dd410d970037529e5bf34 Mon Sep 17 00:00:00 2001
From: Ryan Christian <rchristian@ryanchristian.dev>
Date: Wed, 21 Feb 2024 23:53:35 -0600
Subject: [PATCH] refactor: Skip second loop, delete from bundle earlier

---
 src/prerender.ts | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/src/prerender.ts b/src/prerender.ts
index e2d4d77..d8d9a4e 100644
--- a/src/prerender.ts
+++ b/src/prerender.ts
@@ -221,6 +221,11 @@ export function PrerenderPlugin({
 
 			let prerenderEntry: OutputChunk | undefined;
 			for (const output of Object.keys(bundle)) {
+				// Clean up source maps if the user didn't enable them themselves
+				if (/\.map$/.test(output) && !userEnabledSourceMaps) {
+					delete bundle[output];
+					continue;
+				}
 				if (!/\.js$/.test(output) || bundle[output].type !== "chunk") continue;
 
 				await fs.writeFile(
@@ -407,14 +412,5 @@ export function PrerenderPlugin({
 					});
 			}
 		},
-		async writeBundle(_opts, bundle) {
-			if (!userEnabledSourceMaps) {
-				Object.keys(bundle)
-					.filter(f => /\.map$/.test(f))
-					.forEach(async f => {
-						fs.rm(path.join(viteConfig.root, viteConfig.build.outDir, f));
-					});
-			}
-		},
 	};
 }