README.md

@@ -1,10 +1,3 @@
 # Color
 
-Supported colors: black, white, red, green, blue, yellow, magenta, cyan
-
-```go
-color.Blue.Print("Blue")
-color.Blue.Printf("%s", "Blue")
-color.Blue.Println("Blue")
-blue := color.Blue.String("Blue")
-```
+CLI colors, based on ANSI.

color.go

@@ -3,15 +3,17 @@ package color
 import (
 	"fmt"
 	"os"
-	"strings"
 )
 
-// Color type
-type Color int
+// Whether the terminal supports true color
+var TrueColor = os.Getenv("COLORTERM") == "truecolor"
+
+// Color code
+type Code int
 
 // Color codes
 const (
-	Black Color = iota + 30
+	Black Code = iota + 30
 	Red
 	Green
 	Yellow
@@ -21,55 +23,33 @@ const (
 	White
 )
 
-func (c Color) String(args ...any) string {
-	return fmt.Sprintf("\x1b[%dm%s\x1b[0m", c, fmt.Sprint(args...))
+func (code Code) String(args ...any) string {
+	return fmt.Sprintf("\x1b[%dm%s\x1b[0m", code, fmt.Sprint(args...))
 }
 
-func (c Color) Print(args ...any) {
-	if !supportsColors() {
+func (code Code) Print(args ...any) {
+	if !TrueColor {
 		fmt.Print(args...)
 		return
 	}
 
-	fmt.Printf("\x1b[%dm%s\x1b[0m", c, fmt.Sprint(args...))
+	fmt.Printf("\x1b[%dm%s\x1b[0m", code, fmt.Sprint(args...))
 }
 
-func (c Color) Printf(format string, args ...any) {
-	if !supportsColors() {
+func (code Code) Printf(format string, args ...any) {
+	if !TrueColor {
 		fmt.Printf(format, args...)
 		return
 	}
 
-	fmt.Printf("\x1b[%dm%s\x1b[0m", c, fmt.Sprintf(format, args...))
+	fmt.Printf("\x1b[%dm%s\x1b[0m", code, fmt.Sprintf(format, args...))
 }
 
-func (c Color) Println(args ...any) {
-	if !supportsColors() {
+func (code Code) Println(args ...any) {
+	if !TrueColor {
 		fmt.Println(args...)
 		return
 	}
 
-	fmt.Printf("\x1b[%dm%s\x1b[0m\n", c, fmt.Sprint(args...))
-}
-
-func supportsColors() bool {
-	colorterm := os.Getenv("COLORTERM")
-	term := os.Getenv("TERM")
-
-	// Check for true color (24-bit support)
-	if strings.Contains(colorterm, "truecolor") || strings.Contains(colorterm, "24bit") {
-		return true
-	}
-
-	// Check for 256 colors support
-	if strings.Contains(term, "256color") {
-		return true
-	}
-
-	// Check for basic color support
-	if term == "xterm" || term == "screen" || term == "vt100" || strings.Contains(term, "color") {
-		return true
-	}
-
-	return false
+	fmt.Printf("\x1b[%dm%s\x1b[0m\n", code, fmt.Sprint(args...))
 }