utils: html_builder

krishpranav · krishpranav · commit b9cf48ff2b75 · 2025-02-07T18:57:36.000+05:30
diff --git a/src/utils/html_builder.cr b/src/utils/html_builder.cr
@@ -18,31 +18,65 @@
 # -----------------------------------------------------------------------
 
 module MoonScript
-    class HtmlBuilder
-        def self.build(*, optimize: Bool, doctype: Bool = true, &)
-            html = XML.build_fragment(indent: optimize ? nil : " ") do |xml|
-                builder = new(xml)
-                with builder yield builder
-            end
-        end
+  class HtmlBuilder
+    def self.build(*, optimize : Bool, doctype : Bool = true, &)
+      html =
+        XML.build_fragment(indent: optimize ? nil : "  ") do |xml|
+          builder = new(xml)
 
-        def initialize(@xml: XML::Builder)
+          with builder yield builder
         end
 
-        def tag(tag, attributes, &)
-            @xml.element(tag, attributes) { with self yield }
+      header =
+        if doctype
+          "<!DOCTYPE html>#{optimize ? "" : "\n"}"
         end
 
-        def script(**attributes, &)
-            @xml.element("script", **attributes) { with self yield }
-        end
+      "#{header}#{html}"
+    end
 
-        def script(**attributes)
-            script(**attributes) { text("") }
-        end
+    def initialize(@xml : XML::Builder)
+    end
 
-        def raw(contents)
-            @xml.raw(contents)
-        end
+    def tag(tag, attributes, &)
+      @xml.element(tag, attributes) { with self yield }
     end
-end
+
+    def script(**attributes, &)
+      @xml.element("script", **attributes) { with self yield }
+    end
+
+    def script(**attributes)
+      script(**attributes) { text("") }
+    end
+
+    def a(**attributes, &)
+      @xml.element("a", **attributes) { with self yield }
+    end
+
+    def a(**attributes)
+      a(**attributes) { text("") }
+    end
+
+    def text(contents)
+      @xml.text(contents)
+    end
+
+    def raw(contents)
+      @xml.raw(contents)
+    end
+
+    {% for tag in %w(html head body meta link pre code noscript div span h1 h2
+                    h3 title aside article nav strong) %}
+        def {{tag.id}}(**attributes)
+          @xml.element("{{tag.id}}", **attributes)
+        end
+  
+        def {{tag.id}}(**attributes)
+          @xml.element("{{tag.id}}", **attributes) do
+            with self yield
+          end
+        end
+      {% end %}
+  end
+end
