1 files changed, 68 insertions, 0 deletions
diff --git a/internal/feed/template/template.go b/internal/feed/template/template.go
new file mode 100644
index 0000000..dd31f51
--- /dev/null
+++ b/internal/feed/template/template.go
@@ -0,0 +1,68 @@
+package template
+
+import (
+	"fmt"
+	"html/template"
+	"strconv"
+	"strings"
+
+	"github.com/Necoro/feed2imap-go/internal/log"
+)
+
+func dict(v ...string) map[string]string {
+	dict := map[string]string{}
+	lenv := len(v)
+	for i := 0; i < lenv; i += 2 {
+		key := v[i]
+		if i+1 >= lenv {
+			dict[key] = ""
+			continue
+		}
+		dict[key] = v[i+1]
+	}
+	return dict
+}
+
+func join(sep string, parts []string) string {
+	return strings.Join(parts, sep)
+}
+
+func lastUrlPart(url string) string {
+	split := strings.Split(url, "/")
+	return split[len(split)-1]
+}
+
+func byteCount(str string) string {
+	b, err := strconv.ParseUint(str, 10, 64)
+	if err != nil {
+		log.Printf("Cannot convert '%s' to byte count: %s", str, err)
+	}
+
+	const unit = 1024
+	if b < unit {
+		return fmt.Sprintf("%d B", b)
+	}
+	div, exp := uint64(unit), 0
+	for n := b / unit; n >= unit; n /= unit {
+		div *= unit
+		exp++
+	}
+	return fmt.Sprintf("%.1f %cB", float64(b)/float64(div), "KMGTPE"[exp])
+}
+
+func html(s string) template.HTML {
+	return template.HTML(s)
+}
+
+var funcMap = template.FuncMap{
+	"dict":        dict,
+	"join":        join,
+	"lastUrlPart": lastUrlPart,
+	"byteCount":   byteCount,
+	"html":        html,
+}
+
+func fromString(name, templateStr string) *template.Template {
+	tpl := template.New(name).Funcs(funcMap)
+	return template.Must(tpl.Parse(templateStr))
+}