mirror of
https://github.com/lkeme/BiliHelper-personal.git
synced 2025-12-19 09:30:10 +08:00
427 lines
12 KiB
PHP
427 lines
12 KiB
PHP
<?php
|
||
|
||
/**
|
||
* Website: https://mudew.com/
|
||
* Author: Lkeme
|
||
* License: The MIT License
|
||
* Email: Useri@live.cn
|
||
* Updated: 2020 ~ 2021
|
||
*/
|
||
|
||
namespace BiliHelper\Plugin;
|
||
|
||
use BiliHelper\Core\Log;
|
||
use BiliHelper\Util\TimeLock;
|
||
|
||
use Exception;
|
||
use Socket\Raw\Factory;
|
||
|
||
class ZoneTcpClient
|
||
{
|
||
use TimeLock;
|
||
private static $raffle_id = 0;
|
||
private static $raffle_list = [];
|
||
private static $server_addr = null;
|
||
private static $server_key = null;
|
||
|
||
private static $area_id;
|
||
private static $room_id;
|
||
private static $client;
|
||
private static $client_maps = [];
|
||
private static $trigger_restart = [];
|
||
private static $socket_timeout = 0;
|
||
|
||
|
||
/**
|
||
* @use 入口
|
||
*/
|
||
public static function run()
|
||
{
|
||
if (self::getLock() > time() || getenv('USE_ZONE_SERVER') == 'false') {
|
||
return;
|
||
}
|
||
self::init();
|
||
self::updateConnection();
|
||
self::heartBeat();
|
||
self::receive();
|
||
self::pushHandle();
|
||
}
|
||
|
||
|
||
/**
|
||
* @use 初始化
|
||
*/
|
||
private static function init()
|
||
{
|
||
if (empty(getenv('ZONE_SERVER_ADDR'))) {
|
||
exit('推送服务器信息不完整, 请检查配置文件!');
|
||
}
|
||
if (!self::$server_addr) {
|
||
self::$server_addr = getenv('ZONE_SERVER_ADDR');
|
||
}
|
||
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];
|
||
self::triggerReConnect([
|
||
'area_id' => $area_id,
|
||
'wait_time' => time()
|
||
]);
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* @use 触发重连
|
||
* @param array $area_data
|
||
*/
|
||
private static function triggerReConnect(array $area_data)
|
||
{
|
||
array_push(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']) {
|
||
array_push(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_info = [
|
||
// 'area_id' => $area_id,
|
||
// 'room_id' => 23058
|
||
// ];
|
||
self::update($area_info);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* @use 更新操作
|
||
* @param array $area_info
|
||
*/
|
||
private static function update(array $area_info)
|
||
{
|
||
self::$area_id = $area_info['area_id'];
|
||
self::$room_id = $area_info['room_id'];
|
||
try {
|
||
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;
|
||
self::$client_maps["server" . self::$area_id]['area_id'] = self::$area_id;
|
||
self::$client_maps["server" . self::$area_id]['room_id'] = self::$room_id;
|
||
self::$client_maps["server" . self::$area_id]['heart_beat'] = time() + 20;
|
||
Log::info("连接到 " . self::$client->getPeerName() . "#" . self::$area_id . " 推送服务器");
|
||
} catch (Exception $e) {
|
||
Log::error("连接到 #" . self::$area_id . " 推送服务器失败, {$e->getMessage()}");
|
||
self::triggerReConnect([
|
||
'area_id' => self::$area_id,
|
||
'wait_time' => time() + 60
|
||
]);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 判断字符串是否为 Json 格式
|
||
* @param string $data Json 字符串
|
||
* @param bool $assoc 是否返回对象or关联数组，默认返回关联数组
|
||
* @return array|bool|object 成功返回转换后的对象或数组，失败返回 false
|
||
*/
|
||
private static function analyJson($data = '', $assoc = true)
|
||
{
|
||
$data = json_decode($data, $assoc);
|
||
if (($data && is_object($data)) || (is_array($data) && !empty($data))) {
|
||
return $data;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
|
||
/**
|
||
* @use 响应数据
|
||
* @param $msg
|
||
* @return bool
|
||
*/
|
||
private static function onMessage($msg)
|
||
{
|
||
// 心跳后回复人气
|
||
if (strlen($msg) == 4) {
|
||
// $num = unpack('N', $msg)[1];
|
||
// Log::info("当前直播间现有 {$num} 人聚众搞基!");
|
||
return false;
|
||
}
|
||
$de_raw = self::analyJson($msg, true);
|
||
// 进入房间返回
|
||
if (isset($de_raw['code']) && !$de_raw['code']) {
|
||
return false;
|
||
}
|
||
// 部分cmd抽风
|
||
if (!$de_raw || !isset($de_raw['cmd'])) {
|
||
Log::warning("解析错误: {$msg}");
|
||
return false;
|
||
}
|
||
$data = [];
|
||
switch ($de_raw['cmd']) {
|
||
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 && strpos($msg_common, '节奏风暴') !== false) {
|
||
$data = [
|
||
'room_id' => $real_room_id,
|
||
'raffle_id' => self::$raffle_id++,
|
||
'raffle_title' => '节奏风暴',
|
||
'raffle_type' => 'raffle',
|
||
'source' => $msg
|
||
];
|
||
}
|
||
break;
|
||
case 'GUARD_MSG':
|
||
$data = [
|
||
'room_id' => $de_raw['roomid'],
|
||
'raffle_id' => self::$raffle_id++,
|
||
'raffle_title' => '总督舰长',
|
||
'raffle_type' => 'raffle',
|
||
'source' => $msg
|
||
];
|
||
break;
|
||
case 'SPECIAL_GIFT':
|
||
if (array_key_exists('39', $de_raw['data'])) {
|
||
if ($de_raw['data']['39']['action'] == 'start') {
|
||
$data = [
|
||
'room_id' => $de_raw['roomid'],
|
||
'raffle_id' => self::$raffle_id++,
|
||
'raffle_title' => '节奏风暴',
|
||
'raffle_type' => 'raffle',
|
||
'source' => $msg
|
||
];
|
||
}
|
||
}
|
||
break;
|
||
case 'SYS_GIFT':
|
||
/**
|
||
* 系统礼物消息, 广播
|
||
*/
|
||
break;
|
||
case 'SYS_MSG':
|
||
/**
|
||
* 系统消息, 广播
|
||
*/
|
||
break;
|
||
// TODO 支持更多消息类型
|
||
default:
|
||
$data = [];
|
||
break;
|
||
}
|
||
|
||
if (!empty($data)) {
|
||
unset($data['source']);
|
||
if (!isset(self::$raffle_list[$data['raffle_type']])) {
|
||
self::$raffle_list[$data['raffle_type']] = [];
|
||
}
|
||
array_push(self::$raffle_list[$data['raffle_type']], $data);
|
||
// Log::info("获取到分区 #" . self::$area_id . "# {$data['source']}");
|
||
// print_r($data);
|
||
}
|
||
}
|
||
|
||
|
||
/**
|
||
* @推送到上游处理
|
||
*/
|
||
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()
|
||
{
|
||
return self::writer(self::genHandshakePkg(self::$room_id));
|
||
}
|
||
|
||
/**
|
||
* @use 心跳包
|
||
* @return string
|
||
*/
|
||
private static function genHeartBeatPkg(): string
|
||
{
|
||
return self::packMsg('', 0x0002);
|
||
}
|
||
|
||
|
||
/**
|
||
* @use 握手包
|
||
* @param $room_id
|
||
* @return string
|
||
*/
|
||
private static function genHandshakePkg($room_id): string
|
||
{
|
||
return self::packMsg(json_encode([
|
||
'uid' => random_int(1000000, 2999999),
|
||
'roomid' => intval($room_id),
|
||
]), 0x0007);
|
||
}
|
||
|
||
/**
|
||
* @use 打包数据
|
||
* @param $value
|
||
* @param $option
|
||
* @return string
|
||
*/
|
||
private static function packMsg($value, $option)
|
||
{
|
||
$head = pack('NnnNN', 0x10 + strlen($value), 0x10, 0x01, $option, 0x0001);
|
||
return $head . $value;
|
||
}
|
||
|
||
|
||
/**
|
||
* @use 解包数据
|
||
* @param $value
|
||
* @return int|mixed
|
||
*/
|
||
private static function unPackMsg($value)
|
||
{
|
||
$res = unpack('N', $value);
|
||
return $res[1] - 16;
|
||
}
|
||
|
||
|
||
/**
|
||
* @use 心跳
|
||
*/
|
||
private static function heartBeat()
|
||
{
|
||
foreach (self::$client_maps as $key => $client_info) {
|
||
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
|
||
* @return array|bool|false
|
||
*/
|
||
private static function reader($length)
|
||
{
|
||
$data = false;
|
||
try {
|
||
while (self::$client->selectRead(self::$socket_timeout)) {
|
||
$data = self::$client->read($length);
|
||
if (!$data) {
|
||
throw new Exception("Connection failure");
|
||
}
|
||
if ($length == 16) $data = self::unPackMsg($data);
|
||
break;
|
||
}
|
||
} catch (Exception $exception) {
|
||
self::triggerReConnect([
|
||
'area_id' => self::$area_id,
|
||
'wait_time' => time() + 60
|
||
]);
|
||
}
|
||
return $data;
|
||
}
|
||
|
||
/**
|
||
* @use 写数据
|
||
* @param $data
|
||
* @return bool
|
||
*/
|
||
private static function writer($data)
|
||
{
|
||
$status = false;
|
||
try {
|
||
while (self::$client->selectWrite(self::$socket_timeout)) {
|
||
$status = self::$client->write($data);
|
||
break;
|
||
}
|
||
} catch (Exception $exception) {
|
||
self::triggerReConnect([
|
||
'area_id' => self::$area_id,
|
||
'wait_time' => time() + 60
|
||
]);
|
||
}
|
||
return $status;
|
||
}
|
||
|
||
|
||
/**
|
||
* @use 读取数据
|
||
*/
|
||
private static function receive()
|
||
{
|
||
foreach (self::$client_maps as $client_info) {
|
||
self::$client = $client_info['client'];
|
||
self::$area_id = $client_info['area_id'];
|
||
self::$room_id = $client_info['room_id'];
|
||
$len_body = self::reader(16);
|
||
if (!$len_body) {
|
||
// 长度为0 ，空信息
|
||
continue;
|
||
}
|
||
Log::debug("(len=$len_body)");
|
||
$body = self::reader($len_body);
|
||
self::onMessage($body);
|
||
}
|
||
}
|
||
} |