From bb4e2d8ae71c1c85373a8f8377b4e35f7facef0e Mon Sep 17 00:00:00 2001 From: WJQSERVER Date: Fri, 27 Sep 2024 15:09:55 +0800 Subject: [PATCH] 24w06b --- main.go | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/main.go b/main.go index 8182ff1..d3060d7 100644 --- a/main.go +++ b/main.go @@ -10,7 +10,6 @@ import ( "strconv" "strings" - "ghproxy/auth" "ghproxy/config" "ghproxy/logger" @@ -97,6 +96,14 @@ func api(c *gin.Context) { }) } +func AuthHandler(c *gin.Context) bool { + if cfg.Auth { + authToken := c.Query("auth_token") + return authToken == cfg.AuthToken + } + return true +} + func noRouteHandler(config *config.Config) gin.HandlerFunc { return func(c *gin.Context) { rawPath := strings.TrimPrefix(c.Request.URL.RequestURI(), "/") @@ -115,7 +122,7 @@ func noRouteHandler(config *config.Config) gin.HandlerFunc { rawPath = strings.Replace(rawPath, "/blob/", "/raw/", 1) } - if !auth.AuthHandler(c) { + if !AuthHandler(c) { c.AbortWithStatusJSON(401, gin.H{"error": "Unauthorized"}) logw("Unauthorized request: %s", rawPath) return