36 lines
1.0 KiB
Go
36 lines
1.0 KiB
Go
package middleware
|
|
|
|
import (
|
|
"github.com/casbin/casbin/v2"
|
|
"net/http"
|
|
"schisandra-album-cloud-microservices/common/constant"
|
|
"schisandra-album-cloud-microservices/common/errors"
|
|
"schisandra-album-cloud-microservices/common/xhttp"
|
|
)
|
|
|
|
type CasbinVerifyMiddleware struct {
|
|
casbin *casbin.SyncedCachedEnforcer
|
|
}
|
|
|
|
func NewCasbinVerifyMiddleware(casbin *casbin.SyncedCachedEnforcer) *CasbinVerifyMiddleware {
|
|
return &CasbinVerifyMiddleware{
|
|
casbin: casbin,
|
|
}
|
|
}
|
|
|
|
func (m *CasbinVerifyMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
userId := r.Header.Get(constant.UID_HEADER_KEY)
|
|
if userId == "" {
|
|
xhttp.JsonBaseResponseCtx(r.Context(), w, errors.New(http.StatusNotFound, "user id not found in header"))
|
|
return
|
|
}
|
|
correct, err := m.casbin.Enforce(userId, r.URL.Path, r.Method)
|
|
if err != nil || !correct {
|
|
xhttp.JsonBaseResponseCtx(r.Context(), w, errors.New(http.StatusNotFound, "casbin verify failed"))
|
|
return
|
|
}
|
|
next(w, r)
|
|
}
|
|
}
|