package app

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

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

type articleGenerationTaskEnvelope struct {
	TaskID    int64  `json:"task_id"`
	TaskType  string `json:"task_type"`
	TenantID  int64  `json:"tenant_id"`
	ArticleID int64  `json:"article_id"`
}

type ArticleGenerationTaskEnvelope struct {
	TaskID    int64  `json:"task_id"`
	TaskType  string `json:"task_type"`
	TenantID  int64  `json:"tenant_id"`
	ArticleID int64  `json:"article_id"`
}

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

	payload, err := json.Marshal(envelope)
	if err != nil {
		return fmt.Errorf("marshal generation task envelope: %w", err)
	}

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

func decodeArticleGenerationTaskEnvelope(payload []byte) (*articleGenerationTaskEnvelope, error) {
	var envelope articleGenerationTaskEnvelope
	if err := json.Unmarshal(payload, &envelope); err != nil {
		return nil, fmt.Errorf("decode generation task envelope: %w", err)
	}
	if envelope.TaskID <= 0 {
		return nil, fmt.Errorf("generation task envelope task_id is required")
	}
	return &envelope, nil
}

func DecodeArticleGenerationTaskEnvelope(payload []byte) (*ArticleGenerationTaskEnvelope, error) {
	envelope, err := decodeArticleGenerationTaskEnvelope(payload)
	if err != nil {
		return nil, err
	}
	return &ArticleGenerationTaskEnvelope{
		TaskID:    envelope.TaskID,
		TaskType:  envelope.TaskType,
		TenantID:  envelope.TenantID,
		ArticleID: envelope.ArticleID,
	}, nil
}