mp_audio_manage/app/Services/AIGeneration/AIVideoGenerationService.php

<?php

namespace App\Services\AIGeneration;

use App\Models\MpGenerateVideoTask;
use App\Services\VolcEngineService;
use GuzzleHttp\Client;
use GuzzleHttp\Exception\GuzzleException;

class AIVideoGenerationService
{
    private $volcEngineService;
    private $httpClient;

    public function __construct(VolcEngineService $volcEngineService)
    {
        $this->volcEngineService = $volcEngineService;
        $this->httpClient = new Client([
            'timeout' => 300,
        ]);
    }

    /**
     * 异步创建视频生成任务
     *
     * @param array $params
     * @return MpGenerateVideoTask
     */
    public function createVideoGenerationTask(array $params): MpGenerateVideoTask
    {
        // 生成唯一的任务ID
        $taskId = 'vid_gen_' . time() . '_' . uniqid();

        // 创建任务记录
        $task = MpGenerateVideoTask::create([
            'task_id' => $taskId,
            'prompt' => $params['prompt'] ?? '',
            'first_frame_url' => $params['first_frame_url'] ?? null,
            'tail_frame_url' => $params['tail_frame_url'] ?? null,
            'seed' => $params['seed'] ?? -1,
            'frames' => $params['video_duration'] ? ($params['video_duration'] * 24 + 1) : 121,
            'video_resolution' => $params['video_resolution'] ?? '720P',
            'status' => MpGenerateVideoTask::STATUS_PENDING,
        ]);

        // 发送异步请求到即梦AI API创建任务
        $this->submitTaskToApi($task);

        return $task;
    }

    /**
     * 提交任务到即梦AI API
     *
     * @param MpGenerateVideoTask $task
     * @return void
     */
    private function submitTaskToApi(MpGenerateVideoTask $task): void
    {
        try {
            // 验证环境变量配置
            $accessKey = env('VOLC_AI_AK');
            $secretKey = env('VOLC_AI_SK');
                
            if (empty($accessKey) || empty($secretKey)) {
                $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                    'error_message' => '火山引擎访问密钥未配置，请检查环境变量AK和SK'
                ]);
                return;
            }

            // 根据任务类型选择对应的req_key
            if ($task->first_frame_url && $task->tail_frame_url) {
                $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_tail_v30' : 'jimeng_i2v_first_tail_v30_1080';
            }else if($task->first_frame_url){
                $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_v30' : 'jimeng_i2v_first_v30_1080';
            }else{
                $req_key = $task->video_resolution == '720P' ? 'jimeng_t2v_v30' : 'jimeng_t2v_v30_1080p';
            }
                
            // 构建即梦AI 3.0 视频生成API请求参数
            $apiParams = [
                'req_key' => $req_key,
                'prompt' => $task->prompt,
                'seed' => $task->seed,
                'frames' => $task->frames,
            ];

            // 添加图片URL参数
            if ($task->first_frame_url) {
                $apiParams['image_urls'] = [
                    $task->first_frame_url
                ];

                if ($task->tail_frame_url) $apiParams['image_urls'][] = $task->tail_frame_url;
            }
            
            // 调用即梦AI 3.0 视频生成API提交任务
            $response = $this->volcEngineService->request(
                'POST',
                'visual.volcengineapi.com', // 即梦AI API域名
                '/', // 使用根路径
                [], // Query参数
                json_encode($apiParams),
                $accessKey,
                $secretKey,
                'cv', // 服务标识
                'cn-north-1', // 区域
                'CVSync2AsyncSubmitTask',
                '2022-08-31',
                'application/json'
            );

            $responseData = json_decode($response['body'], true);
            dLog('generate')->info('即梦AI 3.0 视频生成API提交任务响应: ', $responseData);
                
