mp_audio_manage/app/Services/AIGeneration/AIVideoGenerationService.php

<?php

namespace App\Services\AIGeneration;

use App\Libs\Utils;
use App\Models\MpGenerateVideoTask;
use App\Services\VolcEngineService;
use GuzzleHttp\Client;

class AIVideoGenerationService
{
    private $volcEngineService;

    public function __construct(VolcEngineService $volcEngineService)
    {
        $this->volcEngineService = $volcEngineService;
    }

    /**
     * 异步创建视频生成任务
     *
     * @param array $params
     * @return MpGenerateVideoTask
     */
    public function createJimengTask(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,
            'api_type' => 'jimeng',
        ]);

        // 发送异步请求到即梦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' => (int)$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;
            }
            
            dLog('generate')->info('图生视频参数: ', $apiParams);
            
            // 调用即梦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;
                    $updateData['extra_params']['request_id'] = isset($responseData['request_id']) ? $responseData['request_id'] : '';
                }

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

    /**
     * 查询任务状态(即梦AI)
     *
     * @param MpGenerateVideoTask $task
     * @return array
     */
    public function queryJimengTaskStatus(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 ?? []
                ];
            }
    
            // 解析API响应
            $result = $responseData['data'] ?? [];
            $taskStatus = $result['status'] ?? 'failed';
    
            $returnData = [
                'status' => $taskStatus,
                'error_message' => null,
                'result_url' => null,
                'result_json' => $responseData ?? []
            ];
    
            // 如果任务成功，获取结果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) {
            $api_type = getProp($task, 'api_type');
            switch($api_type) {
                case 'seedance':
                    $this->updateSeedanceTask($task);
                    break;
                case 'jimeng':
                    $this->updateJimengTask($task);
                    break;
                case 'keling':
                    $this->updateKelingOmniTask($task);
                    break;
                default:
                    break;
            }
        }
    }

    // 更新即梦AI视频任务状态
    private function updateJimengTask($task) {
        $statusInfo = $this->queryJimengTaskStatus($task);
        if (!isset($statusInfo['status'])) return;

        if ($statusInfo['status'] === 'success') {
            $task->updateStatus(MpGenerateVideoTask::STATUS_SUCCESS, $statusInfo);
        } elseif ($statusInfo['status'] === 'failed') {
            $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, $statusInfo);
        }
        // 如果仍然是处理中状态，不做任何操作
        
        // 处理：如果任务处理超过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();
    }

    /**
     * 创建 Seedance 1.5 Pro 视频生成任务
     *
     * @param array $params
     * @return MpGenerateVideoTask
     */
    public function createSeedanceTask(array $params): MpGenerateVideoTask
    {
        // 生成唯一的任务ID
        $taskId = 'seedance_' . time() . '_' . uniqid();

        // 创建任务记录，复用现有表结构
        $task = MpGenerateVideoTask::create([
            'task_id' => $taskId,
            'prompt' => $params['prompt'] ?? '',
            'first_frame_url' => $params['first_frame_url'] ?? '',
            'tail_frame_url' => $params['tail_frame_url'] ?? '',
            'seed' => $params['seed'] ?? -1,
            'frames' => $params['video_duration'] ? ($params['video_duration'] * 24 + 1) : null,
            'video_resolution' => $params['video_resolution'] ?? '720P',
            'status' => MpGenerateVideoTask::STATUS_PENDING,
            'api_type' => 'seedance', // 标识为 Seedance 1.5 Pro API
            'extra_params' => [
                'model' => $params['model'] ?? 'doubao-seedance-1-5-pro-251215',
                'content' => $params['content'],
                'callback_url' => $params['callback_url'] ?? '',
                'return_last_frame' => $params['return_last_frame'] ?? true,
                'service_tier' => $params['service_tier'] ?? 'default',
                'execution_expires_after' => $params['execution_expires_after'] ?? 172800,
                'generate_audio' => $params['generate_audio'] ?? true,
                'draft' => $params['draft'] ?? false,
                'ratio' => $params['ratio'] ?? '16:9',
                'duration' => $params['video_duration'] ?? 5,
                'camera_fixed' => $params['camera_fixed'] ?? false,
                'watermark' => $params['watermark'] ?? false,
            ]
        ]);

        // 提交任务到火山引擎 API
        $this->submitSeedanceTaskToApi($task);

        return $task;
    }

    /**
     * 提交 Seedance 任务到火山引擎 API
     *
     * @param MpGenerateVideoTask $task
     * @return void
     */
    private function submitSeedanceTaskToApi(MpGenerateVideoTask $task): void
    {
        $client = new Client(['verify'=>false,'timeout'=>120]);
        // try {
            // 验证环境变量配置
            $apiKey = env('VOLC_AI_API_KEY');
                
            if (empty($apiKey)) {
                $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                    'error_message' => '火山引擎访问密钥未配置，请检查环境变量API Key'
                ]);
                return;
            }

            $extraParams = $task->extra_params;
            
            // 构建火山引擎 Seedance API 请求参数
            $apiParams = [
                'model' => $extraParams['model'],
                'content' => $extraParams['content'],
            ];

            // 添加可选参数
            if (!empty($extraParams['callback_url'])) {
                $apiParams['callback_url'] = $extraParams['callback_url'];
            }
            if (isset($extraParams['return_last_frame'])) {
                $apiParams['return_last_frame'] = $extraParams['return_last_frame'];
            }
            if (!empty($extraParams['service_tier'])) {
                $apiParams['service_tier'] = $extraParams['service_tier'];
            }
            if (!empty($extraParams['execution_expires_after'])) {
                $apiParams['execution_expires_after'] = $extraParams['execution_expires_after'];
            }
            if (isset($extraParams['generate_audio'])) {
                $apiParams['generate_audio'] = $extraParams['generate_audio'];
            }
            if (isset($extraParams['draft'])) {
                $apiParams['draft'] = $extraParams['draft'];
            }
            if (!empty($extraParams['video_resolution'])) {
                $apiParams['resolution'] = strtolower($task->video_resolution);
            }
            if (!empty($extraParams['ratio'])) {
                $apiParams['ratio'] = $extraParams['ratio'];
            }
            if (!empty($extraParams['duration'])) {
                $apiParams['duration'] = (int)$extraParams['duration'];
            }
            if (isset($task->frames)) {
                // $apiParams['frames'] = $task->frames;
            }
            if ($task->seed !== -1) {
                $apiParams['seed'] = (int)$task->seed;
            }
            if (isset($extraParams['camera_fixed'])) {
                $apiParams['camera_fixed'] = $extraParams['camera_fixed'];
            }
            if (isset($extraParams['watermark'])) {
                $apiParams['watermark'] = $extraParams['watermark'];
            }
            
            dLog('generate')->info('Seedance 1.5 Pro 视频生成参数: ', $apiParams);
            
            // 调用火山引擎 Seedance API
            $response = $client->post('https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks', [
                'headers' => [
                    'Authorization' => 'Bearer ' . $apiKey,
                    'Content-Type' => 'application/json',
                ],
                'json' => $apiParams
            ]);


            $responseData = json_decode($response->getBody(), true);
            dLog('generate')->info('Seedance 1.5 Pro API 响应: ', $responseData);
                
            if (isset($responseData['error'])) {
                // API返回错误
                $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                    'error_message' => $responseData['error']['message'] ?? 'API Error',
                    'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
                ]);
            } elseif (isset($responseData['id'])) {
                // 任务提交成功，更新状态为处理中
                $updateData = [
                    'task_id' => $responseData['id'], // 使用 API 返回的任务 ID
                    'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
                ];
                $task->updateStatus(MpGenerateVideoTask::STATUS_PROCESSING, $updateData);
            } else {
                $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                    'error_message' => '未知的 API 响应格式',
                    'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
                ]);
            }
        // } catch (\Exception $e) {
        //     // 记录错误
        //     $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
        //         'error_message' => 'API请求失败: ' . $e->getMessage()
        //     ]);
        //     Utils::throwError('1001:'.$e->getMessage());
        // }
    }

    // 更新即梦AI视频任务状态
    private function updateSeedanceTask($task) {
        $statusInfo = $this->querySeedanceTaskStatus($task);
        if (!isset($statusInfo['status'])) return;

        if ($statusInfo['status'] === 'success') {
            $task->updateStatus(MpGenerateVideoTask::STATUS_SUCCESS, $statusInfo);
        } 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 > 48) {
            $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                'error_message' => '任务处理超时（超过48小时）'
            ]);
        }
    }

    /**
     * 查询 Seedance 任务状态
     *
     * @param MpGenerateVideoTask $task
     * @return array
     */
    public function querySeedanceTaskStatus(MpGenerateVideoTask $task): array
    {
        $client = new Client(['verify'=>false,'timeout'=>120]);
        try {

            // 如果任务已完成，直接返回结果
            if (in_array($task->status, [MpGenerateVideoTask::STATUS_SUCCESS, MpGenerateVideoTask::STATUS_FAILED])) {
                return [
                    'success' => true,
                    'data' => [
                        'id' => $task->task_id,
                        'status' => $this->mapTaskStatus($task->status),
                        'result_url' => $task->result_url,
                        'error_message' => $task->error_message,
                        'created_at' => $task->created_at,
                        'started_at' => $task->started_at,
                        'completed_at' => $task->completed_at
                    ]
                ];
            }

            // 调用火山引擎 API 查询任务状态
            $apiKey = env('VOLC_AI_API_KEY');
                
            if (empty($apiKey)) {
                return [
                    'status'    => MpGenerateVideoTask::STATUS_FAILED,
                    'result_json' => [],
                    'error_message' => '火山引擎访问密钥未配置'
                ];
            }

            // 构建查询 URL
            $url = 'https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks/' . $task->task_id;
            
            $response = $client->get($url, [
                'headers' => [
                    'Authorization' => 'Bearer ' . $apiKey,
                    'Content-Type' => 'application/json',
                ]
            ]);

            $responseData = json_decode($response->getBody(), true);
            dLog('generate')->info('Seedance 任务状态查询响应: ', $responseData);

            if (isset($responseData['error'])) {
                return [
                    'status'    => MpGenerateVideoTask::STATUS_FAILED,
                    'result_json' => [],
                    'error_message' => $responseData['error']['message'] ?? 'API查询失败'
                ];
            }

            // 解析任务状态
            $apiStatus = $responseData['status'] ?? 'unknown';
            $taskStatus = $this->mapApiStatusToTaskStatus($apiStatus);
            
            $updateData = ['result_json' => $responseData];
            
            if ($taskStatus === MpGenerateVideoTask::STATUS_SUCCESS) {
                $updateData['status'] = MpGenerateVideoTask::STATUS_SUCCESS;
                // 任务成功，处理结果
                if (isset($responseData['content']['video_url'])) {
                    $video_url = $responseData['content']['video_url'];
                    // 下载并保存视频
                    $video_name = 'ai_generation_' . time() . '_' . uniqid();
                    $video_ext = getVideoExtFromUrl($video_url);
                    $video_name = $video_name . $video_ext;
                    $url = uploadStreamByTos('video', file_get_contents($video_url), $video_name);
                    $updateData['result_url'] = $url;
                }
                
                // 处理尾帧图片（如果启用了 return_last_frame）
                if (isset($responseData['content']['last_frame_url']) && !empty($responseData['content']['last_frame_url'])) {
                    $pic_url = $responseData['content']['last_frame_url'];
                    $pic_name = 'ai_generation_' . time() . '_' . uniqid();
                    $pic_ext = getImgExtFromUrl($pic_url);
                    $pic_name = $pic_name . $pic_ext;
                    
                    if ($pic_ext === '.png') {
                        $comporessed_data = compressRemoteImageUrlToSize($pic_url);
                        $url = uploadStreamByTos('image', $comporessed_data, $pic_name);
                    }else {
                        // 将图片另存到tos
                        $url = uploadStreamByTos('image', file_get_contents($pic_url), $pic_name);
                    }
                    $updateData['last_frame_url'] = $url;
                }
                return $updateData;
            } elseif ($taskStatus === MpGenerateVideoTask::STATUS_FAILED) {
                $updateData = [
                    'status' => MpGenerateVideoTask::STATUS_FAILED,
                    'result_json' => [],
                    'error_message' => $responseData['error_message'] ?? '任务执行失败',
                ];
                return $updateData;
            }
            // 其他状态保持不变

            return [
                'success' => true,
                'data' => [
                    'id' => $task->task_id,
                    'status' => $this->mapTaskStatus($task->status),
                    'result_url' => $task->result_url,
                    'error_message' => $task->error_message,
                    'created_at' => $task->created_at,
                    'started_at' => $task->started_at,
                    'completed_at' => $task->completed_at,
                    'api_response' => $responseData
                ]
            ];
        } catch (\Exception $e) {
            dLog('generate')->info('Seedance 任务状态查询异常: '.$e->getMessage());
            return [
                'success' => false,
                'message' => '查询失败: ' . $e->getMessage()
            ];
        }
    }

    /**
     * 映射 API 状态到任务状态
     *
     * @param string $apiStatus
     * @return string
     */
    private function mapApiStatusToTaskStatus(string $apiStatus): string
    {
        switch ($apiStatus) {
            case 'queued':
            case 'running':
                return MpGenerateVideoTask::STATUS_PROCESSING;
            case 'succeeded':
                return MpGenerateVideoTask::STATUS_SUCCESS;
            case 'failed':
            case 'expired':
                return MpGenerateVideoTask::STATUS_FAILED;
            default:
                return MpGenerateVideoTask::STATUS_PROCESSING;
        }
    }

    /**
     * 生成可灵AI的JWT Token
     *
     * @return string
     */
    private function generateKelingJwtToken(): string
    {
        $ak = env('KELING_AK');
        $sk = env('KELING_SK');
        
        if (empty($ak) || empty($sk)) {
            throw new \Exception('可灵AI访问密钥未配置，请检查环境变量KELING_AK和KELING_SK');
        }

        // JWT头部
        $header = [
            'alg' => 'HS256',
            'typ' => 'JWT'
        ];

        // JWT载荷
        $payload = [
            'iss' => $ak,  // 签发者，使用Access Key
            'exp' => time() + 1800,  // 过期时间，30分钟后过期
            'nbf' => time() - 5  // 生效时间，5秒前开始生效
        ];

        // Base64URL编码
        $headerEncoded = $this->base64UrlEncode(json_encode($header));
        $payloadEncoded = $this->base64UrlEncode(json_encode($payload));

        // 创建签名
        $signature = hash_hmac('sha256', $headerEncoded . '.' . $payloadEncoded, $sk, true);
        $signatureEncoded = $this->base64UrlEncode($signature);

        // 返回完整的JWT token
        return $headerEncoded . '.' . $payloadEncoded . '.' . $signatureEncoded;
    }

    /**
     * Base64URL编码（JWT标准编码）
     *
     * @param string $data
     * @return string
     */
    private function base64UrlEncode(string $data): string
    {
        return rtrim(strtr(base64_encode($data), '+/', '-_'), '=');
    }

    /**
     * 创建可灵AI Omni视频生成任务
     *
     * @param array $params
     * @return MpGenerateVideoTask
     */
    public function createKelingOmniTask(array $params): MpGenerateVideoTask
    {
        // 生成唯一的任务ID
        $taskId = 'keling_omni_' . time() . '_' . uniqid();

        // 创建任务记录，复用现有表结构
        $task = MpGenerateVideoTask::create([
            'task_id' => $taskId,
            'prompt' => $params['prompt'] ?? '',
            'first_frame_url' => $params['first_frame_url'] ?? '',
            'tail_frame_url' => $params['tail_frame_url'] ?? '',
            'seed' => $params['seed'] ?? -1,
            'frames' => $params['video_duration'] ? ($params['video_duration'] * 24 + 1) : null,
            'video_resolution' => $params['video_resolution'] ?? '720P',
            'status' => MpGenerateVideoTask::STATUS_PENDING,
            'api_type' => 'keling', // 标识为可灵AI Omni API
            'extra_params' => [
                'api_type' => 'keling',
                'model' => $params['model'] ?? 'kling-video-o1',
                'prompt' => $params['prompt'] ?? '',
                'multi_prompt' => $params['multi_prompt'] ?? [],
                'shot_type' => $params['shot_type'] ?? 'customize',
                'image_list' => $params['image_list'] ?? [],
                'element_list' => $params['element_list'] ?? [],
                'sound' => $params['sound'] ?? 'off',
                // 'aspect_ratio' => $params['aspect_ratio'] ?? '16:9',
                'duration' => $params['video_duration'] ?? 5,
                'mode' => $params['mode'] ?? 'std',
                'callback_url' => $params['callback_url'] ?? '',
            ]
        ]);

        // 提交任务到可灵AI Omni API
        $this->submitKelingOmniTaskToApi($task);

        return $task;
    }

    /**
     * 提交可灵AI Omni任务到API
     *
     * @param MpGenerateVideoTask $task
     * @return void
     */
    private function submitKelingOmniTaskToApi(MpGenerateVideoTask $task): void
    {
        $client = new Client(['verify'=>false,'timeout'=>120]);
        
        try {
            // 生成JWT token
            $jwtToken = $this->generateKelingJwtToken();
            
            $extraParams = $task->extra_params;
            
            // 构建可灵AI Omni API请求参数
            $apiParams = [
                'model_name' => $extraParams['model'],
                'prompt' => $extraParams['prompt'],
                'mode' => $extraParams['mode'],
                // 'aspect_ratio' => $extraParams['aspect_ratio'],
                'duration' => (int)$extraParams['duration'],
            ];

            // 添加可选参数
            if (!empty($extraParams['multi_prompt'])) {
                $apiParams['multi_prompt'] = $extraParams['multi_prompt'];
            }
            if (!empty($extraParams['shot_type'])) {
                $apiParams['shot_type'] = $extraParams['shot_type'];
            }
            if (!empty($extraParams['image_list'])) {
                $apiParams['image_list'] = $extraParams['image_list'];
            }
            if (!empty($extraParams['element_list'])) {
                $apiParams['element_list'] = $extraParams['element_list'];
            }
            if (isset($extraParams['sound'])) {
                $apiParams['sound'] = $extraParams['sound'];
            }
            // if (!empty($extraParams['callback_url'])) {
            //     $apiParams['callback_url'] = $extraParams['callback_url'];
            // }
            
            dLog('generate')->info('可灵AI Omni视频生成参数: ', $apiParams);
            
            // 调用可灵AI Omni API
            $response = $client->post('https://api-beijing.klingai.com/v1/videos/omni-video', [
                'headers' => [
                    'Authorization' => 'Bearer ' . $jwtToken,
                    'Content-Type' => 'application/json',
                ],
                'json' => $apiParams
            ]);

            $responseData = json_decode($response->getBody(), true);
            dLog('generate')->info('可灵AI Omni API响应: ', $responseData);
                
            if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
                // API返回错误
                $errorMessage = $responseData['error']['message'] ?? $responseData['message'] ?? 'API Error';
                $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                    'error_message' => $errorMessage,
                    'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
                ]);
            } elseif (isset($responseData['data']['task_id'])) {
                // 任务提交成功，更新状态为处理中
                $updateData = [
                    'task_id' => $responseData['data']['task_id'], // 使用API返回的任务ID
                    'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
                ];
                $task->updateStatus(MpGenerateVideoTask::STATUS_PROCESSING, $updateData);
            } else {
                $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                    'error_message' => '未知的API响应格式',
                    'extra_params' => array_merge($task->extra_params, ['api_response' => $responseData])
                ]);
            }
        } catch (\Exception $e) {
            // 记录错误
            $task->updateStatus(MpGenerateVideoTask::STATUS_FAILED, [
                'error_message' => 'API请求失败: ' . $e->getMessage()
            ]);
            dLog('generate')->error('可灵AI Omni API请求异常: ' . $e->getMessage());
        }
    }

    /**
     * 查询可灵AI Omni任务状态
     *
     * @param MpGenerateVideoTask $task
     * @return array
     */
    public function queryKelingOmniTaskStatus(MpGenerateVideoTask $task): array
    {
        $client = new Client(['verify'=>false,'timeout'=>120]);
        
        try {
            // 如果任务已完成，直接返回结果
            if (in_array($task->status, [MpGenerateVideoTask::STATUS_SUCCESS, MpGenerateVideoTask::STATUS_FAILED])) {
                return [
                    'success' => true,
                    'data' => [
                        'task_id' => $task->task_id,
                        'task_status' => $this->mapTaskStatusToKelingStatus($task->status),
                        'result_url' => $task->result_url,
                        'error_message' => $task->error_message,
                        'created_at' => $task->created_at,
                        'updated_at' => $task->updated_at
                    ]
                ];
            }

            // 生成JWT token
            $jwtToken = $this->generateKelingJwtToken();
            
            // 构建查询URL
            $url = 'https://api-beijing.klingai.com/v1/videos/omni-video/' . $task->task_id;
            
            $response = $client->get($url, [
                'headers' => [
                    'Authorization' => 'Bearer ' . $jwtToken,
                    'Content-Type' => 'application/json',
                ]
            ]);

            $responseData = json_decode($response->getBody(), true);
            dLog('generate')->info('可灵AI Omni任务状态查询响应: ', $responseData);

            if (isset($responseData['error']) || (isset($responseData['code']) && $responseData['code'] !== 0)) {
                return [
                    'status' => 'failed',
                    'error_message' => $responseData['error']['message'] ?? $responseData['message'] ?? 'API查询失败',
                    'result_json' => $responseData
                ];
            }

            // 解析任务状态
            $taskData = $responseData['data'] ?? [];
            $taskStatus = $taskData['task_status'] ?? 'unknown';
            
            $updateData = ['result_json' => $responseData];
            
            if ($taskStatus === 'succeed') {
                $updateData['status'] = 'success';
                // 任务成功，处理结果
                if (isset($taskData['task_result']['videos']) && !empty($taskData['task_result']['videos'])) {
                    $video = $taskData['task_result']['videos'][0];
                    $video_url = $video['url'];
                    
                    // 下载并保存视频
                    $video_name = 'ai_generation_' . time() . '_' . uniqid();
                    $video_ext = getVideoExtFromUrl($video_url);
                    $video_name = $video_name . $video_ext;
                    $url = uploadStreamByTos('video', file_get_contents($video_url), $video_name);
                    $updateData['result_url'] = $url;
                }
                return $updateData;
            } elseif ($taskStatus === 'failed') {
                $updateData = [
                    'status' => 'failed',
                    'error_message' => '任务执行失败',
                    'result_json' => $responseData
                ];
                return $updateData;
            }
            // 其他状态（processing等）保持不变

            return [
                'success' => true,
                'data' => [
                    'task_id' => $task->task_id,
                    'task_status' => $taskStatus,
                    'result_url' => $task->result_url,
                    'error_message' => $task->error_message,
                    'created_at' => $task->created_at,
                    'updated_at' => $task->updated_at,
                    'api_response' => $responseData
                ]
            ];
        } catch (\Exception $e) {
            dLog('generate')->error('可灵AI Omni任务状态查询异常: ' . $e->getMessage());
            return [
                'success' => false,
                'message' => '查询失败: ' . $e->getMessage()
            ];
        }
    }

    /**
     * 映射任务状态到可灵AI状态格式
     *
     * @param string $taskStatus
     * @return string
     */
    private function mapTaskStatusToKelingStatus(string $taskStatus): string
    {
        switch ($taskStatus) {
            case MpGenerateVideoTask::STATUS_PENDING:
                return 'submitted';
            case MpGenerateVideoTask::STATUS_PROCESSING:
                return 'processing';
            case MpGenerateVideoTask::STATUS_SUCCESS:
                return 'succeed';
            case MpGenerateVideoTask::STATUS_FAILED:
                return 'failed';
            default:
                return 'unknown';
        }
    }

    /**
     * 更新可灵AI Omni视频任务状态
     *
     * @param MpGenerateVideoTask $task
     * @return void
     */
    private function updateKelingOmniTask($task): void
    {
        $statusInfo = $this->queryKelingOmniTaskStatus($task);
        if (!isset($statusInfo['status'])) return;

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

    /**
     * 映射任务状态到前端显示状态
     *
     * @param string $taskStatus
     * @return string
     */
    private function mapTaskStatus(string $taskStatus): string
    {
        switch ($taskStatus) {
            case MpGenerateVideoTask::STATUS_PENDING:
                return 'queued';
            case MpGenerateVideoTask::STATUS_PROCESSING:
                return 'running';
            case MpGenerateVideoTask::STATUS_SUCCESS:
                return 'succeeded';
            case MpGenerateVideoTask::STATUS_FAILED:
                return 'failed';
            default:
                return 'unknown';
        }
    }
}