package bootstrap

import (
	kRegistry "github.com/go-kratos/kratos/v2/registry"
	"github.com/tx7do/kratos-bootstrap/registry"

	conf "github.com/tx7do/kratos-bootstrap/api/gen/go/conf/v1"

	_ "github.com/tx7do/kratos-bootstrap/registry/consul"
	_ "github.com/tx7do/kratos-bootstrap/registry/etcd"
	_ "github.com/tx7do/kratos-bootstrap/registry/eureka"
	_ "github.com/tx7do/kratos-bootstrap/registry/kubernetes"
	_ "github.com/tx7do/kratos-bootstrap/registry/nacos"
	_ "github.com/tx7do/kratos-bootstrap/registry/servicecomb"
	_ "github.com/tx7do/kratos-bootstrap/registry/zookeeper"
)

// NewRegistry 创建一个注册客户端
func NewRegistry(cfg *conf.Registry) kRegistry.Registrar {
	if cfg == nil {
		return nil
	}

	creator := registry.GetRegistrarCreator(cfg.GetType())
	if creator == nil {
		panic("registrar creator not found:" + cfg.GetType())
		return nil
	}

	return creator(cfg)
}

// NewDiscovery 创建一个发现客户端
func NewDiscovery(cfg *conf.Registry) kRegistry.Discovery {
	if cfg == nil {
		return nil
	}

	creator := registry.GetDiscoveryCreator(cfg.GetType())
	if creator == nil {
		panic("discovery creator not found:" + cfg.GetType())
		return nil
	}

	return creator(cfg)
}