YiTao/middleware/admin.go

package middleware

import (
	"github.com/iris-contrib/middleware/jwt"
	"github.com/kataras/iris/v12"
)

var AdminMiddleware = iris.Handler(func(ctx iris.Context) {

	if token, ok := ctx.Values().Get("jwt").(*jwt.Token); ok {
		// Use the token if needed
		is_admin := token.Claims.(jwt.MapClaims)["is_admin"].(bool)
		if !is_admin {
			ctx.StatusCode(iris.StatusForbidden)
			ctx.JSON(iris.Map{"message": "forbidden"})
			return
		}
	} else {
		ctx.StatusCode(iris.StatusUnauthorized)
		ctx.JSON(iris.Map{"message": "unauthorized"})
		return
	}

	ctx.Next()

})