<?php

/**
 * 
 *   _____   _   _       _   _   _   _____   _       _____   _____   _____
 *  |  _  \ | | | |     | | | | | | | ____| | |     |  _  \ | ____| |  _  \
 *  | |_| | | | | |     | | | |_| | | |__   | |     | |_| | | |__   | |_| |
 *  |  _  { | | | |     | | |  _  | |  __|  | |     |  ___/ |  __|  |  _  /
 *  | |_| | | | | |___  | | | | | | | |___  | |___  | |     | |___  | | \ \
 *  |_____/ |_| |_____| |_| |_| |_| |_____| |_____| |_|     |_____| |_|  \_\
 *
 *  Website: https://mudew.com/
 *  Author: Lkeme
 *  License: The MIT License
 *  Email: Useri@live.cn
 *  Updated: 2022 ~ 2023
 *
 *  &   ／l、
 *    （ﾟ､ ｡ ７
 *   　\、ﾞ ~ヽ   *
 *   　じしf_, )ノ
 *
 */

namespace BiliHelper\Plugin;

use BiliHelper\Core\Log;
use BiliHelper\Util\TimeLock;

use Amp\Delayed;
use Exception;
use JetBrains\PhpStorm\Pure;
use Socket\Raw\Factory;
use function get_class;

class ZoneTcpClient
{
    use TimeLock;

    private static int $raffle_id = 0;
    private static array $raffle_list = [];
    private static array $server = [];
    private static string|null $server_key = null;

    private static int|string $area_id;
    private static int|string $room_id;
    private static $client;
    private static array $client_maps = [];
    private static array $trigger_restart = [];
    private static int $socket_timeout = 0;


    /**
     * @use run
     */
    public static function run()
    {
        if (self::getLock() > time() || !getEnable('zone_monitor')) {
            return;
        }
        self::setPauseStatus();
        self::init();
        self::updateConnection();
        self::heartBeat();
        self::receive();
        self::pushHandle();
    }

    /**
     * @use 初始化
     */
    private static function init()
    {
        if (empty(getConf('server_addr', 'zone_monitor'))) {
            exit('推送服务器信息不完整, 请检查配置文件!');
        }
        if (!self::$client) {
            self::initConnect();
        }
    }

    /**
     * @use 初始化连接
     */
    private static function initConnect()
    {
        $areas = Live::fetchLiveAreas();
        foreach ($areas as $area_id) {
            self::$client_maps["server$area_id"] = ["area_id" => null, "room_id" => null, "client" => null, "heart_beat" => 0, "status" => false];
            self::triggerReConnect([
                'area_id' => $area_id,
                'wait_time' => time()
            ], 'Initialization');
        }
    }

    /**
     * @use 触发重连
     * @param array $area_data
     * @param string $reason
     */
    private static function triggerReConnect(array $area_data, string $reason)
    {
        Log::debug("Reconnect Reason: {$area_data['area_id']} -> $reason");
        self::$client_maps["server" . $area_data['area_id']]['status'] = false;
        self::$trigger_restart[] = $area_data;
    }

    /**
     * @use 更新连接
     */
    private static function updateConnection()
    {
        $num = count(self::$trigger_restart);
        for ($i = 0; $i < $num; $i++) {
            $area_data = array_shift(self::$trigger_restart);
            if (is_null($area_data)) {
                break;
            }
            if (time() < $area_data['wait_time']) {
                self::$trigger_restart[] = $area_data;
                continue;
            }
            Log::notice("update_connections triggered, info: {$area_data['area_id']}");
            $area_info = Live::areaToRid($area_data['area_id']);
//            $area_room_info = [
//                'area_id' => $area_data['area_id'],
//                'room_id' => $room_id
//            ];
            self::update($area_info);
        }
    }

