chore: vendor

vendor/github.com/muesli/termenv/templatehelper.go

@@ -0,0 +1,86 @@
+package termenv
+
+import (
+	"text/template"
+)
+
+// TemplateFuncs returns template helpers for the given output.
+func (o Output) TemplateFuncs() template.FuncMap {
+	return TemplateFuncs(o.Profile)
+}
+
+// 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 := p.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 := p.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 := p.String(values[len(values)-1].(string))
+			if len(values) == 2 {
+				s = s.Background(p.Color(values[0].(string)))
+			}
+
+			return s.String()
+		},
+		"Bold":      styleFunc(p, Style.Bold),
+		"Faint":     styleFunc(p, Style.Faint),
+		"Italic":    styleFunc(p, Style.Italic),
+		"Underline": styleFunc(p, Style.Underline),
+		"Overline":  styleFunc(p, Style.Overline),
+		"Blink":     styleFunc(p, Style.Blink),
+		"Reverse":   styleFunc(p, Style.Reverse),
+		"CrossOut":  styleFunc(p, Style.CrossOut),
+	}
+}
+
+func styleFunc(p Profile, f func(Style) Style) func(...interface{}) string {
+	return func(values ...interface{}) string {
+		s := p.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)
+}