1 files changed, 92 insertions, 0 deletions
diff --git a/logger.go b/logger.go
new file mode 100644
index 0000000..1fcbfeb
--- /dev/null
+++ b/logger.go
@@ -0,0 +1,92 @@
+package main
+
+import (
+	"fmt"
+	"log"
+)
+
+type LevelEnum uint8
+
+const (
+	None LevelEnum = iota
+	Error
+	Warn
+	Info
+	Debug
+)
+
+type levelInfo struct {
+	abbrev string
+	color  string
+}
+
+var (
+	levelInfos = map[LevelEnum]levelInfo{
+		Error: {"ERR", ansiRed},
+		Warn:  {"WARN", ansiYellow},
+		Info:  {"INFO", ansiCyan},
+		Debug: {"DEBUG", ansiGreen},
+	}
+)
+
+func (s LevelEnum) Info() levelInfo {
+	info, ok := levelInfos[s]
+	if !ok {
+		return levelInfo{"UNK", ansiWhite}
+	}
+	return info
+}
+
+func (s LevelEnum) Format() string {
+	info := s.Info()
+	return colorize("["+info.abbrev+"]", info.color)
+}
+
+type Logger struct {
+	Level LevelEnum
+
+	ns string
+}
+
+func NewLogger(level LevelEnum) *Logger {
+	return &Logger{
+		Level: level,
+	}
+}
+
+func (l *Logger) WithNS(ns string) *Logger {
+	return &Logger{
+		Level: l.Level,
+		ns:    formatNS(ns),
+	}
+}
+
+func (l *Logger) Logf(level LevelEnum, format string, args ...any) {
+	if level > l.Level {
+		return
+	}
+	log.Printf("%s%s %s", l.ns, level.Format(), fmt.Sprintf(format, args...))
+}
+
+func (l *Logger) Errorf(format string, args ...any) { l.Logf(Error, format, args...) }
+func (l *Logger) Warnf(format string, args ...any)  { l.Logf(Warn, format, args...) }
+func (l *Logger) Infof(format string, args ...any)  { l.Logf(Info, format, args...) }
+func (l *Logger) Debugf(format string, args ...any) { l.Logf(Debug, format, args...) }
+
+func formatNS(ns string) string {
+	return colorize("["+ns+"]", ansiPurple)
+}
+
+var ansiReset = "\033[0m"
+var ansiRed = "\033[31m"
+var ansiGreen = "\033[32m"
+var ansiYellow = "\033[33m"
+var ansiBlue = "\033[34m"
+var ansiPurple = "\033[35m"
+var ansiCyan = "\033[36m"
+var ansiGray = "\033[37m"
+var ansiWhite = "\033[97m"
+
+func colorize(s string, color string) string {
+	return color + s + ansiReset
+}