Checkpoint

webserver.go

@@ -67,8 +67,21 @@ func rootHandler(w http.ResponseWriter, r *http.Request) {
 		data["DNSRecords"] = entries
 		
 		tree := BuildRecordsTree(entries)
-		htmlTree := tree.ToHTML([]string{})
-		data["Tree"] = template.HTML(htmlTree)
+		htmlElements := tree.ToHTMLElements([]string{}, 0)
+
+		var html string
+		for _, el := range htmlElements {
+			if el.Header {
+				html += el.HTML
+			}
+		}
+		for _, el := range htmlElements {
+			if !el.Header {
+				html += el.HTML
+			}
+		}
+
+		data["Tree"] = template.HTML(html)
 
 		j, _ := json.Marshal(tree)
 		os.WriteFile("/tmp/tree.json", j, 0644)