MessageHandler.php

<?php
/**
 * Created by PhpStorm.
 * User: Hanson
 * Date: 2016/12/14
 * Time: 23:08
 */

namespace Hanson\Vbot\Core;

use Closure;
use Hanson\Vbot\Message\Entity\Emoticon;
use Hanson\Vbot\Message\Entity\Image;
use Hanson\Vbot\Message\Entity\Message;
use Hanson\Vbot\Message\Entity\Text;
use Hanson\Vbot\Message\Entity\Video;
use Hanson\Vbot\Support\Console;
use Hanson\Vbot\Support\System;

class MessageHandler
{
    /**
     * @var MessageHandler
     */
    static $instance = null;

    private $handler;

    private $customHandler;

    private $exitHandler;

    private $exceptionHandler;

    private $onceHandler;

    private $sync;

    private $messageFactory;

    public function __construct()
    {
        $this->sync = new Sync();
        $this->messageFactory = new MessageFactory();
    }

    /**
     * 设置单例模式
     *
     * @return MessageHandler
     */
    public static function getInstance()
    {
        if (static::$instance === null) {
            static::$instance = new MessageHandler();
        }

        return static::$instance;
    }

    /**
     * 消息处理器
     *
     * @param Closure $closure
     * @throws \Exception
     */
    public function setMessageHandler(Closure $closure)
    {
        if (!$closure instanceof Closure) {
            throw new \Exception('message handler must be a closure!');
        }

        $this->handler = $closure;
    }

    /**
     * 自定义处理器
     *
     * @param Closure $closure
     * @throws \Exception
     */
    public function setCustomHandler(Closure $closure)
    {
        if (!$closure instanceof Closure) {
            throw new \Exception('custom handler must be a closure!');
        }

        $this->customHandler = $closure;
    }

    /**
     * 退出处理器
     *
     * @param Closure $closure
     * @throws \Exception
     */
    public function setExitHandler(Closure $closure)
    {
        if (!$closure instanceof Closure) {
            throw new \Exception('exit handler must be a closure!');
        }

        $this->exitHandler = $closure;
    }

    /**
     * 异常处理器
     *
     * @param Closure $closure
     * @throws \Exception
     */
    public function setExceptionHandler(Closure $closure)
    {
        if (!$closure instanceof Closure) {
            throw new \Exception('exit handler must be a closure!');
        }

        $this->exceptionHandler = $closure;
    }

    /**
     * 执行一次的处理器
     *
     * @param Closure $closure
     * @throws \Exception
     */
    public function setOnceHandler(Closure $closure)
    {
        if (!$closure instanceof Closure) {
            throw new \Exception('exit handler must be a closure!');
        }

        $this->onceHandler = $closure;
    }

    /**
     * 轮询消息API接口
     */
    public function listen()
    {
        if ($this->onceHandler instanceof Closure) {
            call_user_func_array($this->onceHandler, []);
        }

        while (true) {
            if ($this->customHandler instanceof Closure) {
                call_user_func_array($this->customHandler, []);
            }

            $time = time();
            list($retCode, $selector) = $this->sync->checkSync();

            if (in_array($retCode, ['1100', '1101'])) { # 微信客户端上登出或者其他设备登录
                Console::log('微信客户端正常退出');
                if ($this->exitHandler) {
                    call_user_func_array($this->exitHandler, []);
                }
                break;
            } elseif ($retCode == 0) {
                $this->handlerMessage($selector);
            } else {
                Console::log('微信客户端异常退出');
                if ($this->exceptionHandler) {
                    call_user_func_array($this->exitHandler, []);
                }
                break;
            }

            $this->sync->checkTime($time);
        }
        Console::log('程序结束');
    }

    /**
     * 处理消息
     *
     * @param $selector
     */
    private function handlerMessage($selector)
    {
        if ($selector === 0) {
            return;
        }

        $message = $this->sync->sync();

        if (count($message['ModContactList']) > 0) {
            foreach ($message['ModContactList'] as $contact) {
                if (str_contains($contact['UserName'], '@@')) {
                    group()->put($contact['UserName'], $contact);
                } else {
                    contact()->put($contact['UserName'], $contact);
                }
            }
        }

        if ($message['AddMsgList']) {
            foreach ($message['AddMsgList'] as $msg) {
                $content = $this->messageFactory->make($msg);
                if ($content) {
                    $this->addToMessageCollection($content);
                    if ($this->handler) {
                        $reply = call_user_func_array($this->handler, [$content]);
                        if ($reply) {
                            if ($reply instanceof Image) {
                                Image::sendByMsgId($content->from['UserName'], $reply->msg['MsgId']);
                            } elseif ($reply instanceof Video) {
                                Video::sendByMsgId($content->from['UserName'], $reply->msg['MsgId']);
                            } elseif ($reply instanceof Emoticon) {
                                Emoticon::sendByMsgId($content->from['UserName'], $reply->msg['MsgId']);
                            } else {
                                Text::send($content->from['UserName'], $reply);
                            }
                        }
                    }
                }
            }
        }
    }

    /**
     * @param $message Message
     */
    private function addToMessageCollection($message)
    {
        message()->put($message->msg['MsgId'], $message);

        if (server()->config['debug']) {
            $file = fopen(System::getPath() . 'message.json', 'a');
            fwrite($file, json_encode($message) . PHP_EOL);
            fclose($file);
        }
    }

}