From bfce13fc748c43551c7b69692ee9e74b3acee04f Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Thu, 25 Jan 2024 14:50:08 +0800
Subject: [PATCH] :bug: Trace collect source issue

---
 pkg/hypertext/proxies.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pkg/hypertext/proxies.go b/pkg/hypertext/proxies.go
index 962b792..0825210 100644
--- a/pkg/hypertext/proxies.go
+++ b/pkg/hypertext/proxies.go
@@ -95,6 +95,8 @@ func UseProxies(app *fiber.App) {
 }
 
 func makeResponse(ctx *fiber.Ctx, region *navi.Region, location *navi.Location, dest *navi.Destination) error {
+	uri := ctx.Request().URI().String()
+
 	// Modify request
 	for _, transformer := range dest.Transformers {
 		if err := transformer.TransformRequest(ctx); err != nil {
@@ -123,7 +125,7 @@ func makeResponse(ctx *fiber.Ctx, region *navi.Region, location *navi.Location,
 			Region:      region.ID,
 			Location:    location.ID,
 			Destination: dest.ID,
-			Uri:         ctx.OriginalURL(),
+			Uri:         uri,
 			IpAddress:   ctx.IP(),
 			UserAgent:   ctx.Get(fiber.HeaderUserAgent),
 			Error:       navi.RoadTraceError{