<?php

/*
 * @Author: your name
 * @Date: 2021-12-24 10:26:10
 * @LastEditTime: 2021-12-24 17:11:44
 * @LastEditors: Please set LastEditors
 * @Description: 打开koroFileHeader查看配置 进行设置: https://github.com/OBKoro1/koro1FileHeader/wiki/%E9%85%8D%E7%BD%AE
 * @FilePath: /hy-websocket/app/Controller/WebsocketController.php
 */
declare (strict_types=1);
namespace App\Controller;

use Hyperf\Contract\OnCloseInterface;
use Hyperf\Contract\OnMessageInterface;
use Hyperf\Contract\OnOpenInterface;
use Hyperf\Contract\OnHandShakeInterface;
use Hyperf\Utils\Codec\Json;
use Swoole\Http\Request;
use Swoole\Http\Response;
use Swoole\WebSocket\Frame;
use Swoole\WebSocket\Server;
use Hyperf\Redis\Redis;
use Hyperf\Di\Annotation\Inject;
use Hyperf\Config\Annotation\Value;
use App\Utils\guid;
use App\Utils\HashMap;
use App\Utils\Http;
use App\Utils\PcmToWave;
use App\Service\TransferService;
use Psr\Container\ContainerInterface;
use Hyperf\Logger\LoggerFactory;
use Hyperf\Contract\ConfigInterface;
class WebsocketController implements OnMessageInterface, OnOpenInterface, OnCloseInterface
{
    use \Hyperf\Di\Aop\ProxyTrait;
    use \Hyperf\Di\Aop\PropertyHandlerTrait;
    /**
     * @Inject
     * @var TransferService
     */
    private $transferService;
    /**
     * 
     * @Inject
     * @var HashMap
     */
    private $HashMap;
    //存放音频写入句柄
    /**
     * 
     * @var \Psr\Log\LoggerInterface
     */
    public $logger;
    /**
     * @Inject()
     * @var ConfigInterface
     */
    private $config;
    private $server_config;
    private $allow_ip;
    public function __construct(LoggerFactory $loggerFactory)
    {
        $this->__handlePropertyHandler(__CLASS__);
        // 第一个参数对应日志的 name, 第二个参数对应 config/autoload/logger.php 内的 key
        $this->logger = $loggerFactory->get('log', 'default');
        //获取 server.php 里的内容
        $this->server_config = $this->config->get('server.servers', '');
        $this->allow_ip = $this->server_config[0]['allow_ip'];
    }
    public function onClose($server, int $fd, int $reactorId) : void
    {
        //客户端异常关闭
        //语音文件传输完毕，关闭文件。
        $host = $this->allow_ip;
        $group_key = $this->transferService->get($host . ':' . $fd . ':group');
        $this->transferService->lrem($group_key, $fd, 0);
        $set_ket = $host . ':' . $fd . ':group';
        $this->transferService->delete($set_ket);
    }
    public function onStart($server) : void
    {
        $host = $this->allow_ip;
        $keyList = $this->transferService->keys("*{$host}*");
        foreach ($keyList as $key => $value) {
            $this->transferService->delete($value);
        }
    }
    /**
     * @param Response|Server $server
     */
    public function onMessage($server, Frame $frame) : void
    {
        $ret = array('code' => 0, 'data' => null);
        $msgData = $this->is_json($frame->data, true);
        if ($msgData) {
            $frameData = $msgData;
            $url = $frameData['url'];
            $data = $frameData['data'];
            $action = substr($url, strrpos($url, "/") + 1);
            switch ($action) {
                case "reg":
                    $groupId = $data['group_id'];
                    $session = $data['session_id'];
                    $this->bind($groupId, '', $frame->fd);
                    $key = "practice_scenes_ws_cur_session_info:" . $session;
                    $node_key = 'practice_scenes_ws_cur_session_node_info:' . $session;
                    $redisData = $this->transferService->get($key);
                    if ($redisData) {
                        $ret['data'] = json_decode($redisData, true);
                        $this->transferService->delete($key);
                    } else {
                        $redisData = $this->transferService->get($node_key);
                        if ($redisData) {
                            $ret['data'] = json_decode($redisData, true);
                        } else {
                            $ret['data'] = '';
                        }
                    }
                    $ret['event'] = "re_bind";
                    $server->push($frame->fd, json_encode($ret));
                    break;
                case "data":
                    $groupId = $data['group_id'];
                    $ret['event'] = "data";
                    $ret['data'] = $data;
                    $connections = array();
                    if (empty($groupId)) {
                        // 获取所有无分组标识在线终端
                        $onlines = count($server->connections);
                        if ($onlines > 0) {
                            $connections = $server->connections;
                        }
                    } else {
                        // 获取所有分组标识在线终端 key = groupID:host
                        $key = $groupId . ':' . $this->allow_ip;
                        $connections = $this->transferService->lrange($key, 0, -1);
                    }
                    $connections = array_unique($connections);
                    // 广播获取的在线终端
                    foreach ($connections as $fd) {
                        $ingfd = (int) $fd;
                        $server->push($ingfd, json_encode($ret));
                    }
                    break;
                case "ping":
                    $server->push($frame->fd, $data . 'pong');
                    break;
                default:
                    $ret['code'] = 404;
                    $ret['msg'] = 'event no existent';
                    $server->push($frame->fd, json_encode($ret));
                    return;
            }
        } else {
            $ret['code'] = -1;
            $ret['msg'] = 'data is null or data no json';
            $server->push($frame->fd, json_encode($ret));
            return;
        }
    }
    public function onOpen($server, Request $request) : void
    {
        // $server->push($request->fd, 'Opened');
        // var_dump($request);
    }
    /**
     * 功能:1、判断是否是json
     * @param string $data
     * @param bool $assoc
     * @return bool|mixed|string
     */
    public function is_json($data = '', $assoc = false)
    {
        $data = json_decode($data, $assoc);
        if ($data && is_object($data) || is_array($data) && !empty(current($data))) {
            return $data;
        }
        return false;
    }
    /**
     * 当前链接的绑定事件。
     * 绑定的几种状态:
     * 1.无分组
     * host -> fds (获取当前节点的在线链接列表)
     * host:fd -> value (获取当前节点当前链接的绑定数据)
     * host:fd:group -> host (获取当前节点当前链接的绑定分组)
     * 2.有分组
     * groupID:host -> fds
     * host:fd -> value
     * host:fd:group -> groupID:host
     * @param string $groupID 需要绑定当前链接所在分组的分组标识，如果为空则使用默认分组
     * @param string $value 需要绑定当前链接对应的业务数据,如果为空则不绑定
     */
    public function bind($groupID = '', $value = '', $fd)
    {
        $host = $this->allow_ip;
        // 绑定分组和当前链接对应的分组标识
        if (empty($groupID)) {
            $this->transferService->rpush($host, $fd);
            $this->transferService->set($host . ':' . $fd . ':group', $host);
        } else {
            $this->transferService->rpush($groupID . ':' . $host, $fd);
            $this->transferService->set($host . ':' . $fd . ':group', $groupID . ':' . $host);
        }
        // 绑定业务数据
        if (!empty($value)) {
            $this->transferService->set($host . ':' . $fd, $value);
        }
    }
}