<?php
/**
 * @remark :
 * @name   :AiVideoTask.php
 * @author :lyh
 * @method :post
 * @time   :2025/4/30 11:18
 */

namespace App\Console\Commands\Ai;

use App\Models\Ai\AiVideo;
use App\Models\Project\AiVideoTask as AiVideoTaskModel;
use App\Models\RouteMap\RouteMap;
use App\Services\AiVideoService;
use App\Services\DingService;
use App\Services\ProjectServer;
use Illuminate\Console\Command;
use Illuminate\Support\Facades\DB;
use Illuminate\Support\Facades\Redis;

class AiVideoTask extends Command
{
    /**
     * The name and signature of the console command.
     *
     * @var string
     */
    protected $signature = 'save_ai_video';

    public $updateProject = [];//需更新的列表
    public $routes = [];//需要更新的路由

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = '查询ai_video是否已经生成';

    /**
     * @return bool
     * @throws \Exception
     */
    public function handle(){
        while (true){
            //获取任务id
            $task_id = $this->getTaskId();
            if(empty($task_id)){
                sleep(300);
                continue;
            }
            $this->_action($task_id);
        }
        return true;
    }

    /**
     * 获取任务id
     * @param int $finish_at
     * @return mixed
     */
    public function getTaskId($finish_at = 2)
    {
        $task_id = Redis::rpop('ai_video_task');
        if (empty($task_id)) {
//            if(!empty($this->updateProject)){
//                $this->updateProject($this->updateProject);
//                $this->updateProject = [];
//            }
//            if(!empty($this->routes)){
//                $this->updateRoutes($this->routes);
//                $this->routes = [];
//            }
            $aiVideoTaskModel = new AiVideoTaskModel();
            $finish_at = date('Y-m-d H:i:s', strtotime('-' . $finish_at . ' hour'));
            $ids = $aiVideoTaskModel->formatQuery(['status'=>$aiVideoTaskModel::STATUS_RUNNING,'updated_at'=>['<=',$finish_at]])->pluck('id');
            if(!empty($ids)){
                foreach ($ids as $id) {
                    Redis::lpush('ai_video_task', $id);
                }
            }
            $task_id = Redis::rpop('ai_video_task');
        }
        return $task_id;
    }

    /**
     * @remark :请求
     * @name   :sendRequest
     * @author :lyh
     * @method :post
     * @time   :2025/4/30 11:31
     */
    public function _action($task_id){
        $aiVideoTaskModel = new AiVideoTaskModel();
        $item = $aiVideoTaskModel->read(['id'=>$task_id]);
        $this->output('ai_video->start：project ID: ' . $item['project_id'] . ',task ID: ' . $task_id);
        $aiVideoService = new AiVideoService($item['project_id']);
        $aiVideoService->task_id = $item['task_id'];
        //拉取文章数据
        $result = $aiVideoService->getVideoDetail();
        if(empty($result['status']) || ($result['status'] != 200)){
            if($item['number'] < 5){
                $aiVideoTaskModel->edit(['number'=>$item['number'] + 1],['id'=>$item['id']]);
            }else{
                $aiVideoTaskModel->edit(['status'=>9],['id'=>$item['id']]);
                // 钉钉通知
                $dingService = new DingService();
                $body = [
                    'keyword' => 'AI_VIDEO获取失败',
                    'msg' => '任务ID:' . $item['task_id'] . PHP_EOL . '返回信息：' . json_encode($result,JSON_UNESCAPED_UNICODE | JSON_UNESCAPED_SLASHES),
                    'isAtAll' => false, // 是否@所有人
                ];
                $dingService->handle($body);
            }
            $this->output('error: 数据获取失败，status：' . $result['status'] . ',message: ' . ($result['message'] ?? 'null'));
            return false;
        }
        //保存当前项目ai_blog数据
        ProjectServer::useProject($item['project_id']);
        $aiVideoModel = new AiVideo();
        $aiVideoInfo = $aiVideoModel->read(['task_id'=>$item['task_id']],['id','route']);
        if($aiVideoInfo === false){
            // 钉钉通知
            $dingService = new DingService();
            $body = [
                'keyword' => 'AI_VIDEO生成错误',
                'msg' => '任务ID：' . $item['task_id'] . ', 子库获取数据失败， 检查子库数据是否被删除！',
                'isAtAll' => false, // 是否@所有人
            ];
            $dingService->handle($body);
            $this->output('error: 子库获取数据失败, task id: ' . $task_id);
            $aiVideoTaskModel->edit(['status'=>9],['id'=>$item['id']]);
            DB::disconnect('custom_mysql');
            return false;
        }
        //拿到返回的路由查看是否重复
        $route = RouteMap::setRoute($result['data']['url'], RouteMap::SOURCE_AI_VIDEO, $aiVideoInfo['id'], $item['project_id']);
        if($route != $result['data']['url']){
            $aiVideoService->updateDetail(['route'=>$route,'task_id'=>$item['task_id']]);
        }
        $saveData = [
            'title'=>$result['data']['title'],
            'image'=>$result['data']['thumb'],
            'video_url'=>$result['data']['video_url'],
            'route'=>$route,
            'author_id'=>$result['data']['author_id'],
            'keyword'=>json_encode($result['data']['keyword'],true),
            'content'=>$result['data']['content'],
            'text'=>$result['data']['"section'],
            'status'=>$aiVideoTaskModel::STATUS_FINISH
        ];
        $aiVideoModel->edit($saveData,['task_id'=>$item['task_id']]);
        DB::disconnect('custom_mysql');
        $aiVideoTaskModel->edit(['status'=>$aiVideoTaskModel::STATUS_FINISH],['id'=>$item['id']]);
        $this->output('success: task id: ' . $task_id);
        return true;
    }

    /**
     * 输入日志
     * @param $message
     * @return bool
     */
    public function output($message)
    {
        $message = date('Y-m-d H:i:s') . ' ' . $message . PHP_EOL;
        echo $message;
        file_put_contents(storage_path('logs/AiVideo/') . date('Ymd') . '.log', $message, FILE_APPEND);
        return true;
    }
}