package paygrpc

import (
	"context"
	"time"

	"chatappgateway/internal/integration/upstream"
	commonpb "gitea.haiyihy.com/hy/chatappcommon/proto"
)

// Client 封装支付服务 gRPC client，并统一超时、重试和节点选择。
type Client struct {
	timeout time.Duration
	client  commonpb.ChatAppPayServiceClient
	pool    *upstream.Pool
}

// New 根据单个底层 gRPC client 构造支付服务调用器。
func New(client commonpb.ChatAppPayServiceClient, timeout time.Duration) *Client {
	return &Client{
		timeout: timeout,
		client:  client,
	}
}

// NewWithPool 根据下游节点池构造支付服务调用器。
func NewWithPool(pool *upstream.Pool) *Client {
	return &Client{pool: pool}
}

// Pay 调用支付服务最小支付接口。
func (c *Client) Pay(ctx context.Context, request *commonpb.PayRequest) (*commonpb.PayResponse, error) {
	if c.pool == nil {
		callCtx, cancel := context.WithTimeout(ctx, c.timeout)
		defer cancel()
		return c.client.Pay(callCtx, request)
	}

	return upstream.Call(ctx, c.pool, func(callCtx context.Context, handle upstream.Handle) (*commonpb.PayResponse, error) {
		return commonpb.NewChatAppPayServiceClient(handle.Conn).Pay(callCtx, request)
	})
}