crawlab/backend/middlewares/auth.go

package middlewares

import (
	"crawlab/constants"
	"crawlab/routes"
	"crawlab/services"
	"github.com/gin-gonic/gin"
	"net/http"
	"strings"
)

func AuthorizationMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		// 如果为登录或注册，不用校验
		//if c.Request.URL.Path == "/login" ||
		//	(c.Request.URL.Path == "/users" && c.Request.Method == "PUT") ||
		//	strings.HasSuffix(c.Request.URL.Path, "download") {
		//	c.Next()
		//	return
		//}

		// 获取token string
		tokenStr := c.GetHeader("Authorization")

		// 校验token
		user, err := services.CheckToken(tokenStr)

		// 校验失败，返回错误响应
		if err != nil {
			c.AbortWithStatusJSON(http.StatusUnauthorized, routes.Response{
				Status:  "ok",
				Message: "unauthorized",
				Error:   "unauthorized",
			})
			return
		}

		// 如果为普通权限，校验请求地址是否符合要求
		if user.Role == constants.RoleNormal {
			if strings.HasPrefix(strings.ToLower(c.Request.URL.Path), "/users") {
				c.AbortWithStatusJSON(http.StatusUnauthorized, routes.Response{
					Status:  "ok",
					Message: "unauthorized",
					Error:   "unauthorized",
				})
				return
			}
		}
		c.Set(constants.ContextUser, &user)

		// 校验成功
		c.Next()
	}
}