-
Notifications
You must be signed in to change notification settings - Fork 1
/
yaml_renderer.go
61 lines (51 loc) · 1.15 KB
/
yaml_renderer.go
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
package htmldelta
import (
_ "embed"
"io"
"github.com/nikolaydubina/htmlyaml"
"gopkg.in/yaml.v3"
)
//go:embed delta_yaml.html
var yamlTemplateHTML []byte
// YAMLRenderer renders delta of YAML in HTML with animations.
// Not idempotent.
// Not safe for concurrent use.
type YAMLRenderer struct {
m *htmlyaml.PageMarshaler
num map[string]float64
}
func NewYAMLRenderer(title string) *YAMLRenderer {
s := YAMLRenderer{
num: make(map[string]float64),
}
r := htmlyaml.DefaultMarshaler
r.Number = s.numberfunc
m := htmlyaml.DefaultPageMarshaler
m.Marshaler = &r
m.Title = title
m.Template = yamlTemplateHTML
s.m = &m
return &s
}
func (s *YAMLRenderer) numberfunc(k string, v float64, sv string) string {
var class string
if prev, ok := s.num[k]; ok {
switch {
case v > prev:
class = "number-up"
case v < prev:
class = "number-down"
default:
class = ""
}
}
s.num[k] = v
return `<div class="yaml-value ` + class + `">` + sv + `</div>`
}
func (s *YAMLRenderer) FromTo(r io.Reader, w io.Writer) error {
var v any
if err := yaml.NewDecoder(r).Decode(&v); err != nil && err != io.EOF {
return err
}
return s.m.MarshalTo(w, v)
}