forked from sneak/simplelog
sneak
24c6ebe407
# Aider chat conversation:
USER: the init() function in `custom_logger.go` is missing, on package initialization it should replace the log/slog default logger.
ASSISTANT: To add the `init()` function in `custom_logger.go` that replaces the default logger in the `log/slog` package, we need to modify the `custom_logger.go` file.
Here is the *SEARCH/REPLACE block*:
custom_logger.go
```go
<<<<<<< SEARCH
func NewCustomLogger() *CustomLogger {
=======
func init() {
slog.SetDefaultLogger(NewCustomLogger())
}
func NewCustomLogger() *CustomLogger {
>>>>>>> REPLACE
```
46 lines
935 B
Go
46 lines
935 B
Go
package simplelog
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/mattn/go-isatty"
|
|
)
|
|
|
|
var (
|
|
relpServerURL = os.Getenv("LOGGER_RELP_URL")
|
|
webhookURL = os.Getenv("LOGGER_WEBHOOK_URL")
|
|
)
|
|
|
|
type CustomLogger struct {
|
|
handlers []Handler
|
|
}
|
|
|
|
func init() {
|
|
slog.SetDefaultLogger(NewCustomLogger())
|
|
}
|
|
|
|
func NewCustomLogger() *CustomLogger {
|
|
cl := &CustomLogger{}
|
|
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
|
|
}
|