package client

import (
	"context"
	"crypto/tls"
	"fmt"
	"time"

	cmapi "certmanager/api/certmanagercontrol"

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

const (
	DefaultWrpcPort int = 20311
	DefaultGrpcPort int = 20312
)

type Access struct {
	Hostname string
	Port     int
	Username string
	Password string
}

type Control struct {
	conn   *grpc.ClientConn
	client cmapi.ControlClient
}

func NewControl(access *Access) (*Control, error) {
	var err error
	cont := &Control{}

	tlsConfig := &tls.Config{
		InsecureSkipVerify: true,
	}

	const dialTimeout time.Duration = 1 * time.Second
	const idleTimeout time.Duration = 5 * time.Second

	authCred := NewAuthCredential(access.Username, access.Password)
	dialOpts := []grpc.DialOption{
		grpc.WithTransportCredentials(credentials.NewTLS(tlsConfig)),
		grpc.WithPerRPCCredentials(authCred),
		grpc.WithBlock(),
		grpc.WithIdleTimeout(idleTimeout),
	}
	address := fmt.Sprintf("%s:%d", access.Hostname, access.Port)
	ctx, _ := context.WithTimeout(context.Background(), dialTimeout)
	conn, err := grpc.DialContext(ctx, address, dialOpts...)
	if err != nil {
		return cont, fmt.Errorf("Dial error: %v", err)
	}
	cont.conn = conn
	cont.client = cmapi.NewControlClient(conn)
	if cont.client == nil {
		return cont, fmt.Errorf("Nil control client")
	}
	return cont, err
}

func (cont *Control) Close() {
	if cont.client != nil {
		cont.conn.Close()
	}
}

func NewClient(access *Access) (cmapi.ControlClient, error) {
	var err error
	var cli cmapi.ControlClient

	tlsConfig := &tls.Config{
		InsecureSkipVerify: true,
	}
	const dialTimeout time.Duration = 1 * time.Second
	const idleTimeout time.Duration = 5 * time.Second

	authCred := NewAuthCredential(access.Username, access.Password)
	dialOpts := []grpc.DialOption{
		grpc.WithTransportCredentials(credentials.NewTLS(tlsConfig)),
		grpc.WithPerRPCCredentials(authCred),
		grpc.WithBlock(),
		grpc.WithIdleTimeout(idleTimeout),
	}
	address := fmt.Sprintf("%s:%d", access.Hostname, access.Port)
	ctx, _ := context.WithTimeout(context.Background(), dialTimeout)
	conn, err := grpc.DialContext(ctx, address, dialOpts...)
	if err != nil {
		return cli, fmt.Errorf("Dial error: %v", err)
	}
	cli = cmapi.NewControlClient(conn)
	return cli, err
}