30 lines
691 B
Go
30 lines
691 B
Go
package middleware
|
|
|
|
import (
|
|
"rijig/utils"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
func RoleMiddleware(allowedRoles ...string) fiber.Handler {
|
|
return func(c *fiber.Ctx) error {
|
|
|
|
if len(allowedRoles) == 0 {
|
|
return utils.GenericResponse(c, fiber.StatusForbidden, "Forbidden: No roles specified")
|
|
}
|
|
|
|
roleID, ok := c.Locals("roleID").(string)
|
|
if !ok || roleID == "" {
|
|
return utils.GenericResponse(c, fiber.StatusUnauthorized, "Unauthorized: Role not found")
|
|
}
|
|
|
|
for _, role := range allowedRoles {
|
|
if role == roleID {
|
|
return c.Next()
|
|
}
|
|
}
|
|
|
|
return utils.GenericResponse(c, fiber.StatusForbidden, "Access Denied: You don't have permission to access this resource")
|
|
}
|
|
}
|