ytdl-web/web/views.go

package web

import (
	"embed"
	"encoding/json"
	"html/template"
	"net/http"
	"net/url"

	"github.com/gofiber/template/html"
	"github.com/htfy96/reformism"
	"go.fifitido.net/ytdl-web/ytdl"
)

//go:embed views/*
var viewsfs embed.FS

func ViewsEngine() *html.Engine {
	engine := html.NewFileSystem(http.FS(viewsfs), ".html")
	engine.AddFunc(
		"unsafe", func(s string) template.HTML {
			return template.HTML(s)
		},
	)
	engine.AddFunc(
		"queryEscape", func(s string) string {
			return url.QueryEscape(s)
		},
	)
	engine.AddFunc(
		"jsonMarshal", func(s any) (string, error) {
			j, err := json.MarshalIndent(s, "", "    ")
			if err != nil {
				return "", err
			}
			return string(j), nil
		},
	)
	engine.AddFunc(
		"downloadContext", func(meta ytdl.Metadata, url, basePath string, format ytdl.Format) map[string]any {
			return map[string]any{
				"Meta":     meta,
				"Url":      url,
				"BasePath": basePath,
				"Format":   format,
			}
		},
	)

	engine.AddFuncMap(reformism.FuncsHTML)
	return engine
}