package api

import (
	"github.com/go-kratos/kratos/v2/log"
	"github.com/go-kratos/kratos/v2/registry"
	grpc2 "github.com/go-kratos/kratos/v2/transport/grpc"
	"github.com/go-kratos/kratos/v2/transport/http"
	v1 "github.com/go-saas/kit/payment/api/subscription/v1"
	_ "github.com/go-saas/kit/payment/i18n"
	"github.com/go-saas/kit/pkg/api"
	"github.com/go-saas/kit/pkg/conf"
	kitdi "github.com/go-saas/kit/pkg/di"
	"google.golang.org/grpc"
)

type GrpcConn grpc.ClientConnInterface
type HttpClient *http.Client

const ServiceName = "payment"

func NewGrpcConn(client *conf.Client, services *conf.Services, dis registry.Discovery, opt *api.Option, tokenMgr api.TokenManager, logger log.Logger, opts []grpc2.ClientOption) (GrpcConn, func()) {
	return api.NewGrpcConn(client, ServiceName, services, dis, opt, tokenMgr, logger, opts)
}

func NewHttpClient(client *conf.Client, services *conf.Services, dis registry.Discovery, opt *api.Option, tokenMgr api.TokenManager, logger log.Logger, opts []http.ClientOption) (HttpClient, func()) {
	return api.NewHttpClient(client, ServiceName, services, dis, opt, tokenMgr, logger, opts)
}

func NewSubscriptionGrpcClient(conn GrpcConn) v1.SubscriptionServiceServer {
	return v1.NewSubscriptionServiceClientProxy(v1.NewSubscriptionServiceClient(conn))
}

func NewSubscriptionInternalGrpcClient(conn GrpcConn) v1.SubscriptionInternalServiceServer {
	return v1.NewSubscriptionInternalServiceClientProxy(v1.NewSubscriptionInternalServiceClient(conn))
}

var GrpcProviderSet = kitdi.NewSet(NewGrpcConn, NewSubscriptionGrpcClient, NewSubscriptionInternalGrpcClient)