levellogger/levellogger.go

package levellogger

import (
	"io"
	"log"

	"github.com/go-errors/errors"
)

var traceable = errors.Errorf("Stack trace request")

func Crash() error {
	return errors.New(traceable)
}

func NewLevelLogger(out io.Writer, level LogLevel) LevelLogger {
	var ll LevelLogger

	ll.CurrentLevel = level

	ll.panicLog = log.New(out, "[PANIC]\t", log.Ldate|log.Ltime|log.Llongfile)
	ll.traceLog = log.New(out, "[TRACE]\t", log.Ldate|log.Ltime|log.Llongfile)
	ll.debugLog = log.New(out, "[DEBUG]\t", log.Ldate|log.Ltime|log.Llongfile)
	ll.infoLog = log.New(out, "[INFO]\t", log.Ldate|log.Ltime)
	ll.warnLog = log.New(out, "[WARN]\t", log.Ldate|log.Ltime)
	ll.errorLog = log.New(out, "[ERROR]\t", log.Ldate|log.Ltime|log.Llongfile)
	ll.fatalLog = log.New(out, "[FATAL]\t", log.Ldate|log.Ltime|log.Llongfile)

	return ll
}

// It's a log -- line output
// so only use Println versions
func (ll LevelLogger) Panic(v ...any) {
	ll.panicLog.Panicln(v...)
}

func (ll LevelLogger) Panicf(format string, v ...any) {
	ll.panicLog.Panicf(format, v...)
}

func (ll LevelLogger) Trace(v ...any) {
	err := Crash()
	ll.traceLog.Println(err.(*errors.Error).ErrorStack())
	ll.traceLog.Println(v...)
}

func (ll LevelLogger) Tracef(format string, v ...any) {
	err := Crash()
	ll.traceLog.Println(err.(*errors.Error).ErrorStack())
	ll.traceLog.Printf(format, v...)
}

func (ll LevelLogger) Debug(v ...any) {
	ll.debugLog.Println(v...)
}

func (ll LevelLogger) Debugf(format string, v ...any) {
	ll.debugLog.Printf(format, v...)
}

func (ll LevelLogger) Info(v ...any) {
	ll.infoLog.Println(v...)
}

func (ll LevelLogger) Infof(format string, v ...any) {
	ll.infoLog.Printf(format, v...)
}

func (ll LevelLogger) Warn(v ...any) {
	ll.warnLog.Println(v...)
}

func (ll LevelLogger) Warnf(format string, v ...any) {
	ll.warnLog.Printf(format, v...)
}

func (ll LevelLogger) Error(v ...any) {
	ll.errorLog.Println(v...)
}

func (ll LevelLogger) Errorf(format string, v ...any) {
	ll.errorLog.Printf(format, v...)
}

func (ll LevelLogger) Fatal(v ...any) {
	ll.fatalLog.Fatalln(v...)
}

func (ll LevelLogger) Fatalf(format string, v ...any) {
	ll.fatalLog.Fatalf(format, v...)
}

// spelling: Panic, Trace, Debug, Info, Warn, Error, Fatal
func (ll LevelLogger) SetPrefix(logname string, prefix string) {
	switch {
	case logname == "Panic":
		ll.panicLog.SetPrefix(prefix)
	case logname == "Trace":
		ll.traceLog.SetPrefix(prefix)
	case logname == "Debug":
		ll.debugLog.SetPrefix(prefix)
	case logname == "Info":
		ll.infoLog.SetPrefix(prefix)
	case logname == "Warn":
		ll.warnLog.SetPrefix(prefix)
	case logname == "Error":
		ll.errorLog.SetPrefix(prefix)
	case logname == "Fatal":
		ll.fatalLog.SetPrefix(prefix)
	}
}

func (ll LevelLogger) SetOutput(logname string, newout io.Writer) {
	switch {
	case logname == "Panic":
		ll.panicLog.SetOutput(newout)
	case logname == "Trace":
		ll.traceLog.SetOutput(newout)
	case logname == "Debug":
		ll.debugLog.SetOutput(newout)
	case logname == "Info":
		ll.infoLog.SetOutput(newout)
	case logname == "Warn":
		ll.warnLog.SetOutput(newout)
	case logname == "Error":
		ll.errorLog.SetOutput(newout)
	case logname == "Fatal":
		ll.fatalLog.SetOutput(newout)
	}
}

func (ll LevelLogger) SetFlags(logname string, newFlags int) {
	switch {
	case logname == "Panic":
		ll.panicLog.SetFlags(newFlags)
	case logname == "Trace":
		ll.traceLog.SetFlags(newFlags)
	case logname == "Debug":
		ll.debugLog.SetFlags(newFlags)
	case logname == "Info":
		ll.infoLog.SetFlags(newFlags)
	case logname == "Warn":
		ll.warnLog.SetFlags(newFlags)
	case logname == "Error":
		ll.errorLog.SetFlags(newFlags)
	case logname == "Fatal":
		ll.fatalLog.SetFlags(newFlags)
	}
}