Compare commits

..

2 Commits

Author SHA1 Message Date
86590435bd Update readme 2024-09-06 08:29:39 -05:00
c93f6e0af0 Change Code to Color, add better terminal support 2024-09-06 08:26:16 -05:00
2 changed files with 45 additions and 18 deletions

View File

@ -1,3 +1,10 @@
# Color
CLI colors, based on ANSI.
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")
```

View File

@ -3,17 +3,15 @@ package color
import (
"fmt"
"os"
"strings"
)
// Whether the terminal supports true color
var TrueColor = os.Getenv("COLORTERM") == "truecolor"
// Color code
type Code int
// Color type
type Color int
// Color codes
const (
Black Code = iota + 30
Black Color = iota + 30
Red
Green
Yellow
@ -23,33 +21,55 @@ const (
White
)
func (code Code) String(args ...any) string {
return fmt.Sprintf("\x1b[%dm%s\x1b[0m", code, fmt.Sprint(args...))
func (c Color) String(args ...any) string {
return fmt.Sprintf("\x1b[%dm%s\x1b[0m", c, fmt.Sprint(args...))
}
func (code Code) Print(args ...any) {
if !TrueColor {
func (c Color) Print(args ...any) {
if !supportsColors() {
fmt.Print(args...)
return
}
fmt.Printf("\x1b[%dm%s\x1b[0m", code, fmt.Sprint(args...))
fmt.Printf("\x1b[%dm%s\x1b[0m", c, fmt.Sprint(args...))
}
func (code Code) Printf(format string, args ...any) {
if !TrueColor {
func (c Color) Printf(format string, args ...any) {
if !supportsColors() {
fmt.Printf(format, args...)
return
}
fmt.Printf("\x1b[%dm%s\x1b[0m", code, fmt.Sprintf(format, args...))
fmt.Printf("\x1b[%dm%s\x1b[0m", c, fmt.Sprintf(format, args...))
}
func (code Code) Println(args ...any) {
if !TrueColor {
func (c Color) Println(args ...any) {
if !supportsColors() {
fmt.Println(args...)
return
}
fmt.Printf("\x1b[%dm%s\x1b[0m\n", code, fmt.Sprint(args...))
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
}