package middleware

import (
	"backend/store"

	"github.com/gin-gonic/gin"
)

func ValidateToken() gin.HandlerFunc {
	return func(c *gin.Context) {
		token := c.GetHeader("Authorization")
		if len(token) > 7 && token[:7] == "Bearer " {
			token = token[7:]
		}
		if token == "" || !store.IsTokenValid(token) {
			c.JSON(401, gin.H{"error": "Authorization header is required"})
			c.Abort()
			return
		}
		tokenInfo, _ := store.GetTokenInfo(token)
		if tokenInfo != nil {
			c.Set("admin", tokenInfo.UserName)
		}
		// 这里可以添加逻辑来验证token的有效性
		// 假设验证通过，继续处理请求
		c.Next()
	}
}