forked from elves/elvish
-
Notifications
You must be signed in to change notification settings - Fork 0
/
transform.go
73 lines (69 loc) · 1.86 KB
/
transform.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
package styled
import (
"strings"
)
// Transform transforms a Text according to a transformer. It does nothing if
// the transformer is not valid.
func Transform(t Text, transformer string) Text {
f := FindTransformer(transformer)
if f == nil {
return t
}
t = t.Clone()
for _, seg := range t {
f(seg)
}
return t
}
// FindTransformer finds the named transformer, a function that mutates a
// *Segment. If the name is not a valid transformer, it returns nil.
func FindTransformer(name string) func(*Segment) {
switch {
// Catch special colors early
case name == "default":
return func(s *Segment) { s.Foreground = "" }
case name == "bg-default":
return func(s *Segment) { s.Background = "" }
case strings.HasPrefix(name, "bg-"):
if color := name[len("bg-"):]; isValidColorName(color) {
return func(s *Segment) { s.Background = color }
}
case strings.HasPrefix(name, "no-"):
if f := boolFieldAccessor(name[len("no-"):]); f != nil {
return func(s *Segment) { *f(s) = false }
}
case strings.HasPrefix(name, "toggle-"):
if f := boolFieldAccessor(name[len("toggle-"):]); f != nil {
return func(s *Segment) {
p := f(s)
*p = !*p
}
}
default:
if isValidColorName(name) {
return func(s *Segment) { s.Foreground = name }
}
if f := boolFieldAccessor(name); f != nil {
return func(s *Segment) { *f(s) = true }
}
}
return nil
}
func boolFieldAccessor(name string) func(*Segment) *bool {
switch name {
case "bold":
return func(s *Segment) *bool { return &s.Bold }
case "dim":
return func(s *Segment) *bool { return &s.Dim }
case "italic":
return func(s *Segment) *bool { return &s.Italic }
case "underlined":
return func(s *Segment) *bool { return &s.Underlined }
case "blink":
return func(s *Segment) *bool { return &s.Blink }
case "inverse":
return func(s *Segment) *bool { return &s.Inverse }
default:
return nil
}
}