debug信息 需要鉴权后显示

This commit is contained in:
bjdgyc 2023-08-24 14:27:12 +08:00
parent 78a8b06467
commit 7714c2a3e8
1 changed files with 7 additions and 1 deletions

View File

@ -82,7 +82,7 @@ func authMiddleware(next http.Handler) http.Handler {
route := mux.CurrentRoute(r) route := mux.CurrentRoute(r)
name := route.GetName() name := route.GetName()
// fmt.Println("bb", r.URL.Path, name) // fmt.Println("bb", r.URL.Path, name)
if utils.InArrStr([]string{"login", "index", "static", "debug"}, name) { if utils.InArrStr([]string{"login", "index", "static"}, name) {
// 不进行鉴权 // 不进行鉴权
next.ServeHTTP(w, r) next.ServeHTTP(w, r)
return return
@ -93,6 +93,12 @@ func authMiddleware(next http.Handler) http.Handler {
if jwtToken == "" { if jwtToken == "" {
jwtToken = r.FormValue("jwt") jwtToken = r.FormValue("jwt")
} }
if jwtToken == "" {
cc, err := r.Cookie("jwt")
if err == nil {
jwtToken = cc.Value
}
}
data, err := GetJwtData(jwtToken) data, err := GetJwtData(jwtToken)
if err != nil || base.Cfg.AdminUser != fmt.Sprint(data["admin_user"]) { if err != nil || base.Cfg.AdminUser != fmt.Sprint(data["admin_user"]) {
w.WriteHeader(http.StatusUnauthorized) w.WriteHeader(http.StatusUnauthorized)