<?php

namespace app\api\logic\sv;


use app\api\logic\service\TokenLogService;
use app\common\model\sv\SvVideoSetting;
use app\common\model\sv\SvVideoTask;
use think\facade\Db;
use think\facade\Log;
use app\common\enum\user\AccountLogEnum;
use app\common\logic\AccountLogLogic;
use app\common\model\user\User;
use app\common\model\user\UserTokensLog;
use app\common\service\FileService;
use TencentCloud\Teo\V20220106\Models\Sv;

/**
 * SvVideoTaskLogic
 * @desc 视频设置逻辑处理
 */
class SvVideoTaskLogic extends SvBaseLogic
{


    const AUDIO_TRAINING_YM = 'audioTrainingYm'; //文字转音频
    const VIDEO_TRAINING_YM = 'videoTrainingYm'; //视频训练

    const AUDIO_TRAINING_YMT = 'audioTrainingYmt'; //文字转音频
    const VIDEO_TRAINING_YMT = 'videoTrainingYmt'; //视频训练
    /**
     * 处理音频合成任务
     * @param string $taskId 任务ID
     * @return void
     */
    public static function compositeAudioCron(string $taskId = '')
    {
        try {
            // 构建查询条件：待处理状态的任务
            $where = [
                ['status', '=', 0], // 待处理
                ['model_version', '=', 4],
                ['tries', '<', 5]
            ];
            
            if (!empty($taskId)) {
                $where[] = ['task_id', '=', $taskId];
            }

            // 获取待处理的任务，限制5条
            $tasks = SvVideoTask::where($where)
                ->order('tries DESC, id ASC')
                ->limit(5)
                ->select();

            if ($tasks->isEmpty()) {
              //  Log::channel('sv')->info('没有需要处理的音频任务');
                return;
            }

            foreach ($tasks as $task) {
                try {
                  //  Log::channel('sv')->info('开始处理音频任务'. json_encode(['task_id' => $task->task_id]));
                    SvVideoSetting::where('id', $task->video_setting_id)->where('status','1')->update(['status' => 1]);
                    // 更新状态为音频合成中

                    switch ($task->model_version) {
                       
                        case 4:
                            $scene = self::AUDIO_TRAINING_YM;
                            break;
                        case 6:
                            $scene = self::AUDIO_TRAINING_YMT;
                            break;
                        default:
                            $scene = self::AUDIO_TRAINING_YM;
                            break;
                    }
                    // 请求合成音频
                    $response = self::requestUrl([
                        'msg' => $task->msg,
                        'voice_id' => $task->voice_id,
                        'language' => 0,
                        'notify_url' => '/api/sv.videoTask/notify'
                    ], $scene, $task->user_id, $task->task_id);

                    if (!isset($response['id']) || empty($response['id'])) {
                        $task->tries = $task->tries +1;
                        if ( $task->tries == 5){
                            $task->remark = '音频合成5次失败';;
                            $task->status = 2;
                        }

                        $task->save();
                        //Log::channel ('sv')->info('音频合成失败'. json_encode($response));
                       return ;
                    }
                    $task->tries = 0;
                    // 更新音频信息
                    $task->audio_id = $response['id'];
                    $task->status = 1; // 更新状态为音频合成成功
                    if (!$task->save()) {
                        throw new \Exception("更新音频结果失败");
                    }


                } catch (\Exception $e) {
                    $task->tries = $task->tries +1;
                  //  Log::channel('sv')->info('音频任务处理失败'.  $task->task_id. $e->getMessage());
                    $task->remark = $e->getMessage();
                    $task->save();
                }
            }

        } catch (\Exception $e) {
        
            Log::channel('sv')->info('批量处理音频任务失败'. $e->getMessage());
        }
    }

