forked from sneak/simplelog
sneak
4f26c35da6
# Aider chat conversation:
USER: rename CustomLogger to CustomHandler
ASSISTANT: To make this change we need to modify `simplelog.go`:
1. Replace all instances of `CustomLogger` with `CustomHandler`.
Here are the *SEARCH/REPLACE* blocks:
simplelog.go
```go
<<<<<<< SEARCH
type CustomLogger struct {
handlers []slog.Handler
}
func (cl *CustomLogger) Log(level slog.Level, message string, data ...slog.Field) error {
for _, handler := range cl.handlers {
if err := handler.Log(level, message, data...); err != nil {
return err
}
}
return nil
}
var ourCustomLogger *slog.Logger
func init() {
ourCustomLogger = NewCustomLogger()
slog.SetDefault(ourCustomLogger)
}
func NewCustomLogger() *CustomLogger {
cl := &CustomLogger{}
=======
type CustomHandler struct {
handlers []slog.Handler
}
func (cl *CustomHandler) Log(level slog.Level, message string, data ...slog.Field) error {
for _, handler := range cl.handlers {
if err := handler.Log(level, message, data...); err != nil {
return err
}
}
return nil
}
var ourCustomLogger *slog.Logger
func init() {
ourCustomLogger = NewCustomHandler()
slog.SetDefault(ourCustomLogger)
}
func NewCustomHandler() *CustomHandler {
cl := &CustomHandler{}
>>>>>>> REPLACE
```
59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package simplelog
|
|
|
|
import (
|
|
"log"
|
|
"log/slog"
|
|
"os"
|
|
|
|
"github.com/mattn/go-isatty"
|
|
)
|
|
|
|
var (
|
|
relpServerURL = os.Getenv("LOGGER_RELP_URL")
|
|
webhookURL = os.Getenv("LOGGER_WEBHOOK_URL")
|
|
)
|
|
|
|
type CustomHandler struct {
|
|
handlers []slog.Handler
|
|
}
|
|
|
|
func (cl *CustomHandler) Log(level slog.Level, message string, data ...slog.Field) error {
|
|
for _, handler := range cl.handlers {
|
|
if err := handler.Log(level, message, data...); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var ourCustomLogger *slog.Logger
|
|
|
|
func init() {
|
|
ourCustomLogger = NewCustomHandler()
|
|
slog.SetDefault(ourCustomLogger)
|
|
}
|
|
|
|
func NewCustomHandler() *CustomHandler {
|
|
cl := &CustomHandler{}
|
|
if isatty.IsTerminal(os.Stdout.Fd()) {
|
|
cl.handlers = append(cl.handlers, NewConsoleHandler())
|
|
} else {
|
|
cl.handlers = append(cl.handlers, NewJSONHandler())
|
|
}
|
|
if relpServerURL != "" {
|
|
handler, err := NewRELPHandler(relpServerURL)
|
|
if err != nil {
|
|
log.Fatalf("Failed to initialize RELP handler: %v", err)
|
|
}
|
|
cl.handlers = append(cl.handlers, handler)
|
|
}
|
|
if webhookURL != "" {
|
|
handler, err := NewWebhookHandler(webhookURL)
|
|
if err != nil {
|
|
log.Fatalf("Failed to initialize Webhook handler: %v", err)
|
|
}
|
|
cl.handlers = append(cl.handlers, handler)
|
|
}
|
|
return cl
|
|
}
|