text_handler.go

package clog

import (
	"log"
	"os"
)

// TextHandler logs messages directly to the console
type TextHandler struct {
	logger *log.Logger
}

// NewTextHandler creates a new handler to send logs to the console
func NewTextHandler(prefix string, flag int) *TextHandler {
	handler := &TextHandler{
		logger: log.New(os.Stdout, prefix, flag),
	}
	return handler
}

// LogEntry sends a log entry with the specified level
func (h *TextHandler) LogEntry(logEntry LogEntry) error {
	return h.logger.Output(logEntry.Calldepth+2, h.levelPrefix(logEntry.Level)+logEntry.GetMessage())
}

// SetPrefix sets a prefix on every log message
func (h *TextHandler) SetPrefix(prefix string) Handler {
	h.logger.SetPrefix(prefix)
	return h
}

func (h *TextHandler) levelPrefix(logLevel LogLevel) string {
	switch logLevel {
	case LevelError:
		return "Error: "
	case LevelWarning:
		return "Warning: "
	default:
		return ""
	}
}

// Verify interface
var (
	_ Handler = &TextHandler{}
)