package data

import (
	"context"
	"database/sql"
	"driver/api/verifyCode"
	"driver/internal/biz"
	"github.com/go-kratos/kratos/contrib/registry/consul/v2"
	"github.com/go-kratos/kratos/v2/transport/grpc"
	"github.com/hashicorp/consul/api"
	"time"
)

type DriverData struct {
	data *Data
}

func NewDriverInterface(data *Data) biz.DriverInterface {
	return &DriverData{data: data}
}

// 获取token的实现
func (dt *DriverData) GetToken(ctx context.Context, tel string) (string, error) {
	// 1.数据表查询
	driver := biz.Driver{}
	if err := dt.data.Mdb.Where("telephone=?", tel).First(&driver).Error; err != nil {
		return "", err
	}

	// 2.返回token
	return driver.Token.String, nil
}

// 获取已存储的验证码验证码
func (dt *DriverData) GetSavedVerifyCode(ctx context.Context, tel string) (string, error) {
	return dt.data.Rdb.Get(ctx, "DVC:"+tel).Result()
}

// 存储token到数据表
func (dt *DriverData) SaveToken(ctx context.Context, tel, token string) error {

	//先获取司机信息
	driver := biz.Driver{}
	if err := dt.data.Mdb.Where("telephone=?", tel).First(&driver).Error; err != nil {
		return err
	}

	//再更新司机信息
	driver.Token = sql.NullString{
		String: token,
		Valid:  true,
	}
	if err := dt.data.Mdb.Save(&driver).Error; err != nil {
		return err
	}

	return nil
}

// 初始化司机信息
func (dt *DriverData) InitDriverInfo(ctx context.Context, tel string) (*biz.Driver, error) {
	// 入库，设置状态未stop
	driver := biz.Driver{}
	driver.Telephone = tel
	driver.Status = sql.NullString{
		String: "stop",
		Valid:  true,
	}
	if err := dt.data.Mdb.Create(&driver).Error; err != nil {
		return nil, err
	}

	return &driver, nil
}

func (dt *DriverData) GetVerifyCode(ctx context.Context, tel string) (string, error) {
	// grpc 请求
	consulConfig := api.DefaultConfig()
	consulConfig.Address = dt.data.cs.Consul.Address
	consulClient, err := api.NewClient(consulConfig)
	dis := consul.New(consulClient)
	if err != nil {
		return "", err
	}
	endpoint := "discovery:///VerifyCode"
	conn, err := grpc.DialInsecure(
		ctx,
		grpc.WithEndpoint(endpoint), // 目标服务的名字
		grpc.WithDiscovery(dis),     // 使用服务发现
	)
	if err != nil {
		return "", err
	}
	//关闭
	defer conn.Close()

	// 2.2,发送获取验证码请求
	client := verifyCode.NewVerifyCodeClient(conn)
	reply, err := client.GetVerifyCode(ctx, &verifyCode.GetVerifyCodeRequest{
		Length: 6,
		Type:   1,
	})
	if err != nil {
		return "", err
	}

	// 三，redis的临时存储
	// 设置key, customer-verify-code
	status := dt.data.Rdb.Set(ctx, "DVC:"+tel, reply.Code, 60*time.Second)
	if _, err := status.Result(); err != nil {
		return "", err
	}
	return reply.Code, nil
}

// 获取号码对应的验证码
func (dt *DriverData) FetchVerifyCode(ctx context.Context, telephone string) (string, error) {
	status := dt.data.Rdb.Get(context.Background(), "DVC:"+telephone)
	code, err := status.Result() // status.String()
	if err != nil {
		return "", err
	}
	return code, nil
}

func (dt *DriverData) FetchInfoByTel(ctx context.Context, tel string) (*biz.Driver, error) {
	driver := &biz.Driver{}
	if err := dt.data.Mdb.Where("telephone=?", tel).First(driver).Error; err != nil {
		return nil, err
	}
	return driver, nil
}