Implement the Stringer interface for ansi color types

muesli · Aug 30, 2021 · 166cf37 · 166cf37
1 parent 1d2849b
commit 166cf37
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/color.go b/color.go
@@ -26,12 +26,24 @@ type Color interface {
 
 type NoColor struct{}
 
+func (nc NoColor) String() string {
+	return ""
+}
+
 // ANSIColor is a color (0-15) as defined by the ANSI Standard.
 type ANSIColor int
 
+func (a ANSIColor) String() string {
+	return ansiHex[a]
+}
+
 // ANSI256Color is a color (16-255) as defined by the ANSI Standard.
 type ANSI256Color int
 
+func (a ANSI256Color) String() string {
+	return ansiHex[a]
+}
+
 // RGBColor is a hex-encoded color, e.g. "#abcdef".
 type RGBColor string
 

diff --git a/examples/hello-world/main.go b/examples/hello-world/main.go
@@ -37,5 +37,7 @@ func main() {
 		termenv.String("gray").Foreground(p.Color("0")).Background(p.Color("#B9BFCA")),
 	)
 
-	fmt.Printf("\n\t%s %t\n", termenv.String("Has dark background?").Bold(), termenv.HasDarkBackground())
+	fmt.Printf("\n\t%s %s\n", termenv.String("Has foreground color").Bold(), termenv.ForegroundColor())
+	fmt.Printf("\t%s %s\n", termenv.String("Has background color").Bold(), termenv.BackgroundColor())
+	fmt.Printf("\t%s %t\n", termenv.String("Has dark background?").Bold(), termenv.HasDarkBackground())
 }