go/echo
go/echo
1
0
Fork 0
mirror of https://github.com/labstack/echo.git synced 2024-12-24 20:14:31 +02:00
Code Issues Releases Activity

[JWT] use local error handler for missing token

Browse Source 
Signed-off-by: Vishal Rana <vr@labstack.com>
This commit is contained in:
Vishal Rana 2018-07-09 11:34:53 -07:00
parent 6007218835
commit d8a6dad02d
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
middleware/jwt.go
View File

@ -61,7 +61,7 @@ type (
JWTSuccessHandler func(echo.Context)
// JWTErrorHandler defines a function which is executed for an invalid token.
JWTErrorHandler func(echo.Context, echo.HandlerFunc) error
JWTErrorHandler func(error, echo.Context, echo.HandlerFunc) error
jwtExtractor func(echo.Context) (string, error)
)
@ -74,7 +74,6 @@ const (
// Errors
var (
ErrJWTMissing = echo.NewHTTPError(http.StatusBadRequest, "missing or malformed jwt")
ErrJWTInvalid = echo.NewHTTPError(http.StatusUnauthorized, "invalid or expired jwt")
)
var (
@ -158,6 +157,9 @@ func JWTWithConfig(config JWTConfig) echo.MiddlewareFunc {
auth, err := extractor(c)
if err != nil {
if config.ErrorHandler != nil {
return config.ErrorHandler(err, c, next)
}
return err
}
token := new(jwt.Token)
@ -178,11 +180,11 @@ func JWTWithConfig(config JWTConfig) echo.MiddlewareFunc {
return next(c)
}
if config.ErrorHandler != nil {
return config.ErrorHandler(c, next)
return config.ErrorHandler(err, c, next)
}
return &echo.HTTPError{
Code: ErrJWTInvalid.Code,
Message: ErrJWTInvalid.Message,
Code: http.StatusUnauthorized,
Message: "Invalid or expired jwt",
Internal: err,
}
}