theme/theme.go at main · gowool/theme · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package theme

import (
	"context"
	"html/template"
	"io"
	"strings"

	et "github.com/gowool/extends-template"
)

type (
	Handler          = et.Handler
	Source           = et.Source
	Loader           = et.Loader
	ChainLoader      = et.ChainLoader
	MemoryLoader     = et.MemoryLoader
	FileSystemLoader = et.FileSystemLoader

	FuncMap func(t Theme) template.FuncMap

	environment struct {
		*et.Environment
	}
)

var (
	NewChainLoader      = et.NewChainLoader
	NewMemoryLoader     = et.NewMemoryLoader
	NewFileSystemLoader = et.NewFileSystemLoader
	NewFSLoaderWithNS   = et.NewFSLoaderWithNS
)

type Theme interface {
	Debug(debug bool) Theme
	Funcs(funcMap FuncMap) Theme
	Global(global ...string) Theme
	Write(ctx context.Context, w io.Writer, name string, data any) error
	HTML(ctx context.Context, name string, data any) (string, error)
}

func New(loader Loader, handlers ...Handler) Theme {
	env := &environment{Environment: et.NewEnvironment(loader, handlers...)}
	env.Environment.Funcs(Funcs)
	return env
}

func (t *environment) Debug(debug bool) Theme {
	t.Environment.Debug(debug)
	return t
}

func (t *environment) Funcs(funcMap FuncMap) Theme {
	t.Environment.Funcs(funcMap(t))
	return t
}

func (t *environment) Global(global ...string) Theme {
	t.Environment.Global(global...)
	return t
}

func (t *environment) Write(ctx context.Context, w io.Writer, name string, data any) error {
	wrap, err := t.Load(ctx, name)
	if err != nil {
		return err
	}

	return wrap.HTML.ExecuteTemplate(w, name, data)
}

func (t *environment) HTML(ctx context.Context, name string, data any) (string, error) {
	var b strings.Builder
	if err := t.Write(ctx, &b, name, data); err != nil {
		return "", err
	}
	return b.String(), nil
}