    /**
     * @use 更新操作
     * @param array $area
     */
    private static function update(array $area)
    {
        self::$area_id = $area['area_id'];
        self::$room_id = $area['room_id'];
        try {
            self::$server = Live::getDanMuInfo(self::$room_id);
            self::$client = (new Factory())->createClient(self::$server['addr'], 40);
            self::$client->setBlocking(false);
            self::sendHandShake();
            self::$client_maps["server" . self::$area_id] = [
                'client' => self::$client,
                'area_id' => self::$area_id,
                'room_id' => self::$room_id,
                'status' => true,
                'heart_beat' => time() + 25,
            ];
            // self::$client->getPeerName()
            Log::info("连接到 @分区 {$area['area_id']}  @房间 {$area['room_id']} @状态 √ @信息 Successful!");
        } catch (Exception $e) {
            Log::error("连接到 @分区 {$area['area_id']}  @房间 {$area['room_id']} @状态 × @信息 {$e->getMessage()}");
            self::triggerReConnect([
                'area_id' => self::$area_id,
                'wait_time' => time() + 60
            ], self::formatErr($e));
        }
    }

    /**
     * 判断字符串是否为 Json 格式
     * @param string $data Json 字符串
     * @param bool $assoc 是否返回对象or关联数组，默认返回关联数组
     * @return array|bool|object 成功返回转换后的对象或数组，失败返回 false
     */
    private static function analyJson(string $data = '', bool $assoc = true): object|bool|array
    {
        if (is_array($data)) {
            return $data;
        }
        $data = json_decode($data, $assoc);
        if (($data && is_object($data)) || (is_array($data) && !empty($data))) {
            return $data;
        }
        return false;
    }

