package client

import (
	"fmt"

	"github.com/tidbcloud/ecosystem-service/config"
	"github.com/tidbcloud/serverless-service/idl/pbgen/proto/rpc"
	tigrpc "github.com/tidbcloud/tiinf/transport/tigrpc/client"
)

func GetBranchSvcClient(conf *config.ConfigWrapper) rpc.BranchServiceClient {
	conn, err := tigrpc.NewClientConn(conf.ServiceConfig.GrpcConfig.ServerlessSvcAddr, tigrpc.WithDefault())
	if err != nil {
		panic(fmt.Sprintf("dial branch-svc: %s", err))
	}

	return rpc.NewBranchServiceClient(conn)
}

func GetBranchUserSvcClient(conf *config.ConfigWrapper) rpc.UserServiceClient {
	conn, err := tigrpc.NewClientConn(conf.ServiceConfig.GrpcConfig.ServerlessSvcAddr, tigrpc.WithDefault())
	if err != nil {
		panic(fmt.Sprintf("dial branch-user-svc: %s", err))
	}

	return rpc.NewUserServiceClient(conn)
}