package color

import (
	"fmt"
	"os"
)

// Whether the terminal supports true color
var TrueColor = os.Getenv("COLORTERM") == "truecolor"

// Color code
type Code int

// Color codes
const (
	Black Code = iota + 30
	Red
	Green
	Yellow
	Blue
	Magenta
	Cyan
	White
)

func (code Code) String(args ...any) string {
	return fmt.Sprintf("\x1b[%dm%s\x1b[0m", code, fmt.Sprint(args...))
}

func (code Code) Print(args ...any) {
	if !TrueColor {
		fmt.Print(args...)
		return
	}

	fmt.Printf("\x1b[%dm%s\x1b[0m", code, fmt.Sprint(args...))
}

func (code Code) Printf(format string, args ...any) {
	if !TrueColor {
		fmt.Printf(format, args...)
		return
	}

	fmt.Printf("\x1b[%dm%s\x1b[0m", code, fmt.Sprintf(format, args...))
}

func (code Code) Println(args ...any) {
	if !TrueColor {
		fmt.Println(args...)
		return
	}

	fmt.Printf("\x1b[%dm%s\x1b[0m\n", code, fmt.Sprint(args...))
}