feat(dom): represents every html element

1 files changed, 100 insertions, 0 deletions

diff --git a/dom/html.go b/dom/html.go
new file mode 100644
index 0000000..f43526c
--- /dev/null
+++ b/dom/html.go
@@ -0,0 +1,100 @@
+package dom
+
+import (
+	"fmt"
+	"html"
+	"html/template"
+)
+
+func render(tag string, attributes map[string]string, endSlash bool) template.HTML {
+	base := fmt.Sprintf(`<%s`, tag)
+	for k, v := range attributes {
+		base += fmt.Sprintf(` %s="%s"`, k, v)
+	}
+	if !endSlash {
+		return template.HTML(base + `>`)
+	}
+	return template.HTML(base + ` />`)
+}
+
+type Element interface {
+	Render() template.HTML
+	HasAttribute(string) bool
+	SetAttribute(string, string) Element
+	RemoveAttribute(string) Element
+}
+
+type LiteralElement struct {
+	Content string
+}
+
+func (e LiteralElement) Render() template.HTML {
+	return template.HTML(html.EscapeString(e.Content))
+}
+
+func (LiteralElement) HasAttribute(string) bool {
+	return false
+}
+
+func (e LiteralElement) SetAttribute(string, string) Element {
+	return e
+}
+
+func (e LiteralElement) RemoveAttribute(string) Element {
+	return e
+}
+
+func NewLiteralElement(s string) LiteralElement {
+	return LiteralElement{s}
+}
+
+type VoidElement struct {
+	Tag        string
+	attributes map[string]string
+}
+
+func (e VoidElement) Render() template.HTML {
+	return render(e.Tag, e.attributes, true)
+}
+
+func (e VoidElement) HasAttribute(k string) bool {
+	_, ok := e.attributes[k]
+	return ok
+}
+
+func (e VoidElement) SetAttribute(k, v string) Element {
+	e.attributes[k] = v
+	return e
+}
+
+func (e VoidElement) RemoveAttribute(k string) Element {
+	delete(e.attributes, k)
+	return e
+}
+
+func NewVoidElement(tag string) VoidElement {
+	return VoidElement{tag, make(map[string]string)}
+}
+
+func NewImg(src, alt string) Element {
+	return NewVoidElement("img").SetAttribute("src", src).SetAttribute("alt", alt)
+}
+
+type ContentElement struct {
+	VoidElement
+	Content Element
+}
+
+func (e ContentElement) Render() template.HTML {
+	base := render(e.Tag, e.attributes, false)
+	base += e.Content.Render()
+	return base + template.HTML(fmt.Sprintf(`</%s>`, e.VoidElement.Tag))
+}
+
+func NewContentElement(tag string, content Element) ContentElement {
+	return ContentElement{NewVoidElement(tag), content}
+}
+
+func NewParagraph(content string) Element {
+	return NewContentElement("p", NewLiteralElement(content))
+}