    /**
     * 处理视频合成任务
     * @param string $taskId 任务ID
     * @return void
     */
    public static function compositeVideoCron(string $taskId = '')
    {
        try {
            // 构建查询条件：音频合成成功的任务
            $where = [
                ['status', '=', 3], // 音频合成成功
                ['model_version', '=', 4],
                ['tries', '<', 5]
            ];
            
            if (!empty($taskId)) {
                $where[] = ['task_id', '=', $taskId];
            }

            // 获取待处理的任务，限制5条
            $tasks = SvVideoTask::where($where)
                ->order('speed DESC, id ASC')
                ->limit(5)
                ->select();

            if ($tasks->isEmpty()) {
                //Log::channel('sv')->info('没有需要处理的视频任务');
                return;
            }

            foreach ($tasks as $task) {
                try {
                  //  Log::channel('sv')->info('开始处理视频任务'.'task_id'. $task->task_id);

                    // 更新状态为视频合成中
                    switch ($task->model_version) {
                     
                        case 4:
                            $scene = self::VIDEO_TRAINING_YM;
                            break;
                        case 6:
                            $scene = self::VIDEO_TRAINING_YMT;
                            break;
                        default:
                            $scene = self::VIDEO_TRAINING_YM;
                            break;
                    }

                    $response = self::requestUrl([
                        'name'      => $task->name,
                        'avatar_id' => $task->anchor_id,
                        'video_url' => $task->upload_video_url,
                        'audio_url' => $task->audio_url,
                        'priority' => $task->speed,
                        'notify_url' => '/api/sv.videoTask/notify'
                    ], $scene, $task->user_id, $task->task_id);

                    if (!isset($response['id']) || empty($response['id'])) {
                        $task->tries = $task->tries + 1;
                        if ( $task->tries == 5){
                            $task->status = 5;
                            $task->remark = '视频合成5次失败';;
                        }
                        $task->save();
                        return;
                    }

                    // 更新视频结果
                    $task->result_id = $response['id'];
                    $task->status = 4;
                    if (!$task->save()) {
                        throw new \Exception("更新视频结果失败");
                    }
                  //  Log::channel('sv')->info('视频合成成功'. 'task_id' .$task->task_id.$task->result_id);

                } catch (\Exception $e) {
                   // Log::channel('sv')->info('视频任务处理失败'.'task_id'.$task->task_id.$e->getMessage());
                    $task->tries = $task->tries + 1;
                    $task->remark = $e->getMessage();
                    $task->save();
                }
            }

        } catch (\Exception $e) {
            Log::channel('sv')->info('批量处理视频任务失败' . $e->getMessage());
        }
    }

