gostyle/style.go at main · arafath-mk/gostyle · 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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package gcstyle

import (
	"fmt"

	"github.com/arafath-mk/gostyle/wcolor"
)

type Style struct {
	Color         *wcolor.Color
	Background    *wcolor.Color
	Bold          bool
	Italic        bool
	Underline     bool
	Strikethrough bool
	Darken        bool
	Lighten       bool
}

func (s Style) ApplyTo(str string, applyStyle bool) StyledString {
	styled := fmt.Sprintf("%s%s%s", s.Start(applyStyle), str, s.End(applyStyle))
	return StyledString{
		str:        styled,
		applyStyle: applyStyle,
	}
}

func (s Style) Start(applyStyle bool) string {
	if !applyStyle {
		return ""
	}

	var str string
	if s.Color != nil {
		str = fmt.Sprintf("%s\033[38;2;%d;%d;%dm", str, s.Color.R, s.Color.G, s.Color.B)
	}
	if s.Background != nil {
		str = fmt.Sprintf("%s\033[48;2;%d;%d;%dm", str, s.Color.R, s.Color.G, s.Color.B)
	}
	if s.Bold {
		str = fmt.Sprintf("%s\033[1m", str)
	}
	if s.Italic {
		str = fmt.Sprintf("%s\033[3m", str)
	}
	if s.Underline {
		str = fmt.Sprintf("%s\033[4m", str)
	}
	if s.Strikethrough {
		str = fmt.Sprintf("%s\033[9m", str)
	}
	if s.Darken {
		str = fmt.Sprintf("%s\033[2m", str)
	}
	if s.Lighten {
		str = fmt.Sprintf("%s\033[1m", str)
	}
	return str
}

func (s Style) End(applyStyle bool) string {
	if !applyStyle {
		return ""
	}

	var str string
	if s.Lighten {
		str = fmt.Sprintf("%s\033[0m", str)
	}
	if s.Darken {
		str = fmt.Sprintf("%s\033[0m", str)
	}
	if s.Strikethrough {
		str = fmt.Sprintf("%s\033[0m", str)
	}
	if s.Underline {
		str = fmt.Sprintf("%s\033[0m", str)
	}
	if s.Italic {
		str = fmt.Sprintf("%s\033[0m", str)
	}
	if s.Bold {
		str = fmt.Sprintf("%s\033[22m", str)
	}
	if s.Background != nil {
		str = fmt.Sprintf("%s\033[m", str)
	}
	if s.Color != nil {
		str = fmt.Sprintf("%s\033[m", str)
	}
	return str
}