From 774a0466734dc2feb0a92b56457b4f9550bb4ca1 Mon Sep 17 00:00:00 2001
From: Mehdikhody <khodayari.khoram@gmail.com>
Date: Sat, 10 Feb 2024 05:21:16 +0330
Subject: [PATCH] Add listener functionality to LogWriter

---
 xray/log_writer.go | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/xray/log_writer.go b/xray/log_writer.go
index 53358ca2..74052708 100644
--- a/xray/log_writer.go
+++ b/xray/log_writer.go
@@ -6,11 +6,14 @@ import (
 )
 
 func NewLogWriter() *LogWriter {
-	return &LogWriter{}
+	return &LogWriter{
+		listeners: &[]func(line string){},
+	}
 }
 
 type LogWriter struct {
-	lastLine string
+	lastLine  string
+	listeners *[]func(line string)
 }
 
 func (lw *LogWriter) Write(m []byte) (n int, err error) {
@@ -33,7 +36,8 @@ func (lw *LogWriter) Write(m []byte) (n int, err error) {
 		endIndex := strings.Index(messageBody, "]")
 		if startIndex != -1 && endIndex != -1 && startIndex < endIndex {
 			level := strings.TrimSpace(messageBody[startIndex+1 : endIndex])
-			msgBody := "XRAY: " + strings.TrimSpace(messageBody[endIndex+1:])
+			rawMsg := strings.TrimSpace(messageBody[endIndex+1:])
+			msgBody := "XRAY: " + rawMsg
 
 			// Map the level to the appropriate logger function
 			switch level {
@@ -48,6 +52,11 @@ func (lw *LogWriter) Write(m []byte) (n int, err error) {
 			default:
 				logger.Debug("XRAY: " + msg)
 			}
+
+			// Notify listeners of the message
+			for _, listener := range *lw.listeners {
+				listener(messageBody)
+			}
 		} else if msg != "" {
 			logger.Debug("XRAY: " + msg)
 			return len(m), nil
@@ -56,3 +65,11 @@ func (lw *LogWriter) Write(m []byte) (n int, err error) {
 
 	return len(m), nil
 }
+
+// SetListener adds a listener to the log writer
+// The listener will be called with each line of the log
+// that is written to the log writer
+// We use this method to prevent reading the log file for better performance
+func (lw *LogWriter) SetListener(listener func(line string)) {
+	*lw.listeners = append(*lw.listeners, listener)
+}