auth.go

package ginmiddleware

import (
	"net/http"

	"github.com/gin-gonic/gin"
)

var AuthPayload string = "auth-payload"

func NewAuthMiddleware(getToken IgetAuthToken, authenticator IAuthenticator) gin.HandlerFunc {
	return func(c *gin.Context) {
		token, err := getToken.GetAuthToken(c.Request)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusUnauthorized, &Error{
				Code:    http.StatusUnauthorized,
				Message: err.Error(),
			})
			return
		}
		if token == "" {
			c.AbortWithStatusJSON(http.StatusUnauthorized, &Error{
				Code:    http.StatusUnauthorized,
				Message: "not found auth token",
			})
			return
		}
		claims, mErr := authenticator.Authenticate(token)
		if mErr != nil {
			c.AbortWithStatusJSON(http.StatusUnauthorized, &Error{
				Code:    http.StatusUnauthorized,
				Message: mErr.Message,
			})
			return
		}
		c.Set(AuthPayload, claims)
		c.Next()
	}
}

func GetAuthPayload(c *gin.Context) (interface{}, bool) {
	return c.Get(AuthPayload)
}

type AuthMiddleware struct {
	GetToken      IgetAuthToken
	Authenticator IAuthenticator
}

func (auth *AuthMiddleware) ValidateAuth() gin.HandlerFunc {
	return func(c *gin.Context) {
		token, err := auth.GetToken.GetAuthToken(c.Request)
		if err != nil {
			c.AbortWithStatusJSON(http.StatusUnauthorized, &Error{
				Code:    http.StatusUnauthorized,
				Message: err.Error(),
			})
			return
		}
		if token == "" {
			c.AbortWithStatusJSON(http.StatusUnauthorized, &Error{
				Code:    http.StatusUnauthorized,
				Message: "not found auth token",
			})
			return
		}
		claims, mErr := auth.Authenticator.Authenticate(token)
		if mErr != nil {
			c.AbortWithStatusJSON(http.StatusUnauthorized, &Error{
				Code:    http.StatusUnauthorized,
				Message: mErr.Message,
			})
			return
		}
		c.Set(AuthPayload, claims)
		c.Next()
	}
}

func (auth *AuthMiddleware) GetAuthPayload(c *gin.Context) (interface{}, bool) {
	return c.Get(AuthPayload)
}