2019-02-19 23:15:00 +08:00
|
|
|
|
# -*- coding: utf-8 -*-
|
2019-02-20 00:25:14 +08:00
|
|
|
|
import asyncio
|
2021-12-12 21:54:07 +08:00
|
|
|
|
import collections
|
|
|
|
|
import enum
|
2019-02-14 18:36:37 +08:00
|
|
|
|
import json
|
2019-02-20 14:53:50 +08:00
|
|
|
|
import logging
|
2019-06-12 22:32:34 +08:00
|
|
|
|
import ssl as ssl_
|
2019-02-19 23:15:00 +08:00
|
|
|
|
import struct
|
2019-09-15 18:46:45 +08:00
|
|
|
|
import zlib
|
2019-06-06 21:50:51 +08:00
|
|
|
|
from typing import *
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2018-06-03 14:06:00 +08:00
|
|
|
|
import aiohttp
|
2018-05-13 21:57:36 +08:00
|
|
|
|
|
2021-12-12 21:54:07 +08:00
|
|
|
|
from . import models
|
|
|
|
|
|
2021-12-12 19:28:12 +08:00
|
|
|
|
logger = logging.getLogger('blivedm')
|
2019-02-20 14:53:50 +08:00
|
|
|
|
|
2020-07-19 22:17:32 +08:00
|
|
|
|
ROOM_INIT_URL = 'https://api.live.bilibili.com/xlive/web-room/v1/index/getInfoByRoom'
|
|
|
|
|
DANMAKU_SERVER_CONF_URL = 'https://api.live.bilibili.com/xlive/web-room/v1/index/getDanmuInfo'
|
2020-10-11 12:43:11 +08:00
|
|
|
|
DEFAULT_DANMAKU_SERVER_LIST = [
|
|
|
|
|
{'host': 'broadcastlv.chat.bilibili.com', 'port': 2243, 'wss_port': 443, 'ws_port': 2244}
|
|
|
|
|
]
|
2018-05-13 21:57:36 +08:00
|
|
|
|
|
2019-04-22 19:47:05 +08:00
|
|
|
|
HEADER_STRUCT = struct.Struct('>I2H2I')
|
2021-12-12 21:54:07 +08:00
|
|
|
|
HeaderTuple = collections.namedtuple('HeaderTuple', ('pack_len', 'raw_header_size', 'ver', 'operation', 'seq_id'))
|
2021-01-31 13:11:20 +08:00
|
|
|
|
WS_BODY_PROTOCOL_VERSION_INFLATE = 0
|
|
|
|
|
WS_BODY_PROTOCOL_VERSION_NORMAL = 1
|
2019-09-15 18:46:45 +08:00
|
|
|
|
WS_BODY_PROTOCOL_VERSION_DEFLATE = 2
|
2019-04-22 19:47:05 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# go-common\app\service\main\broadcast\model\operation.go
|
2021-12-12 21:54:07 +08:00
|
|
|
|
class Operation(enum.IntEnum):
|
2019-04-22 19:47:05 +08:00
|
|
|
|
HANDSHAKE = 0
|
|
|
|
|
HANDSHAKE_REPLY = 1
|
|
|
|
|
HEARTBEAT = 2
|
|
|
|
|
HEARTBEAT_REPLY = 3
|
|
|
|
|
SEND_MSG = 4
|
|
|
|
|
SEND_MSG_REPLY = 5
|
|
|
|
|
DISCONNECT_REPLY = 6
|
2019-02-19 23:15:00 +08:00
|
|
|
|
AUTH = 7
|
2019-04-22 19:47:05 +08:00
|
|
|
|
AUTH_REPLY = 8
|
|
|
|
|
RAW = 9
|
|
|
|
|
PROTO_READY = 10
|
|
|
|
|
PROTO_FINISH = 11
|
|
|
|
|
CHANGE_ROOM = 12
|
|
|
|
|
CHANGE_ROOM_REPLY = 13
|
|
|
|
|
REGISTER = 14
|
|
|
|
|
REGISTER_REPLY = 15
|
|
|
|
|
UNREGISTER = 16
|
|
|
|
|
UNREGISTER_REPLY = 17
|
|
|
|
|
# B站业务自定义OP
|
|
|
|
|
# MinBusinessOp = 1000
|
|
|
|
|
# MaxBusinessOp = 10000
|
2018-05-13 21:57:36 +08:00
|
|
|
|
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-09-15 18:46:45 +08:00
|
|
|
|
class InitError(Exception):
|
|
|
|
|
"""初始化失败"""
|
|
|
|
|
|
|
|
|
|
|
2019-02-19 23:15:00 +08:00
|
|
|
|
class BLiveClient:
|
2019-06-06 21:50:51 +08:00
|
|
|
|
_COMMAND_HANDLERS: Dict[str, Optional[Callable[['BLiveClient', dict], Awaitable]]] = {
|
2019-03-23 23:58:02 +08:00
|
|
|
|
# 收到弹幕
|
2019-04-22 19:47:05 +08:00
|
|
|
|
# go-common\app\service\live\live-dm\service\v1\send.go
|
2021-12-12 21:54:07 +08:00
|
|
|
|
'DANMU_MSG': lambda client, command: client._on_receive_danmaku( # noqa
|
|
|
|
|
models.DanmakuMessage.from_command(command['info'])
|
2019-03-23 23:58:02 +08:00
|
|
|
|
),
|
|
|
|
|
# 有人送礼
|
2021-12-12 21:54:07 +08:00
|
|
|
|
'SEND_GIFT': lambda client, command: client._on_receive_gift( # noqa
|
|
|
|
|
models.GiftMessage.from_command(command['data'])
|
2019-06-06 21:50:51 +08:00
|
|
|
|
),
|
|
|
|
|
# 有人上舰
|
2021-12-12 21:54:07 +08:00
|
|
|
|
'GUARD_BUY': lambda client, command: client._on_buy_guard( # noqa
|
|
|
|
|
models.GuardBuyMessage.from_command(command['data'])
|
2019-09-23 21:43:37 +08:00
|
|
|
|
),
|
|
|
|
|
# 醒目留言
|
2021-12-12 21:54:07 +08:00
|
|
|
|
'SUPER_CHAT_MESSAGE': lambda client, command: client._on_super_chat( # noqa
|
|
|
|
|
models.SuperChatMessage.from_command(command['data'])
|
2019-09-23 21:43:37 +08:00
|
|
|
|
),
|
|
|
|
|
# 删除醒目留言
|
2021-12-12 21:54:07 +08:00
|
|
|
|
'SUPER_CHAT_MESSAGE_DELETE': lambda client, command: client._on_super_chat_delete( # noqa
|
|
|
|
|
models.SuperChatDeleteMessage.from_command(command['data'])
|
2019-03-23 23:58:02 +08:00
|
|
|
|
)
|
|
|
|
|
}
|
2020-08-09 20:28:45 +08:00
|
|
|
|
# 其他常见命令
|
|
|
|
|
for cmd in (
|
|
|
|
|
'INTERACT_WORD', 'ROOM_BANNER', 'ROOM_REAL_TIME_MESSAGE_UPDATE', 'NOTICE_MSG', 'COMBO_SEND',
|
|
|
|
|
'COMBO_END', 'ENTRY_EFFECT', 'WELCOME_GUARD', 'WELCOME', 'ROOM_RANK', 'ACTIVITY_BANNER_UPDATE_V2',
|
|
|
|
|
'PANEL', 'SUPER_CHAT_MESSAGE_JPN', 'USER_TOAST_MSG', 'ROOM_BLOCK_MSG', 'LIVE', 'PREPARING',
|
|
|
|
|
'room_admin_entrance', 'ROOM_ADMINS', 'ROOM_CHANGE'
|
2019-03-23 23:58:02 +08:00
|
|
|
|
):
|
|
|
|
|
_COMMAND_HANDLERS[cmd] = None
|
2019-11-13 23:26:25 +08:00
|
|
|
|
del cmd
|
2019-03-23 23:58:02 +08:00
|
|
|
|
|
2021-12-12 21:54:07 +08:00
|
|
|
|
def __init__(self, room_id, uid=0, session: aiohttp.ClientSession = None,
|
2019-09-15 18:46:45 +08:00
|
|
|
|
heartbeat_interval=30, ssl=True, loop=None):
|
2019-02-19 23:15:00 +08:00
|
|
|
|
"""
|
2019-06-06 21:50:51 +08:00
|
|
|
|
:param room_id: URL中的房间ID,可以为短ID
|
2019-09-15 18:46:45 +08:00
|
|
|
|
:param uid: B站用户ID,0表示未登录
|
|
|
|
|
:param session: cookie、连接池
|
|
|
|
|
:param heartbeat_interval: 发送心跳包的间隔时间(秒)
|
2019-02-19 23:15:00 +08:00
|
|
|
|
:param ssl: True表示用默认的SSLContext验证,False表示不验证,也可以传入SSLContext
|
|
|
|
|
:param loop: 协程事件循环
|
|
|
|
|
"""
|
2019-06-06 21:50:51 +08:00
|
|
|
|
# 用来init_room的临时房间ID
|
|
|
|
|
self._tmp_room_id = room_id
|
|
|
|
|
# 调用init_room后初始化
|
|
|
|
|
self._room_id = self._room_short_id = self._room_owner_uid = None
|
2019-09-15 18:46:45 +08:00
|
|
|
|
# [{host: "tx-bj4-live-comet-04.chat.bilibili.com", port: 2243, wss_port: 443, ws_port: 2244}, ...]
|
|
|
|
|
self._host_server_list = None
|
|
|
|
|
self._host_server_token = None
|
2019-02-20 00:25:14 +08:00
|
|
|
|
self._uid = uid
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-02-20 14:12:11 +08:00
|
|
|
|
if loop is not None:
|
|
|
|
|
self._loop = loop
|
|
|
|
|
elif session is not None:
|
2021-12-12 21:54:07 +08:00
|
|
|
|
self._loop = session.loop # noqa
|
2019-02-20 14:12:11 +08:00
|
|
|
|
else:
|
|
|
|
|
self._loop = asyncio.get_event_loop()
|
2019-06-06 21:50:51 +08:00
|
|
|
|
self._future = None
|
2019-02-20 00:25:14 +08:00
|
|
|
|
|
|
|
|
|
if session is None:
|
2021-03-28 17:47:23 +08:00
|
|
|
|
self._session = aiohttp.ClientSession(loop=self._loop, timeout=aiohttp.ClientTimeout(total=10))
|
2019-02-20 00:25:14 +08:00
|
|
|
|
self._own_session = True
|
|
|
|
|
else:
|
|
|
|
|
self._session = session
|
|
|
|
|
self._own_session = False
|
2021-12-12 21:54:07 +08:00
|
|
|
|
if self._session.loop is not self._loop: # noqa
|
2019-02-20 00:25:14 +08:00
|
|
|
|
raise RuntimeError('BLiveClient and session has to use same event loop')
|
2019-09-15 18:46:45 +08:00
|
|
|
|
|
|
|
|
|
self._heartbeat_interval = heartbeat_interval
|
2021-12-12 21:54:07 +08:00
|
|
|
|
self._ssl = ssl if ssl else ssl_._create_unverified_context() # noqa
|
2019-02-19 23:15:00 +08:00
|
|
|
|
self._websocket = None
|
2021-01-31 13:11:20 +08:00
|
|
|
|
self._heartbeat_timer_handle = None
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-02-20 14:12:11 +08:00
|
|
|
|
@property
|
|
|
|
|
def is_running(self):
|
2019-06-06 21:50:51 +08:00
|
|
|
|
return self._future is not None
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def room_id(self):
|
|
|
|
|
"""
|
|
|
|
|
房间ID,调用init_room后初始化
|
|
|
|
|
"""
|
|
|
|
|
return self._room_id
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def room_short_id(self):
|
|
|
|
|
"""
|
|
|
|
|
房间短ID,没有则为0,调用init_room后初始化
|
|
|
|
|
"""
|
|
|
|
|
return self._room_short_id
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
|
def room_owner_uid(self):
|
|
|
|
|
"""
|
|
|
|
|
主播ID,调用init_room后初始化
|
|
|
|
|
"""
|
|
|
|
|
return self._room_owner_uid
|
2019-02-20 14:12:11 +08:00
|
|
|
|
|
2019-02-20 00:25:14 +08:00
|
|
|
|
async def close(self):
|
|
|
|
|
"""
|
|
|
|
|
如果session是自己创建的则关闭session
|
|
|
|
|
"""
|
|
|
|
|
if self._own_session:
|
|
|
|
|
await self._session.close()
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-06-06 21:50:51 +08:00
|
|
|
|
def start(self):
|
2019-02-19 23:15:00 +08:00
|
|
|
|
"""
|
|
|
|
|
创建相关的协程,不会执行事件循环
|
2019-02-20 14:12:11 +08:00
|
|
|
|
:return: 协程的future
|
2019-02-19 23:15:00 +08:00
|
|
|
|
"""
|
2019-06-06 21:50:51 +08:00
|
|
|
|
if self._future is not None:
|
2019-02-20 14:12:11 +08:00
|
|
|
|
raise RuntimeError('This client is already running')
|
2019-06-06 21:50:51 +08:00
|
|
|
|
self._future = asyncio.ensure_future(self._message_loop(), loop=self._loop)
|
2019-06-12 22:32:34 +08:00
|
|
|
|
self._future.add_done_callback(self.__on_message_loop_done)
|
2019-06-06 21:50:51 +08:00
|
|
|
|
return self._future
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-06-12 22:32:34 +08:00
|
|
|
|
def __on_message_loop_done(self, future):
|
|
|
|
|
self._future = None
|
|
|
|
|
logger.debug('room %s 消息协程结束', self.room_id)
|
|
|
|
|
exception = future.exception()
|
|
|
|
|
if exception is not None:
|
|
|
|
|
logger.exception('room %s 消息协程异常结束:', self.room_id,
|
|
|
|
|
exc_info=(type(exception), exception, exception.__traceback__))
|
|
|
|
|
|
2019-06-06 21:50:51 +08:00
|
|
|
|
def stop(self):
|
|
|
|
|
"""
|
|
|
|
|
停止相关的协程
|
|
|
|
|
:return: 协程的future
|
|
|
|
|
"""
|
|
|
|
|
if self._future is None:
|
|
|
|
|
raise RuntimeError('This client is not running')
|
|
|
|
|
self._future.cancel()
|
|
|
|
|
return self._future
|
|
|
|
|
|
|
|
|
|
async def init_room(self):
|
2020-10-11 12:43:11 +08:00
|
|
|
|
"""
|
|
|
|
|
:return: True代表没有降级,如果需要降级后还可用,重载这个函数返回True
|
|
|
|
|
"""
|
|
|
|
|
res = True
|
|
|
|
|
if not await self._init_room_id_and_owner():
|
|
|
|
|
res = False
|
|
|
|
|
# 失败了则降级
|
|
|
|
|
self._room_id = self._room_short_id = self._tmp_room_id
|
|
|
|
|
self._room_owner_uid = 0
|
|
|
|
|
|
|
|
|
|
if not await self._init_host_server():
|
|
|
|
|
res = False
|
|
|
|
|
# 失败了则降级
|
|
|
|
|
self._host_server_list = DEFAULT_DANMAKU_SERVER_LIST
|
|
|
|
|
self._host_server_token = None
|
|
|
|
|
return res
|
|
|
|
|
|
|
|
|
|
async def _init_room_id_and_owner(self):
|
2019-09-15 18:46:45 +08:00
|
|
|
|
try:
|
2020-07-19 22:17:32 +08:00
|
|
|
|
async with self._session.get(ROOM_INIT_URL, params={'room_id': self._tmp_room_id},
|
2019-09-15 18:46:45 +08:00
|
|
|
|
ssl=self._ssl) as res:
|
|
|
|
|
if res.status != 200:
|
2020-07-19 22:17:32 +08:00
|
|
|
|
logger.warning('room %d init_room失败:%d %s', self._tmp_room_id,
|
2019-09-15 18:46:45 +08:00
|
|
|
|
res.status, res.reason)
|
|
|
|
|
return False
|
|
|
|
|
data = await res.json()
|
|
|
|
|
if data['code'] != 0:
|
2020-10-17 17:09:02 +08:00
|
|
|
|
logger.warning('room %d init_room失败:%s', self._tmp_room_id, data['message'])
|
2019-09-15 18:46:45 +08:00
|
|
|
|
return False
|
|
|
|
|
if not self._parse_room_init(data['data']):
|
|
|
|
|
return False
|
2021-03-28 17:47:23 +08:00
|
|
|
|
except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
|
2020-07-19 22:17:32 +08:00
|
|
|
|
logger.exception('room %d init_room失败:', self._tmp_room_id)
|
2019-09-15 18:46:45 +08:00
|
|
|
|
return False
|
2020-10-11 12:43:11 +08:00
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
|
def _parse_room_init(self, data):
|
|
|
|
|
room_info = data['room_info']
|
|
|
|
|
self._room_id = room_info['room_id']
|
|
|
|
|
self._room_short_id = room_info['short_id']
|
|
|
|
|
self._room_owner_uid = room_info['uid']
|
|
|
|
|
return True
|
2019-09-15 18:46:45 +08:00
|
|
|
|
|
2020-10-11 12:43:11 +08:00
|
|
|
|
async def _init_host_server(self):
|
2019-09-15 18:46:45 +08:00
|
|
|
|
try:
|
2020-07-19 22:17:32 +08:00
|
|
|
|
async with self._session.get(DANMAKU_SERVER_CONF_URL, params={'id': self._room_id, 'type': 0},
|
2019-09-15 18:46:45 +08:00
|
|
|
|
ssl=self._ssl) as res:
|
|
|
|
|
if res.status != 200:
|
2020-07-19 22:17:32 +08:00
|
|
|
|
logger.warning('room %d getConf失败:%d %s', self._room_id,
|
2019-09-15 18:46:45 +08:00
|
|
|
|
res.status, res.reason)
|
|
|
|
|
return False
|
|
|
|
|
data = await res.json()
|
|
|
|
|
if data['code'] != 0:
|
2020-10-17 17:09:02 +08:00
|
|
|
|
logger.warning('room %d getConf失败:%s', self._room_id, data['message'])
|
2019-09-15 18:46:45 +08:00
|
|
|
|
return False
|
2020-10-11 12:43:11 +08:00
|
|
|
|
if not self._parse_danmaku_server_conf(data['data']):
|
2019-09-15 18:46:45 +08:00
|
|
|
|
return False
|
2021-03-28 17:47:23 +08:00
|
|
|
|
except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
|
2020-07-19 22:17:32 +08:00
|
|
|
|
logger.exception('room %d getConf失败:', self._room_id)
|
2019-09-15 18:46:45 +08:00
|
|
|
|
return False
|
|
|
|
|
return True
|
2019-06-06 21:50:51 +08:00
|
|
|
|
|
2020-10-11 12:43:11 +08:00
|
|
|
|
def _parse_danmaku_server_conf(self, data):
|
|
|
|
|
self._host_server_list = data['host_list']
|
|
|
|
|
self._host_server_token = data['token']
|
|
|
|
|
if not self._host_server_list:
|
|
|
|
|
logger.warning('room %d getConf失败:host_server_list为空', self._room_id)
|
|
|
|
|
return False
|
2019-09-15 18:46:45 +08:00
|
|
|
|
return True
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2021-01-31 13:11:20 +08:00
|
|
|
|
@staticmethod
|
|
|
|
|
def _make_packet(data, operation):
|
2019-02-19 23:15:00 +08:00
|
|
|
|
body = json.dumps(data).encode('utf-8')
|
2019-04-22 19:47:05 +08:00
|
|
|
|
header = HEADER_STRUCT.pack(
|
|
|
|
|
HEADER_STRUCT.size + len(body),
|
|
|
|
|
HEADER_STRUCT.size,
|
2019-02-19 23:15:00 +08:00
|
|
|
|
1,
|
|
|
|
|
operation,
|
|
|
|
|
1
|
|
|
|
|
)
|
|
|
|
|
return header + body
|
|
|
|
|
|
|
|
|
|
async def _send_auth(self):
|
|
|
|
|
auth_params = {
|
|
|
|
|
'uid': self._uid,
|
|
|
|
|
'roomid': self._room_id,
|
2019-09-15 18:46:45 +08:00
|
|
|
|
'protover': 2,
|
2019-02-19 23:15:00 +08:00
|
|
|
|
'platform': 'web',
|
2020-07-19 22:17:32 +08:00
|
|
|
|
'clientver': '1.14.3',
|
2020-10-11 12:43:11 +08:00
|
|
|
|
'type': 2
|
2019-02-19 23:15:00 +08:00
|
|
|
|
}
|
2020-10-11 12:43:11 +08:00
|
|
|
|
if self._host_server_token is not None:
|
|
|
|
|
auth_params['key'] = self._host_server_token
|
2019-02-20 00:25:14 +08:00
|
|
|
|
await self._websocket.send_bytes(self._make_packet(auth_params, Operation.AUTH))
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
|
|
|
|
async def _message_loop(self):
|
2019-09-15 18:46:45 +08:00
|
|
|
|
# 如果之前未初始化则初始化
|
|
|
|
|
if self._host_server_token is None:
|
|
|
|
|
if not await self.init_room():
|
|
|
|
|
raise InitError('初始化失败')
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-06-07 20:02:37 +08:00
|
|
|
|
retry_count = 0
|
2018-05-14 01:02:53 +08:00
|
|
|
|
while True:
|
2019-02-19 23:15:00 +08:00
|
|
|
|
try:
|
|
|
|
|
# 连接
|
2019-09-15 18:46:45 +08:00
|
|
|
|
host_server = self._host_server_list[retry_count % len(self._host_server_list)]
|
|
|
|
|
async with self._session.ws_connect(
|
|
|
|
|
f'wss://{host_server["host"]}:{host_server["wss_port"]}/sub',
|
2021-01-31 13:47:32 +08:00
|
|
|
|
receive_timeout=self._heartbeat_interval + 5,
|
2019-09-15 18:46:45 +08:00
|
|
|
|
ssl=self._ssl
|
|
|
|
|
) as websocket:
|
2019-02-19 23:15:00 +08:00
|
|
|
|
self._websocket = websocket
|
|
|
|
|
await self._send_auth()
|
2021-01-31 13:11:20 +08:00
|
|
|
|
self._heartbeat_timer_handle = self._loop.call_later(
|
|
|
|
|
self._heartbeat_interval, self._on_send_heartbeat
|
2019-06-12 22:32:34 +08:00
|
|
|
|
)
|
2018-05-14 01:02:53 +08:00
|
|
|
|
|
2019-02-19 23:15:00 +08:00
|
|
|
|
# 处理消息
|
2021-12-12 21:54:07 +08:00
|
|
|
|
message: aiohttp.WSMessage
|
|
|
|
|
async for message in websocket:
|
2019-11-27 01:10:13 +08:00
|
|
|
|
retry_count = 0
|
2021-01-31 13:11:20 +08:00
|
|
|
|
if message.type != aiohttp.WSMsgType.BINARY:
|
2019-06-06 21:50:51 +08:00
|
|
|
|
logger.warning('room %d 未知的websocket消息:type=%s %s', self.room_id,
|
|
|
|
|
message.type, message.data)
|
2021-01-31 13:11:20 +08:00
|
|
|
|
continue
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
await self._handle_message(message.data)
|
|
|
|
|
except asyncio.CancelledError:
|
|
|
|
|
raise
|
2021-12-12 21:54:07 +08:00
|
|
|
|
except Exception: # noqa
|
2021-01-31 13:11:20 +08:00
|
|
|
|
logger.exception('room %d 处理消息时发生错误:', self.room_id)
|
2018-05-13 23:49:32 +08:00
|
|
|
|
|
2019-02-20 00:25:14 +08:00
|
|
|
|
except asyncio.CancelledError:
|
2019-02-19 23:15:00 +08:00
|
|
|
|
break
|
2019-06-21 17:45:21 +08:00
|
|
|
|
except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
|
2019-02-19 23:15:00 +08:00
|
|
|
|
# 重连
|
2019-06-07 20:02:37 +08:00
|
|
|
|
pass
|
2019-06-12 22:32:34 +08:00
|
|
|
|
except ssl_.SSLError:
|
|
|
|
|
logger.exception('SSL错误:')
|
|
|
|
|
# 证书错误时无法重连
|
|
|
|
|
break
|
2019-02-19 23:15:00 +08:00
|
|
|
|
finally:
|
|
|
|
|
self._websocket = None
|
2021-01-31 13:11:20 +08:00
|
|
|
|
if self._heartbeat_timer_handle is not None:
|
|
|
|
|
self._heartbeat_timer_handle.cancel()
|
|
|
|
|
self._heartbeat_timer_handle = None
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-06-07 20:02:37 +08:00
|
|
|
|
retry_count += 1
|
|
|
|
|
logger.warning('room %d 掉线重连中%d', self.room_id, retry_count)
|
|
|
|
|
try:
|
|
|
|
|
await asyncio.sleep(1)
|
|
|
|
|
except asyncio.CancelledError:
|
|
|
|
|
break
|
|
|
|
|
|
2021-01-31 13:11:20 +08:00
|
|
|
|
def _on_send_heartbeat(self):
|
|
|
|
|
coro = self._websocket.send_bytes(self._make_packet({}, Operation.HEARTBEAT))
|
|
|
|
|
asyncio.ensure_future(coro, loop=self._loop)
|
|
|
|
|
self._heartbeat_timer_handle = self._loop.call_later(self._heartbeat_interval, self._on_send_heartbeat)
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-09-15 18:46:45 +08:00
|
|
|
|
async def _handle_message(self, data):
|
2019-02-19 23:15:00 +08:00
|
|
|
|
offset = 0
|
2019-09-15 18:46:45 +08:00
|
|
|
|
while offset < len(data):
|
2019-02-19 23:15:00 +08:00
|
|
|
|
try:
|
2019-09-15 18:46:45 +08:00
|
|
|
|
header = HeaderTuple(*HEADER_STRUCT.unpack_from(data, offset))
|
2019-02-19 23:15:00 +08:00
|
|
|
|
except struct.error:
|
|
|
|
|
break
|
|
|
|
|
|
2019-04-22 19:47:05 +08:00
|
|
|
|
if header.operation == Operation.HEARTBEAT_REPLY:
|
2019-09-15 18:46:45 +08:00
|
|
|
|
popularity = int.from_bytes(data[offset + HEADER_STRUCT.size:
|
|
|
|
|
offset + HEADER_STRUCT.size + 4],
|
2019-02-19 23:15:00 +08:00
|
|
|
|
'big')
|
2019-06-06 21:50:51 +08:00
|
|
|
|
await self._on_receive_popularity(popularity)
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-04-22 19:47:05 +08:00
|
|
|
|
elif header.operation == Operation.SEND_MSG_REPLY:
|
2019-09-15 18:46:45 +08:00
|
|
|
|
body = data[offset + HEADER_STRUCT.size: offset + header.pack_len]
|
2019-10-06 11:06:51 +08:00
|
|
|
|
if header.ver == WS_BODY_PROTOCOL_VERSION_DEFLATE:
|
2021-01-31 13:11:20 +08:00
|
|
|
|
body = await self._loop.run_in_executor(None, zlib.decompress, body)
|
2019-09-15 18:46:45 +08:00
|
|
|
|
await self._handle_message(body)
|
|
|
|
|
else:
|
2019-11-13 23:26:25 +08:00
|
|
|
|
try:
|
|
|
|
|
body = json.loads(body.decode('utf-8'))
|
|
|
|
|
await self._handle_command(body)
|
2021-01-31 13:11:20 +08:00
|
|
|
|
except Exception:
|
2019-11-13 23:26:25 +08:00
|
|
|
|
logger.error('body: %s', body)
|
|
|
|
|
raise
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-04-22 19:47:05 +08:00
|
|
|
|
elif header.operation == Operation.AUTH_REPLY:
|
|
|
|
|
await self._websocket.send_bytes(self._make_packet({}, Operation.HEARTBEAT))
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
|
|
|
|
else:
|
2019-09-15 18:46:45 +08:00
|
|
|
|
body = data[offset + HEADER_STRUCT.size: offset + header.pack_len]
|
2019-06-06 21:50:51 +08:00
|
|
|
|
logger.warning('room %d 未知包类型:operation=%d %s%s', self.room_id,
|
|
|
|
|
header.operation, header, body)
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-04-22 19:47:05 +08:00
|
|
|
|
offset += header.pack_len
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
|
|
|
|
async def _handle_command(self, command):
|
|
|
|
|
if isinstance(command, list):
|
|
|
|
|
for one_command in command:
|
|
|
|
|
await self._handle_command(one_command)
|
|
|
|
|
return
|
|
|
|
|
|
2019-11-13 23:26:25 +08:00
|
|
|
|
cmd = command.get('cmd', '')
|
2019-05-29 22:28:10 +08:00
|
|
|
|
pos = cmd.find(':') # 2019-5-29 B站弹幕升级新增了参数
|
|
|
|
|
if pos != -1:
|
|
|
|
|
cmd = cmd[:pos]
|
2019-03-23 23:58:02 +08:00
|
|
|
|
if cmd in self._COMMAND_HANDLERS:
|
|
|
|
|
handler = self._COMMAND_HANDLERS[cmd]
|
|
|
|
|
if handler is not None:
|
|
|
|
|
await handler(self, command)
|
2019-02-17 17:01:52 +08:00
|
|
|
|
else:
|
2019-06-06 21:50:51 +08:00
|
|
|
|
logger.warning('room %d 未知命令:cmd=%s %s', self.room_id, cmd, command)
|
2020-08-09 20:28:45 +08:00
|
|
|
|
# 只有第一次遇到未知命令时log
|
2019-09-18 00:09:02 +08:00
|
|
|
|
self._COMMAND_HANDLERS[cmd] = None
|
2019-02-19 23:15:00 +08:00
|
|
|
|
|
2019-06-06 21:50:51 +08:00
|
|
|
|
async def _on_receive_popularity(self, popularity: int):
|
2019-02-19 23:15:00 +08:00
|
|
|
|
"""
|
2019-06-06 21:50:51 +08:00
|
|
|
|
收到人气值
|
2019-02-19 23:15:00 +08:00
|
|
|
|
"""
|
|
|
|
|
pass
|
|
|
|
|
|
2021-12-12 21:54:07 +08:00
|
|
|
|
async def _on_receive_danmaku(self, danmaku: models.DanmakuMessage):
|
2019-02-19 23:15:00 +08:00
|
|
|
|
"""
|
2019-06-06 21:50:51 +08:00
|
|
|
|
收到弹幕
|
2019-02-19 23:15:00 +08:00
|
|
|
|
"""
|
|
|
|
|
pass
|
2019-03-23 23:58:02 +08:00
|
|
|
|
|
2021-12-12 21:54:07 +08:00
|
|
|
|
async def _on_receive_gift(self, gift: models.GiftMessage):
|
2019-03-23 23:58:02 +08:00
|
|
|
|
"""
|
2019-06-06 21:50:51 +08:00
|
|
|
|
收到礼物
|
|
|
|
|
"""
|
|
|
|
|
pass
|
|
|
|
|
|
2021-12-12 21:54:07 +08:00
|
|
|
|
async def _on_buy_guard(self, message: models.GuardBuyMessage):
|
2019-06-06 21:50:51 +08:00
|
|
|
|
"""
|
|
|
|
|
有人上舰
|
2019-03-23 23:58:02 +08:00
|
|
|
|
"""
|
|
|
|
|
pass
|
2019-09-23 21:43:37 +08:00
|
|
|
|
|
2021-12-12 21:54:07 +08:00
|
|
|
|
async def _on_super_chat(self, message: models.SuperChatMessage):
|
2019-09-23 21:43:37 +08:00
|
|
|
|
"""
|
|
|
|
|
醒目留言
|
|
|
|
|
"""
|
|
|
|
|
pass
|
|
|
|
|
|
2021-12-12 21:54:07 +08:00
|
|
|
|
async def _on_super_chat_delete(self, message: models.SuperChatDeleteMessage):
|
2019-09-23 21:43:37 +08:00
|
|
|
|
"""
|
|
|
|
|
删除醒目留言
|
|
|
|
|
"""
|
|
|
|
|
pass
|