Add extractRelevantInfo method

Signed-off-by: Rohanraj123 <[email protected]>

Author: Rohanraj123

kpm.go

@@ -3,6 +3,7 @@
 package main
 
 import (
+	"fmt"
 	"os"
 
 	"github.com/urfave/cli/v2"
@@ -16,7 +17,7 @@ func main() {
 	reporter.InitReporter()
 	kpmcli, err := client.NewKpmClient()
 	if err != nil {
-		reporter.Fatal(err)
+		kpmcli.WriteLog(fmt.Sprintf("Error: %v", err))
 	}
 	app := cli.NewApp()
 	app.Name = "kpm"

pkg/logger/logwriter.go

@@ -3,6 +3,9 @@ package logger
 import (
 	"fmt"
 	"os"
+	"runtime"
+	"strings"
+	"time"
 )
 
 type LogWriter struct {
@@ -26,9 +29,37 @@ func NewLogWriter() *LogWriter {
 
 func (lw *LogWriter) Write(p []byte) (n int, err error) {
 	if lw.Level == DebugLevel {
-		fmt.Fprintf(os.Stdout, "[DEBUG] %s", string(p))
-	} else {
-		fmt.Fprintf(os.Stdout, "[INFO] %s", string(p))
+
+		buf := make([]byte, 1024)
+		runtime.Stack(buf, false)
+
+		debugMessage := fmt.Sprintf(
+			"[DEBUG] %s\nTimestamp: %s\nStack Trace:\n%s\n",
+			string(p),
+			time.Now().Format(time.RFC3339),
+			string(buf),
+		)
+		fmt.Fprint(os.Stdout, debugMessage)
+		return len(debugMessage), nil
 	}
+
+	message := extractRelevantInfo(string(p))
+	fmt.Fprintf(os.Stdout, "[INFO] %s", message)
 	return len(p), nil
 }
+
+func extractRelevantInfo(message string) string {
+	lines := strings.Split(message, "\n") // Split message into lines
+	var filteredLines []string
+
+	for _, line := range lines {
+		// Exclude unnecessary debug/error details
+		if strings.Contains(line, "[FAILED] Expected") ||
+			strings.Contains(line, "In [It] at:") {
+			continue // Skip these lines
+		}
+		filteredLines = append(filteredLines, line)
+	}
+
+	return strings.Join(filteredLines, "\n") // Reconstruct filtered message
+}