package middleware

import (
	"cn/oneao/base-go/utils/jwtx"
	"cn/oneao/base-go/utils/pathx"
	"cn/oneao/base-go/utils/response"
	"context"
	"github.com/cloudwego/hertz/pkg/app"
)

// 私有 key，防止冲突
type ctxKeyUserID struct{}

// Auth 中间件，直接在中间件里过滤白名单路径
func Auth() app.HandlerFunc {
	// 白名单路径，直接写死
	whitelistPaths := []string{
		"/auth/login",
		"/auth/register",
		"/auth/refresh/token",
	}

	return func(ctx context.Context, c *app.RequestContext) {
		path := string(c.Request.URI().Path())

		// 使用工具函数判断是否在白名单
		if pathx.MatchPath(path, whitelistPaths) {
			c.Next(ctx)
			return
		}

		// 获取 token
		token := string(c.GetHeader("token"))

		if token == "" {
			response.Fail(c).Code(response.HttpCode.Unauthorized).Send()
			c.Abort()
			return
		}

		verifyToken := jwtx.VerifyToken(token)

		if verifyToken.IsValid {
			// 设置 userID 到 context
			ctx = setUserID(ctx, verifyToken.Claims.UserID)
			c.Next(ctx)
			return
		}

		// token 无效
		response.Fail(c).Code(response.HttpCode.RefreshToken).Send()
		c.Abort()
	}
}

// GetUserID 从 context 中获取 user_id
func GetUserID(ctx context.Context) int64 {
	userID, ok := ctx.Value(ctxKeyUserID{}).(int64)
	if !ok {
		return 0 // 或者返回 error，根据业务需求
	}
	return userID
}

// 内部中间件设置 user_id
func setUserID(ctx context.Context, userID int64) context.Context {
	return context.WithValue(ctx, ctxKeyUserID{}, userID)
}