<?php

namespace Tests\Feature\Console;

use App\Jobs\PollAiEditRequest;
use App\Jobs\ProcessAiEditRequest;
use App\Models\AiEditRequest;
use App\Models\AiProviderRun;
use App\Models\AiStyle;
use App\Models\Event;
use App\Models\Photo;
use Illuminate\Foundation\Testing\RefreshDatabase;
use Illuminate\Support\Facades\Queue;
use Tests\TestCase;

class AiEditsRecoverStuckCommandTest extends TestCase
{
    use RefreshDatabase;

    public function test_command_requeues_stuck_requests_with_scope_aware_job_selection(): void
    {
        [$queuedRequest, $processingRequest] = $this->createStuckRequests();

        Queue::fake();

        $this->artisan('ai-edits:recover-stuck', [
            '--minutes' => 10,
            '--requeue' => true,
        ])->assertExitCode(0);

        Queue::assertPushed(ProcessAiEditRequest::class, 1);
        Queue::assertPushed(PollAiEditRequest::class, 1);
        $this->assertDatabaseHas('ai_edit_requests', [
            'id' => $queuedRequest->id,
            'status' => AiEditRequest::STATUS_QUEUED,
        ]);
        $this->assertDatabaseHas('ai_edit_requests', [
            'id' => $processingRequest->id,
            'status' => AiEditRequest::STATUS_PROCESSING,
        ]);
    }

    public function test_command_can_mark_stuck_requests_as_failed(): void
    {
        [$queuedRequest, $processingRequest] = $this->createStuckRequests();

        $this->artisan('ai-edits:recover-stuck', [
            '--minutes' => 10,
            '--fail' => true,
        ])->assertExitCode(0);

        $this->assertDatabaseHas('ai_edit_requests', [
            'id' => $queuedRequest->id,
            'status' => AiEditRequest::STATUS_FAILED,
            'failure_code' => 'operator_recovery_marked_failed',
        ]);
        $this->assertDatabaseHas('ai_edit_requests', [
            'id' => $processingRequest->id,
            'status' => AiEditRequest::STATUS_FAILED,
            'failure_code' => 'operator_recovery_marked_failed',
        ]);
    }

    /**
     * @return array{0: AiEditRequest, 1: AiEditRequest}
     */
    private function createStuckRequests(): array
    {
        $event = Event::factory()->create(['status' => 'published']);
        $photo = Photo::factory()->for($event)->create([
            'tenant_id' => $event->tenant_id,
            'status' => 'approved',
        ]);
        $style = AiStyle::query()->create([
            'key' => 'recovery-style',
            'name' => 'Recovery Style',
            'provider' => 'runware',
            'provider_model' => 'runware-default',
            'requires_source_image' => true,
            'is_active' => true,
        ]);

        $queued = AiEditRequest::query()->create([
            'tenant_id' => $event->tenant_id,
            'event_id' => $event->id,
            'photo_id' => $photo->id,
            'style_id' => $style->id,
            'provider' => 'runware',
            'provider_model' => 'runware-default',
            'status' => AiEditRequest::STATUS_QUEUED,
            'safety_state' => 'pending',
            'prompt' => 'Transform image style.',
            'idempotency_key' => 'stuck-queued-1',
            'queued_at' => now()->subMinutes(45),
            'updated_at' => now()->subMinutes(45),
        ]);

        $processing = AiEditRequest::query()->create([
            'tenant_id' => $event->tenant_id,
            'event_id' => $event->id,
            'photo_id' => $photo->id,
            'style_id' => $style->id,
            'provider' => 'runware',
            'provider_model' => 'runware-default',
            'status' => AiEditRequest::STATUS_PROCESSING,
            'safety_state' => 'pending',
            'prompt' => 'Transform image style.',
            'idempotency_key' => 'stuck-processing-1',
            'queued_at' => now()->subMinutes(50),
            'started_at' => now()->subMinutes(40),
            'updated_at' => now()->subMinutes(40),
        ]);

        AiProviderRun::query()->create([
            'request_id' => $processing->id,
            'provider' => 'runware',
            'attempt' => 1,
            'provider_task_id' => 'runware-task-recovery-1',
            'status' => AiProviderRun::STATUS_RUNNING,
            'started_at' => now()->subMinutes(39),
        ]);

        return [$queued, $processing];
    }
}