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

namespace App\Console\Commands\Ai;

use App\Models\Ai\AiBlogAuthor;
use App\Models\Ai\AiVideo;
use App\Models\Ai\AiVideoList;
use App\Models\Project\AiVideoTask as AiVideoTaskModel;
use App\Models\RouteMap\RouteMap;
use App\Services\AiBlogService;
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']]);
        //需要更新的路由
        if (!in_array($result['data']['author_id'], $this->updateProject[$item['project_id']] ?? [])) {
            $this->updateProject[$item['project_id']][] = $result['data']['author_id'];
        }
        if (!in_array($route, $this->routes[$item['project_id']] ?? [])) {
            $this->routes[$item['project_id']][] = $route;
        }
        DB::disconnect('custom_mysql');
        $aiVideoTaskModel->edit(['status'=>$aiVideoTaskModel::STATUS_FINISH],['id'=>$item['id']]);
        $this->output('success: task id: ' . $task_id);
        return true;
    }

    /**
     * @remark :更新项目作者页面及列表页
     * @name   :updateProject
     * @author :lyh
     * @method :post
     * @time   :2025/4/30 15:43
     */
    public function updateProject($updateProject){
        if(empty($updateProject)){
            return true;
        }
        foreach ($updateProject as $project_id => $author){
            ProjectServer::useProject($project_id);
            $this->output('sync: list start, project_id: ' . $project_id);
            $this->updateBlogList($project_id);
            $this->output('sync: list end');
            //更新作者
            $this->output('sync: author start, project_id: ' . $project_id);
            foreach ($author as $val){
                $this->updateAiBlogAuthor($val,$project_id);
            }
            $this->output('sync: author end');
            DB::disconnect('custom_mysql');
        }

        return true;
    }

    /**
     * @remark :更新作者页面
     * @name   :updateAiBlogAuthor
     * @author :lyh
     * @method :post
     * @time   :2025/4/30 15:52
     */
    public function updateAiBlogAuthor($author_id,$project_id){
        if(empty($author_id)){
            return true;
        }
        $aiBlogService = new AiBlogService($project_id);
        $aiBlogService->author_id = $author_id;
        $result = $aiBlogService->getAuthorDetail();
        if(isset($result['status']) && $result['status'] == 200){
            //当前作者的页面
            $aiBlogAuthorModel = new AiBlogAuthor();
            $authorInfo = $aiBlogAuthorModel->read(['author_id'=>$author_id],['id','route']);
            if($authorInfo !== false && !empty($result['data']['section'])){
                //需要更新的路由
                if (!in_array($authorInfo['route'], $this->routes[$project_id] ?? [])) {
                    $this->routes[$project_id][] = $authorInfo['route'];
                }
                $aiBlogAuthorModel->edit(['text'=>$result['data']['section']],['author_id'=>$author_id]);
            }
        }
        return true;
    }

    /**
     * @remark :更新
     * @name   :updateBlogList
     * @author :lyh
     * @method :post
     * @time   :2025/4/30 15:45
     */
    public function updateBlogList($project_id){
        $aiVideoService = new AiVideoService($project_id);
        $page = 1;
        $saveData = [];
        $result = $aiVideoService->getAiVideoList($page,15);
        if(!isset($result['status']) && $result['status'] != 200){
            return true;
        }
        $total_page = $result['data']['total_page'];
        //组装数据保存
        $saveData[] = [
            'route'=>$page,
            'text'=>$result['data']['section'],
        ];
        while ($total_page > $page){
            $page++;
            $result = $aiVideoService->getAiVideoList($page,15);
            if(isset($result['status']) && $result['status'] == 200){
                $saveData[] = [
                    'route'=>$page,
                    'text'=>$result['data']['section'],
                ];
            }
        }
        $aiVideoListModel = new AiVideoList();
        if(!empty($saveData)){
            //写一条路由信息
            $aiVideoListModel->truncate();
            $aiVideoListModel->insertAll($saveData);
        }
        return true;
    }
    /**
     * 输入日志
     * @param $message
     * @return bool
     */
    public function output($message)
    {
        $message = date('Y-m-d H:i:s') . ' ' . $message . PHP_EOL;
        echo $message;
        return true;
    }
}