package core

import (
	"github.com/mattn/go-isatty"
	"os"
)

// https://misc.flogisoft.com/bash/tip_colors_and_formatting
var (
	BOLD = "\033[1m"
	DIM  = "\033[2m"

	RED    = "\033[31m"
	GREEN  = "\033[32m"
	BLUE   = "\033[34m"
	YELLOW = "\033[33m"

	FG_BLACK = "\033[30m"
	FG_WHITE = "\033[97m"

	BG_DGRAY  = "\033[100m"
	BG_RED    = "\033[41m"
	BG_GREEN  = "\033[42m"
	BG_YELLOW = "\033[43m"
	BG_LBLUE  = "\033[104m"

	RESET = "\033[0m"

	NoColors = false
)

func init() {
	NoColors = os.Getenv("TERM") == "dumb" ||
		os.Getenv("TERM") == "" ||
		(!isatty.IsTerminal(os.Stdout.Fd()) && !isatty.IsCygwinTerminal(os.Stdout.Fd()))
	if NoColors {
		BOLD = ""
		DIM = ""
		RED = ""
		GREEN = ""
		BLUE = ""
		YELLOW = ""
		FG_BLACK = ""
		FG_WHITE = ""
		BG_DGRAY = ""
		BG_RED = ""
		BG_GREEN = ""
		BG_YELLOW = ""
		BG_LBLUE = ""
		RESET = ""
	}
}

const (
	DEBUG = iota
	INFO
	IMPORTANT
	WARNING
	ERROR
	FATAL
)

var (
	LogLabels = map[int]string{
		DEBUG:     "dbg",
		INFO:      "inf",
		IMPORTANT: "imp",
		WARNING:   "war",
		ERROR:     "err",
		FATAL:     "!!!",
	}
	LogColors = map[int]string{
		DEBUG:     DIM + FG_BLACK + BG_DGRAY,
		INFO:      FG_WHITE + BG_GREEN,
		IMPORTANT: FG_WHITE + BG_LBLUE,
		WARNING:   FG_WHITE + BG_YELLOW,
		ERROR:     FG_WHITE + BG_RED,
		FATAL:     FG_WHITE + BG_RED + BOLD,
	}
)

// W for Wrap
func W(e, s string) string {
	return e + s + RESET
}

func Bold(s string) string {
	return W(BOLD, s)
}

func Dim(s string) string {
	return W(DIM, s)
}

func Red(s string) string {
	return W(RED, s)
}

func Green(s string) string {
	return W(GREEN, s)
}

func Blue(s string) string {
	return W(BLUE, s)
}

func Yellow(s string) string {
	return W(YELLOW, s)
}