From c7ae18bdd61a7b56d61bf166c0d0fd23d0224bdc Mon Sep 17 00:00:00 2001
From: Spike Lu <luyuanxin1995@gmail.com>
Date: Fri, 30 Aug 2024 22:13:35 -0700
Subject: [PATCH] fix

---
 internal/server/web/proxy/middleware.go | 19 ++++++-------------
 1 file changed, 6 insertions(+), 13 deletions(-)

diff --git a/internal/server/web/proxy/middleware.go b/internal/server/web/proxy/middleware.go
index 4161328..facc646 100644
--- a/internal/server/web/proxy/middleware.go
+++ b/internal/server/web/proxy/middleware.go
@@ -183,20 +183,13 @@ func getMiddleware(cpm CustomProvidersManager, rm routeManager, pm PoliciesManag
 			return
 		}
 
-		for _, ip := range blockList {
-			fmt.Println(c.Request.RemoteAddr)
+		fmt.Println(c.Request.RemoteAddr)
+		fmt.Println(c.Request.UserAgent())
 
-			if strings.Contains(c.Request.RemoteAddr, ip) {
-				telemetry.Incr("bricksllm.proxy.get_middleware.first_block", nil, 1)
-				c.Status(200)
-				return
-			}
-
-			if strings.HasPrefix(c.Request.RemoteAddr, "43.130.32.") {
-				telemetry.Incr("bricksllm.proxy.get_middleware.second_block", nil, 1)
-				c.Status(200)
-				return
-			}
+		if strings.HasPrefix(c.Request.UserAgent(), "Go-http-client") {
+			telemetry.Incr("bricksllm.proxy.get_middleware.block_by_client", nil, 1)
+			c.Status(200)
+			return
 		}
 
 		if removeUserAgent {