package app

import (
	"context"
	"encoding/json"
	"fmt"

	"github.com/geo-platform/tenant-api/internal/shared/messaging/rabbitmq"
)

type kolAssistJob struct {
	TaskID   string `json:"task_id"`
	TenantID int64  `json:"tenant_id"`
}

type KolAssistJob = kolAssistJob

func publishKolAssistTask(ctx context.Context, client *rabbitmq.Client, job kolAssistJob) error {
	if client == nil {
		return fmt.Errorf("rabbitmq client is not configured")
	}

	payload, err := json.Marshal(job)
	if err != nil {
		return fmt.Errorf("marshal kol assist job: %w", err)
	}

	if err := client.PublishKolAssistTask(ctx, payload); err != nil {
		return fmt.Errorf("publish kol assist task: %w", err)
	}
	return nil
}

func decodeKolAssistTask(payload []byte) (*kolAssistJob, error) {
	var job kolAssistJob
	if err := json.Unmarshal(payload, &job); err != nil {
		return nil, fmt.Errorf("decode kol assist job: %w", err)
	}
	if job.TaskID == "" || job.TenantID <= 0 {
		return nil, fmt.Errorf("kol assist job is incomplete")
	}
	return &job, nil
}

func DecodeKolAssistTask(payload []byte) (*KolAssistJob, error) {
	job, err := decodeKolAssistTask(payload)
	if err != nil {
		return nil, err
	}
	return job, nil
}