package middleware

import (
	"fmt"
	"strings"

	"github.com/gin-gonic/gin"
	"github.com/tidbcloud/tiinf/monitor/tilog"
)

func RemoveApiKeyHeaderMiddleware() func(ctx *gin.Context) {
	return func(c *gin.Context) {
		ctx := c.Request.Context()
		authTypeHeader := c.GetHeader("X-Auth-Type")
		tilog.Info(ctx, fmt.Sprintf("get x-auth-type from header is %s", authTypeHeader))
		if strings.ToLower(authTypeHeader) == "console" {
			c.Request.Header.Del("X-Auth-Method")
			c.Request.Header.Del("X-Auth-Content")
		}

		c.Next()
	}

}