package debug

import (
	"fmt"
	"time"

	"github.com/mattn/go-colorable"
	zerolog "github.com/rs/zerolog"
)

var colors = map[string]string{
	"text":  "\x1b[38;5;6m%s\x1b[0m",
	"debug": "\x1b[32mDEBUG\x1b[0m",
	"gray":  "\x1b[38;5;8m%s\x1b[0m",
	"info":  "\x1b[38;5;111mINFO\x1b[0m",
	"error": "\x1b[38;5;204mERROR\x1b[0m",
	"fatal": "\x1b[38;5;52mFATAL\x1b[0m",
}

var output = zerolog.ConsoleWriter{
	Out:        colorable.NewColorableStdout(),
	TimeFormat: time.ANSIC,
	FormatLevel: func(i interface{}) string {
		name := fmt.Sprintf("%s", i)
		coloredName := colors[name]
		return coloredName
	},
	FormatMessage: func(i interface{}) string {
		coloredMsg := fmt.Sprintf(colors["text"], i)
		return coloredMsg
	},
	FormatFieldName: func(i interface{}) string {
		name := fmt.Sprintf("%s", i)
		return fmt.Sprintf(colors["gray"], name+"=")
	},
	FormatFieldValue: func(i interface{}) string {
		return fmt.Sprintf("%s", i)
	},
	NoColor: false,
}

func NewLogger() zerolog.Logger {
	return zerolog.New(output).With().Timestamp().Logger()
}