middleware.go

package middleware

import (
	"errors"
	"fmt"
	"golangtemplate/servers/app/models"
	"net/http"
	"runtime/debug"
	"time"

	"corelab.mkcl.org/MKCLOS/coredevelopmentplatform/corepkgv2/loggermdl"
	"github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
	"github.com/golang-jwt/jwt/v5"
	"github.com/golang-jwt/jwt/v5/request"
)

// Init -Init
func Init(g *gin.Engine) {
	g.Use(cors.Default())
	g.Use(Recovery())
}

type Principal struct {
	UserID            string    `json:"userId"`
	Groups            []string  `json:"groups"`
	SessionExpiration time.Time `json:"sessionExpiration"`
	ClientIP          string    `json:"clientIP"`
	HitsCount         int       `json:"hitsCount"`
	Token             string    `json:"token"`
	Metadata          string    `json:"metadata"`
	jwt.RegisteredClaims
}

var group = []string{"delete"}

// Recovery - a recoverymdl for gin to log the panic and recover
func Recovery() gin.HandlerFunc {
	return func(c *gin.Context) {
		defer func() {
			if err := recover(); err != nil {
				loggermdl.LogError("recovered:", err)
				loggermdl.LogError(string(debug.Stack()))
				// write to access log
				gin.DefaultErrorWriter.Write(debug.Stack())
			}
		}()
		c.Next()
	}
}

func CustomLogger() gin.HandlerFunc {
	return gin.LoggerWithFormatter(func(params gin.LogFormatterParams) string {
		return fmt.Sprintf("", params.ClientIP, params.BodySize, params.Request, params.Method, params.Path)
	})
}

func Auth() gin.HandlerFunc {
	return func(c *gin.Context) {
		_, err := request.ParseFromRequest(c.Request, request.OAuth2Extractor, func(token *jwt.Token) (interface{}, error) {
			b := ([]byte(models.JWTKey))
			return b, nil
		})
		if err != nil {
			c.AbortWithError(401, err)
		}
	}
}

func RolebasedAuth() gin.HandlerFunc {