fix: Log Delegation (#48)

* fix: log setter overwrite Signed-off-by: Martin Buchleitner <[email protected]> * fix: log setter overwrite Signed-off-by: Martin Buchleitner <[email protected]> * fix: log setter overwrite Signed-off-by: Martin Buchleitner <[email protected]> * fix: log setter overwrite Signed-off-by: Martin Buchleitner <[email protected]> * delgation function trial Signed-off-by: Martin Buchleitner <[email protected]> * delgation function - correct message info Signed-off-by: Martin Buchleitner <[email protected]> --------- Signed-off-by: Martin Buchleitner <[email protected]>
mabunixda · Sep 4, 2024 · d7b720b · d7b720b
1 parent f82b7ed
commit d7b720b
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 4 deletions.
diff --git a/logrus_hook.go b/logrus_hook.go
@@ -0,0 +1,20 @@
+package wattpilot
+
+import (
+	log "github.com/sirupsen/logrus"
+)
+
+type CallHook struct {
+	Call      func(string, string)
+	LogLevels []log.Level
+}
+
+func (hook *CallHook) Fire(entry *log.Entry) error {
+	hook.Call(entry.Level.String(), entry.Message)
+	return nil
+}
+
+// Levels define on which log levels this hook would trigger
+func (hook *CallHook) Levels() []log.Level {
+	return hook.LogLevels
+}
diff --git a/wattpilot.go b/wattpilot.go
@@ -10,7 +10,6 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"log"
 	"os"
 	"os/signal"
 	"strconv"
@@ -20,6 +19,7 @@ import (
 	"time"
 
 	"github.com/sirupsen/logrus"
+
 	"golang.org/x/crypto/pbkdf2"
 	"nhooyr.io/websocket"
 )
@@ -116,9 +116,8 @@ func New(host string, password string) *Wattpilot {
 	return w
 
 }
-func (w *Wattpilot) SetLogger(logger *log.Logger) {
-	w.logger.Out = logger.Writer()
-	// w.logger.SetOutput(logger)
+func (w *Wattpilot) SetLogger(delegate func(string, string)) {
+	w.logger.AddHook(&CallHook{Call: delegate, LogLevels: logrus.AllLevels})
 }
 
 func (w *Wattpilot) SetLogLevel(level logrus.Level) {