    /**
     * @use 响应数据
     * @param $msg
     * @param $type
     * @return bool
     */
    private static function onMessage($msg, $type): bool
    {
        // 心跳后回复人气
        if ($type == 3) {
            // $num = unpack('N', $msg)[1];
            // Log::info("当前直播间现有 {$num} 人聚众搞基!");
            return false;
        }
        $de_raw = self::analyJson($msg);
        // 进入房间返回
        if (isset($de_raw['code']) && !$de_raw['code']) {
            return false;
        }
        // 部分cmd抽风
        if (!$de_raw || !isset($de_raw['cmd'])) {
            Log::warning("解析错误: $msg");
            return false;
        }
        $data = [];
        $update_room = false;
        switch ($de_raw['cmd']) {
            case 'POPULARITY_RED_POCKET_WINNER_LIST':
                // data -> lot_id|winner_info[] ->uid|name|user_type|award_name
                Log::debug(json_encode($de_raw, true));
                break;
            case 'POPULARITY_RED_POCKET_START':
                // data [0]?-> * sender_uid|sender_name|join_requirement|danmu|lot_status|lot_id|start_time|end_time|awards
                // awards[] -> gift_id|num
                Log::debug(json_encode($de_raw, true));
                // 关联
                if (count(array_filter(array_keys($de_raw['data']), 'is_string')) > 0) {
                    // 关联数组
                    $temp = $de_raw['data'];
                } else {
                    // 索引数组
                    $temp = $de_raw['data'][0];
                }
                $data = [
                    'room_id' => self::$room_id,
                    'raffle_id' => $temp['lot_id'],
                    'raffle_title' => '利是包',
                    'raffle_type' => 'red_pocket',
                    'source' => $msg
                ];
                break;
            case 'TV_START':
                // 小电视飞船(1)
                $data = [
                    'room_id' => self::$room_id,
                    'raffle_id' => $de_raw['data']['id'],
                    'raffle_title' => $de_raw['data']['title'],
                    'raffle_type' => 'small_tv',
                    'source' => $msg
                ];
                break;
            case 'SPECIAL_GIFT':
                // 节奏风暴(1)
                if (array_key_exists('39', $de_raw['data'])) {
                    if ($de_raw['data']['39']['action'] == 'start') {
                        $data = [
                            'room_id' => self::$room_id,
                            'raffle_id' => $de_raw['data']['39']['id'],
                            'raffle_title' => '节奏风暴(1)',
                            'raffle_type' => 'storm',
                            'source' => $msg
                        ];
                    }
                }
                break;
            case 'GUARD_LOTTERY_START':
                // 舰长(1)
                $data = [
                    'room_id' => self::$room_id,
                    'raffle_id' => $de_raw['data']['id'],
                    'raffle_title' => '总督舰长(1)',
                    'raffle_type' => 'guard',
                    'source' => $msg
                ];
                break;
            case 'GUARD_MSG':
                // 舰长(2)
                // {"buy_type":3,"cmd":"GUARD_MSG","msg":":?淩白夜:? 在本房间开通了舰长"}
                $data = [
                    'room_id' => self::$room_id,
                    'raffle_id' => self::$raffle_id++,
                    'raffle_title' => '总督舰长(2)',
                    'raffle_type' => 'guard',
                    'source' => $msg
                ];
                break;
            case 'LOTTERY_START':
                // 舰长(3)
                $data = [
                    'room_id' => $de_raw['data']['roomid'],
                    'raffle_id' => $de_raw['data']['id'],
                    'raffle_title' => '总督舰长(3)',
                    'raffle_type' => 'guard',
                    'source' => $msg
                ];
                break;
            case 'ANCHOR_LOT_START':
                // 天选时刻(1)
                $data = [
                    'room_id' => $de_raw['data']['room_id'],
                    'raffle_id' => $de_raw['data']['id'],
                    'raffle_title' => '天选时刻(1)',
                    'raffle_type' => 'anchor',
                    'source' => $msg
                ];
                break;
            case 'PK_LOTTERY_START':
                // PK大乱斗(1)
                $data = [
                    'room_id' => $de_raw['data']['room_id'],
                    'raffle_id' => $de_raw['data']['id'],
                    'raffle_title' => 'PK大乱斗(1)',
                    'raffle_type' => 'pk',
                    'source' => $msg
                ];
                break;
            case 'PK_BATTLE_END':
                // PK大乱斗(2)
                $data = [
                    'room_id' => self::$room_id,
                    'raffle_id' => $de_raw['pk_id'],
                    'raffle_title' => 'PK大乱斗(2)',
                    'raffle_type' => 'pk',
                    'source' => $msg
                ];
                break;
            case 'RAFFLE_START':
                // 高能抽奖(1)
                $data = [
                    'room_id' => self::$room_id,
                    'raffle_id' => $de_raw['data']['id'],
                    'raffle_title' => $de_raw['data']['title'], // 高能抽奖(1)
                    'raffle_type' => 'raffle',
                    'source' => $msg
                ];
                break;
            case 'NOTICE_MSG':
                $msg_type = $de_raw['msg_type'];
                $msg_self = $de_raw['msg_self'];
                $msg_common = str_replace(' ', '', $de_raw['msg_common']);
                $real_room_id = $de_raw['real_roomid'];
                if (in_array($msg_type, [2, 8])) {
                    $data = [
                        'room_id' => $real_room_id,
                        'raffle_id' => self::$raffle_id++,
                        'raffle_title' => $msg_self,
                        'raffle_type' => 'raffle',
                        'source' => $msg
                    ];
                    // echo self::$room_id . '--' . $real_room_id . PHP_EOL;
                }
                if ($msg_type == 6 && str_contains($msg_common, '节奏风暴')) {
                    $data = [
                        'room_id' => $real_room_id,
                        'raffle_id' => self::$raffle_id++,
                        'raffle_title' => '节奏风暴',
                        'raffle_type' => 'raffle',
                        'source' => $msg
                    ];

                }
                break;
            case 'DANMU_GIFT_LOTTERY_START':
                // 弹幕抽奖(1)
//                $data = [
//                    'room_id' => $de_raw['data']['room_id'],
//                    'raffle_id' => $de_raw['data']['id'],
//                    'raffle_title' => $de_raw['data']['title'],
//                    'raffle_type' => 'raffle',
//                    'source' => $msg
//                ];
                break;
            case 'PREPARING':
                // 房间内下播消息。
                $update_room = true;
                break;
            case 'CUT_OFF':
                // 房间内被下播消息。
                $update_room = true;
                break;
            case 'WARNING':
                // 房间内管理员警告消息。
                $update_room = true;
                break;
            default:
                $data = [];
                break;
        }
        // 下播处理
        if ($update_room) {
            self::triggerReConnect([
                'area_id' => self::$area_id,
                'wait_time' => time()
            ], 'Interrupt live broadcast');
        }
        // 处理数据
        if (!empty($data)) {
            unset($data['source']);
            if (!isset(self::$raffle_list[$data['raffle_type']])) {
                self::$raffle_list[$data['raffle_type']] = [];
            }
            $data['area_id'] = self::$area_id;
            self::$raffle_list[$data['raffle_type']][] = $data;
            Log::info("监测到 @分区 {$data['area_id']} @房间 {$data['room_id']} @抽奖 {$data['raffle_title']}");
            // print_r($data);
        }
        return true;
    }

