中间件判断逻辑

csznet · Apr 9, 2024 · 2f0d9c6 · 2f0d9c6
1 parent d4cbbdd
commit 2f0d9c6
Showing 1 changed file with 3 additions and 10 deletions.
diff --git a/control/control.go b/control/control.go
@@ -274,16 +274,9 @@ func Middleware(next http.HandlerFunc) http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
 		// 只有当密码设置并且不为"none"时，才进行检查
 		if conf.Pass != "" && conf.Pass != "none" {
-			if cookie, err := r.Cookie("p"); err != nil || cookie.Value != conf.Pass {
-				if r.URL.Path == "/api" {
-					if r.URL.Query().Get("pass") != conf.Pass {
-						http.Error(w, "Unauthorized", http.StatusUnauthorized)
-						return
-					}
-				} else {
-					http.Redirect(w, r, "/pwd", http.StatusSeeOther)
-					return
-				}
+			if cookie, err := r.Cookie("p"); err != nil || cookie.Value != conf.Pass || r.URL.Path == "/api" && r.URL.Query().Get("pass") != conf.Pass {
+				http.Redirect(w, r, "/pwd", http.StatusSeeOther)
+				return
 			}
 		}
 		next(w, r)