package rabbit

import (
	"context"

	rmq "github.com/rabbitmq/rabbitmq-amqp-go-client/pkg/rabbitmqamqp"
)

type rabbitClient struct {
	env  *rmq.Environment
	conn *rmq.AmqpConnection
}

func NewClient(ctx context.Context, endpoints []rmq.Endpoint) (*rabbitClient, error) {

	env := rmq.NewClusterEnvironment(endpoints)
	conn, err := client.env.NewConnection(ctx)
	if err != nil {
		return nil, err
	}

	return &rabbitClient{
		env:  env,
		conn: conn,
	}, nil
}

func (c *rabbitClient) Management() *rmq.AmqpManagement {
	return c.conn.Management()
}

func (c *rabbitClient) NewPublisher(ctx context.Context, destination rmq.ITargetAddress,
	options rmq.IConsumerOptions) (*rmq.Publisher, error) {

	return c.conn.NewPublisher(ctx, destination, options)
}

func (c *rabbitClient) NewConsumer(ctx context.Context, queueName string,
	options rmq.IConsumerOptions) (*rmq.Consumer, error) {

	return c.conn.NewConsumer(ctx, queueName, options)
}

func (c *rabbitClient) Close(ctx context.Context) error {
	return c.env.CloseConnections(ctx)
}