            if ($responseData['code'] !== 10000) {
                // API返回错误
                $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                    'error_message' => $responseData['message'] ?? 'API Error',
                    'extra_params' => $responseData['data'] ?? []
                ]);
            } else {
                // 任务提交成功，更新状态为处理中
                $updateData = [];
                // 存储API返回的任务ID
                if (isset($responseData['data']['task_id'])) {
                    $updateData['task_id'] = $responseData['data']['task_id'];
                    $updateData['extra_params'] = $responseData['data'] ?? [];
                    $updateData['extra_params']['req_key'] = $req_key;
                }

                $task->updateStatus(MpGenerateVideoTask::STATUS_PROCESSING, $updateData);
            }
        } catch (\Exception $e) {
            // 记录错误
            $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                'error_message' => 'API请求失败: ' . $e->getMessage()
            ]);
        }
    }

    /**
     * 查询任务状态
     *
     * @param MpGenerateVideoTask $task
     * @return array
     */
    public function queryTaskStatus(MpGenerateVideoTask $task): array
    {
        try {
            // 获取API任务ID
            $apiTaskId = $task->task_id ?? null;
                
            if (!$apiTaskId) {
                return [
                    'status' => 'failed',
                    'error_message' => 'API任务ID不存在',
                    'result_url' => null
                ];
            }

            // 根据任务类型选择对应的req_key
            if ($task->first_frame_url && $task->tail_frame_url) {
                $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_tail_v30' : 'jimeng_i2v_first_tail_v30_1080';
            }else if($task->first_frame_url){
                $req_key = $task->video_resolution == '720P' ? 'jimeng_i2v_first_v30' : 'jimeng_i2v_first_v30_1080';
            }else{
                $req_key = $task->video_resolution == '720P' ? 'jimeng_t2v_v30' : 'jimeng_t2v_v30_1080p';
            }
    
            // 构建查询参数
            $apiParams = [
                'req_key' => $req_key,
                'task_id' => $apiTaskId,
                'req_json' => '{"return_url":true}' // 返回URL格式
            ];
    
            // 调用即梦AI 3.0 视频生成API查询任务状态
            $response = $this->volcEngineService->request(
                'POST',
                'visual.volcengineapi.com',
                '/', // 使用根路径
                [], // Query参数
                json_encode($apiParams),
                env('VOLC_AI_AK'),
                env('VOLC_AI_SK'),
                'cv',
                'cn-north-1',
                'CVSync2AsyncGetResult',
                '2022-08-31',
                'application/json'
            );
    
            $responseData = json_decode($response['body'], true);
            dLog('generate')->info('即梦AI 3.0 视频生成API查询任务状态响应: ', $responseData);
    
            if ($responseData['code'] !== 10000) {
                // API返回错误
                return [
                    'status' => 'failed',
                    'error_message' => $responseData['message'] ?? 'API Error',
                    'result_url' => null,
                    'result_json' => $responseData['data'] ?? []
                ];
            }
    
            // 解析API响应
            $result = $responseData['data'] ?? [];
            $taskStatus = $result['status'] ?? 'failed';
    
            $returnData = [
                'status' => $taskStatus,
                'error_message' => null,
                'result_url' => null,
                'result_json' => $result
            ];
    
            // 如果任务成功，获取结果URL
            if ($taskStatus === 'done' && isset($result['video_url'])) {
                if (isset($result['video_url'])) {
                    $video_name = 'ai_generation_' . time() . '_' . uniqid();
                    $video_ext = getVideoExtFromUrl($result['video_url']);
                    $video_name = $video_name . $video_ext;
                    // 将视频另存到tos
                    $url = uploadStreamByTos('video', file_get_contents($result['video_url']), $video_name);
                    $returnData['result_url'] = $url;
                }
                
                $returnData['status'] = 'success';
            } elseif (in_array($taskStatus, ['not_found', 'expired'])) {
                $returnData['status'] = 'failed';
                $returnData['error_message'] = '任务未找到或已过期';
            } elseif ($taskStatus === 'failed') {
                $returnData['status'] = 'failed';
                $returnData['error_message'] = '任务执行失败';
            }
            // in_queue 或 generating状态保持原status
    
            return $returnData;
        } catch (\Exception $e) {
            return [
                'status' => 'failed',
                'error_message' => $e->getMessage(),
                'result_url' => null
            ];
        }
    }

    /**
     * 更新所有待处理任务的状态
     *
     * @return void
     */
    public function updatePendingTasks(): void
    {
        // 获取所有处理中的任务
        $tasks = MpGenerateVideoTask::where('status', MpGenerateVideoTask::STATUS_PROCESSING)
            ->get();

        foreach ($tasks as $task) {
            $statusInfo = $this->queryTaskStatus($task);

            if ($statusInfo['status'] === 'success') {
                $task->updateStatus(MpGenerateVideoTask::STATUS_SUCCESS, [
                    'result_url'    => $statusInfo['result_url'],
                    'result_json'   => $statusInfo['result_json']
                ]);
            } elseif ($statusInfo['status'] === 'failed') {
                $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                    'error_message' => $statusInfo['error_message'],
                    'result_json'   => $statusInfo['result_json'] ?? []
                ]);
            }
            // 如果仍然是处理中状态，不做任何操作
            
            // 处理：如果任务处理超过12小时，标记为失败
            $processingTime = now()->diffInHours($task->created_at);
            if ($processingTime > 12) {
                $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                    'error_message' => '任务处理超时（超过12小时）'
                ]);
            }
        }
    }
    
    public function getTaskList($data)
    {
        $task_id = getProp($data, 'task_id');
        $status = getProp($data, 'status');

        $query = MpGenerateVideoTask::select('*');
        if ($task_id) {
            $query->where('task_id', $task_id);
        }
        if ($status) {
            $query->where('status', $status);
        }

        return $query->orderBy('created_at', 'desc')->paginate();
    }
}