package rpc

import (
	"fmt"
	"gitee.com/MinKo-U/keyauth/apps/audit"
	"gitee.com/MinKo-U/keyauth/apps/endpoint"
	"gitee.com/MinKo-U/keyauth/apps/policy"
	"gitee.com/MinKo-U/keyauth/apps/role"
	"gitee.com/MinKo-U/keyauth/apps/token"
	"github.com/infraboard/mcenter/client/rpc"
	"github.com/infraboard/mcenter/client/rpc/auth"
	"github.com/infraboard/mcenter/client/rpc/resolver"
	"github.com/infraboard/mcube/logger"
	"github.com/infraboard/mcube/logger/zap"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
)

var (
	client *ClientSet
)

// SetGlobal todo
func SetGlobal(cli *ClientSet) {
	client = cli
}

// C Global
func C() *ClientSet {
	return client
}

// NewClient todo
func NewClient(conf *rpc.Config) (*ClientSet, error) {
	zap.DevelopmentSetup()
	log := zap.L()

	conn, err := grpc.Dial(
		fmt.Sprintf("%s://%s", resolver.Scheme, "keyauth"),
		grpc.WithTransportCredentials(insecure.NewCredentials()),
		grpc.WithPerRPCCredentials(auth.NewAuthentication(conf.ClientID, conf.ClientSecret)),
		grpc.WithBlock(),
	)

	if err != nil {
		return nil, err
	}

	return &ClientSet{
		conn: conn,
		log:  log,
	}, nil
}

// Client 客户端
type ClientSet struct {
	conn *grpc.ClientConn
	log  logger.Logger
}

// Token服务的SDK
func (c *ClientSet) Token() token.ServiceClient {
	return token.NewServiceClient(c.conn)
}

// Endpoint服务的SDK
func (c *ClientSet) Endpoint() endpoint.ServiceClient {
	return endpoint.NewServiceClient(c.conn)
}

// Endpoint服务的SDK
func (c *ClientSet) Policy() policy.ServiceClient {
	return policy.NewServiceClient(c.conn)
}

// Endpoint服务的SDK
func (c *ClientSet) Role() role.ServiceClient {
	return role.NewServiceClient(c.conn)
}

// Endpoint服务的SDK
func (c *ClientSet) Audit() audit.ServiceClient {
	return audit.NewServiceClient(c.conn)
}