middlewares.go

package common

import (
	"github.com/goccy/go-json"
	"runtime/debug"
	"strconv"
	"time"

	"github.com/gofiber/fiber/v2"
	"github.com/rs/zerolog/log"
)

func GetUserID(c *fiber.Ctx) (int, error) {
	// get user id from header: X-Consumer-Username if through Kong
	username := c.Get("X-Consumer-Username")
	if username != "" {
		id, err := strconv.Atoi(username)
		if err == nil {
			return id, nil
		}
	}

	// get user id from jwt
	// ID and UserID are both valid
	var user struct {
		ID     int `json:"id"`
		UserID int `json:"user_id"`
	}

	token := GetJWTToken(c)
	if token == "" {
		return 0, Unauthorized("Unauthorized")
	}

	err := ParseJWTToken(token, &user)
	if err != nil {
		return 0, Unauthorized("Unauthorized")
	}

	if user.ID != 0 {
		return user.ID, nil
	} else if user.UserID != 0 {
		return user.UserID, nil
	}

	return 0, Unauthorized("Unauthorized")
}

func MiddlewareGetUserID(c *fiber.Ctx) error {
	userID, err := GetUserID(c)
	if err == nil {
		c.Locals("user_id", userID)
	}

	return c.Next()
}

func MiddlewareCustomLogger(c *fiber.Ctx) error {
	startTime := time.Now()
	chainErr := c.Next()

	if chainErr != nil {
		if err := c.App().ErrorHandler(c, chainErr); err != nil {
			_ = c.SendStatus(fiber.StatusInternalServerError)
		}
	}

	latency := time.Since(startTime).Milliseconds()
	userID, ok := c.Locals("user_id").(int)

	contentType := string(c.Request().Header.ContentType())

	output := log.Info().
		Int("status_code", c.Response().StatusCode()).
		Str("method", c.Method()).
		Str("origin_url", c.OriginalURL()).
		Str("remote_ip", c.Get("X-Real-IP")).
		Str("user_agent", c.Get("User-Agent", "")).
		Int64("latency", latency).
		Str("content_type", contentType)

	xForwardFor := c.Get("X-Forwarded-For")
	if xForwardFor != "" {
		output = output.Str("x_forwarded_for", xForwardFor)
	}
	if ok {
		output = output.Int("user_id", userID)
	}
	if chainErr != nil {
		output = output.Err(chainErr)
	}
	if c.Method() == "POST" || c.Method() == "PUT" {
		var body = make(map[string]any)
		err := json.Unmarshal(c.Body(), &body)
		if err != nil {
			output = output.Bytes("body", StripBytes(c.Body(), 32))
		} else {
			delete(body, "password")
			output = output.Any("body", body)
		}
	}
	output.Msg("http log")
	return nil
}

func StackTraceHandler(_ *fiber.Ctx, e any) {
	log.Error().Any("panic", e).Bytes("stack", debug.Stack()).Msg("stacktrace")
}