refactor: deduplicate into log() helper

refactor: deduplicate into log() helper
This commit is contained in:
cubexteam
2026-03-04 18:32:21 +03:00
committed by GitHub
parent c9b068f9f1
commit 8e2d4f7b8c

View File

@@ -2,6 +2,7 @@ package logger
import ( import (
"fmt" "fmt"
"io"
"os" "os"
"time" "time"
) )
@@ -17,40 +18,38 @@ const (
type Logger struct { type Logger struct {
debug bool debug bool
out io.Writer
err io.Writer
} }
func New(debug bool) *Logger { func New(debug bool) *Logger {
return &Logger{debug: debug} return &Logger{debug: debug, out: os.Stdout, err: os.Stderr}
} }
func (l *Logger) timestamp() string { func (l *Logger) timestamp() string {
return colorGray + time.Now().Format("15:04:05") + colorReset return colorGray + time.Now().Format("15:04:05") + colorReset
} }
func (l *Logger) Infof(format string, args ...any) { func (l *Logger) log(w io.Writer, level, color, format string, args ...any) {
fmt.Fprintf(os.Stdout, "%s %sINFO%s %s\n", l.timestamp(), colorCyan, colorReset, fmt.Sprintf(format, args...)) msg := fmt.Sprintf(format, args...)
fmt.Fprintf(w, "%s %s%s%s %s\n", l.timestamp(), color, level, colorReset, msg)
} }
func (l *Logger) Warnf(format string, args ...any) { func (l *Logger) Infof(format string, args ...any) { l.log(l.out, "INFO ", colorCyan, format, args...) }
fmt.Fprintf(os.Stdout, "%s %sWARN%s %s\n", l.timestamp(), colorYellow, colorReset, fmt.Sprintf(format, args...)) func (l *Logger) Warnf(format string, args ...any) { l.log(l.out, "WARN ", colorYellow, format, args...) }
} func (l *Logger) Errorf(format string, args ...any) { l.log(l.err, "ERROR", colorRed, format, args...) }
func (l *Logger) Errorf(format string, args ...any) {
fmt.Fprintf(os.Stderr, "%s %sERROR%s %s\n", l.timestamp(), colorRed, colorReset, fmt.Sprintf(format, args...))
}
func (l *Logger) Debugf(format string, args ...any) { func (l *Logger) Debugf(format string, args ...any) {
if !l.debug { if l.debug {
return l.log(l.out, "DEBUG", colorGray, format, args...)
} }
fmt.Fprintf(os.Stdout, "%s %sDEBUG%s %s\n", l.timestamp(), colorGray, colorReset, fmt.Sprintf(format, args...))
} }
func (l *Logger) Fatalf(format string, args ...any) { func (l *Logger) Fatalf(format string, args ...any) {
fmt.Fprintf(os.Stderr, "%s %sFATAL%s %s\n", l.timestamp(), colorRed, colorReset, fmt.Sprintf(format, args...)) l.log(l.err, "FATAL", colorRed, format, args...)
os.Exit(1) os.Exit(1)
} }
func (l *Logger) Chat(name, message string) { func (l *Logger) Chat(name, message string) {
fmt.Fprintf(os.Stdout, "%s %sCHAT%s <%s> %s\n", l.timestamp(), colorWhite, colorReset, name, message) fmt.Fprintf(l.out, "%s %sCHAT %s<%s> %s\n", l.timestamp(), colorWhite, colorReset, name, message)
} }