From eee1bd41bda257f97b5b956e57796b2cc975f7a2 Mon Sep 17 00:00:00 2001
From: jsip <jsimonroyer@gmail.com>
Date: Fri, 8 Dec 2023 13:23:44 -0500
Subject: [PATCH] feat: use data attr instead of id + prep dist

---
 dist/turbo.es2017-esm.js | 18 +++++++++---------
 dist/turbo.es2017-umd.js | 18 +++++++++---------
 src/util.ts              | 26 +++++++++++++++-----------
 3 files changed, 33 insertions(+), 29 deletions(-)

diff --git a/dist/turbo.es2017-esm.js b/dist/turbo.es2017-esm.js
index d7848d7d2..47b2268c0 100644
--- a/dist/turbo.es2017-esm.js
+++ b/dist/turbo.es2017-esm.js
@@ -360,17 +360,17 @@ function nextMicrotask() {
 function parseHTMLDocument(html = "") {
     const styleMap = {};
     html = html.replace(/<(\w+)([^>]*)style="([^"]*)"/g, (match, tag, otherAttrs, style) => {
-        const id = Math.random().toString(36).substr(2, 9);
-        styleMap[id] = style;
-        return `<${tag}${otherAttrs}id="${id}"`;
+        const uniqueKey = Math.random().toString(36).substr(2, 9);
+        styleMap[uniqueKey] = style;
+        return `<${tag}${otherAttrs.trim()} data-style-attribute="${uniqueKey}"`;
     });
     const doc = new DOMParser().parseFromString(html, "text/html");
-    Object.keys(styleMap).forEach(id => {
-        const element = doc.getElementById(id);
-        if (element) {
-            element.style.cssText = styleMap[id];
-            element.removeAttribute("id");
-        }
+    Object.keys(styleMap).forEach(uniqueKey => {
+        const elements = doc.querySelectorAll(`[data-style-attribute="${uniqueKey}"]`);
+        elements.forEach(element => {
+            element.style.cssText = styleMap[uniqueKey];
+            element.removeAttribute("data-style-attribute");
+        });
     });
     return doc;
 }
diff --git a/dist/turbo.es2017-umd.js b/dist/turbo.es2017-umd.js
index 46a46bb15..2811d3b75 100644
--- a/dist/turbo.es2017-umd.js
+++ b/dist/turbo.es2017-umd.js
@@ -366,17 +366,17 @@ Copyright © 2023 37signals LLC
     function parseHTMLDocument(html = "") {
         const styleMap = {};
         html = html.replace(/<(\w+)([^>]*)style="([^"]*)"/g, (match, tag, otherAttrs, style) => {
-            const id = Math.random().toString(36).substr(2, 9);
-            styleMap[id] = style;
-            return `<${tag}${otherAttrs}id="${id}"`;
+            const uniqueKey = Math.random().toString(36).substr(2, 9);
+            styleMap[uniqueKey] = style;
+            return `<${tag}${otherAttrs.trim()} data-style-attribute="${uniqueKey}"`;
         });
         const doc = new DOMParser().parseFromString(html, "text/html");
-        Object.keys(styleMap).forEach(id => {
-            const element = doc.getElementById(id);
-            if (element) {
-                element.style.cssText = styleMap[id];
-                element.removeAttribute("id");
-            }
+        Object.keys(styleMap).forEach(uniqueKey => {
+            const elements = doc.querySelectorAll(`[data-style-attribute="${uniqueKey}"]`);
+            elements.forEach(element => {
+                element.style.cssText = styleMap[uniqueKey];
+                element.removeAttribute("data-style-attribute");
+            });
         });
         return doc;
     }
diff --git a/src/util.ts b/src/util.ts
index a8e68193d..2230764e0 100644
--- a/src/util.ts
+++ b/src/util.ts
@@ -66,22 +66,26 @@ export function nextMicrotask() {
   return Promise.resolve()
 }
 
-export function parseHTMLDocument(html = "") {
+export function parseHTMLDocument(html: string = ""): Document {
   const styleMap: { [key: string]: string } = {};
+
+  // Replace style attribute with data-style-attribute and ensure proper spacing
   html = html.replace(/<(\w+)([^>]*)style="([^"]*)"/g, (match, tag, otherAttrs, style) => {
-    const id = Math.random().toString(36).substr(2, 9); 
-    styleMap[id] = style;
-    return `<${tag}${otherAttrs}id="${id}"`;
+    const uniqueKey: string = Math.random().toString(36).substr(2, 9); 
+    styleMap[uniqueKey] = style;
+    // Ensure there is a space before data-style-attribute
+    return `<${tag}${otherAttrs.trim()} data-style-attribute="${uniqueKey}"`;
   });
 
-  const doc = new DOMParser().parseFromString(html, "text/html");
+  const doc: Document = new DOMParser().parseFromString(html, "text/html");
 
-  Object.keys(styleMap).forEach(id => {
-      const element = doc.getElementById(id);
-      if (element) {
-          element.style.cssText = styleMap[id];
-          element.removeAttribute("id");
-      }
+  // Apply styles and remove data-style-attribute
+  Object.keys(styleMap).forEach(uniqueKey => {
+    const elements = doc.querySelectorAll(`[data-style-attribute="${uniqueKey}"]`);
+    elements.forEach(element => {
+        (element as HTMLElement).style.cssText = styleMap[uniqueKey];
+        element.removeAttribute("data-style-attribute");
+    });
   });
 
   return doc;