package middleware

import (
	"codeword/internal/models"
	"github.com/gofiber/fiber/v2"
	"github.com/themakers/hlog"
)

func ContextLogger(logger hlog.Logger) fiber.Handler {
	return func(c *fiber.Ctx) error {
		c.Locals(models.LoggerKey, logger)
		return c.Next()
	}
}

func ExtractLogger(c *fiber.Ctx) hlog.Logger {
	logger := c.Context().UserValue(models.LoggerKey).(hlog.Logger)
	return logger
}