mirror of
https://github.com/amir20/dozzle.git
synced 2025-12-21 13:23:07 +01:00
130 lines
3.2 KiB
Go
130 lines
3.2 KiB
Go
package container
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
orderedmap "github.com/wk8/go-ordered-map/v2"
|
|
)
|
|
|
|
var SupportedLogLevels map[string]struct{}
|
|
|
|
// Changing this also needs to change the logContext.ts file
|
|
var logLevels = [][]string{
|
|
{"error", "err"},
|
|
{"warn", "warning"},
|
|
{"info", "inf"},
|
|
{"debug", "dbg"},
|
|
{"trace"},
|
|
{"fatal", "sev", "severe", "crit", "critical"},
|
|
}
|
|
|
|
var plainLevels = map[string][]*regexp.Regexp{}
|
|
var bracketLevels = map[string][]*regexp.Regexp{}
|
|
var timestampRegex = regexp.MustCompile(`^(?:\d{4}[-/]\d{2}[-/]\d{2}(?:[T ](?:\d{2}:\d{2}:\d{2}(?:\.\d+)?Z?|\d{2}:\d{2}(?:AM|PM)))?\s+)`)
|
|
|
|
func init() {
|
|
for _, levelGroup := range logLevels {
|
|
first := levelGroup[0]
|
|
for _, level := range levelGroup {
|
|
plainLevels[first] = append(plainLevels[first], regexp.MustCompile("(?i)^"+level+"[^a-z]"))
|
|
}
|
|
}
|
|
|
|
for _, levelGroup := range logLevels {
|
|
first := levelGroup[0]
|
|
for _, level := range levelGroup {
|
|
bracketLevels[first] = append(bracketLevels[first], regexp.MustCompile("(?i)\\[ ?"+level+" ?\\]"))
|
|
}
|
|
}
|
|
|
|
SupportedLogLevels = make(map[string]struct{}, len(logLevels)+1)
|
|
for _, levelGroup := range logLevels {
|
|
SupportedLogLevels[levelGroup[0]] = struct{}{}
|
|
}
|
|
SupportedLogLevels["unknown"] = struct{}{}
|
|
}
|
|
|
|
func guessLogLevel(logEvent *LogEvent) string {
|
|
switch value := logEvent.Message.(type) {
|
|
case string:
|
|
value = stripANSI(value)
|
|
value = timestampRegex.ReplaceAllString(value, "")
|
|
for _, levelGroup := range logLevels {
|
|
first := levelGroup[0]
|
|
// Look for the level at the beginning of the message
|
|
for _, regex := range plainLevels[first] {
|
|
if regex.MatchString(value) {
|
|
return first
|
|
}
|
|
}
|
|
|
|
// Look for the level in brackets
|
|
for _, regex := range bracketLevels[first] {
|
|
if regex.MatchString(value) {
|
|
return first
|
|
}
|
|
}
|
|
|
|
// Look for the level in the middle of the message that are uppercase and surrounded by quotes
|
|
if strings.Contains(value, "\""+strings.ToUpper(first)+"\"") {
|
|
return first
|
|
}
|
|
|
|
// Look for the level in the middle of the message that are uppercase
|
|
if strings.Contains(value, " "+strings.ToUpper(first)+" ") {
|
|
return first
|
|
}
|
|
}
|
|
|
|
return "unknown"
|
|
|
|
case *orderedmap.OrderedMap[string, any]:
|
|
if value == nil {
|
|
return "unknown"
|
|
}
|
|
|
|
if level, ok := value.Get("level"); ok {
|
|
if level, ok := level.(string); ok {
|
|
return normalizeLogLevel(level)
|
|
}
|
|
} else if severity, ok := value.Get("severity"); ok {
|
|
if severity, ok := severity.(string); ok {
|
|
return normalizeLogLevel(severity)
|
|
}
|
|
}
|
|
|
|
case *orderedmap.OrderedMap[string, string]:
|
|
if value == nil {
|
|
return "unknown"
|
|
}
|
|
if level, ok := value.Get("level"); ok {
|
|
return normalizeLogLevel(level)
|
|
} else if severity, ok := value.Get("severity"); ok {
|
|
return normalizeLogLevel(severity)
|
|
}
|
|
|
|
case map[string]interface{}:
|
|
panic("not implemented")
|
|
|
|
case map[string]string:
|
|
panic("not implemented")
|
|
|
|
default:
|
|
log.Debug().Type("type", value).Msg("unknown logEvent type")
|
|
}
|
|
|
|
return "unknown"
|
|
}
|
|
|
|
func normalizeLogLevel(level string) string {
|
|
level = stripANSI(level)
|
|
level = strings.ToLower(level)
|
|
if _, ok := SupportedLogLevels[level]; ok {
|
|
return level
|
|
}
|
|
|
|
return "unknown"
|
|
}
|