From c6834728952aa53bbc615cf322ec2be4456eb82d Mon Sep 17 00:00:00 2001
From: augustin64 <me.git@augustin64.fr>
Date: Fri, 29 Mar 2024 16:53:59 +0100
Subject: [PATCH] Fix ANSI code formatting

---
 Flask/templates/logs.html | 25 +++++++++++++++++++------
 1 file changed, 19 insertions(+), 6 deletions(-)

diff --git a/Flask/templates/logs.html b/Flask/templates/logs.html
index 43e27dd..c034d57 100644
--- a/Flask/templates/logs.html
+++ b/Flask/templates/logs.html
@@ -13,16 +13,29 @@
     {% endfor %}
 </select>
 <br><br>
-<iframe type="text/html" src="{{url_for('static', filename='logs/1.txt')}}"  width="100%" height="85%" id="embed">
+<iframe type="text/html" src="{{url_for('static', filename='logs/1.txt')}}"  width="100%" height="85%" id="embed"></iframe>
 
-<script type="text/javascript">
-    const subdoc = document.getElementsByTagName("iframe")[0].contentWindow.document;
+<script defer>
+    const iframe = document.getElementsByTagName("iframe")[0];
     var script = document.createElement('script');
+
+    // Wait until ansi_up load
     script.onload = function () {
-        const ansiOutput = subdoc.getElementsByTagName("body")[0].getElementsByTagName("pre")[0]
-        const ansi_up = new AnsiUp();
-        ansiOutput.innerHTML = ansi_up.ansi_to_html(ansiOutput.innerText)
+        // Wait until iframe load
+        iframe.onload = function() {
+            const subdoc = iframe.contentWindow.document;
+            const subBody = subdoc.getElementsByTagName("body")[0]
+            let ansiOutput = subBody;
+            // Depending on the content encoding (and maybe on the browser)
+            // a <pre> is added around the content of the file
+            if (subBody.getElementsByTagName("pre").length > 0) {
+                ansiOutput = subBody.getElementsByTagName("pre")[0];
+            }
+            const ansi_up = new AnsiUp();
+            ansiOutput.innerHTML = ansi_up.ansi_to_html(ansiOutput.innerText);
+        }
     };
+    
     script.src = "https://cdn.jsdelivr.net/npm/ansi_up@4.0.4/ansi_up.js";
     document.head.appendChild(script);
 </script>
-- 
2.47.1