    /**
     * @推送到上游处理
     */
    private static function pushHandle()
    {
        foreach (self::$raffle_list as $type => $data) {
            $temp_room_id = 0;
            foreach (self::$raffle_list[$type] as $raffle) {
                if ($temp_room_id != $raffle['room_id']) {
                    DataTreating::distribute($raffle);
                    $temp_room_id = $raffle['room_id'];
                }
            }
        }
        self::$raffle_list = [];
    }

    /**
     * @use 响应关闭
     * @param $client
     */
    private static function onClosed($client)
    {
    }

    /**
     * @use 发送握手包
     * @return bool
     */
    private static function sendHandShake(): bool
    {
        return self::writer(self::genHandshakePkg(self::$room_id));
    }

    /**
     * @use 心跳包
     * @return string
     */
    #[Pure]
    private static function genHeartBeatPkg(): string
    {
        return self::packMsg('', 0x0002);
    }

    /**
     * @use 握手包
     * @param $room_id
     * @return string
     */
    private static function genHandshakePkg($room_id): string
    {
        // {"uid":123,"roomid":123,"protover":3,"platform":"web","type":2,"key":""}
        return self::packMsg(json_encode([
            "uid" => 0,
            "roomid" => intval($room_id),
            "platform" => "web",
            "clientver" => "1.10.6",
            "protover" => 2,
            "type" => 2,
            "key" => self::$server['token'],
        ]), 0x0007);
    }

    /**
     * @use 打包数据
     * @param $value
     * @param $option
     * @return string
     */
    private static function packMsg($value, $option): string
    {
        $head = pack('NnnNN', 0x10 + strlen($value), 0x10, 0x01, $option, 0x0001);
        return $head . $value;
    }

    /**
     * @use 解包数据
     * @param $value
     * @return array|false
     */
    private static function unPackMsg($value): bool|array
    {
        if (strlen($value) < 4) {
            Log::warning("unPackMsg: 包头异常 " . strlen($value));
            return [];
        }
        // Log::info(json_encode($head, true));
        return unpack('Npacklen/nheadlen/nver/Nop/Nseq', $value);
    }

    /**
     * @use 心跳
     */
    private static function heartBeat()
    {
        foreach (self::$client_maps as $key => $client_info) {
            // 如果重连状态 跳过
            if (!$client_info['status']) {
                continue;
            }
            if ($client_info['heart_beat'] > time()) {
                continue;
            }
            self::$client = $client_info['client'];
            self::$area_id = $client_info['area_id'];
            self::$room_id = $client_info['room_id'];
            self::writer(self::genHeartBeatPkg());
            self::$client_maps[$key]['heart_beat'] = time() + 20;
        }
    }

    /**
     * @use 读数据
     * @param $length
     * @param bool $is_header
     * @return bool|array|string
     */
    private static function reader($length, bool $is_header = false): bool|array|string
    {
        $data = false;
        try {
            if (self::$client->selectRead(self::$socket_timeout)) {
                $ret = 0;
                $socket = self::$client->getResource();
                while ($length) {
                    if ($length < 1 || $length > 65535) {
                        Log::warning("Socket error: [$ret] [$length]" . PHP_EOL);
                        throw new Exception("Socket error: [$ret] [$length]");
                    }
                    $cnt = 0;
                    $w = NULL;
                    $e = NULL;
                    while ($cnt++ < 60) {
                        // reset read fdset when timeout
                        $r = array($socket);
                        $ret = socket_select($r, $w, $e, 1);
                        if ($ret === false)
                            throw new Exception("Socket error: ret == false");
                        if ($ret)
                            break;
                        Log::debug("Socket debug: select timeout" . PHP_EOL);
                    }
                    // Todo unable to read from socket[104]: Connection reset by peer
                    $ret = socket_recv($socket, $buffer, $length, 0);
                    if ($ret < 1) {
                        Log::warning("Socket error: [$ret] [$length]" . PHP_EOL);
                        throw new Exception("Socket error: [$ret] [$length]");
                    }
                    $data .= $buffer;
                    unset($buffer);
                    $length -= $ret;
                }
                if ($is_header) $data = self::unPackMsg($data);
            }
        } catch (Exception $e) {
            self::triggerReConnect([
                'area_id' => self::$area_id,
                'wait_time' => time() + 60
            ], self::formatErr($e));
            $data = false;
        }
        return $data;
    }

