package server

import (
	"context"
	"driver/internal/service"
	"github.com/go-kratos/kratos/v2/errors"
	"github.com/go-kratos/kratos/v2/middleware"
	"github.com/go-kratos/kratos/v2/middleware/auth/jwt"
	"github.com/go-kratos/kratos/v2/transport"
	jwtv4 "github.com/golang-jwt/jwt/v4"
	"strings"
)

// 返回中间件的函数
func DriverToken(service *service.DriverService) middleware.Middleware {
	return func(handler middleware.Handler) middleware.Handler {
		return func(ctx context.Context, req interface{}) (interface{}, error) {
			// 1.校验JWT，获取其中的司机标识tel
			claims, ok := jwt.FromContext(ctx)
			if !ok {
				return nil, errors.Unauthorized("Unauthorized", "claims not found")
			}
			claimsMap := claims.(jwtv4.MapClaims)
			tel := claimsMap["jti"]

			// 2.利用tel，获取存储在司机表（缓存）中的token
			token, err := service.Bz.DI.GetToken(ctx, tel.(string))
			if err != nil {
				return nil, errors.Unauthorized("Unauthorized", "driver token not found")
			}

			// 3.比对两个token(和请求头中的）
			header, _ := transport.FromServerContext(ctx)
			auths := strings.SplitN(header.RequestHeader().Get("Authorization"), " ", 2)
			reqToken := auths[1]
			if token != reqToken {
				return nil, errors.Unauthorized("Unauthorized", "token was updated")
			}

			// 4.记录登录司机信息
			driver, err := service.Bz.DI.FetchInfoByTel(ctx, tel.(string))
			if err != nil {
				return nil, errors.Unauthorized("Unauthorized", "driver was found")
			}
			// 基于当前的ctx，构建新的带有值的ctx
			ctxWithDriver := context.WithValue(ctx, "driver", driver)
			//ctxWithDriver.Value("driver")

			// 5.jwt校验通过
			return handler(ctxWithDriver, req)
		}
	}
}