mp_audio_manage/app/Console/Commands/CheckVideoGenerationTasksCommand.php

<?php

namespace App\Console\Commands;

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

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

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

    /**
     * Execute the console command.
     *
     * @param AIVideoGenerationService $videoGenerationService
     * @return int
     */
    public function handle(AIVideoGenerationService $videoGenerationService)
    {
        dLog('command')->info('开始检查视频生成任务状态...');

        // 执行50s
        $time_start = time();
        try {
            $count = DB::table('mp_generate_video_tasks')->where('status', 'processing')->count('id');
            while ($count > 0) {
                $time_diff = time() - $time_start;
                sleep(3);
                if ($time_diff > 50) break;
                $videoGenerationService->updatePendingTasks();
                $count = DB::table('mp_generate_video_tasks')->where('status', 'processing')->count('id');
            }
            
            // 更新分镜表中的视频任务状态
            $this->updateSegmentVideoStatus();
            
            dLog('command')->info('视频任务状态检查完成');
        } catch (\Exception $e) {
            dLog('command')->error('视频任务状态检查失败: ' . $e->getMessage());
            logDB('command', 'error', '定时任务：视频生成任务状态检查失败', ['error' => $e->getMessage()]);
            return 1;
        }

        return 0;
    }

    /**
     * 更新分镜表中的视频任务状态
     * 
     * @return void
     */
    private function updateSegmentVideoStatus()
    {
        try {
            // 获取所有状态为"生成中"的分镜
            $segments = DB::table('mp_episode_segments')
                ->where('video_task_status', '生成中')
                ->whereNotNull('video_task_id')
                ->where('video_task_id', '<>', '')
                ->select('segment_id', 'video_task_id')
                ->get();

            if ($segments->isEmpty()) {
                return;
            }

            dLog('command')->info('开始更新分镜视频任务状态', ['count' => $segments->count()]);

            $successCount = 0;
            $failedCount = 0;
            $processingCount = 0;

            foreach ($segments as $segment) {
                $videoTaskId = $segment->video_task_id;
                $segmentId = $segment->segment_id;

                // 查询视频生成任务状态
                $videoTask = DB::table('mp_generate_video_tasks')
                    ->where('id', $videoTaskId)
                    ->select('id', 'status', 'result_url', 'last_frame_url', 'error_message', 'result_json', 'extra_params')
                    ->first();

                if (!$videoTask) {
                    dLog('command')->warning('分镜关联的视频任务不存在', [
                        'segment_id' => $segmentId,
                        'video_task_id' => $videoTaskId
                    ]);
                    continue;
                }

                $updateData = ['updated_at' => date('Y-m-d H:i:s')];

                // 根据任务状态更新分镜表
                if ($videoTask->status === 'success') {
                    $updateData['video_task_status'] = '已完成';
                    $updateData['video_url'] = $videoTask->result_url;
                    $updateData['current_type'] = 2;
                    
                    // 如果有尾帧图片，也更新
                    if (!empty($videoTask->last_frame_url)) {
                        $updateData['last_frame_url'] = $videoTask->last_frame_url;
                    }
                    
                    // 计算视频时长
                    $video_duration = 0;
                    if (!empty($videoTask->result_json)) {
                        $resultJson = is_array($videoTask->result_json) ? $videoTask->result_json : json_decode($videoTask->result_json, true);
                        
                        if (isset($resultJson['duration'])) {
                            // 如果直接返回duration，使用该值
                            $video_duration = (int)$resultJson['duration'];
                        } elseif (isset($resultJson['frames']) && isset($resultJson['framespersecond'])) {
                            // 如果返回frames和framespersecond，计算时长
                            $frames = (int)$resultJson['frames'];
                            $fps = (int)$resultJson['framespersec8ond'];
                            if ($fps > 0) {
                                $video_duration = (int)floor($frames / $fps);
                            }
                        }
                    }
                    
                    // 如果没有计算出时长，使用任务表中的设定值
                    if ($video_duration <= 0 && !empty($videoTask->extra_params)) {
                        $extraParams = is_array($videoTask->extra_params) ? $videoTask->extra_params : json_decode($videoTask->extra_params, true);
                        if (isset($extraParams['duration'])) {
                            $video_duration = (int)$extraParams['duration'];
                        }
                    }
                    
                    // 只有当video_duration > 0时才添加到更新数据中
                    if ($video_duration > 0) {
                        $updateData['video_duration'] = $video_duration;
                        $updateData['video_time_point_start'] = 0;
                        // $updateData['video_time_point_end'] = $video_duration;
                        $updateData['video_time_point_end'] = $segment->audio_duration ?? 0;
                    }
                    
                    $successCount++;
                    
                    // 更新分镜视频成功后新增对话记录
                    $this->addVideoGenerationRecords($segmentId, $videoTask->result_url);
                    
                    dLog('command')->info('分镜视频生成成功', [
                        'segment_id' => $segmentId,
                        'video_task_id' => $videoTaskId,
                        'video_url' => $videoTask->result_url,
                        'video_duration' => $video_duration
                    ]);
                    
                } elseif ($videoTask->status === 'failed') {
                    $updateData['video_task_status'] = '失败';
                    
                    $failedCount++;
                    
                    dLog('command')->error('分镜视频生成失败', [
                        'segment_id' => $segmentId,
                        'video_task_id' => $videoTaskId,
                        'error' => $videoTask->error_message
                    ]);
                    logDB('command', 'error', '分镜视频生成失败', [
                        'segment_id' => $segmentId,
                        'video_task_id' => $videoTaskId,
                        'error' => $videoTask->error_message
                    ]);
                    
                } else {
                    // 仍在处理中，不更新
                    $processingCount++;
                    continue;
                }

                // 更新分镜表
                DB::table('mp_episode_segments')
                    ->where('segment_id', $segmentId)
                    ->update($updateData);
            }

            dLog('command')->info('分镜视频任务状态更新完成', [
                'total' => $segments->count(),
                'success' => $successCount,
                'failed' => $failedCount,
                'processing' => $processingCount
            ]);

        } catch (\Exception $e) {
            dLog('command')->error('更新分镜视频任务状态异常: ' . $e->getMessage());
            logDB('command', 'error', '更新分镜视频任务状态异常', [
                'error' => $e->getMessage(),
                'trace' => $e->getTraceAsString()
            ]);
        }
    }

    /**
     * 添加视频生成对话记录
     * 
     * @param int $segmentId 分镜ID
     * @param string $videoUrl 视频URL
     * @return void
     */
    private function addVideoGenerationRecords($segmentId, $videoUrl)
    {
        try {
            // 获取分镜信息
            $segment = DB::table('mp_episode_segments')
                ->where('segment_id', $segmentId)
                ->select('uid', 'anime_id', 'episode_number', 'tail_frame')
                ->first();

            if (!$segment) {
                dLog('command')->warning('分镜信息不存在，无法添加对话记录', ['segment_id' => $segmentId]);
                return;
            }

            $now = date('Y-m-d H:i:s');
            
            // 保存对话记录
            $records = [
                [
                    'uid'           => $segment->uid,
                    'anime_id'      => $segment->anime_id,
                    'sequence'      => $segment->episode_number,
                    'role'          => 'user',
                    'content'       => '图片转视频',
                    'segment_id'    => $segmentId,
                    'video_url'     => '',
                    'created_at'    => $now,
                    'updated_at'    => $now
                ],
                [
                    'uid'           => $segment->uid,
                    'anime_id'      => $segment->anime_id,
                    'sequence'      => $segment->episode_number,
                    'role'          => 'assistant',
                    'content'       => $segment->tail_frame ?: '',
                    'segment_id'    => $segmentId,
                    'video_url'     => $videoUrl,
                    'created_at'    => $now,
                    'updated_at'    => $now
                ]
            ];

            DB::table('mp_anime_records')->insert($records);
            
            dLog('command')->info('视频生成对话记录添加成功', [
                'segment_id' => $segmentId,
                'video_url' => $videoUrl,
                'uid' => $segment->uid,
                'anime_id' => $segment->anime_id
            ]);

        } catch (\Exception $e) {
            dLog('command')->error('添加视频生成对话记录失败: ' . $e->getMessage(), [
                'segment_id' => $segmentId,
                'video_url' => $videoUrl,
                'error' => $e->getMessage()
            ]);
        }
    }
}