    /**
     * 查询音频合成结果任务
     * @param string $taskId 任务ID
     * @return void
     */
    public static function queryAudioCron(string $taskId = '')
    {
        try {
            // 构建查询条件：音频合成成功的任务
            $where = [
                ['status', '=', 1], // 音频已合成
                ['model_version', '=', 4],
                ['tries', '<', 20]
            ];

            if (!empty($taskId)) {
                $where[] = ['task_id', '=', $taskId];
            }

            // 获取待处理的任务，限制5条
            $tasks = SvVideoTask::where($where)
                ->order('tries DESC, id ASC')
                ->limit(2)
                ->select()->toArray();

            if (!$tasks) {
                //Log::channel('sv')->info('没有需要查询音频任务');
                return;
            }
            $methodMap = [
                4 => 'detailYm',
                6 => 'detailYmt',
            ];

            $typeMap = [
                4 => AccountLogEnum::TOKENS_DEC_HUMAN_AUDIO_YM,
                6 => AccountLogEnum::TOKENS_DEC_HUMAN_AUDIO_YMT,
            ];

            foreach ($tasks as $task) {
                try {
                   // Log::channel('sv')->info('开始查询音频任务' . $task->task_id);
                    unset($task['create_time']);
                    $task['update_time'] = time();
                    $method = $methodMap[$task['model_version']] ?? 'detailYmt';
                    $response = \app\common\service\ToolsService::Human()->$method([
                        'type' => 3,
                        'id' =>  $task['audio_id']
                    ]);

                    if (isset($response['data']['status']) && $response['data']['status'] == 3) {
                        $upload_url = FileService::downloadFileBySource($response['data']['speech_url'], 'audio');
                        $task['tries'] = 0;
                        $task['status'] = 3;
                        $task['audio_result_url'] = $response['data']['speech_url'];
                        $task['audio_url'] =  $upload_url;

                    }elseif(isset($response['data']['status']) && $response['data']['status'] == 4) {
                        $typeID = $typeMap[$task['model_version']] ?? AccountLogEnum::TOKENS_DEC_HUMAN_AUDIO_YMT;
                        //查询是否已返还
                        if (UserTokensLog::where('user_id',  $task['user_id'])->where('change_type', $typeID)->where('action', 1)->where('task_id',  $task['task_id'])->count() == 0) {

                            $points = UserTokensLog::where('user_id',  $task['user_id'])->where('change_type', $typeID)->where('task_id',  $task['task_id'])->value('change_amount') ?? 0;

                            AccountLogLogic::recordUserTokensLog(false,  $task['user_id'], $typeID, $points,  $task['task_id']);
                        }
                        $task['status'] = 2;

                    }
                    $task['tries'] =  $task['tries'] + 1;
                    SvVideoTask::update($task, ['id' => $task['id']]);

                } catch (\Exception $e) {
                    $task['tries'] = $task['tries'] + 1;
                   // Log::channel('sv')->info('视频任务处理失败'. $task['task_id'] .$e->getMessage());
                    $task['remark'] = $e->getMessage();
                    SvVideoTask::update($task, ['id' => $task['id']]);
                }
            }

        } catch (\Exception $e) {
            Log::channel('sv')->info('批量处理视频任务失败'.  $e->getMessage());
        }
    }

    /**
     * 请求上游接口与计费
     * @param array $request
     * @param string $scene
     * @param int $userId
     * @param string $taskId
     * @return array
     * @throws \Exception
     */
    private static function requestUrl(array $request, string $scene, int $userId, string $taskId): array
    {

        $requestService = \app\common\service\ToolsService::Human();

        [$tokenScene, $tokenCode] = match ($scene) {
            self::AUDIO_TRAINING_YM => ['human_audio_ym', AccountLogEnum::TOKENS_DEC_HUMAN_AUDIO_YM],
            self::VIDEO_TRAINING_YM => ['human_video_ym', AccountLogEnum::TOKENS_DEC_HUMAN_VIDEO_YM],
            self::AUDIO_TRAINING_YMT => ['human_audio_ymt', AccountLogEnum::TOKENS_DEC_HUMAN_AUDIO_YMT],
            self::VIDEO_TRAINING_YMT => ['human_video_ymt', AccountLogEnum::TOKENS_DEC_HUMAN_VIDEO_YMT],
        };

        //计费
        $unit = TokenLogService::checkToken($userId, $tokenScene);

        // 添加辅助参数
        $request['task_id'] = $taskId;
        $request['user_id'] = $userId;
        $request['now'] = time();
        switch ($scene) {
      
            case self::AUDIO_TRAINING_YM:

                $response = $requestService->audioTrainingYm($request);

                break;
            case self::VIDEO_TRAINING_YM:
                $response = $requestService->videoTrainingYm($request);
                break; 
            case self::AUDIO_TRAINING_YMT:
                $response = $requestService->audioTrainingYmt($request);
                break;  
            case self::VIDEO_TRAINING_YMT:
                $response = $requestService->videoTrainingYmt($request);
                break;  
            default:
        }

        //成功响应，需要扣费
        if (isset($response['code']) && $response['code'] == 10000) {

            $points = $unit;

            if ($points > 0) {

                $extra = [];

                //合成视频按时长扣费
                if (in_array($scene, [
                   self::AUDIO_TRAINING_YM, self::VIDEO_TRAINING_YM,
                   self::AUDIO_TRAINING_YMT, self::VIDEO_TRAINING_YMT
                ])) {

                    $duration = $response['data']['duration'] ?? 1;

                    $points = ceil($duration * $unit);

                    $extra = ['音视频时长' => $duration, '算力单价' => $unit, '实际消耗算力' => $points];
                }

                //token扣除
                User::userTokensChange($userId, $points);

                //记录日志
                AccountLogLogic::recordUserTokensLog(true, $userId, $tokenCode, $points, $taskId, $extra);
            }
        }

        return $response['data'] ?? [];
    }

