package token

import (
	"context"
	"fmt"
	"log"
	"time"

	pb "schain/proto"

	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
	"google.golang.org/grpc/keepalive"
)

const (
	dialTimeout        = 10 * time.Second
	maxRecvMessageSize = 100 * 1024 * 1024 // 100 MiB
	maxSendMessageSize = 100 * 1024 * 1024 // 100 MiB

)

// 建立与grpc服务器的连接
func NewClient(address string) (*grpc.ClientConn, error) {
	var conn *grpc.ClientConn
	//Keepalive 时间、超时时间和允许无流的连接
	kaOpts := keepalive.ClientParameters{
		Time:                1 * time.Minute,
		Timeout:             20 * time.Second,
		PermitWithoutStream: true,
	}

	//gRPC 连接选项
	dialOpts := []grpc.DialOption{
		grpc.WithKeepaliveParams(kaOpts),
		grpc.WithBlock(),
		grpc.FailOnNonTempDialError(true),
		grpc.WithDefaultCallOptions(
			grpc.MaxCallRecvMsgSize(maxRecvMessageSize),
			grpc.MaxCallSendMsgSize(maxSendMessageSize),
		),
	}
	//取消证书认证
	dialOpts = append(dialOpts, grpc.WithTransportCredentials(insecure.NewCredentials()))

	//ctx, cancel := context.WithTimeout(context.Background(), dialTimeout)
	//defer cancel()
	conn, err := grpc.Dial(address, dialOpts...)
	//conn, err := grpc.DialContext(ctx, address, dialOpts...)
	if err != nil {
		log.Fatalf("did not connect: %s", err)
	}

	//调用服务器上的 Login RPC 方法
	client := pb.NewPingClient(conn)

	//对username和password进行aes加密
	usr, err := encrypt("Wuxinyu", key)
	if err != nil {
		panic("encrypt error")
	}
	pwd, err := encrypt("123456", key)
	if err != nil {
		panic("encrypt error")
	}
	loginReply, err := client.Login(context.Background(), &pb.LoginRequest{Username: usr, Password: pwd})
	if err != nil {
		log.Fatalf("Error when calling SayHello: %s", err)
		panic("wrong")
	}
	fmt.Println("Login Reply:", loginReply)
	conn.Close()

	//Call SayHello,第二次连接
	requestToken := new(AuthToken)
	requestToken.Token = loginReply.Token

	//创建一个超时的上下文连接
	ctx, cancel := context.WithTimeout(context.Background(), dialTimeout)
	defer cancel()
	dialOpts = append(dialOpts, grpc.WithPerRPCCredentials(requestToken))
	conn, err = grpc.DialContext(ctx, address, dialOpts...)
	if err != nil {
		log.Fatalf("did not connect: %s", err)
	}

	client = pb.NewPingClient(conn)

	helloreply, err := client.SayHello(context.Background(), &pb.PingMessage{Greeting: "foo"})
	if err != nil {
		log.Fatalf("Error when calling SayHello: %s", err)
		panic("connect error")
	}
	log.Printf("Response from server: %s", helloreply.Greeting)

	//通过验证，返回客户端连接
	return conn, err
}

func NewRegisterClient(conn *grpc.ClientConn) (pb.ChaincodeSupportClient, error) {
	return pb.NewChaincodeSupportClient(conn), nil
}