<?php

namespace App\Console\Commands;

use App\Services\AIGeneration\AIImageGenerationService;
use Illuminate\Console\Command;
use Illuminate\Support\Facades\DB;

class CheckImageGenerationTasksCommand extends Command
{
    /**
     * The name and signature of the console command.
     *
     * @var string
     */
    protected $signature = 'AIGeneration:checkImgTasks';

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = '检查图片生成任务状态并更新';

    /**
     * Execute the console command.
     *
     * @param AIImageGenerationService $aiImageGenerationService
     * @return int
     */
    public function handle(AIImageGenerationService $aiImageGenerationService)
    {
        dLog('generate')->info('开始检查图片生成任务状态...');

        // 执行40s
        $time_start = time();
        try {
            $count = DB::table('mp_generate_pic_tasks')->where('status', 'processing')->count('id');
            while ($count > 0) {
                $time_diff = time() - $time_start;
                sleep(3);
                if ($time_diff > 50) break;
                $aiImageGenerationService->updatePendingTasks();
                $count = DB::table('mp_generate_pic_tasks')->where('status', 'processing')->count('id');
            }
            dLog('generate')->info('任务状态检查完成');
        } catch (\Exception $e) {
            dLog('generate')->error('任务状态检查失败: ' . $e->getMessage());
            return 1;
        }

        return 0;
    }
}