fix: log setter overwrite

Signed-off-by: Martin Buchleitner <[email protected]>
mabunixda · Sep 4, 2024 · 37747c7 · 37747c7
1 parent c402157
commit 37747c7
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/wattpilot.go b/wattpilot.go
@@ -20,6 +20,8 @@ import (
 	"time"
 
 	"github.com/sirupsen/logrus"
+	logruswriter "github.com/sirupsen/logrus/hooks/writer"
+
 	"golang.org/x/crypto/pbkdf2"
 	"nhooyr.io/websocket"
 )
@@ -117,8 +119,17 @@ func New(host string, password string) *Wattpilot {
 
 }
 func (w *Wattpilot) SetLogger(logger *log.Logger) {
-	w.logger.SetOutput(logger.Writer())
 	w.logger.SetFormatter(&logrus.TextFormatter{})
+
+	w.logger.AddHook(&logruswriter.Hook{
+		Writer: logger.Writer(),
+		LogLevels: []logrus.Level{
+			logrus.PanicLevel,
+			logrus.FatalLevel,
+			logrus.ErrorLevel,
+			logrus.WarnLevel,
+		},
+	})
 }
 
 func (w *Wattpilot) SetLogLevel(level logrus.Level) {