From b46f59be2e6c23fc8b01d8af35e7a00074f1526e Mon Sep 17 00:00:00 2001
From: Kuba Orlik <kontakt@kuba-orlik.name>
Date: Sun, 11 Feb 2024 14:01:22 +0100
Subject: [PATCH] Fix styles refreshing issue with old styles still being there

---
 src/back/html.ts | 47 ++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 36 insertions(+), 11 deletions(-)

diff --git a/src/back/html.ts b/src/back/html.ts
index 1dab02a..2c8c77e 100644
--- a/src/back/html.ts
+++ b/src/back/html.ts
@@ -39,19 +39,44 @@ export default function html(
 				${(navbar || default_navbar)(ctx)} ${body}
 				${autoRefreshCSS
 					? /* HTML */ `<script>
+							function make_new_link() {
+								const new_link = document.createElement("link");
+								new_link.rel = "stylesheet";
+								new_link.href = \`/dist/main.css?\${Math.random()}+\${Math.random()}\`;
+								new_link.type = "text/css";
+								return new_link;
+							}
+
+							function getStyles() {
+								return Array.from(
+									document.querySelectorAll("head link")
+								).filter(
+									(e) => new URL(e.href).pathname == "/dist/main.css"
+								);
+							}
+
 							function refresh_css() {
-								Array.from(document.querySelectorAll("link"))
-									.filter(
-										(e) =>
-											new URL(e.href).pathname == "/dist/main.css"
-									)
-									.forEach((e) => {
-										const url = new URL(e.href);
-										url.search =
-											"?" + Math.random() + "" + Math.random();
-										e.href = url.toString();
-									});
+								const new_link = make_new_link();
+
+								//only remove old css once the new one is loaded to prevent blink of unstyled content
+								new_link.onload = () => {
+									console.log("clearing styles");
+									getStyles()
+										.slice(0, -1)
+										.forEach((style) => {
+											console.log({ style, new_link });
+											if (style !== new_link) {
+												style.parentElement.removeChild(style);
+											}
+										});
+								};
+
+								document.querySelector("head").appendChild(new_link);
 							}
+							document.documentElement.addEventListener(
+								"turbo:morph",
+								refresh_css
+							);
 							const socket = new WebSocket("ws://localhost:60808");
 							socket.onmessage = refresh_css;
 					  </script>`