forked from gitea/gitea
117 lines
2.0 KiB
Go
117 lines
2.0 KiB
Go
// Copyright 2019 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package log
|
|
|
|
import (
|
|
"bytes"
|
|
"strings"
|
|
|
|
"code.gitea.io/gitea/modules/json"
|
|
)
|
|
|
|
// Level is the level of the logger
|
|
type Level int
|
|
|
|
const (
|
|
UNDEFINED Level = iota
|
|
TRACE
|
|
DEBUG
|
|
INFO
|
|
WARN
|
|
ERROR
|
|
FATAL
|
|
NONE
|
|
)
|
|
|
|
const CRITICAL = ERROR // most logger frameworks doesn't support CRITICAL, and it doesn't seem useful
|
|
|
|
var toString = map[Level]string{
|
|
UNDEFINED: "undefined",
|
|
|
|
TRACE: "trace",
|
|
DEBUG: "debug",
|
|
INFO: "info",
|
|
WARN: "warn",
|
|
ERROR: "error",
|
|
|
|
FATAL: "fatal",
|
|
NONE: "none",
|
|
}
|
|
|
|
var toLevel = map[string]Level{
|
|
"undefined": UNDEFINED,
|
|
|
|
"trace": TRACE,
|
|
"debug": DEBUG,
|
|
"info": INFO,
|
|
"warn": WARN,
|
|
"warning": WARN,
|
|
"error": ERROR,
|
|
|
|
"fatal": FATAL,
|
|
"none": NONE,
|
|
}
|
|
|
|
var levelToColor = map[Level][]ColorAttribute{
|
|
TRACE: {Bold, FgCyan},
|
|
DEBUG: {Bold, FgBlue},
|
|
INFO: {Bold, FgGreen},
|
|
WARN: {Bold, FgYellow},
|
|
ERROR: {Bold, FgRed},
|
|
FATAL: {Bold, BgRed},
|
|
NONE: {Reset},
|
|
}
|
|
|
|
func (l Level) String() string {
|
|
s, ok := toString[l]
|
|
if ok {
|
|
return s
|
|
}
|
|
return "info"
|
|
}
|
|
|
|
func (l Level) ColorAttributes() []ColorAttribute {
|
|
color, ok := levelToColor[l]
|
|
if ok {
|
|
return color
|
|
}
|
|
none := levelToColor[NONE]
|
|
return none
|
|
}
|
|
|
|
// MarshalJSON takes a Level and turns it into text
|
|
func (l Level) MarshalJSON() ([]byte, error) {
|
|
buffer := bytes.NewBufferString(`"`)
|
|
buffer.WriteString(toString[l])
|
|
buffer.WriteString(`"`)
|
|
return buffer.Bytes(), nil
|
|
}
|
|
|
|
// UnmarshalJSON takes text and turns it into a Level
|
|
func (l *Level) UnmarshalJSON(b []byte) error {
|
|
var tmp any
|
|
err := json.Unmarshal(b, &tmp)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
switch v := tmp.(type) {
|
|
case string:
|
|
*l = LevelFromString(v)
|
|
case int:
|
|
*l = LevelFromString(Level(v).String())
|
|
default:
|
|
*l = INFO
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// LevelFromString takes a level string and returns a Level
|
|
func LevelFromString(level string) Level {
|
|
if l, ok := toLevel[strings.ToLower(level)]; ok {
|
|
return l
|
|
}
|
|
return INFO
|
|
}
|