<?php

//error_reporting();

use Swoole\Process;



include_once __DIR__."/../vendor/autoload.php";


function start(){

// 删除停止运行的值
    redis()->delete(RUNNING_REDIS_KEY);

    // 进程管理器
    $pm = new Process\Manager();

    // 启动一个进程来管理定时
    $pm->add(function (Process\Pool $pool, int $workerId){
        _echo("定时进程（{$workerId}）启动成功");


        // 进行阻塞，否则定时器无法运行
        while (true){
            co::sleep(9999);
        }

    },true);


// 协程配置
    \co::set([
        'max_coroutine'=>COROUTINE_MAX_NUM, // 最大携程数量
        'hook_flags'=>SWOOLE_HOOK_TCP,  //  redis需要的配置
    ]);

    // 启动业务进程
    $pm->addBatch(WORKER_NUM,function (Process\Pool $pool, int $worker_id){
        _echo("业务进程（{$worker_id}）启动成功");

        $start_num = 0;// 启动的协程数量

        // 循环阻塞
        while (true){
            // 是否到了协程配置的数量上限
            if($start_num < COROUTINE_MAX_NUM){
                // 需要同步的id
                $id = redis()->lPop('sync_email_lists');

                if(!$id){
                    co::sleep(1);
                }else{
                    // 占用当前的id,占用2小时
                    redis()->add('just_sync_'.$id,time(),7200);
                    // 启动一个协程
                    go(function () use (&$start_num,$worker_id,$id){
                        $start_num++;

                        // 开始同步
                        try {
                            sync($id,$worker_id);
                        }catch (\Throwable $e){
                            // 重新发布同步任务,如果失败了是否重新发布
                            redis()->rPush('sync_email_lists',$id);

//                            _echo($e->getMessage());
                            logs(
                                $e->getMessage().PHP_EOL.$e->getTraceAsString(),
                                LOG_PATH.'/'.$worker_id.'_'.co::getCid().'.log'
                            );
                        }

                        // 协程完成后执行的函数
                        co::defer(function () use (&$start_num,$worker_id,$id){
//                        _echo('正常关闭进程（'.$worker_id.'）下的协程（'.co::getCid().'）');
                            $start_num--;
                            // 消除占用
                            redis()->delete('just_sync_'.$id);
                            // 写入日志
                            \Lib\Log::getInstance()->write();

                            // 关闭数据库链接
                            db()->close();
                            // 关闭redis链接
                            redis()->close();

                        });

                    });

                }
            }else{
                // 协程到了最大的数量，阻塞1秒
                co::sleep(1);
            }


        }


    },true);


    // 启动管理器
    $pm->start();

}


/**
 * 开始同步, 这里是主要的业务代码
 * @param $email_id
 * @param $worker_id
 * @return int
 * @author：dc
 * @time 2023/3/10 10:19
 */
function sync($email_id,$worker_id){

    $email = db()->first(\Model\emailSql::first($email_id));
    if(!$email){
        return 0;
    }

    if($email['pwd_error']){
        return 1;
    }

    $mailServer = new Lib\Mail\Mail($email['email'],base64_decode($email['password']),$email['imap']);

        // 登录服务器
    if(!$mailServer->login()){
        return 2;
    }


    // 同步文件夹
    $mailServer->syncFolder($email_id,db());


    // 读取到邮箱中的文件夹
    $folders = db()->all(\Model\folderSql::all($email['id']));
    if(!$folders){
        return 3;
    }
    $folders = list_to_tree($folders);
    foreach ($folders as $folder){
        try {
            if(empty($folder['_child'])){
                // 同步父文件夹
                $mailServer->syncMail($email_id,$folder['id'],$folder['origin_folder']);
            }else{
                foreach ($folder as $item){
                    // 同步子文件夹
                    $mailServer->syncMail($email_id,$item['id'],$item['origin_folder']);
                }
            }

        }catch (Throwable $e){
            logs(
                $e->getMessage(),
                LOG_PATH.'/imap/'.$email['email'].'.error.log'
            );
        }
    }


    $email = null;
    $mailServer = null;
}




switch ($argv[1]){
    case 'start':{
        start();
        break;
    }
    case 'stop':{
        \Co\run(function (){
            echo "正在退出程序...\n非必要请不要强制kill掉进程\n";
            redis()->set(RUNNING_REDIS_KEY,'stop');
            while (true){
                $num = exec("ps -ef | grep \"sync.php start\" | grep -v grep | wc -l");
                if(!$num){
                    break;
                }
                co::sleep(0.5);
            }
            echo "已退出程序\n";
        });
        break;
    }
    default:{
        break;
    }
}