From 5ab50614bfda7a743bcbfd4527881a116dfcfe02 Mon Sep 17 00:00:00 2001
From: kevkevinpal <oapallikunnel@gmail.com>
Date: Thu, 19 Dec 2024 16:06:50 -0500
Subject: [PATCH] added xgo and started using Trap to intercept function calls

---
 config/config.go      |   2 +-
 go.mod                |   1 +
 go.sum                |   2 +
 main.go               |  20 ++++++++
 routes/people.go      |   2 +-
 routes/test_routes.go |  40 ++++++++--------
 trace.out             | Bin 0 -> 4073 bytes
 utils/logger.go       |  14 +++---
 utils/trace.go        | 104 ++++++++++++++++++++++++++++--------------
 9 files changed, 120 insertions(+), 65 deletions(-)
 create mode 100644 trace.out

diff --git a/config/config.go b/config/config.go
index 860c66005..a1662fe49 100644
--- a/config/config.go
+++ b/config/config.go
@@ -68,7 +68,7 @@ func InitConfig() {
 	V2BotUrl = os.Getenv("V2_BOT_URL")
 	V2BotToken = os.Getenv("V2_BOT_TOKEN")
 	FfWebsocket = os.Getenv("FF_WEBSOCKET") == "true"
-	LogLevel = os.Getenv("LOG_LEVEL")
+	LogLevel = strings.ToUpper(os.Getenv("LOG_LEVEL"))
 
 	// Add to super admins
 	SuperAdmins = StripSuperAdmins(AdminStrings)
diff --git a/go.mod b/go.mod
index 8b0898986..3a073f373 100644
--- a/go.mod
+++ b/go.mod
@@ -61,6 +61,7 @@ require (
 	github.com/onsi/gomega v1.26.0 // indirect
 	github.com/robfig/cron v1.2.0
 	github.com/urfave/negroni v1.0.0 // indirect
+	github.com/xhd2015/xgo/runtime v1.0.52 // indirect
 	golang.org/x/exp v0.0.0-20230315142452-642cacee5cc0 // indirect
 	gopkg.in/go-playground/assert.v1 v1.2.1 // indirect
 )
diff --git a/go.sum b/go.sum
index d389d674c..d4ae95b46 100644
--- a/go.sum
+++ b/go.sum
@@ -2168,6 +2168,8 @@ github.com/urfave/cli v1.22.5/go.mod h1:Gos4lmkARVdJ6EkW0WaNv/tZAAMe9V7XWyB60NtX
 github.com/urfave/cli v1.22.9/go.mod h1:Gos4lmkARVdJ6EkW0WaNv/tZAAMe9V7XWyB60NtXRu0=
 github.com/urfave/negroni v1.0.0 h1:kIimOitoypq34K7TG7DUaJ9kq/N4Ofuwi1sjz0KipXc=
 github.com/urfave/negroni v1.0.0/go.mod h1:Meg73S6kFm/4PpbYdq35yYWoCZ9mS/YSx+lKnmiohz4=
+github.com/xhd2015/xgo/runtime v1.0.52 h1:njcRzY3Xo2AFu/qQSC4ak9+JN7xFmaI3iEUyJxoErWM=
+github.com/xhd2015/xgo/runtime v1.0.52/go.mod h1:9GBQ2SzJCzpD3T+HRN+2C0TUOGv7qIz4s0mad1xJ8Jo=
 github.com/xi2/xz v0.0.0-20171230120015-48954b6210f8 h1:nIPpBwaJSVYIxUFsDv3M8ofmx9yWTog9BfvIu0q41lo=
 github.com/xi2/xz v0.0.0-20171230120015-48954b6210f8/go.mod h1:HUYIGzjTL3rfEspMxjDjgmT5uz5wzYJKVo23qUhYTos=
 github.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2 h1:eY9dn8+vbi4tKz5Qo6v2eYzo7kUS51QINcR5jNpbZS8=
diff --git a/main.go b/main.go
index 58843ddf5..3c5361d00 100644
--- a/main.go
+++ b/main.go
@@ -7,6 +7,7 @@ import (
 	"os/signal"
 	"syscall"
 	"time"
+	"strings"
 
 	"github.com/joho/godotenv"
 	"github.com/robfig/cron"
@@ -17,8 +18,27 @@ import (
 	"github.com/stakwork/sphinx-tribes/routes"
 	"github.com/stakwork/sphinx-tribes/websocket"
 	"gopkg.in/go-playground/validator.v9"
+	"github.com/xhd2015/xgo/runtime/core"
+    "github.com/xhd2015/xgo/runtime/trap"
 )
 
+func init() {
+    trap.AddInterceptor(&trap.Interceptor{
+        Pre: func(ctx context.Context, f *core.FuncInfo, args core.Object, results core.Object) (interface{}, error) {
+						//if strings.Contains(f.Pkg, "sphinx-tribes") {
+										index := strings.Index(f.File, "sphinx-tribes")
+										trimmed := f.File
+										if index != -1 {
+								trimmed = f.File[index:]
+				}
+						fmt.Printf("%s:%d %s\n", trimmed, f.Line, f.Name)
+		//}
+
+
+            return nil, nil
+        },
+    })
+}
 func main() {
 	if err := godotenv.Load(); err != nil {
 		fmt.Println("no .env file")
diff --git a/routes/people.go b/routes/people.go
index 1f11e5735..341987462 100644
--- a/routes/people.go
+++ b/routes/people.go
@@ -15,7 +15,7 @@ func PeopleRoutes() chi.Router {
 
 	peopleHandler := handlers.NewPeopleHandler(db.DB)
 	r.Group(func(r chi.Router) {
-		r.Get("/", utils.TraceWithLogging(peopleHandler.GetListedPeople))
+		r.Get("/", peopleHandler.GetListedPeople)
 		r.Get("/search", utils.TraceWithLogging(peopleHandler.GetPeopleBySearch))
 		r.Get("/posts", utils.TraceWithLogging(handlers.GetListedPosts))
 		r.Get("/wanteds/assigned/{uuid}", bountyHandler.GetPersonAssignedBounties)
diff --git a/routes/test_routes.go b/routes/test_routes.go
index 5b944adf5..bd20f2208 100644
--- a/routes/test_routes.go
+++ b/routes/test_routes.go
@@ -2,10 +2,10 @@ package routes
 
 import (
 	"net/http"
-	"runtime/trace"
-	"log"
-//	"os"
-	"bytes"
+//	"runtime/trace"
+//	"log"
+////	"os"
+//	"bytes"
 
 	"github.com/go-chi/chi"
 )
@@ -14,23 +14,23 @@ func TestRoutes() chi.Router {
 	r := chi.NewRouter()
 
 	r.Get("/internal-server-error", func(w http.ResponseWriter, r *http.Request) {
-					// Enable tracing
-	//f, err := os.Create("trace.out")
-	var buf bytes.Buffer
-	//if err != nil {
-	//	log.Fatalf("Failed to create trace output file: %v", err)
-	//}
-	//defer f.Close()
+	//				// Enable tracing
+	////f, err := os.Create("trace.out")
+	//var buf bytes.Buffer
+	////if err != nil {
+	////	log.Fatalf("Failed to create trace output file: %v", err)
+	////}
+	////defer f.Close()
 
-	if err := trace.Start(&buf); err != nil {
-		log.Fatalf("Failed to start trace: %v", err)
-	}
-	defer func() {
-					trace.Stop()
-					log.Println("Trace Data:")
-	log.Println(buf.String())
-	}()
-		//panic("Forced internal server error")
+	//if err := trace.Start(&buf); err != nil {
+	//	log.Fatalf("Failed to start trace: %v", err)
+	//}
+	//defer func() {
+	//				trace.Stop()
+	//				log.Println("Trace Data:")
+	//log.Println(buf.String())
+	//}()
+		panic("Forced internal server error")
 	})
 
 	return r
diff --git a/trace.out b/trace.out
new file mode 100644
index 0000000000000000000000000000000000000000..83de2b6f90d346ecfc7970fd38ab2f86b2bd7198
GIT binary patch
literal 4073
zcmdT{U2GIp6z<II-Q8)~wzTwTyQK`YTXsu3(_Ltx3W=ut#)702c_h=B+uccbXO=s&
zr8O}rf*7AHR1IqhC^3eh3F1$bM2$WO3Q_yuPlzUhVl)PYN1-HmW_G8Y0i-BT-SqD6
zckZ3<eCM8X&)p_8F2={>oIykh7ZnvzCAaJsmAmoL+3~l7&x{scMgHU8o%T^BHxG|o
zziLg5?Xjl<M<|*Kma8yFvlpmxsxDGRHwWJO@dHo5S40PV6dhn0I^gxt0S3XR7e0M3
z{4iJ;#dN?^0`_?@0x(Kpl+o<gawha46@GH^@9RwI^VC7ggG#rLQ(j-`LoZN%HvBR(
zdFloe?wtG$-n%Eqr<ich<jE<<8OO7GJ~rccbFVEnI6+f(l(#5PkfognF;2vIoQOtF
z<ncNY?Q<fJ--(zCC!#Bzh*{u7o+>AztDT5h;zErsRO3RmE);a3Iwzt-PQ=u^&_Wk#
za-n4|)a^p+Txh)uZE&HDPK0{DK0Lw%y}KMnXCmL(ZzXc-IXjUw<EJiOy5g^{8KoF6
zHF_{d*9Nb788&$G;uU`#Sg##A%7*G|e*g3wv#?>R{M1Xwe2chu85@s%`>Jm-xBk-p
zW4<Na`?PJoeOd0aBWz<+n?3!@>qpseb8dS1=f{t*OCu*Jd;a?~6Ku34Y{M@;nP8W-
zcG>XD8xw3B-($nCzA+{2`19m>Mrb%wd-lb1Onbx4GR8x_@<xu1#akIFbpD4Ny?n(!
z`s$@abVu+n>ervl=$@Lp!7-upV+ZKYl^2S~FPkBa(sMo}x^CakW++$Zes)mky%RZl
zRlN0(m7J&NesX-F%Lj6F7xaeLtdRE3`H-A_@$?_oZ9G0^_eBq)innM6w_78Fm>{}z
zLpK+Zv8+g<f#oP-%1pxm-1&y0sbVUMJgjL^<P?q~RFYNo43?Au1<T0m8_)<I0vexB
z#fC|V{9YZWuw)>1zCNN#e59pYQ{^bXLsSe@>@jVqgq4z_%IR2@(Y$%qcY-CX3}F<g
z2WM%K4058bD>~?e=Qwa<2}Y%LFx@Xog9)N#RhiRwW1PV-X|k79=_9&<(_BJ>Swm5A
z-d@gT2*&9Q#E0frP7}(wKbznrF_l6U6_z!t8cK=-4<JlOl}JwkJ6cdi%!xGSH)y()
z1d^%}-4Kb9|7C=#gN1p)`lZBgGw1k$tSZG&O+?7*L`Z4iMo4IaP9(v#y6{XwL$y)c
zO{B#ldQ=_~RS6T|GjNo*To!19jkD0kd0?!obh*k4L!qvOVkEQuyriWCARRQ*Cg_=@
zq7JtiMCr%6U~V7Q`97>0z4@>CaOzhJY1I()q)4#5QznoxVePsn)^6&(D}ewc#?As6
z2wK<xRU#w}Bg1bLRVXO39(IY}hRG1#*w?oeH8kBBm;H#Ko6Cq9*rH}f2nnrCN-6>j
zsI|@6;V5r8wD`C+LQ+d(VGs|&pQ4&6$P6Zgv?dF8rFdj0#&_@?LOMHaI$C^hM>E2g
zMD9vwE7}ajYL`UgBaWr_-hV|^jBbdl4~h@ckWW4zLrp#ZH!MYlUEdAa6U>4(IU&mM
z!wyA67bK17*108m)J{~}K>Bu5p%$B0-<OyIuergoc0QJO0?wCiayXxDI}A*qC0W}}
z))AtSLKzjLL^}Ulc`B-mtvpBD9U^85N=Q%d7vGB|4ayG+Fn<(Uw%k|PGi1}HHO(nN
z(<^YoMl4%1s|K?k_;!%8D+>p!M@uAZIZza>k3G1cRGt#lwB65+STZ)PE$j&Y_=5`|
z`OtPMvRRk{XCHn(fB5Zm&m6q=>ggqs7D-c8Xq0B3K(R%3JGGkWHnS=BVo~niy1k$`
z-h5}XaW%}il*yVhmN&SldNc`cPPh!x;&6s&l8#oi%*JFB09!|e4J=K4=6~0yv&uHY
m5g<li67SGRN=7Re|2wnk+myG;zJK^TB1K(|Ht#(ym45))uWNb$

literal 0
HcmV?d00001

diff --git a/utils/logger.go b/utils/logger.go
index e32d9f06f..e4685b0d4 100644
--- a/utils/logger.go
+++ b/utils/logger.go
@@ -3,7 +3,7 @@ package utils
 import (
 	"log"
 	"os"
-	"strings"
+	"github.com/stakwork/sphinx-tribes/config"
 )
 
 type Logger struct {
@@ -12,7 +12,6 @@ type Logger struct {
 	errorLogger   *log.Logger
 	debugLogger   *log.Logger
 	machineLogger *log.Logger
-	logLevel      string
 }
 
 var Log = Logger{
@@ -21,35 +20,34 @@ var Log = Logger{
 	errorLogger:   log.New(os.Stderr, "ERROR: ", log.Ldate|log.Ltime|log.Lshortfile),
 	debugLogger:   log.New(os.Stdout, "DEBUG: ", log.Ldate|log.Ltime|log.Lshortfile),
 	machineLogger: log.New(os.Stdout, "MACHINE: ", log.Ldate|log.Ltime|log.Lshortfile),
-	logLevel:      strings.ToUpper(os.Getenv("LOG_LEVEL")),
 }
 
 func (l *Logger) Machine(format string, v ...interface{}) {
-	if l.logLevel == "MACHINE" {
+	if config.LogLevel == "MACHINE" {
 		l.machineLogger.Printf(format, v...)
 	}
 }
 
 func (l *Logger) Debug(format string, v ...interface{}) {
-	if l.logLevel == "MACHINE" || l.logLevel == "DEBUG" {
+	if config.LogLevel == "MACHINE" || config.LogLevel == "DEBUG" {
 		l.debugLogger.Printf(format, v...)
 	}
 }
 
 func (l *Logger) Info(format string, v ...interface{}) {
-	if l.logLevel == "MACHINE" || l.logLevel == "DEBUG" || l.logLevel == "INFO" {
+	if config.LogLevel == "MACHINE" || config.LogLevel == "DEBUG" || config.LogLevel == "INFO" {
 		l.infoLogger.Printf(format, v...)
 	}
 }
 
 func (l *Logger) Warning(format string, v ...interface{}) {
-	if l.logLevel == "MACHINE" || l.logLevel == "DEBUG" || l.logLevel == "INFO" || l.logLevel == "WARNING" {
+	if config.LogLevel == "MACHINE" || config.LogLevel == "DEBUG" || config.LogLevel == "INFO" || config.LogLevel == "WARNING" {
 		l.warningLogger.Printf(format, v...)
 	}
 }
 
 func (l *Logger) Error(format string, v ...interface{}) {
-	if l.logLevel == "MACHINE" || l.logLevel == "DEBUG" || l.logLevel == "INFO" || l.logLevel == "WARNING" || l.logLevel == "ERROR" {
+	if config.LogLevel == "MACHINE" || config.LogLevel == "DEBUG" || config.LogLevel == "INFO" || config.LogLevel == "WARNING" || config.LogLevel == "ERROR" {
 		l.errorLogger.Printf(format, v...)
 	}
 }
diff --git a/utils/trace.go b/utils/trace.go
index d7570c89c..d35c6ff8c 100644
--- a/utils/trace.go
+++ b/utils/trace.go
@@ -1,52 +1,86 @@
 package utils
 
 import (
-	"bytes"
-	"log"
+	//"bytes"
+//	"log"
 	"net/http"
 	"regexp"
-	"runtime/trace"
+//	"runtime/trace"
+  "runtime"
+	"fmt"
 	"strings"
 )
 
+func AutoLog(fn http.HandlerFunc) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		defer logCallStack()
+		fn(w, r)
+	}
+}
+
+func logCallStack() {
+	// Retrieve the stack trace
+	pc := make([]uintptr, 50) // Adjust size as needed
+	n := runtime.Callers(2, pc)
+	frames := runtime.CallersFrames(pc[:n])
+
+	fmt.Println("Call stack:")
+	for {
+		frame, more := frames.Next()
+		fmt.Printf("- %s (line %d)\n", frame.Function, frame.Line)
+		if !more {
+			break
+		}
+	}
+	fmt.Println()
+}
+
 func TraceWithLogging(next http.HandlerFunc) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
-		var buf bytes.Buffer
+		//var buf bytes.Buffer
 
 		// Start trace capturing
-		if err := trace.Start(&buf); err != nil {
-			http.Error(w, "Failed to start trace", http.StatusInternalServerError)
-			log.Fatalf("Failed to start trace: %v", err)
-			return
+		//if err := trace.Start(&buf); err != nil {
+		//	http.Error(w, "Failed to start trace", http.StatusInternalServerError)
+		//	log.Fatalf("Failed to start trace: %v", err)
+		//	return
+		//}
+						Log.Info("HI")
+						wrappedFn := func() {
+										defer func() {
+						buf := make([]byte, 4096)
+						n := runtime.Stack(buf, true)
+						stackTrace := string(buf[:n])
+						Log.Info("%s", stackTrace)
+			//trace.Stop()
+
+			//// Extract and process trace output
+			//traceOutput := buf.String()
+			////log.Println("Raw Trace Output (for debugging):", traceOutput)
+			//// Split the trace output into lines
+			//traceLines := strings.Split(traceOutput, "\n")
+			//startedFormatting := false
+
+			//// Log each trace line with a line number and format
+			//for _, line := range traceLines {
+			//	if !startedFormatting && strings.Contains(line, "start trace") {
+			//		startedFormatting = true
+			//	} else {
+			//		continue
+			//	}
+
+			//	if line != "" {
+			//	  traceFormatted := formatTrace(line)
+			//		for j, formattedLine := range traceFormatted {
+			//					log.Printf("%03d %s", j+1, formattedLine)
+			//	}
+			//	}
+			//}
+			}()
+		next(w, r)
 		}
-		defer func() {
-			trace.Stop()
-
-			// Extract and process trace output
-			traceOutput := buf.String()
-			//log.Println("Raw Trace Output (for debugging):", traceOutput)
-			// Split the trace output into lines
-			traceLines := strings.Split(traceOutput, "\n")
-			startedFormatting := false
-
-			// Log each trace line with a line number and format
-			for _, line := range traceLines {
-				if !startedFormatting && strings.Contains(line, "start trace") {
-					startedFormatting = true
-				} else {
-					continue
-				}
-
-				if line != "" {
-				  traceFormatted := formatTrace(line)
-					for j, formattedLine := range traceFormatted {
-								log.Printf("%03d %s", j+1, formattedLine)
-				}
-				}
-			}
-		}()
+		wrappedFn()
 
-		next(w, r)
 	}
 }