From a95c745ae24c53d1cd77e9b832e5d55d49ca179f Mon Sep 17 00:00:00 2001
From: Adam Bem <adam.bem@zoho.eu>
Date: Thu, 23 Feb 2023 10:42:10 +0100
Subject: [PATCH] Formatter now accepts XMLs with prologs

---
 Backend-libXML/Parser.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/Backend-libXML/Parser.py b/Backend-libXML/Parser.py
index 66901a6..827b356 100644
--- a/Backend-libXML/Parser.py
+++ b/Backend-libXML/Parser.py
@@ -7,8 +7,15 @@ def prettify(source: str) -> str:
     :param source: XML
     :return: prettified XML
     """
+    prolog = ""
+    prolog_start = source.find("<?")
+    
+    if prolog_start != -1:
+        prolog_end = source.find("?>") + 2
+        prolog = source[prolog_start:prolog_end] + "\n"
+        source = source[prolog_end: ]
     xml = etree.XML(source)
-    return etree.tostring(xml, pretty_print=True).decode()
+    return prolog + etree.tostring(xml, pretty_print=True).decode()
 
 def minimize(source: str) -> str:
     """Method used to minimize XML by deleting not needed whitespaces.