package queue

import (
	"context"
	"fmt"
	"log/slog"
	awsc "queryorchestration/internal/serviceconfig/aws"
	"time"

	"github.com/aws/aws-sdk-go-v2/aws"
	"github.com/aws/aws-sdk-go-v2/config"
	"github.com/aws/aws-sdk-go-v2/service/sqs"
	"github.com/aws/aws-sdk-go-v2/service/sqs/types"
)

type QueueConfig struct {
	AWSEndpointUrlSQS string `env:"AWS_ENDPOINT_URL_SQS"`
	QueueClient       SQSClient
}

type ConfigProvider interface {
	awsc.ConfigProvider
	SetQueueClient(context.Context) error
	GetQueueClient() SQSClient
	PingQueueByURL(context.Context, string) error
	SendToQueue(context.Context, *SendParams) error
	ReceiveFromQueue(context.Context, *ReceiveParams) (*sqs.ReceiveMessageOutput, error)
	DeleteFromQueue(context.Context, *DeleteParams) error
	SetSQSEndpoint(string)
	GetSQSEndpoint() string
}

func (c *QueueConfig) GetQueueClient() SQSClient {
	return c.QueueClient
}

func (c *QueueConfig) GetSQSEndpoint() string {
	return c.AWSEndpointUrlSQS
}

func (c *QueueConfig) PingQueueByURL(ctx context.Context, url string) error {
	timeout := time.After(30 * time.Second)
	tick := time.NewTicker(2 * time.Second)
	defer tick.Stop()

	input := &sqs.GetQueueAttributesInput{
		QueueUrl:       aws.String(url),
		AttributeNames: []types.QueueAttributeName{types.QueueAttributeNameApproximateNumberOfMessages},
	}

	for {
		select {
		case <-timeout:
			return fmt.Errorf("unable to ping queue by url: %s", url)
		case <-tick.C:
			r, err := c.QueueClient.GetQueueAttributes(ctx, input)
			if err == nil {
				slog.Info("Successful queue ping", "url", url, "approx_msgs", r.Attributes[string(types.QueueAttributeNameApproximateNumberOfMessages)])
				return nil
			}
			slog.Debug("Attempted queue ping", "url", url, "address", c.GetSQSEndpoint())
		case <-ctx.Done():
			return ctx.Err()
		}
	}
}

func (c *QueueConfig) SetQueueClient(ctx context.Context) error {
	cfg, err := config.LoadDefaultConfig(ctx)
	if err != nil {
		return fmt.Errorf("unable to load SDK config: %v", err)
	}

	c.QueueClient = sqs.NewFromConfig(cfg)

	return nil
}

func (c *QueueConfig) SetSQSEndpoint(e string) {
	c.AWSEndpointUrlSQS = e
}