Html Generator + Quote

diff --git a/LICENSE b/LICENSE
index 278bf201c6554fc5211171e4ce9d4d19313d67d1..4e918404cb99c15647ce3c8bad7ead52407bc9ae 100644 (file)
--- a/LICENSE
+++ b/LICENSE
@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2023 Mattia Cabrini
+Copyright (c) 2023-2024 Mattia Cabrini
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

diff --git a/html_generator.go b/html_generator.go
new file mode 100644 (file)
index 0000000..7c04252
--- /dev/null
+++ b/html_generator.go
@@ -0,0 +1,157 @@
+// Copyright (c) 2024 Mattia Cabrini
+// SPDX-License-Identifier: MIT
+
+package utility
+
+import (
+       "fmt"
+       "os"
+)
+
+type HtmlGenerator struct {
+       fp *os.File
+
+       partial      int
+       writtenBytes int
+       openedTags   []string
+
+       defined bool
+
+       style map[string]string
+
+       err error
+}
+
+func NewHtmlGenerator(fp *os.File) (h *HtmlGenerator) {
+       if fp == nil {
+               fp = os.Stdout
+       }
+
+       return &HtmlGenerator{
+               fp:    fp,
+               style: make(map[string]string),
+       }
+}
+
+func (h *HtmlGenerator) push(tag string) {
+       h.openedTags = append(h.openedTags, tag)
+}
+
+func (h *HtmlGenerator) pop() (tag string) {
+       if l := len(h.openedTags); l > 0 {
+               tag = h.openedTags[l-1]
+               h.openedTags = h.openedTags[:l-1]
+               h.defined = true
+       }
+
+       return
+}
+
+func (h *HtmlGenerator) printf(format string, args ...any) (w int, _ error) {
+       w, h.err = fmt.Fprintf(h.fp, format, args...)
+       h.writtenBytes += w
+       h.partial += w
+       return w, h.err
+}
+
+func (h *HtmlGenerator) Err() error {
+       return h.err
+}
+
+func (h *HtmlGenerator) Open(tag string) (w int, _ error) {
+       h.closeDefinition()
+
+       w, h.err = h.printf("<%s ", tag)
+
+       if h.err == nil {
+               h.push(tag)
+               h.defined = false
+       }
+
+       return w, h.err
+}
+
+func (h *HtmlGenerator) AddAttribute(name string, value string) {
+       _, h.err = h.printf(" %s=\"%s\" ", name, value)
+}
+
+func (h *HtmlGenerator) AddStyle(name string, value string) {
+       h.style[name] = value
+}
+
+func (h *HtmlGenerator) AddText(text string) {
+       h.closeDefinition()
+
+       if h.err != nil {
+               h.printf("%s", text)
+       }
+}
+
+func (h *HtmlGenerator) closeDefinition() {
+       if len(h.openedTags) == 0 {
+               return
+       }
+
+       if h.defined {
+               return
+       }
+
+       h.printStyle()
+
+       if h.err == nil {
+               h.printf(">")
+       }
+
+       h.style = make(map[string]string)
+       h.defined = true
+}
+
+func (h *HtmlGenerator) printStyle() {
+       if !h.defined && len(h.style) > 0 {
+               h.printf("style=\"")
+
+               for k, v := range h.style {
+                       h.printf("%s: %s; ", k, v)
+               }
+
+               h.printf("\"")
+       }
+}
+
+func (h *HtmlGenerator) OpenInput(class string, typ string, name string, placeholder string, value string) {
+       h.Open("input")
+
+       h.AddAttribute("class", Quote(class))
+       h.AddAttribute("type", Quote(typ))
+       h.AddAttribute("name", Quote(name))
+       h.AddAttribute("placeholder", Quote(placeholder))
+       h.AddAttribute("value", Quote(value))
+}
+
+func (h *HtmlGenerator) Close() {
+       if len(h.openedTags) == 0 {
+               return
+       }
+
+       h.closeDefinition()
+
+       h.printf("</%s>", h.pop())
+}
+
+func (h *HtmlGenerator) CloseAll() {
+       for len(h.openedTags) != 0 {
+               h.Close()
+       }
+}
+
+func (h *HtmlGenerator) GetPartialBytesCount() int {
+       return h.partial
+}
+
+func (h *HtmlGenerator) GetWrittenBytesCount() int {
+       return h.writtenBytes
+}
+
+func (h *HtmlGenerator) ResetPartialBytesCount() {
+       h.partial = 0
+}

diff --git a/strings.go b/strings.go
index 70f5f3b602f58a77d0f8993ea186ccad5715cdb6..3f7f7c4ddd86ed0c248f24e589f89a67a5439d00 100644 (file)
--- a/strings.go
+++ b/strings.go
@@ -3,6 +3,8 @@
 
 package utility
 
+import "strings"
+
 // Returns str without last n characters
 func ButLastN(str string, n int) string {
        lenStr := len(str)
@@ -37,3 +39,21 @@ func EndsWith(str string, suffix string) bool {
 
        return str[lenStr-lenSuffix:] == suffix
 }
+
+func Quote(str string) string {
+       var sb strings.Builder
+
+       sb.WriteRune('"')
+
+       for _, rx := range str {
+               if rx == '"' {
+                       sb.WriteString("\\\"")
+               } else {
+                       sb.WriteRune(rx)
+               }
+       }
+
+       sb.WriteRune('"')
+
+       return sb.String()
+}