// SPDX-FileCopyrightText: 2023-2025 Steffen Vogel <post@steffenvogel.de>
// SPDX-License-Identifier: Apache-2.0

package tty

import (
	"fmt"
	"io"
	"os"
	"strings"
)

const (
	RunesAlpha        = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
	RunesAlphaNumeric = RunesAlpha + "0123456789"
)

const (
	FgBlack         = "\x1b[30m"
	FgRed           = "\x1b[31m"
	FgGreen         = "\x1b[32m"
	FgYellow        = "\x1b[33m"
	FgBlue          = "\x1b[34m"
	FgMagenta       = "\x1b[35m"
	FgCyan          = "\x1b[36m"
	FgWhite         = "\x1b[37m"
	FgDefault       = "\x1b[39m"
	BgBlack         = "\x1b[40m"
	BgRed           = "\x1b[41m"
	BgGreen         = "\x1b[42m"
	BgYellow        = "\x1b[43m"
	BgBlue          = "\x1b[44m"
	BgMagenta       = "\x1b[45m"
	BgCyan          = "\x1b[46m"
	BgWhite         = "\x1b[47m"
	BgDefault       = "\x1b[49m"
	FgBrightBlack   = "\x1b[90m"
	FgBrightRed     = "\x1b[91m"
	FgBrightGreen   = "\x1b[92m"
	FgBrightYellow  = "\x1b[93m"
	FgBrightBlue    = "\x1b[94m"
	FgBrightMagenta = "\x1b[95m"
	FgBrightCyan    = "\x1b[96m"
	FgBrightWhite   = "\x1b[97m"
	BgBrightBlack   = "\x1b[100m"
	BgBrightRed     = "\x1b[101m"
	BgBrightGreen   = "\x1b[102m"
	BgBrightYellow  = "\x1b[103m"
	BgBrightBlue    = "\x1b[104m"
	BgBrightMagenta = "\x1b[105m"
	BgBrightCyan    = "\x1b[106m"
	BgBrightWhite   = "\x1b[107m"
	Bold            = "\x1b[1m"
	NoBold          = "\x1b[22m"
	Underline       = "\x1b[4m"
	NoUnderline     = "\x1b[24m"
	Reset           = "\x1b[0m"
)

func TrueColor(r, g, b byte) string {
	return fmt.Sprintf("\x1b[38;2;%d;%d;%dm", r, g, b)
}

func Color256(b byte) string {
	return fmt.Sprintf("\x1b[38;5;%dm", b)
}

func Mods(str string, mods ...string) string {
	return strings.Join(mods, "") + str + Reset
}

func FprintKV(wr io.Writer, k string, v ...any) (int, error) {
	switch {
	case len(v) == 0:
		return fmt.Fprintf(wr, Mods("%s", Bold)+":\n", k)
	case len(v) == 1:
		return fmt.Fprintf(wr, Mods("%s", Bold)+": %v\n", k, v[0])
	case len(v) > 1:
		return fmt.Fprintf(wr, Mods("%s", Bold)+": %v\n", k, v)
	default:
		return 0, nil
	}
}

func IsATTY(f *os.File) bool {
	fi, err := f.Stat()
	if err != nil {
		panic(fmt.Errorf("failed to stat stdout: %w", err))
	}

	return (fi.Mode() & os.ModeCharDevice) != 0
}