package etcd

import (
	"context"
	"fmt"
	"time"

	"github.com/RichardKnop/machinery/v2/log"
	"github.com/RichardKnop/machinery/v2/tasks"
	clientv3 "go.etcd.io/etcd/client/v3"
)

type Delivery interface {
	Ack()
	Nack()
	Body() string
	Signature() *tasks.Signature
}

type deliver struct {
	ctx       context.Context
	client    *clientv3.Client
	signature *tasks.Signature
	value     string
}

func NewDelivery(ctx context.Context, key string) (Delivery, error) {
	return &deliver{}, nil
}

func (d *deliver) Ack() {
	keyPrefix := fmt.Sprintf("/machinery/v2/broker/pending_tasks/%s/%s", d.signature.RoutingKey, d.signature.UUID)
	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
	defer cancel()

	_, err := d.client.Delete(ctx, keyPrefix, clientv3.WithPrefix())
	if err != nil {
		log.ERROR.Printf("ack task %s err: %s", d.value, err)
	}
}

func (d *deliver) Nack() {
	keyPrefix := fmt.Sprintf("/machinery/v2/broker/pending_tasks/%s/%s/state", d.signature.RoutingKey, d.signature.UUID)
	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
	defer cancel()

	_, err := d.client.Delete(ctx, keyPrefix)
	if err != nil {
		log.ERROR.Printf("nack task %s err: %s", d.value, err)
	}
}

func (d *deliver) Signature() *tasks.Signature {
	return d.signature
}

func (d *deliver) Body() string {
	return d.value
}