 /**
     * 更新音频
     * @param array $data
     * @param string $modelVersion
     * @return bool
     * @throws DataNotFoundException
     * @throws DbException
     * @throws ModelNotFoundException
     */
    public static function updateAudio(array $data, string $modelVersion): bool
    {

        $model = SvVideoTask::where('model_version', $modelVersion)->where('status', 1);
        if (in_array($modelVersion,[4,6])) {
            $model = $model->where('audio_id', $data['audio_id']);
        } elseif ($modelVersion == 2) {
            // 目前是同步的  没有回调
            return true;
        } else{
            return false;
        }

        $model->select()
            ->each(function ($item) use ($data) {

                if (in_array($item->model_version,[4,6])) {

                    $item->status = ($data['url'] != "") ? 1 : 2;
                    $item->url = FileService::downloadFileBySource($data['url'], 'audio');
                    $scene = $item->model_version == 4 ? "human_audio_ym" : "human_audio_ymt";
                    // TODO 失败退费
                    if ($item->status == 2) {
                        self::refundTokens($item->user_id, $item->anchor_id, $item->task_id, $scene);
                    }

                    // 更新视频
                    SvVideoTask::where('task_id', $item->task_id)->update([
                        'audio_url' => FileService::setFileUrl($item->url)
                    ]);
                }
                $item->save();
            });

        return true;
    }

        /**
     * 更新视频
     * @param array $data
     * @param string $modelVersion
     * @return bool
     * @throws DataNotFoundException
     * @throws DbException
     * @throws ModelNotFoundException
     */
    public static function updateVideo(array $data, string $modelVersion): bool
    {
        //查询形象
        $model = SvVideoTask::where('model_version', $modelVersion)->where('status', 4);
        if (in_array($modelVersion,[4,6])) {
            $model = $model->where('result_id', $data['job_id']);
        }elseif ($modelVersion == 2) {
            return true;
        } else {

            return false;
        }
        $model->select()
            ->each(function ($item) use ($data) {

                if (in_array($item->model_version,[4,6])) { //高级版
                   //这里对应 status=3 或 status=4） 3成功 4失败
                    if (in_array($data['status'], [3,4])) {
                        $item->status = ($data['status'] == 3) ? 6 : 5;
                        $scene = $item->model_version == 4 ? "human_video_ym" : "human_video_ymt";

                        $videoSetting = SvVideoSetting::where('id', $item->video_setting_id)->find();
                        if($item->status == 5){
                            self::refundTokens($item->user_id, $item->result_id, $item->task_id, $scene);
                            $videoSetting->error_num += 1;
                        }else{
                            $videoSetting->success_num += 1;
                        }
                        $num = $videoSetting->video_count -  $videoSetting->success_num;
                        if ( $videoSetting->error_num == $num){
                            $videoSetting->status = 5;
                        }

                        if ($videoSetting->success_num == $videoSetting->video_count){
                            $videoSetting->status = 3;
                        }
                        if ($videoSetting->error_num == $videoSetting->video_count){
                            $videoSetting->status = 4;
                        }

                        $videoSetting->save();


                    } else {
                        $item->status = 4;
                    }
                    $item->video_result_url   = FileService::downloadFileBySource($data['video_Url'], 'video');
                    $item->remark       = $data['message'] ?? '';
                }
                $item->save();
            });

        return true;
    }


