// Package middleware provides HTTP middleware for metrics collection.
package middleware

import (
	"strconv"
	"time"

	"github.com/Wei-Shaw/sub2api/internal/pkg/metrics"
	"github.com/gin-gonic/gin"
)

// PrometheusMetrics returns a Gin middleware that records Prometheus metrics
func PrometheusMetrics() gin.HandlerFunc {
	return func(c *gin.Context) {
		start := time.Now()
		path := c.FullPath()
		if path == "" {
			path = "unknown"
		}

		c.Next()

		duration := time.Since(start).Seconds()
		status := strconv.Itoa(c.Writer.Status())

		// Record HTTP metrics
		metrics.RecordHTTPRequest(c.Request.Method, path, status, duration)
	}
}

// GatewayMetrics returns a Gin middleware that records gateway-specific metrics
func GatewayMetrics() gin.HandlerFunc {
	return func(c *gin.Context) {
		start := time.Now()
		platform := c.GetString("platform")
		model := c.GetString("model")

		c.Next()

		duration := time.Since(start).Seconds()
		status := strconv.Itoa(c.Writer.Status())

		if platform != "" {
			metrics.RecordGatewayRequest(platform, model, status, duration)
		}
	}
}