    /**
     * @use 写数据
     * @param $data
     * @return bool
     */
    private static function writer($data): bool
    {
        $status = false;
        try {
            while (self::$client->selectWrite(self::$socket_timeout)) {
                $status = self::$client->write($data);
                break;
            }
        } catch (Exception $e) {
            self::triggerReConnect([
                'area_id' => self::$area_id,
                'wait_time' => time() + 60
            ], self::formatErr($e));
        }
        return $status;
    }

    /**
     * @use 读取数据
     */
    private static function receive()
    {
        foreach (self::$client_maps as $client_info) {
            // 如果重连状态 就跳过
            if (!$client_info['status']) {
                continue;
            }
            self::$client = $client_info['client'];
            self::$area_id = $client_info['area_id'];
            self::$room_id = $client_info['room_id'];
            $head = self::reader(16, true);
            if (!$head) {
                // 长度为0 ，空信息
                continue;
            }
            $length = $head['packlen'] ?? 16;
            $type = $head['op'] ?? 0x0000;
            $len_body = $length - 16;
            Log::debug("(AreaId={$client_info['area_id']} -> RoomId={$client_info['room_id']} -> Len=$len_body)");
            if (!$len_body)
                continue;
            $body = self::reader($len_body);
            if ($body) {
                if ($head['ver'] == 2) {
                    $data_list = self::v2_split($body, $len_body);
                    foreach ($data_list as $body) {
                        self::onMessage($body, $type);
                    }
                } else {
                    self::onMessage($body, $type);
                }
            }
        }
    }

    /**
     * @use V2切割
     * @param $bin
     * @param $total
     * @return array
     */
    private static function v2_split($bin, $total): array
    {
        $list = [];
        $step = 0;
        $data = gzuncompress($bin);
        $total = strlen($data);
        while (true) {
            if ($step > 165535) {
                Log::warning("v2_split: 数据step异常 $step");
                break;
            }
            if ($step == $total) break;
            $bin = substr($data, $step, 16);
            $head = self::unPackMsg($bin);
            $length = $head['packlen'] ?? 16;
            $body = substr($data, $step + 16, $length - 16);
            $step += $length;
            $list[] = $body;
        }
        return $list;
    }

    /**
     * @param $object
     * @return string
     */
    private static function getClass($object): string
    {
        $class = get_class($object);
        return 'c' === $class[0] && str_starts_with($class, "class@anonymous\0") ? get_parent_class($class) . '@anonymous' : $class;
    }

    /**
     * @param $e
     * @return string
     */
    private static function formatErr($e): string
    {
        return sprintf('Uncaught Exception %s: "%s" at %s line %s', self::getClass($e), $e->getMessage(), $e->getFile(), $e->getLine());
    }

    /*
     * @use replace delay by select
     */
    public static function Delayed(): Delayed
    {
        $r = [];
        $w = NULL;
        $e = NULL;
        $delay = 0;
        if (self::getLock() > time())
            return new Delayed(1000);
        try {
            foreach (self::$client_maps as $client_info) {
                if ($client_info['client'])
                    $r[] = $client_info['client']->getResource();
            }
            if (count($r) !== 0)
                socket_select($r, $w, $e, 1);
            else
                $delay = 50;
        } catch (Exception $exception) {
            $delay = 1000;
        }
        return new Delayed($delay);
    }
}