81f03c7d07
Customization of the original library will continue from here to make it leaner and more suited for general use in MimirTech projects.
110 lines
2.5 KiB
Go
110 lines
2.5 KiB
Go
// +build !windows
|
|
|
|
// Copyright 2013, Örjan Persson. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package logging
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
)
|
|
|
|
type color int
|
|
|
|
const (
|
|
ColorBlack = iota + 30
|
|
ColorRed
|
|
ColorGreen
|
|
ColorYellow
|
|
ColorBlue
|
|
ColorMagenta
|
|
ColorCyan
|
|
ColorWhite
|
|
)
|
|
|
|
var (
|
|
colors = []string{
|
|
CRITICAL: ColorSeq(ColorMagenta),
|
|
ERROR: ColorSeq(ColorRed),
|
|
WARNING: ColorSeq(ColorYellow),
|
|
NOTICE: ColorSeq(ColorGreen),
|
|
DEBUG: ColorSeq(ColorCyan),
|
|
}
|
|
boldcolors = []string{
|
|
CRITICAL: ColorSeqBold(ColorMagenta),
|
|
ERROR: ColorSeqBold(ColorRed),
|
|
WARNING: ColorSeqBold(ColorYellow),
|
|
NOTICE: ColorSeqBold(ColorGreen),
|
|
DEBUG: ColorSeqBold(ColorCyan),
|
|
}
|
|
)
|
|
|
|
// LogBackend utilizes the standard log module.
|
|
type LogBackend struct {
|
|
Logger *log.Logger
|
|
Color bool
|
|
ColorConfig []string
|
|
}
|
|
|
|
// NewLogBackend creates a new LogBackend.
|
|
func NewLogBackend(out io.Writer, prefix string, flag int) *LogBackend {
|
|
return &LogBackend{Logger: log.New(out, prefix, flag)}
|
|
}
|
|
|
|
// Log implements the Backend interface.
|
|
func (b *LogBackend) Log(level Level, calldepth int, rec *Record) error {
|
|
if b.Color {
|
|
col := colors[level]
|
|
if len(b.ColorConfig) > int(level) && b.ColorConfig[level] != "" {
|
|
col = b.ColorConfig[level]
|
|
}
|
|
|
|
buf := &bytes.Buffer{}
|
|
buf.Write([]byte(col))
|
|
buf.Write([]byte(rec.Formatted(calldepth + 1)))
|
|
buf.Write([]byte("\033[0m"))
|
|
// For some reason, the Go logger arbitrarily decided "2" was the correct
|
|
// call depth...
|
|
return b.Logger.Output(calldepth+2, buf.String())
|
|
}
|
|
|
|
return b.Logger.Output(calldepth+2, rec.Formatted(calldepth+1))
|
|
}
|
|
|
|
// ConvertColors takes a list of ints representing colors for log levels and
|
|
// converts them into strings for ANSI color formatting
|
|
func ConvertColors(colors []int, bold bool) []string {
|
|
converted := []string{}
|
|
for _, i := range colors {
|
|
if bold {
|
|
converted = append(converted, ColorSeqBold(color(i)))
|
|
} else {
|
|
converted = append(converted, ColorSeq(color(i)))
|
|
}
|
|
}
|
|
|
|
return converted
|
|
}
|
|
|
|
func ColorSeq(color color) string {
|
|
return fmt.Sprintf("\033[%dm", int(color))
|
|
}
|
|
|
|
func ColorSeqBold(color color) string {
|
|
return fmt.Sprintf("\033[%d;1m", int(color))
|
|
}
|
|
|
|
func doFmtVerbLevelColor(layout string, level Level, output io.Writer) {
|
|
if layout == "bold" {
|
|
output.Write([]byte(boldcolors[level]))
|
|
} else if layout == "reset" {
|
|
output.Write([]byte("\033[0m"))
|
|
} else {
|
|
output.Write([]byte(colors[level]))
|
|
}
|
|
}
|