From 7e7fa772e164d6eaa79e94926aa59737ac6c7731 Mon Sep 17 00:00:00 2001
From: "M. Mert Yildiran" <me@mertyildiran.com>
Date: Fri, 30 Dec 2022 08:30:48 +0300
Subject: [PATCH] :bug: Have a short caller (file:line) log

---
 kubeshark.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/kubeshark.go b/kubeshark.go
index d5949ae64..e10bdcc46 100644
--- a/kubeshark.go
+++ b/kubeshark.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"os"
+	"strconv"
 	"time"
 
 	"github.com/kubeshark/kubeshark/cmd"
@@ -11,6 +12,20 @@ import (
 
 func main() {
 	zerolog.SetGlobalLevel(zerolog.InfoLevel)
+
+	// Short caller (file:line)
+	zerolog.CallerMarshalFunc = func(pc uintptr, file string, line int) string {
+		short := file
+		for i := len(file) - 1; i > 0; i-- {
+			if file[i] == '/' {
+				short = file[i+1:]
+				break
+			}
+		}
+		file = short
+		return file + ":" + strconv.Itoa(line)
+	}
+
 	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr, TimeFormat: time.RFC3339}).With().Caller().Logger()
 	cmd.Execute()
 }