/
templatehelper.go
80 lines (71 loc) · 1.94 KB
/
templatehelper.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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
package termenv
import (
"text/template"
)
// TemplateFuncs contains a few useful template helpers
func TemplateFuncs(p Profile) template.FuncMap {
if p == Ascii {
return noopTemplateFuncs
}
return template.FuncMap{
"Color": func(values ...interface{}) string {
s := String(values[len(values)-1].(string))
switch len(values) {
case 2:
s = s.Foreground(p.Color(values[0].(string)))
case 3:
s = s.
Foreground(p.Color(values[0].(string))).
Background(p.Color(values[1].(string)))
}
return s.String()
},
"Foreground": func(values ...interface{}) string {
s := String(values[len(values)-1].(string))
if len(values) == 2 {
s = s.Foreground(p.Color(values[0].(string)))
}
return s.String()
},
"Background": func(values ...interface{}) string {
s := String(values[len(values)-1].(string))
if len(values) == 2 {
s = s.Background(p.Color(values[0].(string)))
}
return s.String()
},
"Bold": styleFunc(Style.Bold),
"Faint": styleFunc(Style.Faint),
"Italic": styleFunc(Style.Italic),
"Underline": styleFunc(Style.Underline),
"Overline": styleFunc(Style.Overline),
"Blink": styleFunc(Style.Blink),
"Reverse": styleFunc(Style.Reverse),
"CrossOut": styleFunc(Style.CrossOut),
}
}
func styleFunc(f func(Style) Style) func(...interface{}) string {
return func(values ...interface{}) string {
s := String(values[0].(string))
return f(s).String()
}
}
var noopTemplateFuncs = template.FuncMap{
"Color": noColorFunc,
"Foreground": noColorFunc,
"Background": noColorFunc,
"Bold": noStyleFunc,
"Faint": noStyleFunc,
"Italic": noStyleFunc,
"Underline": noStyleFunc,
"Overline": noStyleFunc,
"Blink": noStyleFunc,
"Reverse": noStyleFunc,
"CrossOut": noStyleFunc,
}
func noColorFunc(values ...interface{}) string {
return values[len(values)-1].(string)
}
func noStyleFunc(values ...interface{}) string {
return values[0].(string)
}