     /**
     * @desc 退费
     * @param int $userId
     * @param int $id
     * @param string $taskId
     * @param string $type
     * @return bool
     */
    public static function refundTokens(int $userId, string $id, string $taskId, string $type): bool
    {

        try {

            [$typeIndex, $typeID] = match ($type) {
                'human_audio_ym' => [3, AccountLogEnum::TOKENS_DEC_HUMAN_AUDIO_YM],
                'human_video_ym' => [4, AccountLogEnum::TOKENS_DEC_HUMAN_VIDEO_YM],
                'human_audio_ymt' => [3, AccountLogEnum::TOKENS_DEC_HUMAN_AUDIO_YMT],
                'human_video_ymt' => [4, AccountLogEnum::TOKENS_DEC_HUMAN_VIDEO_YMT],
            };
            // 请求查询接口
            $requestParams = [
                'id' => $id,
                'task_id' => $taskId,
                'type' => $typeIndex
            ];

            if (strpos($type, '_ymt') !== false) {
                $response = \app\common\service\ToolsService::Human()->detailYmt($requestParams);
            }elseif (strpos($type, '_ym') !== false) {
                $response = \app\common\service\ToolsService::Human()->detailYm($requestParams);
            } else {
                $response = \app\common\service\ToolsService::Human()->detail($requestParams);
            }
           
            if(isset($response['data']['task_status']) && $response['data']['task_status'] == 1) {
                return true;
            }

            //查询是否已返还
            if (UserTokensLog::where('user_id', $userId)->where('change_type', $typeID)->where('action', 1)->where('task_id', $taskId)->count() == 0) {

                $points = UserTokensLog::where('user_id', $userId)->where('change_type', $typeID)->where('task_id', $taskId)->value('change_amount') ?? 0;

                AccountLogLogic::recordUserTokensLog(false, $userId, $typeID, $points, $taskId);
            }

            return true;
        } catch (\Throwable $e) {
            return false;
        }
    }


    public static function updateSvVideoTask(array $params)
    {
        try {
           
            $task = SvVideoTask::where('id',$params['id'])->where('user_id', self::$uid)
                ->find();
            if (!$task) {
                self::setError('视频不存在');
                return false;
            }
            unset($params['status'],$params['user_id']);
            $task->update($params);
            self::$returnData = $task->refresh()->toArray();
            return true;
        } catch (\Exception $e) {
            self::setError($e->getMessage());
            return false;
        }
    }

    public static function deleteSvVideoTask(int $id)
    {
        try {
            $task = SvVideoTask::where('id',$id)->whereIn('status',[3,5,6])->where('user_id', self::$uid)
                ->find();
            if (!$task) {
                self::setError('视频不存在');
                return false;
            };
            $task->delete();
            return true;
        } catch (\Exception $e) {
            self::setError($e->getMessage());
            return false;
        }
    }


    public static function retrySvVideoTask(array $params){

        Db::startTrans();

        try {
            $task = SvVideoTask::where('id', $params['id'])->whereIn('status',[2,5])->where('user_id', self::$uid)
                ->find();
            if (!$task) {
                self::setError('视频不存在');
                return false;
            };
            $setting = SvVideoSetting::where('id', $task->video_setting_id)->field('id,error_num,status')->find();
            if (!$setting) {
                self::setError('任务不存在');
                return false;
            };
            if ($task['status'] == 2){
                $update['status'] = 0;
            }else{
                $update['status'] = 3;
            }
            $update['tries'] = 0;
            $update['id'] =  $params['id'];
            $task->update($update);


            $set['error_num'] = $setting['error_num'] -1;
            $set['status'] = 2;
            $set['id'] = $setting['id'];
            $setting->update($set);
            Db::commit();
            return true;
        } catch (\Exception $e) {
            Db::rollback();
            self::setError($e->getMessage());
            return false;
        }


    }

}