更新版本号v1.8.0

close #120

.dockerignore

@@ -18,4 +18,7 @@ README.md
 # runtime data
 data/*
 !data/config.example.ini
+!data/emoticons/
+data/emoticons/*
+!data/emoticons/.gitkeep
 log/*

.gitignore

@@ -107,4 +107,7 @@ venv.bak/
 .idea/
 data/*
 !data/config.example.ini
+!data/emoticons/
+data/emoticons/*
+!data/emoticons/.gitkeep
 log/*

Dockerfile

@@ -1,35 +1,47 @@
-# 运行时
-FROM python:3.7.10-slim-stretch
-RUN mv /etc/apt/sources.list /etc/apt/sources.list.bak \
-    && echo "deb http://mirrors.tuna.tsinghua.edu.cn/debian/ stretch main contrib non-free">>/etc/apt/sources.list \
-    && echo "deb http://mirrors.tuna.tsinghua.edu.cn/debian/ stretch-updates main contrib non-free">>/etc/apt/sources.list \
-    && echo "deb http://mirrors.tuna.tsinghua.edu.cn/debian/ stretch-backports main contrib non-free">>/etc/apt/sources.list \
-    && echo "deb http://mirrors.tuna.tsinghua.edu.cn/debian-security stretch/updates main contrib non-free">>/etc/apt/sources.list \
-    && apt-get update \
-    && apt-get install -y wget tar xz-utils
-RUN wget https://nodejs.org/dist/v10.16.0/node-v10.16.0-linux-x64.tar.xz \
-    && tar -xvf node-v10.16.0-linux-x64.tar.xz \
-    && rm node-v10.16.0-linux-x64.tar.xz \
-    && ln -s /node-v10.16.0-linux-x64/bin/node /usr/local/bin/node \
-    && ln -s /node-v10.16.0-linux-x64/bin/npm /usr/local/bin/npm
+#
+# 构建前端
+#
 
-# 后端依赖
-WORKDIR /blivechat
-COPY requirements.txt ./
-RUN pip3 install --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
+FROM node:18.17.0-bullseye AS builder
+ARG BASE_PATH='/root/blivechat'
+WORKDIR "${BASE_PATH}/frontend"
 
 # 前端依赖
-WORKDIR ./frontend
 COPY frontend/package.json ./
-RUN npm i --registry=https://registry.npm.taobao.org
+RUN npm i
 
 # 编译前端
-COPY . ../
+COPY frontend ./
 RUN npm run build
 
+#
+# 准备后端
+#
+
+FROM python:3.8.12-bullseye
+ARG BASE_PATH='/root/blivechat'
+ARG EXT_DATA_PATH='/mnt/data'
+WORKDIR "${BASE_PATH}"
+
+# 后端依赖
+COPY blivedm/requirements.txt blivedm/
+COPY requirements.txt ./
+RUN pip3 install --no-cache-dir -i https://mirrors.aliyun.com/pypi/simple -r requirements.txt
+
+# 数据目录
+COPY . ./
+RUN mkdir -p "${EXT_DATA_PATH}/frontend/dist" \
+    && mv data "${EXT_DATA_PATH}/data" \
+    && ln -s "${EXT_DATA_PATH}/data" data \
+    && mv log "${EXT_DATA_PATH}/log" \
+    && ln -s "${EXT_DATA_PATH}/log" log \
+    && ln -s "${EXT_DATA_PATH}/frontend/dist" frontend/dist
+
+# 编译好的前端
+COPY --from=builder "${BASE_PATH}/frontend/dist" "${EXT_DATA_PATH}/frontend/dist/"
+
 # 运行
-WORKDIR ..
-VOLUME /blivechat/data /blivechat/log /blivechat/frontend/dist
+VOLUME "${EXT_DATA_PATH}"
 EXPOSE 12450
 ENTRYPOINT ["python3", "main.py"]
 CMD ["--host", "0.0.0.0", "--port", "12450"]

README.md

@@ -1,89 +1,116 @@
 # blivechat
+
 用于OBS的仿YouTube风格的bilibili直播评论栏
 
-最近喜欢看VTuber，想为此写些程序，于是有了这个东西。~~写到一半发现有类似项目了：[bilibili-live-chat](https://github.com/Tsuk1ko/bilibili-live-chat)、[BiliChat](https://github.com/3Shain/BiliChat)~~
+![OBS截图](./screenshots/obs.png)
 
-![OBS截图](https://github.com/xfgryujk/blivechat/blob/master/screenshots/obs.png)  
-![Chrome截图](https://github.com/xfgryujk/blivechat/blob/master/screenshots/chrome.png)  
-![样式生成器截图](https://github.com/xfgryujk/blivechat/blob/master/screenshots/stylegen.png)  
+![Chrome截图](./screenshots/chrome.png)
+
+![样式生成器截图](./screenshots/stylegen.png)
 
 ## 特性
+
 * 兼容YouTube直播评论栏的样式
-* 金瓜子礼物模仿醒目留言显示
+* 付费礼物模仿醒目留言显示
 * 高亮舰队、房管、主播的用户名
 * 支持屏蔽弹幕、合并相似弹幕等设置
-* 自带样式生成器
-* 支持自动翻译弹幕、醒目留言到日语
-* 支持标注打赏用户名的读音（拼音和日文假名）
+* 自带两种样式生成器，经典YouTube风格和仿微信风格
+* 支持前端直连B站服务器或者通过后端转发
+* 支持自动翻译弹幕、醒目留言到日语，可以在后台配置翻译目标语言
+* 支持标注打赏用户名的读音，可选拼音或日文假名
+* 支持配置自定义表情，不需要开通B站官方表情
 
 ## 使用方法
+
+以下几种方式任选一种即可
+
 ### 一、本地使用
+
 1. 下载[发布版](https://github.com/xfgryujk/blivechat/releases)（仅提供x64 Windows版）
 2. 双击`blivechat.exe`运行服务器，或者用命令行可以指定host和端口号：
-   ```bat
-   blivechat.exe --host 127.0.0.1 --port 12450
-   ```
-3. 用浏览器打开[http://localhost:12450](http://localhost:12450)，输入房间ID，复制房间URL
+
+    ```sh
+    blivechat.exe --host 127.0.0.1 --port 12450
+    ```
+
+   或者也可以在配置文件里指定host和端口号
+
+3. 用浏览器打开[http://localhost:12450](http://localhost:12450)，输入主播在开始直播时获得的身份码，复制房间URL
 4. 用样式生成器生成样式，复制CSS
 5. 在OBS中添加浏览器源，输入URL和自定义CSS
 
 **注意事项：**
 
-* 本地使用时不要关闭blivechat.exe那个黑框，否则不能继续获取头像或弹幕
+* 本地使用时不要关闭blivechat.exe那个黑框，否则不能继续获取弹幕
 * 样式生成器没有列出所有本地字体，但是可以手动输入本地字体
+* 如果需要使用翻译功能，建议看[配置官方翻译接口傻瓜式教程](https://www.bilibili.com/read/cv14663633)
 
 ### 二、公共服务器
+
 请优先在本地使用，使用公共服务器会有更大的延迟，而且服务器故障时可能发生直播事故
 
 * [公共服务器](http://chat.bilisc.com/)
-* [仅样式生成器](https://style.vtbs.moe/)
 
 ### 三、源代码版（自建服务器或在Windows以外平台）
+
 0. 由于使用了git子模块，clone时需要加上`--recursive`参数：
-   ```sh
-   git clone --recursive https://github.com/xfgryujk/blivechat.git
-   ```
-   如果已经clone，拉子模块的方法：
-   ```sh
-   git submodule update --init --recursive
-   ```
+
+    ```sh
+    git clone --recursive https://github.com/xfgryujk/blivechat.git
+    ```
+
+    如果已经clone，拉子模块的方法：
+
+    ```sh
+    git submodule update --init --recursive
+    ```
+
 1. 编译前端（需要安装Node.js）：
-   ```sh
-   cd frontend
-   npm i
-   npm run build
-   ```
-2. 运行服务器（需要Python3.6以上版本）：
-   ```sh
-   pip3 install -r requirements.txt
-   python3 main.py
-   ```
-   或者可以指定host和端口号：
-   ```sh
-   python3 main.py --host 127.0.0.1 --port 12450
-   ```
+
+    ```sh
+    cd frontend
+    npm i
+    npm run build
+    ```
+
+2. 运行服务器（需要Python3.8以上版本）：
+
+    ```sh
+    pip3 install -r requirements.txt
+    python3 main.py
+    ```
+
+    或者可以指定host和端口号：
+
+    ```sh
+    python3 main.py --host 127.0.0.1 --port 12450
+    ```
+
 3. 用浏览器打开[http://localhost:12450](http://localhost:12450)，以下略
 
 ### 四、Docker（自建服务器）
-1. ```sh
-   docker run --name blivechat -d -p 12450:12450 \
-     --mount source=blc-data,target=/blivechat/data \
-     --mount source=blc-log,target=/blivechat/log \
-     --mount source=blc-frontend,target=/blivechat/frontend/dist \
-     xfgryujk/blivechat:latest
-   ```
+
+1.  ```sh
+    docker run --name blivechat -d -p 12450:12450 \
+      --mount source=blivechat-data,target=/mnt/data \
+      xfgryujk/blivechat:latest
+    ```
+
 2. 用浏览器打开[http://localhost:12450](http://localhost:12450)，以下略
 
 ## 自建服务器相关补充
+
 ### 服务器配置
+
 服务器配置在`data/config.ini`，可以配置数据库和允许自动翻译等，编辑后要重启生效
 
 **自建服务器时强烈建议不使用加载器**，否则可能因为混合HTTP和HTTPS等原因加载不出来
 
 ### 参考nginx配置
+
 `sudo vim /etc/nginx/sites-enabled/blivechat.conf`
 
-```conf
+```nginx
 upstream blivechat {
 	keepalive 8;
 	# blivechat地址
@@ -108,6 +135,11 @@ server {
 	ssl_certificate /PATH/TO/CERT.crt;
 	ssl_certificate_key /PATH/TO/CERT_KEY.key;
 
+	set $blivechat_path /PATH/TO/BLIVECHAT;
+
+	client_body_buffer_size 256k;
+	client_max_body_size 1.1m;
+
 	# 代理header
 	proxy_http_version 1.1;
 	proxy_set_header Host $host;
@@ -117,9 +149,20 @@ server {
 
 	# 静态文件
 	location / {
-		root /PATH/TO/BLIVECHAT/frontend/dist;
-		# 如果文件不存在，交给前端路由
-		try_files $uri $uri/ /index.html;
+		root $blivechat_path/frontend/dist;
+		try_files $uri $uri/ @index;
+	}
+	# 不存在的文件请求转发到index.html，交给前端路由
+	location @index {
+		rewrite ^ /index.html last;
+	}
+	location = /index.html {
+		root $blivechat_path/frontend/dist;
+		# index.html不缓存，防止更新后前端还是旧版
+		add_header Cache-Control no-cache;
+	}
+	location /emoticons {
+		alias $blivechat_path/data/emoticons;
 	}
 	# 动态API
 	location /api {

api/base.py

@@ -1,31 +1,21 @@
 # -*- coding: utf-8 -*-
-
 import json
+from typing import *
 
 import tornado.web
 
 
-# noinspection PyAbstractClass
 class ApiHandler(tornado.web.RequestHandler):
-    def set_default_headers(self):
-        # 跨域测试用
-        if not self.application.settings['debug']:
-            return
-        self.set_header('Access-Control-Allow-Origin', '*')
-        self.set_header('Access-Control-Allow-Methods', 'OPTIONS, PUT, POST, GET, DELETE')
-        if 'Access-Control-Request-Headers' in self.request.headers:
-            self.set_header('Access-Control-Allow-Headers',
-                            self.request.headers['Access-Control-Request-Headers'])
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.json_args: Optional[dict] = None
 
     def prepare(self):
-        if self.request.headers.get('Content-Type', '').startswith('application/json'):
-            try:
-                self.json_args = json.loads(self.request.body)
-            except json.JSONDecodeError:
-                self.json_args = None
-        else:
-            self.json_args = None
+        self.set_header('Cache-Control', 'no-cache')
 
-    async def options(self, *_args, **_kwargs):
-        # 跨域测试用
-        self.set_status(204 if self.application.settings['debug'] else 405)
+        if not self.request.headers.get('Content-Type', '').startswith('application/json'):
+            return
+        try:
+            self.json_args = json.loads(self.request.body)
+        except json.JSONDecodeError:
+            pass

api/chat.py

@@ -1,5 +1,4 @@
 # -*- coding: utf-8 -*-
-
 import asyncio
 import enum
 import json
@@ -13,10 +12,12 @@ import aiohttp
 import tornado.websocket
 
 import api.base
-import blivedm.blivedm as blivedm
+import blivedm.blivedm.clients.web as dm_web_cli
 import config
-import models.avatar
-import models.translate
+import services.avatar
+import services.chat
+import services.translate
+import utils.request
 
 logger = logging.getLogger(__name__)
 
@@ -32,242 +33,43 @@ class Command(enum.IntEnum):
     UPDATE_TRANSLATION = 7
 
 
-_http_session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=10))
-
-room_manager: Optional['RoomManager'] = None
+class ContentType(enum.IntEnum):
+    TEXT = 0
+    EMOTICON = 1
 
 
-def init():
-    global room_manager
-    room_manager = RoomManager()
+def make_message_body(cmd, data):
+    return json.dumps(
+        {
+            'cmd': cmd,
+            'data': data
+        }
+    ).encode('utf-8')
 
 
-class Room(blivedm.BLiveClient):
-    HEARTBEAT_INTERVAL = 10
-
-    # 重新定义parse_XXX是为了减少对字段名的依赖，防止B站改字段名
-    def __parse_danmaku(self, command):
-        info = command['info']
-        if info[3]:
-            room_id = info[3][3]
-            medal_level = info[3][0]
-        else:
-            room_id = medal_level = 0
-        return self._on_receive_danmaku(blivedm.DanmakuMessage(
-            None, None, None, info[0][4], None, None, info[0][9], None,
-            info[1],
-            info[2][0], info[2][1], info[2][2], None, None, info[2][5], info[2][6], None,
-            medal_level, None, None, room_id, None, None,
-            info[4][0], None, None,
-            None, None,
-            info[7]
-        ))
-
-    def __parse_gift(self, command):
-        data = command['data']
-        return self._on_receive_gift(blivedm.GiftMessage(
-            data['giftName'], data['num'], data['uname'], data['face'], None,
-            data['uid'], data['timestamp'], None, None,
-            None, None, None, data['coin_type'], data['total_coin']
-        ))
-
-    def __parse_buy_guard(self, command):
-        data = command['data']
-        return self._on_buy_guard(blivedm.GuardBuyMessage(
-            data['uid'], data['username'], data['guard_level'], None, None,
-            None, None, data['start_time'], None
-        ))
-
-    def __parse_super_chat(self, command):
-        data = command['data']
-        return self._on_super_chat(blivedm.SuperChatMessage(
-            data['price'], data['message'], None, data['start_time'],
-            None, None, data['id'], None,
-            None, data['uid'], data['user_info']['uname'],
-            data['user_info']['face'], None,
-            None, None,
-            None, None, None,
-            None
-        ))
-
-    _COMMAND_HANDLERS = {
-        **blivedm.BLiveClient._COMMAND_HANDLERS,
-        'DANMU_MSG': __parse_danmaku,
-        'SEND_GIFT': __parse_gift,
-        'GUARD_BUY': __parse_buy_guard,
-        'SUPER_CHAT_MESSAGE': __parse_super_chat
-    }
-
-    def __init__(self, room_id):
-        super().__init__(room_id, session=_http_session, heartbeat_interval=self.HEARTBEAT_INTERVAL)
-        self.clients: List['ChatHandler'] = []
-        self.auto_translate_count = 0
-
-    async def init_room(self):
-        await super().init_room()
-        return True
-
-    def stop_and_close(self):
-        if self.is_running:
-            future = self.stop()
-            future.add_done_callback(lambda _future: asyncio.ensure_future(self.close()))
-        else:
-            asyncio.ensure_future(self.close())
-
-    def send_message(self, cmd, data):
-        body = json.dumps({'cmd': cmd, 'data': data})
-        for client in self.clients:
-            try:
-                client.write_message(body)
-            except tornado.websocket.WebSocketClosedError:
-                room_manager.del_client(self.room_id, client)
-
-    def send_message_if(self, can_send_func: Callable[['ChatHandler'], bool], cmd, data):
-        body = json.dumps({'cmd': cmd, 'data': data})
-        for client in filter(can_send_func, self.clients):
-            try:
-                client.write_message(body)
-            except tornado.websocket.WebSocketClosedError:
-                room_manager.del_client(self.room_id, client)
-
-    async def _on_receive_danmaku(self, danmaku: blivedm.DanmakuMessage):
-        asyncio.ensure_future(self.__on_receive_danmaku(danmaku))
-
-    async def __on_receive_danmaku(self, danmaku: blivedm.DanmakuMessage):
-        if danmaku.uid == self.room_owner_uid:
-            author_type = 3  # 主播
-        elif danmaku.admin:
-            author_type = 2  # 房管
-        elif danmaku.privilege_type != 0:  # 1总督，2提督，3舰长
-            author_type = 1  # 舰队
-        else:
-            author_type = 0
-
-        need_translate = self._need_translate(danmaku.msg)
-        if need_translate:
-            translation = models.translate.get_translation_from_cache(danmaku.msg)
-            if translation is None:
-                # 没有缓存，需要后面异步翻译后通知
-                translation = ''
-            else:
-                need_translate = False
-        else:
-            translation = ''
-
-        id_ = uuid.uuid4().hex
-        # 为了节省带宽用list而不是dict
-        self.send_message(Command.ADD_TEXT, make_text_message(
-            await models.avatar.get_avatar_url(danmaku.uid),
-            int(danmaku.timestamp / 1000),
-            danmaku.uname,
-            author_type,
-            danmaku.msg,
-            danmaku.privilege_type,
-            danmaku.msg_type,
-            danmaku.user_level,
-            danmaku.urank < 10000,
-            danmaku.mobile_verify,
-            0 if danmaku.room_id != self.room_id else danmaku.medal_level,
-            id_,
-            translation
-        ))
-
-        if need_translate:
-            await self._translate_and_response(danmaku.msg, id_)
-
-    async def _on_receive_gift(self, gift: blivedm.GiftMessage):
-        avatar_url = models.avatar.process_avatar_url(gift.face)
-        models.avatar.update_avatar_cache(gift.uid, avatar_url)
-        if gift.coin_type != 'gold':  # 丢人
-            return
-        id_ = uuid.uuid4().hex
-        self.send_message(Command.ADD_GIFT, {
-            'id': id_,
-            'avatarUrl': avatar_url,
-            'timestamp': gift.timestamp,
-            'authorName': gift.uname,
-            'totalCoin': gift.total_coin,
-            'giftName': gift.gift_name,
-            'num': gift.num
-        })
-
-    async def _on_buy_guard(self, message: blivedm.GuardBuyMessage):
-        asyncio.ensure_future(self.__on_buy_guard(message))
-
-    async def __on_buy_guard(self, message: blivedm.GuardBuyMessage):
-        id_ = uuid.uuid4().hex
-        self.send_message(Command.ADD_MEMBER, {
-            'id': id_,
-            'avatarUrl': await models.avatar.get_avatar_url(message.uid),
-            'timestamp': message.start_time,
-            'authorName': message.username,
-            'privilegeType': message.guard_level
-        })
-
-    async def _on_super_chat(self, message: blivedm.SuperChatMessage):
-        avatar_url = models.avatar.process_avatar_url(message.face)
-        models.avatar.update_avatar_cache(message.uid, avatar_url)
-
-        need_translate = self._need_translate(message.message)
-        if need_translate:
-            translation = models.translate.get_translation_from_cache(message.message)
-            if translation is None:
-                # 没有缓存，需要后面异步翻译后通知
-                translation = ''
-            else:
-                need_translate = False
-        else:
-            translation = ''
-
-        id_ = str(message.id)
-        self.send_message(Command.ADD_SUPER_CHAT, {
-            'id': id_,
-            'avatarUrl': avatar_url,
-            'timestamp': message.start_time,
-            'authorName': message.uname,
-            'price': message.price,
-            'content': message.message,
-            'translation': translation
-        })
-
-        if need_translate:
-            asyncio.ensure_future(self._translate_and_response(message.message, id_))
-
-    async def _on_super_chat_delete(self, message: blivedm.SuperChatDeleteMessage):
-        self.send_message(Command.ADD_SUPER_CHAT, {
-            'ids': list(map(str, message.ids))
-        })
-
-    def _need_translate(self, text):
-        cfg = config.get_config()
-        return (
-            cfg.enable_translate
-            and (not cfg.allow_translate_rooms or self.room_id in cfg.allow_translate_rooms)
-            and self.auto_translate_count > 0
-            and models.translate.need_translate(text)
-        )
-
-    async def _translate_and_response(self, text, msg_id):
-        translation = await models.translate.translate(text)
-        if translation is None:
-            return
-        self.send_message_if(
-            lambda client: client.auto_translate,
-            Command.UPDATE_TRANSLATION, make_translation_message(
-                msg_id,
-                translation
-            )
-        )
-
-
-def make_text_message(avatar_url, timestamp, author_name, author_type, content, privilege_type,
-                      is_gift_danmaku, author_level, is_newbie, is_mobile_verified, medal_level,
-                      id_, translation):
+def make_text_message_data(
+    avatar_url: str = services.avatar.DEFAULT_AVATAR_URL,
+    timestamp: int = None,
+    author_name: str = '',
+    author_type: int = 0,
+    content: str = '',
+    privilege_type: int = 0,
+    is_gift_danmaku: bool = False,
+    author_level: int = 1,
+    is_newbie: bool = False,
+    is_mobile_verified: bool = True,
+    medal_level: int = 0,
+    id_: str = None,
+    translation: str = '',
+    content_type: int = ContentType.TEXT,
+    content_type_params: list = None,
+):
+    # 为了节省带宽用list而不是dict
     return [
         # 0: avatarUrl
         avatar_url,
         # 1: timestamp
-        timestamp,
+        timestamp if timestamp is not None else int(time.time()),
         # 2: authorName
         author_name,
         # 3: authorType
@@ -287,85 +89,34 @@ def make_text_message(avatar_url, timestamp, author_name, author_type, content,
         # 10: medalLevel
         medal_level,
         # 11: id
-        id_,
+        id_ if id_ is not None else uuid.uuid4().hex,
         # 12: translation
-        translation
+        translation,
+        # 13: contentType
+        content_type,
+        # 14: contentTypeParams
+        content_type_params if content_type_params is not None else [],
+        # 15: textEmoticons
+        [],  # 已废弃，保留
     ]
 
 
-def make_translation_message(msg_id, translation):
+def make_emoticon_params(url):
+    return [
+        # 0: url
+        url,
+    ]
+
+
+def make_translation_message_data(msg_id, translation):
     return [
         # 0: id
         msg_id,
         # 1: translation
-        translation
+        translation,
     ]
 
 
-class RoomManager:
-    def __init__(self):
-        self._rooms: Dict[int, Room] = {}
-
-    async def add_client(self, room_id, client: 'ChatHandler'):
-        if room_id not in self._rooms:
-            if not await self._add_room(room_id):
-                client.close()
-                return
-        room = self._rooms.get(room_id, None)
-        if room is None:
-            return
-
-        room.clients.append(client)
-        logger.info('%d clients in room %s', len(room.clients), room_id)
-        if client.auto_translate:
-            room.auto_translate_count += 1
-
-        await client.on_join_room()
-
-    def del_client(self, room_id, client: 'ChatHandler'):
-        room = self._rooms.get(room_id, None)
-        if room is None:
-            return
-
-        try:
-            room.clients.remove(client)
-        except ValueError:
-            # _add_room未完成，没有执行到room.clients.append
-            pass
-        else:
-            logger.info('%d clients in room %s', len(room.clients), room_id)
-            if client.auto_translate:
-                room.auto_translate_count = max(0, room.auto_translate_count - 1)
-
-        if not room.clients:
-            self._del_room(room_id)
-
-    async def _add_room(self, room_id):
-        if room_id in self._rooms:
-            return True
-        logger.info('Creating room %d', room_id)
-        self._rooms[room_id] = room = Room(room_id)
-        if await room.init_room():
-            room.start()
-            logger.info('%d rooms', len(self._rooms))
-            return True
-        else:
-            self._del_room(room_id)
-            return False
-
-    def _del_room(self, room_id):
-        room = self._rooms.get(room_id, None)
-        if room is None:
-            return
-        logger.info('Removing room %d', room_id)
-        for client in room.clients:
-            client.close()
-        room.stop_and_close()
-        self._rooms.pop(room_id, None)
-        logger.info('%d rooms', len(self._rooms))
-
-
-# noinspection PyAbstractClass
 class ChatHandler(tornado.websocket.WebSocketHandler):
     HEARTBEAT_INTERVAL = 10
     RECEIVE_TIMEOUT = HEARTBEAT_INTERVAL + 5
@@ -375,38 +126,38 @@ class ChatHandler(tornado.websocket.WebSocketHandler):
         self._heartbeat_timer_handle = None
         self._receive_timeout_timer_handle = None
 
-        self.room_id = None
+        self.room_key: Optional[services.chat.RoomKey] = None
         self.auto_translate = False
 
     def open(self):
-        logger.info('Websocket connected %s', self.request.remote_ip)
-        self._heartbeat_timer_handle = asyncio.get_event_loop().call_later(
+        logger.info('client=%s connected', self.request.remote_ip)
+        self._heartbeat_timer_handle = asyncio.get_running_loop().call_later(
             self.HEARTBEAT_INTERVAL, self._on_send_heartbeat
         )
         self._refresh_receive_timeout_timer()
 
     def _on_send_heartbeat(self):
-        self.send_message(Command.HEARTBEAT, {})
-        self._heartbeat_timer_handle = asyncio.get_event_loop().call_later(
+        self.send_cmd_data(Command.HEARTBEAT, {})
+        self._heartbeat_timer_handle = asyncio.get_running_loop().call_later(
             self.HEARTBEAT_INTERVAL, self._on_send_heartbeat
         )
 
     def _refresh_receive_timeout_timer(self):
         if self._receive_timeout_timer_handle is not None:
             self._receive_timeout_timer_handle.cancel()
-        self._receive_timeout_timer_handle = asyncio.get_event_loop().call_later(
+        self._receive_timeout_timer_handle = asyncio.get_running_loop().call_later(
             self.RECEIVE_TIMEOUT, self._on_receive_timeout
         )
 
     def _on_receive_timeout(self):
-        logger.warning('Client %s timed out', self.request.remote_ip)
+        logger.info('client=%s timed out', self.request.remote_ip)
         self._receive_timeout_timer_handle = None
         self.close()
 
     def on_close(self):
-        logger.info('Websocket disconnected %s room: %s', self.request.remote_ip, str(self.room_id))
+        logger.info('client=%s disconnected, room=%s', self.request.remote_ip, self.room_key)
         if self.has_joined_room:
-            room_manager.del_client(self.room_id, self)
+            services.chat.client_room_manager.del_client(self.room_key, self)
         if self._heartbeat_timer_handle is not None:
             self._heartbeat_timer_handle.cancel()
             self._heartbeat_timer_handle = None
@@ -416,32 +167,57 @@ class ChatHandler(tornado.websocket.WebSocketHandler):
 
     def on_message(self, message):
         try:
-            # 超时没有加入房间也断开
-            if self.has_joined_room:
-                self._refresh_receive_timeout_timer()
-
             body = json.loads(message)
-            cmd = body['cmd']
+            cmd = int(body['cmd'])
+
             if cmd == Command.HEARTBEAT:
-                pass
-            elif cmd == Command.JOIN_ROOM:
+                # 超时没有加入房间也断开
                 if self.has_joined_room:
-                    return
-                self._refresh_receive_timeout_timer()
+                    self._refresh_receive_timeout_timer()
 
-                self.room_id = int(body['data']['roomId'])
-                logger.info('Client %s is joining room %d', self.request.remote_ip, self.room_id)
-                try:
-                    cfg = body['data']['config']
-                    self.auto_translate = cfg['autoTranslate']
-                except KeyError:
-                    pass
+            elif cmd == Command.JOIN_ROOM:
+                self._on_join_room_req(body)
 
-                asyncio.ensure_future(room_manager.add_client(self.room_id, self))
             else:
-                logger.warning('Unknown cmd, client: %s, cmd: %d, body: %s', self.request.remote_ip, cmd, body)
-        except Exception:
-            logger.exception('on_message error, client: %s, message: %s', self.request.remote_ip, message)
+                logger.warning('client=%s unknown cmd=%d, body=%s', self.request.remote_ip, cmd, body)
+
+        except Exception:  # noqa
+            logger.exception('client=%s on_message error, message=%s', self.request.remote_ip, message)
+
+    def _on_join_room_req(self, body: dict):
+        if self.has_joined_room:
+            return
+        data = body['data']
+
+        room_key_dict = data.get('roomKey', None)
+        if room_key_dict is not None:
+            room_key_type = services.chat.RoomKeyType(room_key_dict['type'])
+            room_key_value = room_key_dict['value']
+            if room_key_type == services.chat.RoomKeyType.ROOM_ID:
+                if not isinstance(room_key_value, int):
+                    raise TypeError(f'Room key value type error, value={room_key_value}')
+            elif room_key_type == services.chat.RoomKeyType.AUTH_CODE:
+                if not isinstance(room_key_value, str):
+                    raise TypeError(f'Room key value type error, value={room_key_value}')
+            else:
+                raise ValueError(f'Unknown RoomKeyType={room_key_type}')
+        else:
+            # 兼容旧版客户端 TODO 过几个版本可以移除
+            room_key_type = services.chat.RoomKeyType.ROOM_ID
+            room_key_value = int(data['roomId'])
+        self.room_key = services.chat.RoomKey(room_key_type, room_key_value)
+        logger.info('client=%s joining room %s', self.request.remote_ip, self.room_key)
+
+        try:
+            cfg = data['config']
+            self.auto_translate = bool(cfg['autoTranslate'])
+        except KeyError:
+            pass
+
+        services.chat.client_room_manager.add_client(self.room_key, self)
+        asyncio.create_task(self._on_joined_room())
+
+        self._refresh_receive_timeout_timer()
 
     # 跨域测试用
     def check_origin(self, origin):
@@ -451,60 +227,50 @@ class ChatHandler(tornado.websocket.WebSocketHandler):
 
     @property
     def has_joined_room(self):
-        return self.room_id is not None
+        return self.room_key is not None
 
-    def send_message(self, cmd, data):
-        body = json.dumps({'cmd': cmd, 'data': data})
+    def send_cmd_data(self, cmd, data):
+        self.send_body_no_raise(make_message_body(cmd, data))
+
+    def send_body_no_raise(self, body: Union[bytes, str, Dict[str, Any]]):
         try:
             self.write_message(body)
         except tornado.websocket.WebSocketClosedError:
             self.close()
 
-    async def on_join_room(self):
-        if self.application.settings['debug']:
-            await self.send_test_message()
+    async def _on_joined_room(self):
+        if self.settings['debug']:
+            await self._send_test_message()
 
         # 不允许自动翻译的提示
         if self.auto_translate:
             cfg = config.get_config()
-            if cfg.allow_translate_rooms and self.room_id not in cfg.allow_translate_rooms:
-                self.send_message(Command.ADD_TEXT, make_text_message(
-                    models.avatar.DEFAULT_AVATAR_URL,
-                    int(time.time()),
-                    'blivechat',
-                    2,
-                    'Translation is not allowed in this room. Please download to use translation',
-                    0,
-                    False,
-                    60,
-                    False,
-                    True,
-                    0,
-                    uuid.uuid4().hex,
-                    ''
+            if (
+                cfg.allow_translate_rooms
+                # 身份码就不管了吧，反正配置正确的情况下不会看到这个提示
+                and self.room_key.type == services.chat.RoomKeyType.ROOM_ID
+                and self.room_key.value not in cfg.allow_translate_rooms
+            ):
+                self.send_cmd_data(Command.ADD_TEXT, make_text_message_data(
+                    author_name='blivechat',
+                    author_type=2,
+                    content='Translation is not allowed in this room. Please download to use translation',
+                    author_level=60,
                 ))
 
     # 测试用
-    async def send_test_message(self):
+    async def _send_test_message(self):
         base_data = {
-            'avatarUrl': await models.avatar.get_avatar_url(300474),
+            'avatarUrl': await services.avatar.get_avatar_url(300474),
             'timestamp': int(time.time()),
             'authorName': 'xfgryujk',
         }
-        text_data = make_text_message(
-            base_data['avatarUrl'],
-            base_data['timestamp'],
-            base_data['authorName'],
-            0,
-            '我能吞下玻璃而不伤身体',
-            0,
-            False,
-            20,
-            False,
-            True,
-            0,
-            uuid.uuid4().hex,
-            ''
+        text_data = make_text_message_data(
+            avatar_url=base_data['avatarUrl'],
+            timestamp=base_data['timestamp'],
+            author_name=base_data['authorName'],
+            content='我能吞下玻璃而不伤身体',
+            author_level=60,
         )
         member_data = {
             **base_data,
@@ -525,35 +291,36 @@ class ChatHandler(tornado.websocket.WebSocketHandler):
             'content': 'The quick brown fox jumps over the lazy dog',
             'translation': ''
         }
-        self.send_message(Command.ADD_TEXT, text_data)
+        self.send_cmd_data(Command.ADD_TEXT, text_data)
+        text_data[4] = 'te[dog]st'
+        text_data[11] = uuid.uuid4().hex
+        self.send_cmd_data(Command.ADD_TEXT, text_data)
         text_data[2] = '主播'
         text_data[3] = 3
         text_data[4] = "I can eat glass, it doesn't hurt me."
         text_data[11] = uuid.uuid4().hex
-        self.send_message(Command.ADD_TEXT, text_data)
-        self.send_message(Command.ADD_MEMBER, member_data)
-        self.send_message(Command.ADD_SUPER_CHAT, sc_data)
+        self.send_cmd_data(Command.ADD_TEXT, text_data)
+        self.send_cmd_data(Command.ADD_MEMBER, member_data)
+        self.send_cmd_data(Command.ADD_SUPER_CHAT, sc_data)
         sc_data['id'] = str(random.randint(1, 65535))
         sc_data['price'] = 100
         sc_data['content'] = '敏捷的棕色狐狸跳过了懒狗'
-        self.send_message(Command.ADD_SUPER_CHAT, sc_data)
-        # self.send_message(Command.DEL_SUPER_CHAT, {'ids': [sc_data['id']]})
-        self.send_message(Command.ADD_GIFT, gift_data)
+        self.send_cmd_data(Command.ADD_SUPER_CHAT, sc_data)
+        # self.send_cmd_data(Command.DEL_SUPER_CHAT, {'ids': [sc_data['id']]})
+        self.send_cmd_data(Command.ADD_GIFT, gift_data)
         gift_data['id'] = uuid.uuid4().hex
         gift_data['totalCoin'] = 1245000
         gift_data['giftName'] = '小电视飞船'
-        self.send_message(Command.ADD_GIFT, gift_data)
+        self.send_cmd_data(Command.ADD_GIFT, gift_data)
 
 
-# noinspection PyAbstractClass
 class RoomInfoHandler(api.base.ApiHandler):
-    _host_server_list_cache = blivedm.DEFAULT_DANMAKU_SERVER_LIST
-
     async def get(self):
         room_id = int(self.get_query_argument('roomId'))
-        logger.info('Client %s is getting room info %d', self.request.remote_ip, room_id)
+        logger.info('client=%s getting room info, room=%d', self.request.remote_ip, room_id)
         room_id, owner_uid = await self._get_room_info(room_id)
-        host_server_list = await self._get_server_host_list(room_id)
+        # 连接其他host必须要key
+        host_server_list = dm_web_cli.DEFAULT_DANMAKU_SERVER_LIST
         if owner_uid == 0:
             # 缓存3分钟
             self.set_header('Cache-Control', 'private, max-age=180')
@@ -569,66 +336,59 @@ class RoomInfoHandler(api.base.ApiHandler):
     @staticmethod
     async def _get_room_info(room_id):
         try:
-            async with _http_session.get(blivedm.ROOM_INIT_URL, params={'room_id': room_id}
-                                         ) as res:
+            async with utils.request.http_session.get(
+                dm_web_cli.ROOM_INIT_URL,
+                headers={
+                    **utils.request.BILIBILI_COMMON_HEADERS,
+                    'Origin': 'https://live.bilibili.com',
+                    'Referer': f'https://live.bilibili.com/{room_id}'
+                },
+                params={
+                    'room_id': room_id
+                }
+            ) as res:
                 if res.status != 200:
-                    logger.warning('room %d _get_room_info failed: %d %s', room_id,
+                    logger.warning('room=%d _get_room_info failed: %d %s', room_id,
                                    res.status, res.reason)
                     return room_id, 0
                 data = await res.json()
         except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
-            logger.exception('room %d _get_room_info failed', room_id)
+            logger.exception('room=%d _get_room_info failed', room_id)
             return room_id, 0
 
         if data['code'] != 0:
-            logger.warning('room %d _get_room_info failed: %s', room_id, data['message'])
+            logger.warning('room=%d _get_room_info failed: %s', room_id, data['message'])
             return room_id, 0
 
         room_info = data['data']['room_info']
         return room_info['room_id'], room_info['uid']
 
-    @classmethod
-    async def _get_server_host_list(cls, _room_id):
-        return cls._host_server_list_cache
 
-        # 连接其他host必须要key
-        # try:
-        #     async with _http_session.get(blivedm.DANMAKU_SERVER_CONF_URL, params={'id': room_id, 'type': 0}
-        #                                  ) as res:
-        #         if res.status != 200:
-        #             logger.warning('room %d _get_server_host_list failed: %d %s', room_id,
-        #                            res.status, res.reason)
-        #             return cls._host_server_list_cache
-        #         data = await res.json()
-        # except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
-        #     logger.exception('room %d _get_server_host_list failed', room_id)
-        #     return cls._host_server_list_cache
-        #
-        # if data['code'] != 0:
-        #     logger.warning('room %d _get_server_host_list failed: %s', room_id, data['message'])
-        #     return cls._host_server_list_cache
-        #
-        # host_server_list = data['data']['host_list']
-        # if not host_server_list:
-        #     logger.warning('room %d _get_server_host_list failed: host_server_list is empty')
-        #     return cls._host_server_list_cache
-        #
-        # cls._host_server_list_cache = host_server_list
-        # return host_server_list
-
-
-# noinspection PyAbstractClass
 class AvatarHandler(api.base.ApiHandler):
     async def get(self):
         uid = int(self.get_query_argument('uid'))
-        avatar_url = await models.avatar.get_avatar_url_or_none(uid)
+        avatar_url = await services.avatar.get_avatar_url_or_none(uid)
         if avatar_url is None:
-            avatar_url = models.avatar.DEFAULT_AVATAR_URL
+            avatar_url = services.avatar.DEFAULT_AVATAR_URL
             # 缓存3分钟
             self.set_header('Cache-Control', 'private, max-age=180')
         else:
             # 缓存1天
             self.set_header('Cache-Control', 'private, max-age=86400')
-        self.write({
-            'avatarUrl': avatar_url
-        })
+        self.write({'avatarUrl': avatar_url})
+
+
+class TextEmoticonMappingsHandler(api.base.ApiHandler):
+    async def get(self):
+        # 缓存1天
+        self.set_header('Cache-Control', 'private, max-age=86400')
+        cfg = config.get_config()
+        self.write({'textEmoticons': cfg.text_emoticons})
+
+
+ROUTES = [
+    (r'/api/chat', ChatHandler),
+    (r'/api/room_info', RoomInfoHandler),
+    (r'/api/avatar_url', AvatarHandler),
+    (r'/api/text_emoticon_mappings', TextEmoticonMappingsHandler),
+]

api/main.py

@@ -1,4 +1,8 @@
 # -*- coding: utf-8 -*-
+import asyncio
+import hashlib
+import logging
+import os
 
 import tornado.web
 
@@ -6,21 +10,33 @@ import api.base
 import config
 import update
 
+logger = logging.getLogger(__name__)
+
+EMOTICON_UPLOAD_PATH = os.path.join(config.DATA_PATH, 'emoticons')
+EMOTICON_BASE_URL = '/emoticons'
+
 
-# noinspection PyAbstractClass
 class MainHandler(tornado.web.StaticFileHandler):
     """为了使用Vue Router的history模式，把不存在的文件请求转发到index.html"""
     async def get(self, path, include_body=True):
+        if path == '':
+            await self._get_index(include_body)
+            return
+
         try:
             await super().get(path, include_body)
         except tornado.web.HTTPError as e:
             if e.status_code != 404:
                 raise
             # 不存在的文件请求转发到index.html，交给前端路由
-            await super().get('index.html', include_body)
+            await self._get_index(include_body)
+
+    async def _get_index(self, include_body=True):
+        # index.html不缓存，防止更新后前端还是旧版
+        self.set_header('Cache-Control', 'no-cache')
+        await super().get('index.html', include_body)
 
 
-# noinspection PyAbstractClass
 class ServerInfoHandler(api.base.ApiHandler):
     async def get(self):
         cfg = config.get_config()
@@ -28,6 +44,53 @@ class ServerInfoHandler(api.base.ApiHandler):
             'version': update.VERSION,
             'config': {
                 'enableTranslate': cfg.enable_translate,
+                'enableUploadFile': cfg.enable_upload_file,
                 'loaderUrl': cfg.loader_url
             }
         })
+
+
+class UploadEmoticonHandler(api.base.ApiHandler):
+    async def post(self):
+        cfg = config.get_config()
+        if not cfg.enable_upload_file:
+            raise tornado.web.HTTPError(403)
+
+        try:
+            file = self.request.files['file'][0]
+        except LookupError:
+            raise tornado.web.MissingArgumentError('file')
+        if len(file.body) > 1024 * 1024:
+            raise tornado.web.HTTPError(413, 'file is too large, size=%d', len(file.body))
+        if not file.content_type.lower().startswith('image/'):
+            raise tornado.web.HTTPError(415)
+
+        url = await asyncio.get_running_loop().run_in_executor(
+            None, self._save_file, file.body, self.request.remote_ip
+        )
+        self.write({'url': url})
+
+    @staticmethod
+    def _save_file(body, client):
+        md5 = hashlib.md5(body).hexdigest()
+        filename = md5 + '.png'
+        path = os.path.join(EMOTICON_UPLOAD_PATH, filename)
+        logger.info('client=%s uploaded file, path=%s, size=%d', client, path, len(body))
+
+        tmp_path = path + '.tmp'
+        with open(tmp_path, 'wb') as f:
+            f.write(body)
+        os.replace(tmp_path, path)
+
+        return f'{EMOTICON_BASE_URL}/{filename}'
+
+
+ROUTES = [
+    (r'/api/server_info', ServerInfoHandler),
+    (r'/api/emoticon', UploadEmoticonHandler),
+]
+# 通配的放在最后
+LAST_ROUTES = [
+    (rf'{EMOTICON_BASE_URL}/(.*)', tornado.web.StaticFileHandler, {'path': EMOTICON_UPLOAD_PATH}),
+    (r'/(.*)', MainHandler, {'path': config.WEB_ROOT}),
+]

api/open_live.py

@@ -0,0 +1,194 @@
+# -*- coding: utf-8 -*-
+import asyncio
+import datetime
+import hashlib
+import hmac
+import json
+import logging
+import random
+from typing import *
+
+import aiohttp
+import tornado.web
+
+import api.base
+import config
+import utils.request
+
+logger = logging.getLogger(__name__)
+
+START_GAME_OPEN_LIVE_URL = 'https://live-open.biliapi.com/v2/app/start'
+END_GAME_OPEN_LIVE_URL = 'https://live-open.biliapi.com/v2/app/end'
+GAME_HEARTBEAT_OPEN_LIVE_URL = 'https://live-open.biliapi.com/v2/app/heartbeat'
+
+COMMON_SERVER_BASE_URL = 'https://chat.bilisc.com'
+START_GAME_COMMON_SERVER_URL = COMMON_SERVER_BASE_URL + '/api/internal/open_live/start_game'
+END_GAME_COMMON_SERVER_URL = COMMON_SERVER_BASE_URL + '/api/internal/open_live/end_game'
+GAME_HEARTBEAT_COMMON_SERVER_URL = COMMON_SERVER_BASE_URL + '/api/internal/open_live/game_heartbeat'
+
+
+class TransportError(Exception):
+    """网络错误或HTTP状态码错误"""
+
+
+class BusinessError(Exception):
+    """业务返回码错误"""
+    def __init__(self, data: dict):
+        super().__init__(f"message={data['message']}, request_id={data['request_id']}")
+        self.data = data
+
+    @property
+    def code(self) -> int:
+        return self.data['code']
+
+
+async def request_open_live_or_common_server(open_live_url, common_server_url, body: dict) -> dict:
+    """如果配置了开放平台，则直接请求，否则转发请求到公共服务器的内部接口"""
+    cfg = config.get_config()
+    if cfg.is_open_live_configured:
+        return await _request_open_live(open_live_url, body)
+
+    try:
+        req_ctx_mgr = utils.request.http_session.post(common_server_url, json=body)
+        return await _read_response(req_ctx_mgr)
+    except TransportError:
+        logger.exception('Request common server failed:')
+        raise
+    except BusinessError as e:
+        logger.warning('Request common server failed: %s', e)
+        raise
+
+
+async def _request_open_live(url, body: dict) -> dict:
+    cfg = config.get_config()
+    assert cfg.is_open_live_configured
+
+    body_bytes = json.dumps(body).encode('utf-8')
+    headers = {
+        'x-bili-accesskeyid': cfg.open_live_access_key_id,
+        'x-bili-content-md5': hashlib.md5(body_bytes).hexdigest(),
+        'x-bili-signature-method': 'HMAC-SHA256',
+        'x-bili-signature-nonce': str(random.randint(0, 999999999)),
+        'x-bili-signature-version': '1.0',
+        'x-bili-timestamp': str(int(datetime.datetime.now().timestamp())),
+    }
+
+    str_to_sign = '\n'.join(
+        f'{key}:{value}'
+        for key, value in headers.items()
+    )
+    signature = hmac.new(
+        cfg.open_live_access_key_secret.encode('utf-8'), str_to_sign.encode('utf-8'), hashlib.sha256
+    ).hexdigest()
+    headers['Authorization'] = signature
+
+    headers['Content-Type'] = 'application/json'
+    headers['Accept'] = 'application/json'
+
+    try:
+        req_ctx_mgr = utils.request.http_session.post(url, headers=headers, data=body_bytes)
+        return await _read_response(req_ctx_mgr)
+    except TransportError:
+        logger.exception('Request open live failed:')
+        raise
+    except BusinessError as e:
+        logger.warning('Request open live failed: %s', e)
+        raise
+
+
+async def _read_response(req_ctx_mgr: AsyncContextManager[aiohttp.ClientResponse]) -> dict:
+    try:
+        async with req_ctx_mgr as r:
+            r.raise_for_status()
+            data = await r.json()
+            code = data['code']
+            if code != 0:
+                raise BusinessError(data)
+            return data
+    except (aiohttp.ClientError, asyncio.TimeoutError) as e:
+        raise TransportError(f'{type(e).__name__}: {e}')
+
+
+class _OpenLiveHandlerBase(api.base.ApiHandler):
+    def prepare(self):
+        super().prepare()
+        if not isinstance(self.json_args, dict):
+            raise tornado.web.MissingArgumentError('body')
+
+        if 'app_id' in self.json_args:
+            cfg = config.get_config()
+            self.json_args['app_id'] = cfg.open_live_app_id
+
+        logger.info('client=%s requesting open live, cls=%s', self.request.remote_ip, type(self).__name__)
+
+
+class _PublicHandlerBase(_OpenLiveHandlerBase):
+    """外部接口，如果配置了开放平台，则直接请求，否则转发请求到公共服务器的内部接口"""
+    _OPEN_LIVE_URL: str
+    _COMMON_SERVER_URL: str
+
+    async def post(self):
+        try:
+            res = await request_open_live_or_common_server(
+                self._OPEN_LIVE_URL, self._COMMON_SERVER_URL, self.json_args
+            )
+        except TransportError:
+            raise tornado.web.HTTPError(500)
+        except BusinessError as e:
+            res = e.data
+        self.write(res)
+
+
+class _PrivateHandlerBase(_OpenLiveHandlerBase):
+    """内部接口，如果配置了开放平台，则直接请求，否则响应错误"""
+    _OPEN_LIVE_URL: str
+
+    async def post(self):
+        cfg = config.get_config()
+        if not cfg.is_open_live_configured:
+            raise tornado.web.HTTPError(501)
+
+        try:
+            res = await _request_open_live(self._OPEN_LIVE_URL, self.json_args)
+        except TransportError:
+            raise tornado.web.HTTPError(500)
+        except BusinessError as e:
+            res = e.data
+        self.write(res)
+
+
+class StartGamePublicHandler(_PublicHandlerBase):
+    _OPEN_LIVE_URL = START_GAME_OPEN_LIVE_URL
+    _COMMON_SERVER_URL = START_GAME_COMMON_SERVER_URL
+
+
+class StartGamePrivateHandler(_PrivateHandlerBase):
+    _OPEN_LIVE_URL = START_GAME_OPEN_LIVE_URL
+
+
+class EndGamePublicHandler(_PublicHandlerBase):
+    _OPEN_LIVE_URL = END_GAME_OPEN_LIVE_URL
+    _COMMON_SERVER_URL = END_GAME_COMMON_SERVER_URL
+
+
+class EndGamePrivateHandler(_PrivateHandlerBase):
+    _OPEN_LIVE_URL = END_GAME_OPEN_LIVE_URL
+
+
+class GameHeartbeatPublicHandler(_PublicHandlerBase):
+    _OPEN_LIVE_URL = GAME_HEARTBEAT_OPEN_LIVE_URL
+    _COMMON_SERVER_URL = GAME_HEARTBEAT_COMMON_SERVER_URL
+
+
+class GameHeartbeatPrivateHandler(_PrivateHandlerBase):
+    _OPEN_LIVE_URL = GAME_HEARTBEAT_OPEN_LIVE_URL
+
+
+ROUTES = [
+    (r'/api/open_live/start_game', StartGamePublicHandler),
+    (r'/api/internal/open_live/start_game', StartGamePrivateHandler),
+    (r'/api/open_live/end_game', EndGamePublicHandler),
+    (r'/api/internal/open_live/end_game', EndGamePrivateHandler),
+    (r'/api/open_live/game_heartbeat', GameHeartbeatPublicHandler),
+    (r'/api/internal/open_live/game_heartbeat', GameHeartbeatPrivateHandler),
+]

config.py

@@ -1,5 +1,4 @@
 # -*- coding: utf-8 -*-
-
 import configparser
 import logging
 import os
@@ -7,9 +6,13 @@ from typing import *
 
 logger = logging.getLogger(__name__)
 
+BASE_PATH = os.path.dirname(os.path.realpath(__file__))
+WEB_ROOT = os.path.join(BASE_PATH, 'frontend', 'dist')
+DATA_PATH = os.path.join(BASE_PATH, 'data')
+
 CONFIG_PATH_LIST = [
-    os.path.join('data', 'config.ini'),
-    os.path.join('data', 'config.example.ini')
+    os.path.join(DATA_PATH, 'config.ini'),
+    os.path.join(DATA_PATH, 'config.example.ini')
 ]
 
 _config: Optional['AppConfig'] = None
@@ -46,82 +49,122 @@ def get_config():
 
 class AppConfig:
     def __init__(self):
+        self.host = '127.0.0.1'
+        self.port = 12450
         self.database_url = 'sqlite:///data/database.db'
         self.tornado_xheaders = False
         self.loader_url = ''
+        self.open_browser_at_startup = True
+        self.enable_upload_file = True
 
-        self.fetch_avatar_interval = 3.5
-        self.fetch_avatar_max_queue_size = 2
-        self.avatar_cache_size = 50000
+        self.fetch_avatar_max_queue_size = 4
+        self.avatar_cache_size = 10000
+
+        self.open_live_access_key_id = ''
+        self.open_live_access_key_secret = ''
+        self.open_live_app_id = 0
 
         self.enable_translate = True
         self.allow_translate_rooms = set()
+        self.translate_max_queue_size = 10
         self.translation_cache_size = 50000
         self.translator_configs = []
 
+        self.text_emoticons = []
+
+    @property
+    def is_open_live_configured(self):
+        return (
+            self.open_live_access_key_id != '' and self.open_live_access_key_secret != '' and self.open_live_app_id != 0
+        )
+
     def load(self, path):
         try:
             config = configparser.ConfigParser()
-            config.read(path, 'utf-8')
+            config.read(path, 'utf-8-sig')
 
             self._load_app_config(config)
             self._load_translator_configs(config)
-        except Exception:
+            self._load_text_emoticons(config)
+        except Exception:  # noqa
             logger.exception('Failed to load config:')
             return False
         return True
 
-    def _load_app_config(self, config):
+    def _load_app_config(self, config: configparser.ConfigParser):
         app_section = config['app']
-        self.database_url = app_section['database_url']
-        self.tornado_xheaders = app_section.getboolean('tornado_xheaders')
-        self.loader_url = app_section['loader_url']
+        self.host = app_section.get('host', self.host)
+        self.port = app_section.getint('port', self.port)
+        self.database_url = app_section.get('database_url', self.database_url)
+        self.tornado_xheaders = app_section.getboolean('tornado_xheaders', self.tornado_xheaders)
+        self.loader_url = app_section.get('loader_url', self.loader_url)
+        self.open_browser_at_startup = app_section.getboolean('open_browser_at_startup', self.open_browser_at_startup)
+        self.enable_upload_file = app_section.getboolean('enable_upload_file', self.enable_upload_file)
 
-        self.fetch_avatar_interval = app_section.getfloat('fetch_avatar_interval')
-        self.fetch_avatar_max_queue_size = app_section.getint('fetch_avatar_max_queue_size')
-        self.avatar_cache_size = app_section.getint('avatar_cache_size')
+        self.fetch_avatar_max_queue_size = app_section.getint(
+            'fetch_avatar_max_queue_size', self.fetch_avatar_max_queue_size
+        )
+        self.avatar_cache_size = app_section.getint('avatar_cache_size', self.avatar_cache_size)
 
-        self.enable_translate = app_section.getboolean('enable_translate')
-        self.allow_translate_rooms = _str_to_list(app_section['allow_translate_rooms'], int, set)
-        self.translation_cache_size = app_section.getint('translation_cache_size')
+        self.open_live_access_key_id = app_section.get('open_live_access_key_id', self.open_live_access_key_id)
+        self.open_live_access_key_secret = app_section.get(
+            'open_live_access_key_secret', self.open_live_access_key_secret
+        )
+        self.open_live_app_id = app_section.getint('open_live_app_id', self.open_live_app_id)
 
-    def _load_translator_configs(self, config):
+        self.enable_translate = app_section.getboolean('enable_translate', self.enable_translate)
+        self.allow_translate_rooms = _str_to_list(app_section.get('allow_translate_rooms', ''), int, set)
+        self.translate_max_queue_size = app_section.getint('translate_max_queue_size', self.translate_max_queue_size)
+        self.translation_cache_size = app_section.getint('translation_cache_size', self.translation_cache_size)
+
+    def _load_translator_configs(self, config: configparser.ConfigParser):
         app_section = config['app']
-        section_names = _str_to_list(app_section['translator_configs'])
+        section_names = _str_to_list(app_section.get('translator_configs', ''))
         translator_configs = []
         for section_name in section_names:
-            section = config[section_name]
-            type_ = section['type']
+            try:
+                section = config[section_name]
+                type_ = section['type']
 
-            translator_config = {
-                'type': type_,
-                'query_interval': section.getfloat('query_interval'),
-                'max_queue_size': section.getint('max_queue_size')
-            }
-            if type_ == 'TencentTranslateFree':
-                translator_config['source_language'] = section['source_language']
-                translator_config['target_language'] = section['target_language']
-            elif type_ == 'BilibiliTranslateFree':
-                pass
-            elif type_ == 'TencentTranslate':
-                translator_config['source_language'] = section['source_language']
-                translator_config['target_language'] = section['target_language']
-                translator_config['secret_id'] = section['secret_id']
-                translator_config['secret_key'] = section['secret_key']
-                translator_config['region'] = section['region']
-            elif type_ == 'BaiduTranslate':
-                translator_config['source_language'] = section['source_language']
-                translator_config['target_language'] = section['target_language']
-                translator_config['app_id'] = section['app_id']
-                translator_config['secret'] = section['secret']
-            else:
-                raise ValueError(f'Invalid translator type: {type_}')
+                translator_config = {
+                    'type': type_,
+                    'query_interval': section.getfloat('query_interval'),
+                }
+                if type_ == 'TencentTranslateFree':
+                    translator_config['source_language'] = section['source_language']
+                    translator_config['target_language'] = section['target_language']
+                elif type_ == 'BilibiliTranslateFree':
+                    pass
+                elif type_ == 'TencentTranslate':
+                    translator_config['source_language'] = section['source_language']
+                    translator_config['target_language'] = section['target_language']
+                    translator_config['secret_id'] = section['secret_id']
+                    translator_config['secret_key'] = section['secret_key']
+                    translator_config['region'] = section['region']
+                elif type_ == 'BaiduTranslate':
+                    translator_config['source_language'] = section['source_language']
+                    translator_config['target_language'] = section['target_language']
+                    translator_config['app_id'] = section['app_id']
+                    translator_config['secret'] = section['secret']
+                else:
+                    raise ValueError(f'Invalid translator type: {type_}')
+            except Exception:  # noqa
+                logger.exception('Failed to load translator=%s config:', section_name)
+                continue
 
             translator_configs.append(translator_config)
         self.translator_configs = translator_configs
 
+    def _load_text_emoticons(self, config: configparser.ConfigParser):
+        mappings_section = config['text_emoticon_mappings']
+        text_emoticons = []
+        for value in mappings_section.values():
+            keyword, _, url = value.partition(',')
+            text_emoticons.append({'keyword': keyword, 'url': url})
+        self.text_emoticons = text_emoticons
 
-def _str_to_list(value, item_type: Type=str, container_type: Type=list):
+
+def _str_to_list(value, item_type: Type = str, container_type: Type = list):
     value = value.strip()
     if value == '':
         return container_type()

data/config.example.ini

@@ -2,8 +2,13 @@
 # If you want to modify the configuration, copy this file and rename it to "config.ini" and edit
 
 [app]
-# 数据库配置，见https://docs.sqlalchemy.org/en/13/core/engines.html#database-urls
-# See https://docs.sqlalchemy.org/en/13/core/engines.html#database-urls
+# 服务器监听的地址
+# The address the server listens on
+host = 127.0.0.1
+port = 12450
+
+# 数据库配置，见 https://docs.sqlalchemy.org/en/20/core/engines.html#database-urls
+# See https://docs.sqlalchemy.org/en/20/core/engines.html#database-urls
 database_url = sqlite:///data/database.db
 
 # 如果使用了nginx之类的反向代理服务器，设置为true
@@ -15,18 +20,22 @@ tornado_xheaders = false
 # Use a loader so that you can run OBS before blivechat. If empty, no loader is used
 loader_url = https://xfgryujk.sinacloud.net/blivechat/loader.html
 
+# 启动时打开浏览器
+# Open browser at startup
+open_browser_at_startup = true
 
-# 获取头像间隔时间（秒）。如果小于3秒有很大概率被服务器拉黑
-# Interval between fetching avatar (s). At least 3 seconds is recommended
-fetch_avatar_interval = 3.5
+# 允许上传自定义表情文件
+# Enable uploading custom emote file
+enable_upload_file = true
 
-# 获取头像最大队列长度，注意最长等待时间等于 最大队列长度 * 请求间隔时间
+
+# 获取头像最大队列长度
 # Maximum queue length for fetching avatar
-fetch_avatar_max_queue_size = 2
+fetch_avatar_max_queue_size = 4
 
-# 头像缓存数量
-# Number of avatar caches
-avatar_cache_size = 50000
+# 内存中头像缓存数量
+# Number of avatar caches in memory
+avatar_cache_size = 10000
 
 
 # 允许自动翻译到日语
@@ -34,10 +43,14 @@ avatar_cache_size = 50000
 enable_translate = true
 
 # 允许翻译的房间ID，以逗号分隔。如果为空，允许所有房间
-# Comma separated room IDs in which translation are not allowed. If empty, all are allowed
+# Comma separated room IDs in which translation are allowed. If empty, all rooms are allowed
 # Example: allow_translate_rooms = 4895312,22347054,21693691
 allow_translate_rooms =
 
+# 翻译最大队列长度
+# Maximum queue length for translating
+translate_max_queue_size = 10
+
 # 翻译缓存数量
 # Number of translation caches
 translation_cache_size = 50000
@@ -48,20 +61,25 @@ translation_cache_size = 50000
 # **The following is for translation team. Leave it default if you don't know its meaning**
 # -------------------------------------------------------------------------------------------------
 
+# 在B站直播开放平台申请的开发者密钥，如果不填，会把请求转发到作者的服务器
+open_live_access_key_id =
+open_live_access_key_secret =
+# 在B站直播开放平台创建的项目ID，如果不填，会把请求转发到作者的服务器
+open_live_app_id = 0
+
+
 # 翻译器配置，索引到下面的配置节。可以以逗号分隔配置多个翻译器，翻译时会自动负载均衡
 # 配置多个翻译器可以增加额度、增加QPS、容灾
 # 不同配置可以使用同一个类型，但要使用不同的账号，否则还是会遇到额度、调用频率限制
-translator_configs = tencent_translate_free,bilibili_translate_free
+translator_configs = tencent_translate_free
 
 
 [tencent_translate_free]
 # 类型：腾讯翻译白嫖版。使用了网页版的接口，**将来可能失效**
 type = TencentTranslateFree
 
-# 请求间隔时间（秒），等于 1 / QPS。目前没有遇到此接口有调用频率限制，10QPS应该够用了
-query_interval = 0.1
-# 最大队列长度，注意最长等待时间等于 最大队列长度 * 请求间隔时间
-max_queue_size = 100
+# 请求间隔时间（秒），等于 1 / QPS
+query_interval = 1
 
 # 自动：auto；中文：zh；日语：jp；英语：en；韩语：kr
 # 完整语言列表见文档：https://cloud.tencent.com/document/product/551/15619
@@ -71,16 +89,6 @@ source_language = zh
 target_language = jp
 
 
-[bilibili_translate_free]
-# 类型：B站翻译白嫖版。使用了B站直播网页的接口，**将来可能失效**。目前B站翻译后端是百度翻译
-type = BilibiliTranslateFree
-
-# 请求间隔时间（秒），等于 1 / QPS。目前此接口频率限制是3秒一次
-query_interval = 3.1
-# 最大队列长度，注意最长等待时间等于 最大队列长度 * 请求间隔时间
-max_queue_size = 3
-
-
 [tencent_translate]
 # 文档：https://cloud.tencent.com/product/tmt
 # 定价：https://cloud.tencent.com/document/product/551/35017
@@ -95,21 +103,19 @@ type = TencentTranslate
 
 # 请求间隔时间（秒），等于 1 / QPS。理论上最高QPS为5，实际测试是3
 query_interval = 0.333
-# 最大队列长度，注意最长等待时间等于 最大队列长度 * 请求间隔时间
-max_queue_size = 30
 
-# 自动：auto；中文：zh；日语：jp；英语：en；韩语：kr
+# 自动：auto；中文：zh；日语：ja；英语：en；韩语：ko
 # 完整语言列表见文档：https://cloud.tencent.com/document/product/551/15619
 # 源语言
 source_language = zh
 # 目标语言
-target_language = jp
+target_language = ja
 
 # 腾讯云API密钥
 secret_id =
 secret_key =
 
-# 腾讯云地域参数，用来标识希望操作哪个地域的数据
+# 腾讯云地域参数，用来标识希望操作哪个地域的数据，建议按照运行blivechat的机器所在地区就近选择
 # 北京：ap-beijing；上海：ap-shanghai；香港：ap-hongkong；首尔：ap-seoul
 # 完整地域列表见文档：https://cloud.tencent.com/document/api/551/15615#.E5.9C.B0.E5.9F.9F.E5.88.97.E8.A1.A8
 region = ap-shanghai
@@ -118,17 +124,15 @@ region = ap-shanghai
 [baidu_translate]
 # 文档：https://fanyi-api.baidu.com/
 # 定价：https://fanyi-api.baidu.com/product/112
-# * 标准版完全免费，不限使用字符量（QPS=1）
-# * 高级版每月前200万字符免费，超出后仅收取超出部分费用（QPS=10），49元/百万字符
-# * 尊享版每月前200万字符免费，超出后仅收取超出部分费用（QPS=100），49元/百万字符
+# * 标准版每月前5万字符免费，超出仅收取超出部分费用（QPS=1），按49元/百万字符计费
+# * 高级版每月前100万字符免费，超出仅收取超出部分费用（QPS=10），按49元/百万字符计费
+# * 尊享版每月前200万字符免费，超出后仅收取超出部分费用（QPS=100），按49元/百万字符计费
 
 # 类型：百度翻译
 type = BaiduTranslate
 
 # 请求间隔时间（秒），等于 1 / QPS
 query_interval = 1.5
-# 最大队列长度，注意最长等待时间等于 最大队列长度 * 请求间隔时间
-max_queue_size = 9
 
 # 自动：auto；中文：zh；日语：jp；英语：en；韩语：kor
 # 完整语言列表见文档：https://fanyi-api.baidu.com/doc/21
@@ -140,3 +144,89 @@ target_language = jp
 # 百度翻译开放平台应用ID和密钥
 app_id =
 secret =
+
+
+# 傻逼B站，获取表情都要登录，开放平台也不发文本表情的URL，我服了
+[text_emoticon_mappings]
+1 = [dog],http://i0.hdslb.com/bfs/live/4428c84e694fbf4e0ef6c06e958d9352c3582740.png
+2 = [花],http://i0.hdslb.com/bfs/live/7dd2ef03e13998575e4d8a803c6e12909f94e72b.png
+3 = [妙],http://i0.hdslb.com/bfs/live/08f735d950a0fba267dda140673c9ab2edf6410d.png
+4 = [哇],http://i0.hdslb.com/bfs/live/650c3e22c06edcbca9756365754d38952fc019c3.png
+5 = [爱],http://i0.hdslb.com/bfs/live/1daaa5d284dafaa16c51409447da851ff1ec557f.png
+6 = [手机],http://i0.hdslb.com/bfs/live/b159f90431148a973824f596288e7ad6a8db014b.png
+7 = [撇嘴],http://i0.hdslb.com/bfs/live/4255ce6ed5d15b60311728a803d03dd9a24366b2.png
+8 = [委屈],http://i0.hdslb.com/bfs/live/69312e99a00d1db2de34ef2db9220c5686643a3f.png
+9 = [抓狂],http://i0.hdslb.com/bfs/live/a7feb260bb5b15f97d7119b444fc698e82516b9f.png
+10 = [比心],http://i0.hdslb.com/bfs/live/4e029593562283f00d39b99e0557878c4199c71d.png
+11 = [赞],http://i0.hdslb.com/bfs/live/2dd666d3651bafe8683acf770b7f4163a5f49809.png
+12 = [滑稽],http://i0.hdslb.com/bfs/live/8624fd172037573c8600b2597e3731ef0e5ea983.png
+13 = [吃瓜],http://i0.hdslb.com/bfs/live/ffb53c252b085d042173379ac724694ce3196194.png
+14 = [笑哭],http://i0.hdslb.com/bfs/live/c5436c6806c32b28d471bb23d42f0f8f164a187a.png
+15 = [捂脸],http://i0.hdslb.com/bfs/live/e6073c6849f735ae6cb7af3a20ff7dcec962b4c5.png
+16 = [喝彩],http://i0.hdslb.com/bfs/live/b51824125d09923a4ca064f0c0b49fc97d3fab79.png
+17 = [偷笑],http://i0.hdslb.com/bfs/live/e2ba16f947a23179cdc00420b71cc1d627d8ae25.png
+18 = [大笑],http://i0.hdslb.com/bfs/live/e2589d086df0db8a7b5ca2b1273c02d31d4433d4.png
+19 = [惊喜],http://i0.hdslb.com/bfs/live/9c75761c5b6e1ff59b29577deb8e6ad996b86bd7.png
+20 = [傲娇],http://i0.hdslb.com/bfs/live/b5b44f099059a1bafb2c2722cfe9a6f62c1dc531.png
+21 = [疼],http://i0.hdslb.com/bfs/live/492b10d03545b7863919033db7d1ae3ef342df2f.png
+22 = [吓],http://i0.hdslb.com/bfs/live/c6bed64ffb78c97c93a83fbd22f6fdf951400f31.png
+23 = [阴险],http://i0.hdslb.com/bfs/live/a4df45c035b0ca0c58f162b5fb5058cf273d0d09.png
+24 = [惊讶],http://i0.hdslb.com/bfs/live/bc26f29f62340091737c82109b8b91f32e6675ad.png
+25 = [生病],http://i0.hdslb.com/bfs/live/84c92239591e5ece0f986c75a39050a5c61c803c.png
+26 = [嘘],http://i0.hdslb.com/bfs/live/b6226219384befa5da1d437cb2ff4ba06c303844.png
+27 = [奸笑],http://i0.hdslb.com/bfs/live/5935e6a4103d024955f749d428311f39e120a58a.png
+28 = [囧],http://i0.hdslb.com/bfs/live/204413d3cf330e122230dcc99d29056f2a60e6f2.png
+29 = [捂脸2],http://i0.hdslb.com/bfs/live/a2ad0cc7e390a303f6d243821479452d31902a5f.png
+30 = [出窍],http://i0.hdslb.com/bfs/live/bb8e95fa54512ffea07023ea4f2abee4a163e7a0.png
+31 = [吐了啊],http://i0.hdslb.com/bfs/live/2b6b4cc33be42c3257dc1f6ef3a39d666b6b4b1a.png
+32 = [鼻子],http://i0.hdslb.com/bfs/live/f4ed20a70d0cb85a22c0c59c628aedfe30566b37.png
+33 = [调皮],http://i0.hdslb.com/bfs/live/84fe12ecde5d3875e1090d83ac9027cb7d7fba9f.png
+34 = [酸],http://i0.hdslb.com/bfs/live/98fd92c6115b0d305f544b209c78ec322e4bb4ff.png
+35 = [冷],http://i0.hdslb.com/bfs/live/b804118a1bdb8f3bec67d9b108d5ade6e3aa93a9.png
+36 = [OK],http://i0.hdslb.com/bfs/live/86268b09e35fbe4215815a28ef3cf25ec71c124f.png
+37 = [微笑],http://i0.hdslb.com/bfs/live/f605dd8229fa0115e57d2f16cb019da28545452b.png
+38 = [藏狐],http://i0.hdslb.com/bfs/live/05ef7849e7313e9c32887df922613a7c1ad27f12.png
+39 = [龇牙],http://i0.hdslb.com/bfs/live/8b99266ea7b9e86cf9d25c3d1151d80c5ba5c9a1.png
+40 = [防护],http://i0.hdslb.com/bfs/live/17435e60dcc28ce306762103a2a646046ff10b0a.png
+41 = [笑],http://i0.hdslb.com/bfs/live/a91a27f83c38b5576f4cd08d4e11a2880de78918.png
+42 = [一般],http://i0.hdslb.com/bfs/live/8d436de0c3701d87e4ca9c1be01c01b199ac198e.png
+43 = [嫌弃],http://i0.hdslb.com/bfs/live/c409425ba1ad2c6534f0df7de350ba83a9c949e5.png
+44 = [无语],http://i0.hdslb.com/bfs/live/4781a77be9c8f0d4658274eb4e3012c47a159f23.png
+45 = [哈欠],http://i0.hdslb.com/bfs/live/6e496946725cd66e7ff1b53021bf1cc0fc240288.png
+46 = [可怜],http://i0.hdslb.com/bfs/live/8e88e6a137463703e96d4f27629f878efa323456.png
+47 = [歪嘴笑],http://i0.hdslb.com/bfs/live/bea1f0497888f3e9056d3ce14ba452885a485c02.png
+48 = [亲亲],http://i0.hdslb.com/bfs/live/10662d9c0d6ddb3203ecf50e77788b959d4d1928.png
+49 = [问号],http://i0.hdslb.com/bfs/live/a0c456b6d9e3187399327828a9783901323bfdb5.png
+50 = [波吉],http://i0.hdslb.com/bfs/live/57dee478868ed9f1ce3cf25a36bc50bde489c404.png
+51 = [OH],http://i0.hdslb.com/bfs/live/0d5123cddf389302df6f605087189fd10919dc3c.png
+52 = [再见],http://i0.hdslb.com/bfs/live/f408e2af700adcc2baeca15510ef620bed8d4c43.png
+53 = [白眼],http://i0.hdslb.com/bfs/live/7fa907ae85fa6327a0466e123aee1ac32d7c85f7.png
+54 = [鼓掌],http://i0.hdslb.com/bfs/live/d581d0bc30c8f9712b46ec02303579840c72c42d.png
+55 = [大哭],http://i0.hdslb.com/bfs/live/816402551e6ce30d08b37a917f76dea8851fe529.png
+56 = [呆],http://i0.hdslb.com/bfs/live/179c7e2d232cd74f30b672e12fc728f8f62be9ec.png
+57 = [流汗],http://i0.hdslb.com/bfs/live/b00e2e02904096377061ec5f93bf0dd3321f1964.png
+58 = [生气],http://i0.hdslb.com/bfs/live/2c69dad2e5c0f72f01b92746bc9d148aee1993b2.png
+59 = [加油],http://i0.hdslb.com/bfs/live/fbc3c8bc4152a65bbf4a9fd5a5d27710fbff2119.png
+60 = [害羞],http://i0.hdslb.com/bfs/live/d8ce9b05c0e40cec61a15ba1979c8517edd270bf.png
+61 = [虎年],http://i0.hdslb.com/bfs/live/a51af0d7d9e60ce24f139c468a3853f9ba9bb184.png
+62 = [doge2],http://i0.hdslb.com/bfs/live/f547cc853cf43e70f1e39095d9b3b5ac1bf70a8d.png
+63 = [金钱豹],http://i0.hdslb.com/bfs/live/b6e8131897a9a718ee280f2510bfa92f1d84429b.png
+64 = [瓜子],http://i0.hdslb.com/bfs/live/fd35718ac5a278fd05fe5287ebd41de40a59259d.png
+65 = [墨镜],http://i0.hdslb.com/bfs/live/5e01c237642c8b662a69e21b8e0fbe6e7dbc2aa1.png
+66 = [难过],http://i0.hdslb.com/bfs/live/5776481e380648c0fb3d4ad6173475f69f1ce149.png
+67 = [抱抱],http://i0.hdslb.com/bfs/live/abddb0b621b389fc8c2322b1cfcf122d8936ba91.png
+68 = [跪了],http://i0.hdslb.com/bfs/live/4f2155b108047d60c1fa9dccdc4d7abba18379a0.png
+69 = [摊手],http://i0.hdslb.com/bfs/live/1e0a2baf088a34d56e2cc226b2de36a5f8d6c926.png
+70 = [热],http://i0.hdslb.com/bfs/live/6df760280b17a6cbac8c1874d357298f982ba4cf.png
+71 = [三星堆],http://i0.hdslb.com/bfs/live/0a1ab3f0f2f2e29de35c702ac1ecfec7f90e325d.png
+72 = [鼠],http://i0.hdslb.com/bfs/live/98f842994035505c728e32e32045d649e371ecd6.png
+73 = [汤圆],http://i0.hdslb.com/bfs/live/23ae12d3a71b9d7a22c8773343969fcbb94b20d0.png
+74 = [泼水],http://i0.hdslb.com/bfs/live/29533893115c4609a4af336f49060ea13173ca78.png
+75 = [鬼魂],http://i0.hdslb.com/bfs/live/5d86d55ba9a2f99856b523d8311cf75cfdcccdbc.png
+76 = [不行],http://i0.hdslb.com/bfs/live/607f74ccf5eec7d2b17d91b9bb36be61a5dd196b.png
+77 = [响指],http://i0.hdslb.com/bfs/live/3b2fedf09b0ac79679b5a47f5eb3e8a38e702387.png
+78 = [牛],http://i0.hdslb.com/bfs/live/5e61223561203c50340b4c9b41ba7e4b05e48ae2.png
+79 = [保佑],http://i0.hdslb.com/bfs/live/241b13adb4933e38b7ea6f5204e0648725e76fbf.png
+80 = [抱拳],http://i0.hdslb.com/bfs/live/3f170894dd08827ee293afcb5a3d2b60aecdb5b1.png
+81 = [给力],http://i0.hdslb.com/bfs/live/d1ba5f4c54332a21ed2ca0dcecaedd2add587839.png
+82 = [耶],http://i0.hdslb.com/bfs/live/eb2d84ba623e2335a48f73fb5bef87bcf53c1239.png

frontend/.eslintignore

@@ -0,0 +1,2 @@
+brotli_decode.js
+pronunciation/dict*.js

frontend/.eslintrc.js

@@ -0,0 +1,81 @@
+module.exports = {
+  "root": true,
+  "env": {
+    "browser": true,
+    "node": true
+  },
+  "parserOptions": {
+    "parser": "@babel/eslint-parser"
+  },
+  "extends": [
+    "plugin:vue/essential",
+    "eslint:recommended"
+  ],
+  "rules": {
+    "array-bracket-spacing": ["error", "never"], // 数组括号内不加空格
+    "arrow-parens": ["error", "as-needed"], // 箭头函数单个参数不加括号
+    "arrow-spacing": "error", // 箭头前后加空格
+    "block-spacing": "error", // 块大括号内加空格
+    "brace-style": "error", // 大括号不独占一行
+    "comma-spacing": "error", // 逗号前面不加空格，后面加空格
+    "comma-style": "error", // 逗号在语句后面而不是下一条的前面
+    "computed-property-spacing": "error", // 计算属性名前后不加空格
+    "curly": "error", // 禁止省略大括号
+    "dot-notation": "error", // 使用点访问成员
+    "eol-last": "error", // 文件末尾加换行符
+    "func-call-spacing": "error", // 调用函数名和括号间不加空格
+    "func-style": ["error", "declaration", { "allowArrowFunctions": true }], // 使用函数定义语法，而不是把函数表达式赋值到变量
+    "indent": ["error", 2], // 缩进2空格
+    "key-spacing": ["error", { "mode": "minimum" }],
+    "keyword-spacing": "error", // 关键词前后加空格
+    "lines-between-class-members": "error", // 类成员定义间加空格
+    "max-lines-per-function": ["error", 150], // 每个函数最多行数
+    "max-nested-callbacks": ["error", 3], // 每个函数最多嵌套回调数
+    "new-parens": "error", // new调用构造函数加空格
+    "no-array-constructor": "error", // 使用数组字面量，而不是数组构造函数
+    "no-floating-decimal": "error", // 禁止省略浮点数首尾的0
+    "no-implicit-coercion": "error", // 禁止隐式转换
+    "no-empty": ["error", { "allowEmptyCatch": true }], // 禁止空的块，除了catch
+    "no-extra-parens": ["error", "all", { "nestedBinaryExpressions": false }], // 禁止多余的括号
+    "no-labels": "error", // 禁止使用标签
+    "no-lone-blocks": "error", // 禁止没用的块
+    "no-mixed-operators": "error", // 禁止混用不同优先级的操作符而不加括号
+    "no-multi-spaces": ["error", { "ignoreEOLComments": true }], // 禁止多个空格，除了行尾注释前
+    "no-multiple-empty-lines": "error", // 最多2个连续空行
+    "no-nested-ternary": "error", // 禁止嵌套三元表达式
+    "no-sequences": "error", // 禁止使用逗号操作符
+    "no-tabs": "error", // 禁止使用tab
+    "no-trailing-spaces": ["error", { "skipBlankLines": true }], // 禁止行尾的空格，除了空行
+    "no-unused-expressions": "error", // 禁止没用的表达式
+    "no-useless-concat": "error", // 禁止没用的字符串连接
+    "no-useless-rename": "error", // 禁止没用的模块导入重命名、解构赋值重命名
+    "no-useless-return": "error", // 禁止没用的return
+    "no-var": "error", // 禁止使用var声明变量
+    "no-void": "error", // 禁止使用void
+    "no-whitespace-before-property": "error", // 禁止访问属性的点前后加空格
+    "object-curly-spacing": ["error", "always"], // 对象字面量括号内加空格
+    "operator-assignment": "error", // 尽量使用+=
+    "operator-linebreak": ["error", "before"], // 操作符放行首
+    "prefer-object-spread": "error", // 使用{...obj}，而不是Object.assign
+    "prefer-rest-params": "error", // 使用...args，而不是arguments
+    "prefer-spread": "error", // 使用func(...args)，而不是apply
+    "prefer-template": "error", // 使用模板字符串，而不是字符串连接
+    "rest-spread-spacing": ["error", "never"], // 解包操作符不加空格
+    "semi": ["error", "never"], // 禁止使用多余的分号
+    "semi-spacing": "error", // 分号前面不加空格，后面加空格
+    "semi-style": "error", // 分号在语句后面而不是下一条的前面
+    "space-before-blocks": "error", // 块大括号前加空格
+    "space-before-function-paren": ["error", "never"], // 函数定义名称和括号间不加空格
+    "space-in-parens": "error", // 括号内不加空格
+    "space-infix-ops": "error", // 二元操作符前后加空格
+    "space-unary-ops": "error", // 关键词一元操作符后加空格，符号一元操作符不加
+    "spaced-comment": ["error", "always", { "block": { "balanced": true } }], // 注释前面加空格
+    "template-curly-spacing": "error", // 模板字符串中变量大括号内不加空格
+
+    "no-shadow": "warn", // 变量名和外部作用域重复
+
+    "no-console": "off", // 线上尽量不要用console输出，看不到的
+
+    "vue/multi-word-component-names": "off", // Vue组件名允许用1个单词
+  }
+}

frontend/jsconfig.json

@@ -0,0 +1,17 @@
+{
+  "compilerOptions": {
+    "target": "es2015",
+    "module": "esnext",
+    "baseUrl": "./",
+    "paths": {
+      "@/*": ["src/*"]
+    }
+  },
+  "vueCompilerOptions": {
+    "target": 2.7,
+  },
+  "include": [
+    "./src/**/*.js",
+    "./src/**/*.vue"
+  ]
+}

frontend/package.json

@@ -1,6 +1,6 @@
 {
   "name": "blivechat",
-  "version": "0.1.0",
+  "version": "1.8.0",
   "private": true,
   "scripts": {
     "serve": "vue-cli-service serve",
@@ -8,47 +8,29 @@
     "lint": "vue-cli-service lint"
   },
   "dependencies": {
-    "axios": "^0.21.1",
-    "core-js": "^3.6.5",
+    "axios": "^1.4.0",
+    "core-js": "^3.8.3",
     "downloadjs": "^1.4.7",
-    "element-ui": "^2.9.1",
-    "lodash": "^4.17.19",
-    "pako": "^1.0.11",
-    "vue": "^2.6.10",
-    "vue-i18n": "^8.11.2",
-    "vue-router": "^3.0.6"
+    "element-ui": "^2.15.13",
+    "lodash": "^4.17.21",
+    "pako": "^2.1.0",
+    "vue": "^2.7.14",
+    "vue-i18n": "^8.28.2",
+    "vue-router": "^3.6.5"
   },
   "devDependencies": {
-    "@vue/cli-plugin-babel": "^4.5.12",
-    "@vue/cli-plugin-eslint": "^4.5.12",
-    "@vue/cli-service": "~4.5.12",
-    "babel-eslint": "^10.1.0",
+    "@babel/core": "^7.12.16",
+    "@babel/eslint-parser": "^7.12.16",
+    "@vue/cli-plugin-babel": "~5.0.0",
+    "@vue/cli-plugin-eslint": "~5.0.0",
+    "@vue/cli-service": "~5.0.0",
     "babel-plugin-component": "^1.1.1",
-    "eslint": "^6.7.2",
-    "eslint-plugin-vue": "^6.2.2",
-    "vue-template-compiler": "^2.5.21"
-  },
-  "eslintConfig": {
-    "root": true,
-    "env": {
-      "node": true
-    },
-    "extends": [
-      "plugin:vue/essential",
-      "eslint:recommended"
-    ],
-    "rules": {},
-    "parserOptions": {
-      "parser": "babel-eslint"
-    }
-  },
-  "postcss": {
-    "plugins": {
-      "autoprefixer": {}
-    }
+    "eslint": "^7.32.0",
+    "eslint-plugin-vue": "^9.16.1"
   },
   "browserslist": [
     "> 1%",
-    "last 2 versions"
+    "last 2 versions",
+    "not dead"
   ]
 }

frontend/src/api/chat/ChatClientDirect.js

@@ -1,372 +0,0 @@
-import axios from 'axios'
-import * as pako from 'pako'
-
-import {getUuid4Hex} from '@/utils'
-import * as avatar from './avatar'
-
-const HEADER_SIZE = 16
-
-// const WS_BODY_PROTOCOL_VERSION_INFLATE = 0
-// const WS_BODY_PROTOCOL_VERSION_NORMAL = 1
-const WS_BODY_PROTOCOL_VERSION_DEFLATE = 2
-
-// const OP_HANDSHAKE = 0
-// const OP_HANDSHAKE_REPLY = 1
-const OP_HEARTBEAT = 2
-const OP_HEARTBEAT_REPLY = 3
-// const OP_SEND_MSG = 4
-const OP_SEND_MSG_REPLY = 5
-// const OP_DISCONNECT_REPLY = 6
-const OP_AUTH = 7
-const OP_AUTH_REPLY = 8
-// const OP_RAW = 9
-// const OP_PROTO_READY = 10
-// const OP_PROTO_FINISH = 11
-// const OP_CHANGE_ROOM = 12
-// const OP_CHANGE_ROOM_REPLY = 13
-// const OP_REGISTER = 14
-// const OP_REGISTER_REPLY = 15
-// const OP_UNREGISTER = 16
-// const OP_UNREGISTER_REPLY = 17
-// B站业务自定义OP
-// const MinBusinessOp = 1000
-// const MaxBusinessOp = 10000
-
-const HEARTBEAT_INTERVAL = 10 * 1000
-const RECEIVE_TIMEOUT = HEARTBEAT_INTERVAL + 5 * 1000
-
-let textEncoder = new TextEncoder()
-let textDecoder = new TextDecoder()
-
-export default class ChatClientDirect {
-  constructor (roomId) {
-    // 调用initRoom后初始化，如果失败，使用这里的默认值
-    this.roomId = roomId
-    this.roomOwnerUid = 0
-    this.hostServerList = [
-      {host: "broadcastlv.chat.bilibili.com", port: 2243, wss_port: 443, ws_port: 2244}
-    ]
-
-    this.onAddText = null
-    this.onAddGift = null
-    this.onAddMember = null
-    this.onAddSuperChat = null
-    this.onDelSuperChat = null
-    this.onUpdateTranslation = null
-
-    this.websocket = null
-    this.retryCount = 0
-    this.isDestroying = false
-    this.heartbeatTimerId = null
-    this.receiveTimeoutTimerId = null
-  }
-
-  async start () {
-    await this.initRoom()
-    this.wsConnect()
-  }
-
-  stop () {
-    this.isDestroying = true
-    if (this.websocket) {
-      this.websocket.close()
-    }
-  }
-
-  async initRoom () {
-    let res
-    try {
-      res = (await axios.get('/api/room_info', {params: {
-        roomId: this.roomId
-      }})).data
-    } catch {
-      return
-    }
-    this.roomId = res.roomId
-    this.roomOwnerUid = res.ownerUid
-    if (res.hostServerList.length !== 0) {
-      this.hostServerList = res.hostServerList
-    }
-  }
-
-  makePacket (data, operation) {
-    let body = textEncoder.encode(JSON.stringify(data))
-    let header = new ArrayBuffer(HEADER_SIZE)
-    let headerView = new DataView(header)
-    headerView.setUint32(0, HEADER_SIZE + body.byteLength)   // pack_len
-    headerView.setUint16(4, HEADER_SIZE)                     // raw_header_size
-    headerView.setUint16(6, 1)                               // ver
-    headerView.setUint32(8, operation)                       // operation
-    headerView.setUint32(12, 1)                              // seq_id
-    return new Blob([header, body])
-  }
-
-  sendAuth () {
-    let authParams = {
-      uid: 0,
-      roomid: this.roomId,
-      protover: 2,
-      platform: 'web',
-      clientver: '1.14.3',
-      type: 2
-    }
-    this.websocket.send(this.makePacket(authParams, OP_AUTH))
-  }
-
-  wsConnect () {
-    if (this.isDestroying) {
-      return
-    }
-    let hostServer = this.hostServerList[this.retryCount % this.hostServerList.length]
-    const url = `wss://${hostServer.host}:${hostServer.wss_port}/sub`
-    this.websocket = new WebSocket(url)
-    this.websocket.binaryType = 'arraybuffer'
-    this.websocket.onopen = this.onWsOpen.bind(this)
-    this.websocket.onclose = this.onWsClose.bind(this)
-    this.websocket.onmessage = this.onWsMessage.bind(this)
-  }
-
-  onWsOpen () {
-    this.sendAuth()
-    this.heartbeatTimerId = window.setInterval(this.sendHeartbeat.bind(this), HEARTBEAT_INTERVAL)
-    this.refreshReceiveTimeoutTimer()
-  }
-
-  sendHeartbeat () {
-    this.websocket.send(this.makePacket({}, OP_HEARTBEAT))
-  }
-
-  refreshReceiveTimeoutTimer() {
-    if (this.receiveTimeoutTimerId) {
-      window.clearTimeout(this.receiveTimeoutTimerId)
-    }
-    this.receiveTimeoutTimerId = window.setTimeout(this.onReceiveTimeout.bind(this), RECEIVE_TIMEOUT)
-  }
-
-  onReceiveTimeout() {
-    window.console.warn('接收消息超时')
-    this.receiveTimeoutTimerId = null
-
-    // 直接丢弃阻塞的websocket，不等onclose回调了
-    this.websocket.onopen = this.websocket.onclose = this.websocket.onmessage = null
-    this.websocket.close()
-    this.onWsClose()
-  }
-
-  onWsClose () {
-    this.websocket = null
-    if (this.heartbeatTimerId) {
-      window.clearInterval(this.heartbeatTimerId)
-      this.heartbeatTimerId = null
-    }
-    if (this.receiveTimeoutTimerId) {
-      window.clearTimeout(this.receiveTimeoutTimerId)
-      this.receiveTimeoutTimerId = null
-    }
-
-    if (this.isDestroying) {
-      return
-    }
-    window.console.warn(`掉线重连中${++this.retryCount}`)
-    window.setTimeout(this.wsConnect.bind(this), 1000)
-  }
-
-  onWsMessage (event) {
-    this.refreshReceiveTimeoutTimer()
-    this.retryCount = 0
-    if (!(event.data instanceof ArrayBuffer)) {
-      window.console.warn('未知的websocket消息：', event.data)
-      return
-    }
-
-    let data = new Uint8Array(event.data)
-    this.handlerMessage(data)
-  }
-
-  handlerMessage (data) {
-    let offset = 0
-    while (offset < data.byteLength) {
-      let dataView = new DataView(data.buffer, offset)
-      let packLen = dataView.getUint32(0)
-      // let rawHeaderSize = dataView.getUint16(4)
-      let ver = dataView.getUint16(6)
-      let operation = dataView.getUint32(8)
-      // let seqId = dataView.getUint32(12)
-      
-      switch (operation) {
-      case OP_HEARTBEAT_REPLY: {
-        // 人气值没用
-        break
-      }
-      case OP_SEND_MSG_REPLY: {
-        let body = new Uint8Array(data.buffer, offset + HEADER_SIZE, packLen - HEADER_SIZE)
-        if (ver == WS_BODY_PROTOCOL_VERSION_DEFLATE) {
-          body = pako.inflate(body)
-          this.handlerMessage(body)
-        } else {
-          try {
-            body = JSON.parse(textDecoder.decode(body))
-            this.handlerCommand(body)
-          } catch (e) {
-            window.console.warn('body:', body)
-            throw e
-          }
-        }
-        break
-      }
-      case OP_AUTH_REPLY: {
-        this.sendHeartbeat()
-        break
-      }
-      default: {
-        let body = new Uint8Array(data.buffer, offset + HEADER_SIZE, packLen - HEADER_SIZE)
-        window.console.warn('未知包类型：operation=', operation, body)
-        break
-      }
-      }
-
-      offset += packLen
-    }
-  }
-
-  handlerCommand (command) {
-    if (command instanceof Array) {
-      for (let oneCommand of command) {
-        this.handlerCommand(oneCommand)
-      }
-      return
-    }
-
-    let cmd = command.cmd || ''
-    let pos = cmd.indexOf(':')
-    if (pos != -1) {
-      cmd = cmd.substr(0, pos)
-    }
-    let handler = COMMAND_HANDLERS[cmd]
-    if (handler) {
-      handler.call(this, command)
-    }
-  }
-
-  async onReceiveDanmaku (command) {
-    if (!this.onAddText) {
-      return
-    }
-    let info = command.info
-
-    let roomId, medalLevel
-    if (info[3]) {
-      roomId = info[3][3]
-      medalLevel = info[3][0]
-    } else {
-      roomId = medalLevel = 0
-    }
-
-    let uid = info[2][0]
-    let isAdmin = info[2][2]
-    let privilegeType = info[7]
-    let authorType
-    if (uid === this.roomOwnerUid) {
-      authorType = 3
-    } else if (isAdmin) {
-      authorType = 2
-    } else if (privilegeType !== 0) {
-      authorType = 1
-    } else {
-      authorType = 0
-    }
-
-    let urank = info[2][5]
-    let data = {
-      avatarUrl: await avatar.getAvatarUrl(uid),
-      timestamp: info[0][4] / 1000,
-      authorName: info[2][1],
-      authorType: authorType,
-      content: info[1],
-      privilegeType: privilegeType,
-      isGiftDanmaku: !!info[0][9],
-      authorLevel: info[4][0],
-      isNewbie: urank < 10000,
-      isMobileVerified: !!info[2][6],
-      medalLevel: roomId === this.roomId ? medalLevel : 0,
-      id: getUuid4Hex(),
-      translation: ''
-    }
-    this.onAddText(data)
-  }
-
-  onReceiveGift (command) {
-    if (!this.onAddGift) {
-      return
-    }
-    let data = command.data
-    if (data.coin_type !== 'gold') { // 丢人
-      return
-    }
-
-    data = {
-      id: getUuid4Hex(),
-      avatarUrl: avatar.processAvatarUrl(data.face),
-      timestamp: data.timestamp,
-      authorName: data.uname,
-      totalCoin: data.total_coin,
-      giftName: data.giftName,
-      num: data.num
-    }
-    this.onAddGift(data)
-  }
-
-  async onBuyGuard (command) {
-    if (!this.onAddMember) {
-      return
-    }
-
-    let data = command.data
-    data = {
-      id: getUuid4Hex(),
-      avatarUrl: await avatar.getAvatarUrl(data.uid),
-      timestamp: data.start_time,
-      authorName: data.username,
-      privilegeType: data.guard_level
-    }
-    this.onAddMember(data)
-  }
-
-  onSuperChat (command) {
-    if (!this.onAddSuperChat) {
-      return
-    }
-
-    let data = command.data
-    data = {
-      id: data.id.toString(),
-      avatarUrl: avatar.processAvatarUrl(data.user_info.face),
-      timestamp: data.start_time,
-      authorName: data.user_info.uname,
-      price: data.price,
-      content: data.message,
-      translation: ''
-    }
-    this.onAddSuperChat(data)
-  }
-
-  onSuperChatDelete (command) {
-    if (!this.onDelSuperChat) {
-      return
-    }
-
-    let ids = []
-    for (let id of command.data.ids) {
-      ids.push(id.toString())
-    }
-    this.onDelSuperChat({ids})
-  }
-}
-
-const COMMAND_HANDLERS = {
-  DANMU_MSG: ChatClientDirect.prototype.onReceiveDanmaku,
-  SEND_GIFT: ChatClientDirect.prototype.onReceiveGift,
-  GUARD_BUY: ChatClientDirect.prototype.onBuyGuard,
-  SUPER_CHAT_MESSAGE: ChatClientDirect.prototype.onSuperChat,
-  SUPER_CHAT_MESSAGE_DELETE: ChatClientDirect.prototype.onSuperChatDelete
-}

frontend/src/api/chat/ChatClientDirectOpenLive.js

@@ -0,0 +1,255 @@
+import axios from 'axios'
+
+import * as chat from '.'
+import * as base from './ChatClientOfficialBase'
+import ChatClientOfficialBase from './ChatClientOfficialBase'
+
+const GAME_HEARTBEAT_INTERVAL = 20 * 1000
+
+export default class ChatClientDirectOpenLive extends ChatClientOfficialBase {
+  constructor(roomOwnerAuthCode) {
+    super()
+    this.CMD_CALLBACK_MAP = CMD_CALLBACK_MAP
+
+    this.roomOwnerAuthCode = roomOwnerAuthCode
+
+    // 调用initRoom后初始化
+    this.roomOwnerUid = null
+    this.hostServerUrlList = []
+    this.authBody = null
+    this.gameId = null
+
+    this.gameHeartbeatTimerId = null
+  }
+
+  stop() {
+    super.stop()
+
+    if (this.gameHeartbeatTimerId) {
+      window.clearInterval(this.gameHeartbeatTimerId)
+      this.gameHeartbeatTimerId = null
+    }
+    this.endGame()
+  }
+
+  async initRoom() {
+    if (!await this.startGame()) {
+      return false
+    }
+
+    if (this.gameId && this.gameHeartbeatTimerId === null) {
+      this.gameHeartbeatTimerId = window.setInterval(this.sendGameHeartbeat.bind(this), GAME_HEARTBEAT_INTERVAL)
+    }
+    return true
+  }
+
+  async startGame() {
+    let res
+    try {
+      res = (await axios.post('/api/open_live/start_game', {
+        code: this.roomOwnerAuthCode,
+        app_id: 0
+      })).data
+      if (res.code !== 0) {
+        throw Error(`code=${res.code}, message=${res.message}, request_id=${res.request_id}`)
+      }
+    } catch (e) {
+      console.error('startGame failed:', e)
+      return false
+    }
+
+    let data = res.data
+    this.gameId = data.game_info.game_id
+    let websocketInfo = data.websocket_info
+    this.authBody = websocketInfo.auth_body
+    this.hostServerUrlList = websocketInfo.wss_link
+    let anchorInfo = data.anchor_info
+    // this.roomId = anchorInfo.room_id
+    this.roomOwnerUid = anchorInfo.uid
+    return true
+  }
+
+  async endGame() {
+    if (!this.gameId) {
+      return true
+    }
+
+    try {
+      let res = (await axios.post('/api/open_live/end_game', {
+        app_id: 0,
+        game_id: this.gameId
+      })).data
+      if (res.code !== 0) {
+        if (res.code === 7000 || res.code === 7003) {
+          // 项目已经关闭了也算成功
+          return true
+        }
+        throw Error(`code=${res.code}, message=${res.message}, request_id=${res.request_id}`)
+      }
+    } catch (e) {
+      console.error('endGame failed:', e)
+      return false
+    }
+    return true
+  }
+
+  async sendGameHeartbeat() {
+    if (!this.gameId) {
+      return false
+    }
+
+    // 保存一下，防止await之后gameId改变
+    let gameId = this.gameId
+    try {
+      let res = (await axios.post('/api/open_live/game_heartbeat', {
+        game_id: this.gameId
+      })).data
+      if (res.code !== 0) {
+        console.error(`sendGameHeartbeat failed: code=${res.code}, message=${res.message}, request_id=${res.request_id}`)
+
+        if (res.code === 7003 && this.gameId === gameId) {
+          // 项目异常关闭，可能是心跳超时，需要重新开启项目
+          this.needInitRoom = true
+          this.discardWebsocket()
+        }
+
+        return false
+      }
+    } catch (e) {
+      console.error('sendGameHeartbeat failed:', e)
+      return false
+    }
+    return true
+  }
+
+  async onBeforeWsConnect() {
+    // 重连次数太多则重新init_room，保险
+    let reinitPeriod = Math.max(3, (this.hostServerUrlList || []).length)
+    if (this.retryCount > 0 && this.retryCount % reinitPeriod === 0) {
+      this.needInitRoom = true
+    }
+    return super.onBeforeWsConnect()
+  }
+
+  getWsUrl() {
+    return this.hostServerUrlList[this.retryCount % this.hostServerUrlList.length]
+  }
+
+  sendAuth() {
+    this.websocket.send(this.makePacket(this.authBody, base.OP_AUTH))
+  }
+
+  async dmCallback(command) {
+    if (!this.onAddText) {
+      return
+    }
+    let data = command.data
+
+    let authorType
+    if (data.uid === this.roomOwnerUid) {
+      authorType = 3
+    } else if (data.guard_level !== 0) {
+      authorType = 1
+    } else {
+      authorType = 0
+    }
+
+    let emoticon = null
+    if (data.dm_type === 1) {
+      emoticon = data.emoji_img_url
+    }
+
+    data = {
+      avatarUrl: chat.processAvatarUrl(data.uface),
+      timestamp: data.timestamp,
+      authorName: data.uname,
+      authorType: authorType,
+      content: data.msg,
+      privilegeType: data.guard_level,
+      isGiftDanmaku: false,
+      authorLevel: 1,
+      isNewbie: false,
+      isMobileVerified: true,
+      medalLevel: data.fans_medal_wearing_status ? data.fans_medal_level : 0,
+      id: data.msg_id,
+      translation: '',
+      emoticon: emoticon,
+    }
+    this.onAddText(data)
+  }
+
+  sendGiftCallback(command) {
+    if (!this.onAddGift) {
+      return
+    }
+    let data = command.data
+    if (!data.paid) { // 丢人
+      return
+    }
+
+    data = {
+      id: data.msg_id,
+      avatarUrl: chat.processAvatarUrl(data.uface),
+      timestamp: data.timestamp,
+      authorName: data.uname,
+      totalCoin: data.price,
+      giftName: data.gift_name,
+      num: data.gift_num
+    }
+    this.onAddGift(data)
+  }
+
+  async guardCallback(command) {
+    if (!this.onAddMember) {
+      return
+    }
+
+    let data = command.data
+    data = {
+      id: data.msg_id,
+      avatarUrl: chat.processAvatarUrl(data.user_info.uface),
+      timestamp: data.timestamp,
+      authorName: data.user_info.uname,
+      privilegeType: data.guard_level
+    }
+    this.onAddMember(data)
+  }
+
+  superChatCallback(command) {
+    if (!this.onAddSuperChat) {
+      return
+    }
+
+    let data = command.data
+    data = {
+      id: data.message_id.toString(),
+      avatarUrl: chat.processAvatarUrl(data.uface),
+      timestamp: data.start_time,
+      authorName: data.uname,
+      price: data.rmb,
+      content: data.message,
+      translation: ''
+    }
+    this.onAddSuperChat(data)
+  }
+
+  superChatDelCallback(command) {
+    if (!this.onDelSuperChat) {
+      return
+    }
+
+    let ids = []
+    for (let id of command.data.message_ids) {
+      ids.push(id.toString())
+    }
+    this.onDelSuperChat({ ids })
+  }
+}
+
+const CMD_CALLBACK_MAP = {
+  LIVE_OPEN_PLATFORM_DM: ChatClientDirectOpenLive.prototype.dmCallback,
+  LIVE_OPEN_PLATFORM_SEND_GIFT: ChatClientDirectOpenLive.prototype.sendGiftCallback,
+  LIVE_OPEN_PLATFORM_GUARD: ChatClientDirectOpenLive.prototype.guardCallback,
+  LIVE_OPEN_PLATFORM_SUPER_CHAT: ChatClientDirectOpenLive.prototype.superChatCallback,
+  LIVE_OPEN_PLATFORM_SUPER_CHAT_DEL: ChatClientDirectOpenLive.prototype.superChatDelCallback
+}

frontend/src/api/chat/ChatClientDirectWeb.js

@@ -0,0 +1,185 @@
+import axios from 'axios'
+
+import { getUuid4Hex } from '@/utils'
+import * as chat from '.'
+import * as base from './ChatClientOfficialBase'
+import ChatClientOfficialBase from './ChatClientOfficialBase'
+
+export default class ChatClientDirectWeb extends ChatClientOfficialBase {
+  constructor(roomId) {
+    super()
+    this.CMD_CALLBACK_MAP = CMD_CALLBACK_MAP
+
+    // 调用initRoom后初始化，如果失败，使用这里的默认值
+    this.roomId = roomId
+    this.roomOwnerUid = -1
+    this.hostServerList = [
+      { host: "broadcastlv.chat.bilibili.com", port: 2243, wss_port: 443, ws_port: 2244 }
+    ]
+  }
+
+  async initRoom() {
+    let res
+    try {
+      res = (await axios.get('/api/room_info', { params: {
+        roomId: this.roomId
+      } })).data
+    } catch {
+      return true
+    }
+    this.roomId = res.roomId
+    this.roomOwnerUid = res.ownerUid
+    if (res.hostServerList.length !== 0) {
+      this.hostServerList = res.hostServerList
+    }
+    return true
+  }
+
+  async onBeforeWsConnect() {
+    // 重连次数太多则重新init_room，保险
+    let reinitPeriod = Math.max(3, (this.hostServerList || []).length)
+    if (this.retryCount > 0 && this.retryCount % reinitPeriod === 0) {
+      this.needInitRoom = true
+    }
+    return super.onBeforeWsConnect()
+  }
+
+  getWsUrl() {
+    let hostServer = this.hostServerList[this.retryCount % this.hostServerList.length]
+    return `wss://${hostServer.host}:${hostServer.wss_port}/sub`
+  }
+
+  sendAuth() {
+    let authParams = {
+      uid: 0,
+      roomid: this.roomId,
+      protover: 3,
+      platform: 'web',
+      type: 2,
+      buvid: '',
+    }
+    this.websocket.send(this.makePacket(authParams, base.OP_AUTH))
+  }
+
+  async danmuMsgCallback(command) {
+    if (!this.onAddText) {
+      return
+    }
+    let info = command.info
+
+    let roomId, medalLevel
+    if (info[3]) {
+      roomId = info[3][3]
+      medalLevel = info[3][0]
+    } else {
+      roomId = medalLevel = 0
+    }
+
+    let uid = info[2][0]
+    let isAdmin = info[2][2]
+    let privilegeType = info[7]
+    let authorType
+    if (uid === this.roomOwnerUid) {
+      authorType = 3
+    } else if (isAdmin) {
+      authorType = 2
+    } else if (privilegeType !== 0) {
+      authorType = 1
+    } else {
+      authorType = 0
+    }
+
+    let data = {
+      avatarUrl: await chat.getAvatarUrl(uid),
+      timestamp: info[0][4] / 1000,
+      authorName: info[2][1],
+      authorType: authorType,
+      content: info[1],
+      privilegeType: privilegeType,
+      isGiftDanmaku: Boolean(info[0][9]),
+      authorLevel: info[4][0],
+      isNewbie: info[2][5] < 10000,
+      isMobileVerified: Boolean(info[2][6]),
+      medalLevel: roomId === this.roomId ? medalLevel : 0,
+      id: getUuid4Hex(),
+      translation: '',
+      emoticon: info[0][13].url || null,
+    }
+    this.onAddText(data)
+  }
+
+  sendGiftCallback(command) {
+    if (!this.onAddGift) {
+      return
+    }
+    let data = command.data
+    if (data.coin_type !== 'gold') { // 丢人
+      return
+    }
+
+    data = {
+      id: getUuid4Hex(),
+      avatarUrl: chat.processAvatarUrl(data.face),
+      timestamp: data.timestamp,
+      authorName: data.uname,
+      totalCoin: data.total_coin,
+      giftName: data.giftName,
+      num: data.num
+    }
+    this.onAddGift(data)
+  }
+
+  async guardBuyCallback(command) {
+    if (!this.onAddMember) {
+      return
+    }
+
+    let data = command.data
+    data = {
+      id: getUuid4Hex(),
+      avatarUrl: await chat.getAvatarUrl(data.uid),
+      timestamp: data.start_time,
+      authorName: data.username,
+      privilegeType: data.guard_level
+    }
+    this.onAddMember(data)
+  }
+
+  superChatMessageCallback(command) {
+    if (!this.onAddSuperChat) {
+      return
+    }
+
+    let data = command.data
+    data = {
+      id: data.id.toString(),
+      avatarUrl: chat.processAvatarUrl(data.user_info.face),
+      timestamp: data.start_time,
+      authorName: data.user_info.uname,
+      price: data.price,
+      content: data.message,
+      translation: ''
+    }
+    this.onAddSuperChat(data)
+  }
+
+  superChatMessageDeleteCallback(command) {
+    if (!this.onDelSuperChat) {
+      return
+    }
+
+    let ids = []
+    for (let id of command.data.ids) {
+      ids.push(id.toString())
+    }
+    this.onDelSuperChat({ ids })
+  }
+}
+
+const CMD_CALLBACK_MAP = {
+  DANMU_MSG: ChatClientDirectWeb.prototype.danmuMsgCallback,
+  SEND_GIFT: ChatClientDirectWeb.prototype.sendGiftCallback,
+  GUARD_BUY: ChatClientDirectWeb.prototype.guardBuyCallback,
+  SUPER_CHAT_MESSAGE: ChatClientDirectWeb.prototype.superChatMessageCallback,
+  SUPER_CHAT_MESSAGE_DELETE: ChatClientDirectWeb.prototype.superChatMessageDeleteCallback
+}

frontend/src/api/chat/ChatClientOfficialBase/index.js

@@ -0,0 +1,298 @@
+import { BrotliDecode } from './brotli_decode'
+import { inflate } from 'pako'
+
+const HEADER_SIZE = 16
+
+export const WS_BODY_PROTOCOL_VERSION_NORMAL = 0
+export const WS_BODY_PROTOCOL_VERSION_HEARTBEAT = 1
+export const WS_BODY_PROTOCOL_VERSION_DEFLATE = 2
+export const WS_BODY_PROTOCOL_VERSION_BROTLI = 3
+
+export const OP_HANDSHAKE = 0
+export const OP_HANDSHAKE_REPLY = 1
+export const OP_HEARTBEAT = 2
+export const OP_HEARTBEAT_REPLY = 3
+export const OP_SEND_MSG = 4
+export const OP_SEND_MSG_REPLY = 5
+export const OP_DISCONNECT_REPLY = 6
+export const OP_AUTH = 7
+export const OP_AUTH_REPLY = 8
+export const OP_RAW = 9
+export const OP_PROTO_READY = 10
+export const OP_PROTO_FINISH = 11
+export const OP_CHANGE_ROOM = 12
+export const OP_CHANGE_ROOM_REPLY = 13
+export const OP_REGISTER = 14
+export const OP_REGISTER_REPLY = 15
+export const OP_UNREGISTER = 16
+export const OP_UNREGISTER_REPLY = 17
+// B站业务自定义OP
+// export const MinBusinessOp = 1000
+// export const MaxBusinessOp = 10000
+
+export const AUTH_REPLY_CODE_OK = 0
+export const AUTH_REPLY_CODE_TOKEN_ERROR = -101
+
+const HEARTBEAT_INTERVAL = 10 * 1000
+const RECEIVE_TIMEOUT = HEARTBEAT_INTERVAL + (5 * 1000)
+
+let textEncoder = new TextEncoder()
+let textDecoder = new TextDecoder()
+
+export default class ChatClientOfficialBase {
+  constructor() {
+    this.CMD_CALLBACK_MAP = {}
+
+    this.onAddText = null
+    this.onAddGift = null
+    this.onAddMember = null
+    this.onAddSuperChat = null
+    this.onDelSuperChat = null
+    this.onUpdateTranslation = null
+
+    this.needInitRoom = true
+    this.websocket = null
+    this.retryCount = 0
+    this.isDestroying = false
+    this.heartbeatTimerId = null
+    this.receiveTimeoutTimerId = null
+  }
+
+  start() {
+    this.wsConnect()
+  }
+
+  stop() {
+    this.isDestroying = true
+    if (this.websocket) {
+      this.websocket.close()
+    }
+  }
+
+  async initRoom() {
+    throw Error('Not implemented')
+  }
+
+  makePacket(data, operation) {
+    let body
+    if (typeof data === 'object') {
+      body = textEncoder.encode(JSON.stringify(data))
+    } else { // string
+      body = textEncoder.encode(data)
+    }
+    let header = new ArrayBuffer(HEADER_SIZE)
+    let headerView = new DataView(header)
+    headerView.setUint32(0, HEADER_SIZE + body.byteLength)   // pack_len
+    headerView.setUint16(4, HEADER_SIZE)                     // raw_header_size
+    headerView.setUint16(6, 1)                               // ver
+    headerView.setUint32(8, operation)                       // operation
+    headerView.setUint32(12, 1)                              // seq_id
+    return new Blob([header, body])
+  }
+
+  sendAuth() {
+    throw Error('Not implemented')
+  }
+
+  async wsConnect() {
+    if (this.isDestroying) {
+      return
+    }
+
+    await this.onBeforeWsConnect()
+    if (this.isDestroying) {
+      return
+    }
+
+    this.websocket = new WebSocket(this.getWsUrl())
+    this.websocket.binaryType = 'arraybuffer'
+    this.websocket.onopen = this.onWsOpen.bind(this)
+    this.websocket.onclose = this.onWsClose.bind(this)
+    this.websocket.onmessage = this.onWsMessage.bind(this)
+  }
+
+  async onBeforeWsConnect() {
+    if (!this.needInitRoom) {
+      return
+    }
+
+    if (!await this.initRoom()) {
+      this.onWsClose()
+      throw Error('initRoom failed')
+    }
+    this.needInitRoom = false
+  }
+
+  getWsUrl() {
+    throw Error('Not implemented')
+  }
+
+  onWsOpen() {
+    this.sendAuth()
+    this.heartbeatTimerId = window.setInterval(this.sendHeartbeat.bind(this), HEARTBEAT_INTERVAL)
+    this.refreshReceiveTimeoutTimer()
+  }
+
+  sendHeartbeat() {
+    this.websocket.send(this.makePacket({}, OP_HEARTBEAT))
+  }
+
+  refreshReceiveTimeoutTimer() {
+    if (this.receiveTimeoutTimerId) {
+      window.clearTimeout(this.receiveTimeoutTimerId)
+    }
+    this.receiveTimeoutTimerId = window.setTimeout(this.onReceiveTimeout.bind(this), RECEIVE_TIMEOUT)
+  }
+
+  onReceiveTimeout() {
+    console.warn('接收消息超时')
+    this.discardWebsocket()
+  }
+
+  discardWebsocket() {
+    if (this.receiveTimeoutTimerId) {
+      window.clearTimeout(this.receiveTimeoutTimerId)
+      this.receiveTimeoutTimerId = null
+    }
+
+    // 直接丢弃阻塞的websocket，不等onclose回调了
+    this.websocket.onopen = this.websocket.onclose = this.websocket.onmessage = null
+    this.websocket.close()
+    this.onWsClose()
+  }
+
+  onWsClose() {
+    this.websocket = null
+    if (this.heartbeatTimerId) {
+      window.clearInterval(this.heartbeatTimerId)
+      this.heartbeatTimerId = null
+    }
+    if (this.receiveTimeoutTimerId) {
+      window.clearTimeout(this.receiveTimeoutTimerId)
+      this.receiveTimeoutTimerId = null
+    }
+
+    if (this.isDestroying) {
+      return
+    }
+    this.retryCount++
+    console.warn('掉线重连中', this.retryCount)
+    window.setTimeout(this.wsConnect.bind(this), 1000)
+  }
+
+  onWsMessage(event) {
+    if (!(event.data instanceof ArrayBuffer)) {
+      console.warn('未知的websocket消息类型，data=', event.data)
+      return
+    }
+
+    let data = new Uint8Array(event.data)
+    this.parseWsMessage(data)
+
+    // 至少成功处理1条消息
+    this.retryCount = 0
+  }
+
+  parseWsMessage(data) {
+    let offset = 0
+    let dataView = new DataView(data.buffer)
+    let packLen = dataView.getUint32(0)
+    let rawHeaderSize = dataView.getUint16(4)
+    // let ver = dataView.getUint16(6)
+    let operation = dataView.getUint32(8)
+    // let seqId = dataView.getUint32(12)
+
+    switch (operation) {
+    case OP_AUTH_REPLY:
+    case OP_SEND_MSG_REPLY: {
+      // 业务消息，可能有多个包一起发，需要分包
+      while (true) { // eslint-disable-line no-constant-condition
+        let body = new Uint8Array(data.buffer, offset + rawHeaderSize, packLen - rawHeaderSize)
+        this.parseBusinessMessage(dataView, body)
+
+        offset += packLen
+        if (offset >= data.byteLength) {
+          break
+        }
+
+        dataView = new DataView(data.buffer, offset)
+        packLen = dataView.getUint32(0)
+        rawHeaderSize = dataView.getUint16(4)
+      }
+      break
+    }
+    case OP_HEARTBEAT_REPLY: {
+      // 服务器心跳包，包含人气值，这里没用
+      this.refreshReceiveTimeoutTimer()
+      break
+    }
+    default: {
+      // 未知消息
+      let body = new Uint8Array(data.buffer, offset + rawHeaderSize, packLen - rawHeaderSize)
+      console.warn('未知包类型，operation=', operation, dataView, body)
+      break
+    }
+    }
+  }
+
+  parseBusinessMessage(dataView, body) {
+    let ver = dataView.getUint16(6)
+    let operation = dataView.getUint32(8)
+
+    switch (operation) {
+    case OP_SEND_MSG_REPLY: {
+      // 业务消息
+      if (ver == WS_BODY_PROTOCOL_VERSION_BROTLI) {
+        // 压缩过的先解压
+        body = BrotliDecode(body)
+        this.parseWsMessage(body)
+      } else if (ver == WS_BODY_PROTOCOL_VERSION_DEFLATE) {
+        // web端已经不用zlib压缩了，但是开放平台会用
+        body = inflate(body)
+        this.parseWsMessage(body)
+      } else {
+        // 没压缩过的直接反序列化
+        if (body.length !== 0) {
+          try {
+            body = JSON.parse(textDecoder.decode(body))
+            this.handlerCommand(body)
+          } catch (e) {
+            console.error('body=', body)
+            throw e
+          }
+        }
+      }
+      break
+    }
+    case OP_AUTH_REPLY: {
+      // 认证响应
+      body = JSON.parse(textDecoder.decode(body))
+      if (body.code !== AUTH_REPLY_CODE_OK) {
+        console.error('认证响应错误，body=', body)
+        this.needInitRoom = true
+        this.discardWebsocket()
+        throw new Error('认证响应错误')
+      }
+      this.sendHeartbeat()
+      break
+    }
+    default: {
+      // 未知消息
+      console.warn('未知包类型，operation=', operation, dataView, body)
+      break
+    }
+    }
+  }
+
+  handlerCommand(command) {
+    let cmd = command.cmd || ''
+    let pos = cmd.indexOf(':')
+    if (pos != -1) {
+      cmd = cmd.substr(0, pos)
+    }
+    let callback = this.CMD_CALLBACK_MAP[cmd]
+    if (callback) {
+      callback.call(this, command)
+    }
+  }
+}

frontend/src/api/chat/ChatClientRelay.js

@@ -7,12 +7,14 @@ const COMMAND_ADD_SUPER_CHAT = 5
 const COMMAND_DEL_SUPER_CHAT = 6
 const COMMAND_UPDATE_TRANSLATION = 7
 
-const HEARTBEAT_INTERVAL = 10 * 1000
-const RECEIVE_TIMEOUT = HEARTBEAT_INTERVAL + 5 * 1000
+// const CONTENT_TYPE_TEXT = 0
+const CONTENT_TYPE_EMOTICON = 1
+
+const RECEIVE_TIMEOUT = 15 * 1000
 
 export default class ChatClientRelay {
-  constructor (roomId, autoTranslate) {
-    this.roomId = roomId
+  constructor(roomKey, autoTranslate) {
+    this.roomKey = roomKey
     this.autoTranslate = autoTranslate
 
     this.onAddText = null
@@ -25,56 +27,46 @@ export default class ChatClientRelay {
     this.websocket = null
     this.retryCount = 0
     this.isDestroying = false
-    this.heartbeatTimerId = null
     this.receiveTimeoutTimerId = null
   }
 
-  start () {
+  start() {
     this.wsConnect()
   }
 
-  stop () {
+  stop() {
     this.isDestroying = true
     if (this.websocket) {
       this.websocket.close()
     }
   }
 
-  wsConnect () {
+  wsConnect() {
     if (this.isDestroying) {
       return
     }
     const protocol = window.location.protocol === 'https:' ? 'wss' : 'ws'
-    // 开发时使用localhost:12450
-    const host = process.env.NODE_ENV === 'development' ? 'localhost:12450' : window.location.host
-    const url = `${protocol}://${host}/api/chat`
+    const url = `${protocol}://${window.location.host}/api/chat`
     this.websocket = new WebSocket(url)
     this.websocket.onopen = this.onWsOpen.bind(this)
     this.websocket.onclose = this.onWsClose.bind(this)
     this.websocket.onmessage = this.onWsMessage.bind(this)
   }
 
-  onWsOpen () {
+  onWsOpen() {
     this.retryCount = 0
     this.websocket.send(JSON.stringify({
       cmd: COMMAND_JOIN_ROOM,
       data: {
-        roomId: this.roomId,
+        roomKey: this.roomKey,
         config: {
           autoTranslate: this.autoTranslate
         }
       }
     }))
-    this.heartbeatTimerId = window.setInterval(this.sendHeartbeat.bind(this), HEARTBEAT_INTERVAL)
     this.refreshReceiveTimeoutTimer()
   }
 
-  sendHeartbeat () {
-    this.websocket.send(JSON.stringify({
-      cmd: COMMAND_HEARTBEAT
-    }))
-  }
-
   refreshReceiveTimeoutTimer() {
     if (this.receiveTimeoutTimerId) {
       window.clearTimeout(this.receiveTimeoutTimerId)
@@ -83,7 +75,7 @@ export default class ChatClientRelay {
   }
 
   onReceiveTimeout() {
-    window.console.warn('接收消息超时')
+    console.warn('接收消息超时')
     this.receiveTimeoutTimerId = null
 
     // 直接丢弃阻塞的websocket，不等onclose回调了
@@ -92,12 +84,8 @@ export default class ChatClientRelay {
     this.onWsClose()
   }
 
-  onWsClose () {
+  onWsClose() {
     this.websocket = null
-    if (this.heartbeatTimerId) {
-      window.clearInterval(this.heartbeatTimerId)
-      this.heartbeatTimerId = null
-    }
     if (this.receiveTimeoutTimerId) {
       window.clearTimeout(this.receiveTimeoutTimerId)
       this.receiveTimeoutTimerId = null
@@ -106,22 +94,34 @@ export default class ChatClientRelay {
     if (this.isDestroying) {
       return
     }
-    window.console.warn(`掉线重连中${++this.retryCount}`)
+    console.warn(`掉线重连中${++this.retryCount}`)
     window.setTimeout(this.wsConnect.bind(this), 1000)
   }
 
-  onWsMessage (event) {
-    this.refreshReceiveTimeoutTimer()
-
-    let {cmd, data} = JSON.parse(event.data)
+  onWsMessage(event) {
+    let { cmd, data } = JSON.parse(event.data)
     switch (cmd) {
     case COMMAND_HEARTBEAT: {
+      this.refreshReceiveTimeoutTimer()
+
+      // 不能由定时器触发发心跳包，因为浏览器会把不活动页面的定时器调到1分钟以上
+      this.websocket.send(JSON.stringify({
+        cmd: COMMAND_HEARTBEAT
+      }))
       break
     }
     case COMMAND_ADD_TEXT: {
       if (!this.onAddText) {
         break
       }
+
+      let emoticon = null
+      let contentType = data[13]
+      let contentTypeParams = data[14]
+      if (contentType === CONTENT_TYPE_EMOTICON) {
+        emoticon = contentTypeParams[0]
+      }
+
       data = {
         avatarUrl: data[0],
         timestamp: data[1],
@@ -129,13 +129,14 @@ export default class ChatClientRelay {
         authorType: data[3],
         content: data[4],
         privilegeType: data[5],
-        isGiftDanmaku: !!data[6],
+        isGiftDanmaku: Boolean(data[6]),
         authorLevel: data[7],
-        isNewbie: !!data[8],
-        isMobileVerified: !!data[9],
+        isNewbie: Boolean(data[8]),
+        isMobileVerified: Boolean(data[9]),
         medalLevel: data[10],
         id: data[11],
-        translation: data[12]
+        translation: data[12],
+        emoticon: emoticon,
       }
       this.onAddText(data)
       break

frontend/src/api/chat/ChatClientTest.js

@@ -1,49 +1,94 @@
-import {getUuid4Hex} from '@/utils'
+import { getUuid4Hex } from '@/utils'
 import * as constants from '@/components/ChatRenderer/constants'
-import * as avatar from './avatar'
+import * as chat from '.'
 
 const NAMES = [
-  'xfgryujk', 'Simon', 'Il Harper', 'Kinori', 'shugen', 'yuyuyzl', '3Shain', '光羊', '黑炎', 'Misty', '孤梦星影',
-  'ジョナサン・ジョースター', 'ジョセフ・ジョースター', 'ディオ・ブランドー', '空條承太郎', '博丽灵梦', '雾雨魔理沙',
-  'Rick Astley'
+  '光羊',
+  '黑炎',
+  '孤梦星影',
+  '博丽灵梦',
+  '雾雨魔理沙',
+  '空條承太郎',
+  'ディオ・ブランドー',
+  'ジョセフ・ジョースター',
+  'ジョナサン・ジョースター',
+  'Simon',
+  'Misty',
+  'Kinori',
+  'shugen',
+  '3Shain',
+  'yuyuyzl',
+  'xfgryujk',
+  'Il Harper',
+  'Rick Astley',
 ]
 
 const CONTENTS = [
-  '草', 'kksk', '8888888888', '888888888888888888888888888888', '老板大气，老板身体健康',
-  'The quick brown fox jumps over the lazy dog', "I can eat glass, it doesn't hurt me",
-  '我不做人了，JOJO', '無駄無駄無駄無駄無駄無駄無駄無駄', '欧啦欧啦欧啦欧啦欧啦欧啦欧啦欧啦', '逃げるんだよォ！',
-  '嚯，朝我走过来了吗，没有选择逃跑而是主动接近我么', '不要停下来啊', '已经没有什么好怕的了',
-  'I am the bone of my sword. Steel is my body, and fire is my blood.', '言いたいことがあるんだよ！',
-  '我忘不掉夏小姐了。如果不是知道了夏小姐，说不定我已经对这个世界没有留恋了', '迷えば、敗れる',
-  'Farewell, ashen one. May the flame guide thee', '竜神の剣を喰らえ！', '竜が我が敌を喰らう！',
-  '有一说一，这件事大家懂的都懂，不懂的，说了你也不明白，不如不说', '让我看看', '我柜子动了，我不玩了'
+  '草',
+  '让我看看',
+  '不要停下来啊',
+  '我不做人了，JOJO',
+  '已经没有什么好怕的了',
+  '我柜子动了，我不玩了',
+  '老板大气，老板身体健康',
+  '我醉提酒游寒山，爽滑慢舔',
+  '無駄無駄無駄無駄無駄無駄無駄無駄',
+  '欧啦欧啦欧啦欧啦欧啦欧啦欧啦欧啦',
+  '所有没好全部康复呀，我的癌也全部康复呀',
+  '嚯，朝我走过来了吗，没有选择逃跑而是主动接近我么',
+  '有一说一，这件事大家懂的都懂，不懂的，说了你也不明白，不如不说',
+  '如来来了吗？如来嘛~他真来了吗？如~来~到底来没来？如来~如来他真来了吗？如来~你看看，来没来？如~来~',
+  '迷えば、敗れる',
+  '逃げるんだよォ！',
+  '竜神の剣を喰らえ！',
+  '竜が我が敌を喰らう！',
+  '言いたいことがあるんだよ！',
+  '知らず知らず隠してた 本当の声を響かせてよほら',
+  'kksk',
+  '8888888888',
+  'Never gonna give you up',
+  'Never gonna let you down',
+  '888888888888888888888888888888',
+  'I am the storm that is approaching',
+  "I can eat glass, it doesn't hurt me",
+  'The quick brown fox jumps over the lazy dog',
+  'Farewell, ashen one. May the flame guide thee',
+  'I am the bone of my sword. Steel is my body, and fire is my blood.',
+]
+
+const EMOTICONS = [
+  '/static/img/emoticons/233.png',
+  '/static/img/emoticons/miaoa.png',
+  '/static/img/emoticons/lipu.png'
 ]
 
 const AUTHOR_TYPES = [
-  {weight: 10, value: constants.AUTHRO_TYPE_NORMAL},
-  {weight: 5, value: constants.AUTHRO_TYPE_MEMBER},
-  {weight: 2, value: constants.AUTHRO_TYPE_ADMIN},
-  {weight: 1, value: constants.AUTHRO_TYPE_OWNER}
+  { weight: 10, value: constants.AUTHRO_TYPE_NORMAL },
+  { weight: 5, value: constants.AUTHRO_TYPE_MEMBER },
+  { weight: 2, value: constants.AUTHRO_TYPE_ADMIN },
+  { weight: 1, value: constants.AUTHRO_TYPE_OWNER }
 ]
 
-function randGuardInfo () {
+function randGuardInfo() {
   let authorType = randomChoose(AUTHOR_TYPES)
   let privilegeType
-  if (authorType === constants.AUTHRO_TYPE_MEMBER || authorType === constants.AUTHRO_TYPE_ADMIN) {
+  if (authorType === constants.AUTHRO_TYPE_MEMBER) {
     privilegeType = randInt(1, 3)
+  } else if (authorType === constants.AUTHRO_TYPE_ADMIN) {
+    privilegeType = randInt(0, 3)
   } else {
     privilegeType = 0
   }
-  return {authorType, privilegeType}
+  return { authorType, privilegeType }
 }
 
 const GIFT_INFO_LIST = [
-  {giftName: 'B坷垃', totalCoin: 9900},
-  {giftName: '礼花', totalCoin: 28000},
-  {giftName: '花式夸夸', totalCoin: 39000},
-  {giftName: '天空之翼', totalCoin: 100000},
-  {giftName: '摩天大楼', totalCoin: 450000},
-  {giftName: '小电视飞船', totalCoin: 1245000}
+  { giftName: 'B坷垃', totalCoin: 9900 },
+  { giftName: '礼花', totalCoin: 28000 },
+  { giftName: '花式夸夸', totalCoin: 39000 },
+  { giftName: '天空之翼', totalCoin: 100000 },
+  { giftName: '摩天大楼', totalCoin: 450000 },
+  { giftName: '小电视飞船', totalCoin: 1245000 }
 ]
 
 const SC_PRICES = [
@@ -59,17 +104,42 @@ const MESSAGE_GENERATORS = [
         type: constants.MESSAGE_TYPE_TEXT,
         message: {
           ...randGuardInfo(),
-          avatarUrl: avatar.DEFAULT_AVATAR_URL,
+          avatarUrl: chat.DEFAULT_AVATAR_URL,
           timestamp: new Date().getTime() / 1000,
           authorName: randomChoose(NAMES),
           content: randomChoose(CONTENTS),
           isGiftDanmaku: randInt(1, 10) <= 1,
           authorLevel: randInt(0, 60),
-          isNewbie: randInt(1, 10) <= 9,
+          isNewbie: randInt(1, 10) <= 1,
           isMobileVerified: randInt(1, 10) <= 9,
           medalLevel: randInt(0, 40),
           id: getUuid4Hex(),
-          translation: ''
+          translation: '',
+          emoticon: null,
+        }
+      }
+    }
+  },
+  // 表情
+  {
+    weight: 5,
+    value() {
+      return {
+        type: constants.MESSAGE_TYPE_TEXT,
+        message: {
+          ...randGuardInfo(),
+          avatarUrl: chat.DEFAULT_AVATAR_URL,
+          timestamp: new Date().getTime() / 1000,
+          authorName: randomChoose(NAMES),
+          content: '',
+          isGiftDanmaku: false,
+          authorLevel: randInt(0, 60),
+          isNewbie: randInt(1, 10) <= 1,
+          isMobileVerified: randInt(1, 10) <= 9,
+          medalLevel: randInt(0, 40),
+          id: getUuid4Hex(),
+          translation: '',
+          emoticon: randomChoose(EMOTICONS),
         }
       }
     }
@@ -83,7 +153,7 @@ const MESSAGE_GENERATORS = [
         message: {
           ...randomChoose(GIFT_INFO_LIST),
           id: getUuid4Hex(),
-          avatarUrl: avatar.DEFAULT_AVATAR_URL,
+          avatarUrl: chat.DEFAULT_AVATAR_URL,
           timestamp: new Date().getTime() / 1000,
           authorName: randomChoose(NAMES),
           num: 1
@@ -99,7 +169,7 @@ const MESSAGE_GENERATORS = [
         type: constants.MESSAGE_TYPE_SUPER_CHAT,
         message: {
           id: getUuid4Hex(),
-          avatarUrl: avatar.DEFAULT_AVATAR_URL,
+          avatarUrl: chat.DEFAULT_AVATAR_URL,
           timestamp: new Date().getTime() / 1000,
           authorName: randomChoose(NAMES),
           price: randomChoose(SC_PRICES),
@@ -117,7 +187,7 @@ const MESSAGE_GENERATORS = [
         type: constants.MESSAGE_TYPE_MEMBER,
         message: {
           id: getUuid4Hex(),
-          avatarUrl: avatar.DEFAULT_AVATAR_URL,
+          avatarUrl: chat.DEFAULT_AVATAR_URL,
           timestamp: new Date().getTime() / 1000,
           authorName: randomChoose(NAMES),
           privilegeType: randInt(1, 3)
@@ -127,7 +197,7 @@ const MESSAGE_GENERATORS = [
   }
 ]
 
-function randomChoose (nodes) {
+function randomChoose(nodes) {
   if (nodes.length === 0) {
     return null
   }
@@ -155,15 +225,12 @@ function randomChoose (nodes) {
   return null
 }
 
-function randInt (min, max) {
-  return Math.floor(min + (max - min + 1) * Math.random())
+function randInt(min, max) {
+  return Math.floor(min + ((max - min + 1) * Math.random()))
 }
 
 export default class ChatClientTest {
-  constructor () {
-    this.minSleepTime = 800
-    this.maxSleepTime = 1200
-
+  constructor() {
     this.onAddText = null
     this.onAddGift = null
     this.onAddMember = null
@@ -174,25 +241,32 @@ export default class ChatClientTest {
     this.timerId = null
   }
 
-  start () {
+  start() {
     this.refreshTimer()
   }
 
-  stop () {
+  stop() {
     if (this.timerId) {
       window.clearTimeout(this.timerId)
       this.timerId = null
     }
   }
 
-  refreshTimer () {
-    this.timerId = window.setTimeout(this.onTimeout.bind(this), randInt(this.minSleepTime, this.maxSleepTime))
+  refreshTimer() {
+    // 模仿B站的消息间隔模式
+    let sleepTime
+    if (randInt(0, 4) == 0) {
+      sleepTime = randInt(1000, 2000)
+    } else {
+      sleepTime = randInt(0, 400)
+    }
+    this.timerId = window.setTimeout(this.onTimeout.bind(this), sleepTime)
   }
 
-  onTimeout () {
+  onTimeout() {
     this.refreshTimer()
 
-    let {type, message} = randomChoose(MESSAGE_GENERATORS)()
+    let { type, message } = randomChoose(MESSAGE_GENERATORS)()
     switch (type) {
     case constants.MESSAGE_TYPE_TEXT:
       this.onAddText(message)

frontend/src/api/chat/index.js

@@ -2,27 +2,33 @@ import axios from 'axios'
 
 export const DEFAULT_AVATAR_URL = '//static.hdslb.com/images/member/noface.gif'
 
-export function processAvatarUrl (avatarUrl) {
+export function processAvatarUrl(avatarUrl) {
   // 去掉协议，兼容HTTP、HTTPS
   let m = avatarUrl.match(/(?:https?:)?(.*)/)
   if (m) {
     avatarUrl = m[1]
   }
-  // 缩小图片加快传输
-  if (!avatarUrl.endsWith('noface.gif')) {
-    avatarUrl += '@48w_48h'
-  }
   return avatarUrl
 }
 
-export async function getAvatarUrl (uid) {
+export async function getAvatarUrl(uid) {
   let res
   try {
-    res = (await axios.get('/api/avatar_url', {params: {
+    res = (await axios.get('/api/avatar_url', { params: {
       uid: uid
-    }})).data
+    } })).data
   } catch {
     return DEFAULT_AVATAR_URL
   }
   return res.avatarUrl
 }
+
+export async function getTextEmoticons() {
+  let res
+  try {
+    res = (await axios.get('/api/text_emoticon_mappings')).data
+  } catch {
+    return []
+  }
+  return res.textEmoticons
+}

frontend/src/api/chatConfig.js

@@ -1,4 +1,6 @@
-import {mergeConfig} from '@/utils'
+import _ from 'lodash'
+
+import { mergeConfig } from '@/utils'
 
 export const DEFAULT_CONFIG = {
   minGiftPrice: 7, // $1
@@ -19,18 +21,48 @@ export const DEFAULT_CONFIG = {
 
   relayMessagesByServer: false,
   autoTranslate: false,
-  giftUsernamePronunciation: ''
+  giftUsernamePronunciation: '',
+
+  emoticons: [] // [{ keyword: '', url: '' }, ...]
 }
 
-export function setLocalConfig (config) {
+export function deepCloneDefaultConfig() {
+  return _.cloneDeep(DEFAULT_CONFIG)
+}
+
+export function setLocalConfig(config) {
   config = mergeConfig(config, DEFAULT_CONFIG)
   window.localStorage.config = JSON.stringify(config)
 }
 
-export function getLocalConfig () {
+export function getLocalConfig() {
   try {
-    return mergeConfig(JSON.parse(window.localStorage.config), DEFAULT_CONFIG)
+    let config = JSON.parse(window.localStorage.config)
+    config = mergeConfig(config, deepCloneDefaultConfig())
+    sanitizeConfig(config)
+    return config
   } catch {
-    return {...DEFAULT_CONFIG}
+    return deepCloneDefaultConfig()
   }
 }
+
+export function sanitizeConfig(config) {
+  let newEmoticons = []
+  if (config.emoticons instanceof Array) {
+    for (let emoticon of config.emoticons) {
+      try {
+        let newEmoticon = {
+          keyword: emoticon.keyword,
+          url: emoticon.url
+        }
+        if ((typeof newEmoticon.keyword !== 'string') || (typeof newEmoticon.url !== 'string')) {
+          continue
+        }
+        newEmoticons.push(newEmoticon)
+      } catch {
+        continue
+      }
+    }
+  }
+  config.emoticons = newEmoticons
+}

frontend/src/api/main.js

@@ -0,0 +1,11 @@
+import axios from 'axios'
+
+export async function getServerInfo() {
+  return (await axios.get('/api/server_info')).data
+}
+
+export async function uploadEmoticon(file) {
+  let body = new FormData()
+  body.set('file', file)
+  return (await axios.post('/api/emoticon', body)).data
+}

frontend/src/assets/css/youtube/yt-live-chat-membership-item-renderer.css

@@ -314,7 +314,9 @@ yt-live-chat-membership-item-renderer[dashboard-money-feed] #content.yt-live-cha
 }
 
 #content.yt-live-chat-membership-item-renderer img.yt-live-chat-membership-item-renderer {
-  width: var(--yt-live-chat-emoji-size);
+  /* B站表情有不是正方形的 */
+  /* width: var(--yt-live-chat-emoji-size); */
+  width: auto;
   height: var(--yt-live-chat-emoji-size);
 
   margin: -1px 2px 1px 2px;

frontend/src/assets/css/youtube/yt-live-chat-paid-message-renderer.css

@@ -312,7 +312,9 @@ yt-live-chat-paid-message-renderer[allow-animations] #content.yt-live-chat-paid-
 }
 
 #content.yt-live-chat-paid-message-renderer img.yt-live-chat-paid-message-renderer {
-  width: var(--yt-live-chat-emoji-size);
+  /* B站表情有不是正方形的 */
+  /* width: var(--yt-live-chat-emoji-size); */
+  width: auto;
   height: var(--yt-live-chat-emoji-size);
   margin: -1px 2px 1px 2px;
   vertical-align: middle;

frontend/src/assets/css/youtube/yt-live-chat-text-message-renderer.css

@@ -159,7 +159,9 @@ yt-live-chat-author-chip.yt-live-chat-text-message-renderer {
 }
 
 #message.yt-live-chat-text-message-renderer .emoji.yt-live-chat-text-message-renderer {
-  width: var(--yt-live-chat-emoji-size);
+  /* B站表情有不是正方形的 */
+  /* width: var(--yt-live-chat-emoji-size); */
+  width: auto;
   height: var(--yt-live-chat-emoji-size);
   margin: -1px 2px 1px 2px;
   vertical-align: middle;

frontend/src/components/ChatRenderer/AuthorBadge.vue

@@ -1,5 +1,5 @@
 <template>
-  <yt-live-chat-author-badge-renderer :type="authorTypeText" v-if="isAdmin || privilegeType > 0">
+  <yt-live-chat-author-badge-renderer :type="authorTypeText">
     <el-tooltip :content="readableAuthorTypeText" placement="top">
       <div id="image" class="style-scope yt-live-chat-author-badge-renderer">
         <yt-icon v-if="isAdmin" class="style-scope yt-live-chat-author-badge-renderer">
@@ -14,7 +14,8 @@
           </svg>
         </yt-icon>
         <img v-else :src="`/static/img/icons/guard-level-${privilegeType}.png`"
-          class="style-scope yt-live-chat-author-badge-renderer" :alt="readableAuthorTypeText">
+          class="style-scope yt-live-chat-author-badge-renderer" :alt="readableAuthorTypeText"
+        >
       </div>
     </el-tooltip>
   </yt-live-chat-author-badge-renderer>
@@ -38,9 +39,9 @@ export default {
     },
     readableAuthorTypeText() {
       if (this.isAdmin) {
-        return '管理员'
+        return this.$t('chat.moderator')
       }
-      return constants.GUARD_LEVEL_TO_TEXT[this.privilegeType]
+      return constants.getShowGuardLevelText(this.privilegeType)
     }
   }
 }

frontend/src/components/ChatRenderer/AuthorChip.vue

@@ -0,0 +1,54 @@
+<template>
+  <yt-live-chat-author-chip>
+    <span id="author-name" dir="auto" class="style-scope yt-live-chat-author-chip" :class="{ member: isInMemberMessage }"
+      :type="authorTypeText"
+    >
+      <template>{{ authorName }}</template>
+      <!-- 这里是已验证勋章 -->
+      <span id="chip-badges" class="style-scope yt-live-chat-author-chip"></span>
+    </span>
+    <span id="chat-badges" class="style-scope yt-live-chat-author-chip">
+      <author-badge v-if="isInMemberMessage" class="style-scope yt-live-chat-author-chip"
+        :isAdmin="false" :privilegeType="privilegeType"
+      ></author-badge>
+      <template v-else>
+        <author-badge v-if="authorType === AUTHRO_TYPE_ADMIN" class="style-scope yt-live-chat-author-chip"
+          isAdmin :privilegeType="0"
+        ></author-badge>
+        <author-badge v-if="privilegeType > 0" class="style-scope yt-live-chat-author-chip"
+          :isAdmin="false" :privilegeType="privilegeType"
+        ></author-badge>
+      </template>
+    </span>
+  </yt-live-chat-author-chip>
+</template>
+
+<script>
+import AuthorBadge from './AuthorBadge'
+import * as constants from './constants'
+
+export default {
+  name: 'AuthorChip',
+  components: {
+    AuthorBadge
+  },
+  props: {
+    isInMemberMessage: Boolean,
+    authorName: String,
+    authorType: Number,
+    privilegeType: Number
+  },
+  data() {
+    return {
+      AUTHRO_TYPE_ADMIN: constants.AUTHRO_TYPE_ADMIN
+    }
+  },
+  computed: {
+    authorTypeText() {
+      return constants.AUTHOR_TYPE_TO_TEXT[this.authorType]
+    }
+  }
+}
+</script>
+
+<style src="@/assets/css/youtube/yt-live-chat-author-chip.css"></style>

frontend/src/components/ChatRenderer/ImgShadow.vue

@@ -5,7 +5,7 @@
 </template>
 
 <script>
-import * as avatar from '@/api/chat/avatar'
+import * as chat from '@/api/chat'
 
 export default {
   name: 'ImgShadow',
@@ -26,8 +26,8 @@ export default {
   },
   methods: {
     onLoadError() {
-      if (this.showImgUrl !== avatar.DEFAULT_AVATAR_URL) {
-        this.showImgUrl = avatar.DEFAULT_AVATAR_URL
+      if (this.showImgUrl !== chat.DEFAULT_AVATAR_URL) {
+        this.showImgUrl = chat.DEFAULT_AVATAR_URL
       }
     }
   }

frontend/src/components/ChatRenderer/MembershipItem.vue

@@ -8,22 +8,13 @@
         <div id="header-content" class="style-scope yt-live-chat-membership-item-renderer">
           <div id="header-content-primary-column" class="style-scope yt-live-chat-membership-item-renderer">
             <div id="header-content-inner-column" class="style-scope yt-live-chat-membership-item-renderer">
-              <yt-live-chat-author-chip class="style-scope yt-live-chat-membership-item-renderer">
-                <span id="author-name" dir="auto" class="member style-scope yt-live-chat-author-chip">{{
-                  authorName
-                  }}<!-- 这里是已验证勋章 -->
-                  <span id="chip-badges" class="style-scope yt-live-chat-author-chip"></span>
-                </span>
-                <span id="chat-badges" class="style-scope yt-live-chat-author-chip">
-                  <author-badge class="style-scope yt-live-chat-author-chip"
-                    :isAdmin="false" :privilegeType="privilegeType"
-                  ></author-badge>
-                </span>
-              </yt-live-chat-author-chip>
+              <author-chip class="style-scope yt-live-chat-membership-item-renderer"
+                isInMemberMessage :authorName="authorName" :authorType="0" :privilegeType="privilegeType"
+              ></author-chip>
             </div>
-            <div id="header-subtext" class="style-scope yt-live-chat-membership-item-renderer">{{title}}</div>
+            <div id="header-subtext" class="style-scope yt-live-chat-membership-item-renderer">{{ title }}</div>
           </div>
-          <div id="timestamp" class="style-scope yt-live-chat-membership-item-renderer">{{timeText}}</div>
+          <div id="timestamp" class="style-scope yt-live-chat-membership-item-renderer">{{ timeText }}</div>
         </div>
       </div>
     </div>
@@ -31,15 +22,15 @@
 </template>
 
 <script>
-import ImgShadow from './ImgShadow.vue'
-import AuthorBadge from './AuthorBadge.vue'
+import ImgShadow from './ImgShadow'
+import AuthorChip from './AuthorChip'
 import * as utils from '@/utils'
 
 export default {
   name: 'MembershipItem',
   components: {
     ImgShadow,
-    AuthorBadge
+    AuthorChip
   },
   props: {
     avatarUrl: String,

frontend/src/components/ChatRenderer/PaidMessage.vue

@@ -16,23 +16,21 @@
         ></img-shadow>
         <div id="header-content" class="style-scope yt-live-chat-paid-message-renderer">
           <div id="header-content-primary-column" class="style-scope yt-live-chat-paid-message-renderer">
-            <div id="author-name" class="style-scope yt-live-chat-paid-message-renderer">{{authorName}}</div>
-            <div id="purchase-amount" class="style-scope yt-live-chat-paid-message-renderer">{{priceText}}</div>
+            <div id="author-name" class="style-scope yt-live-chat-paid-message-renderer">{{ authorName }}</div>
+            <div id="purchase-amount" class="style-scope yt-live-chat-paid-message-renderer">{{ priceText }}</div>
           </div>
-          <span id="timestamp" class="style-scope yt-live-chat-paid-message-renderer">{{timeText}}</span>
+          <span id="timestamp" class="style-scope yt-live-chat-paid-message-renderer">{{ timeText }}</span>
         </div>
       </div>
       <div id="content" class="style-scope yt-live-chat-paid-message-renderer">
-        <div id="message" dir="auto" class="style-scope yt-live-chat-paid-message-renderer">{{
-          content
-        }}</div>
+        <div id="message" dir="auto" class="style-scope yt-live-chat-paid-message-renderer">{{ content }}</div>
       </div>
     </div>
   </yt-live-chat-paid-message-renderer>
 </template>
 
 <script>
-import ImgShadow from './ImgShadow.vue'
+import ImgShadow from './ImgShadow'
 import * as constants from './constants'
 import * as utils from '@/utils'
 
@@ -53,7 +51,7 @@ export default {
       return constants.getPriceConfig(this.price).colors
     },
     priceText() {
-      return 'CN¥' + utils.formatCurrency(this.price)
+      return `CN¥${utils.formatCurrency(this.price)}`
     },
     timeText() {
       return utils.getTimeTextHourMin(this.time)

frontend/src/components/ChatRenderer/TextMessage.vue

@@ -4,23 +4,20 @@
       :imgUrl="avatarUrl"
     ></img-shadow>
     <div id="content" class="style-scope yt-live-chat-text-message-renderer">
-      <span id="timestamp" class="style-scope yt-live-chat-text-message-renderer">{{timeText}}</span>
-      <yt-live-chat-author-chip class="style-scope yt-live-chat-text-message-renderer">
-        <span id="author-name" dir="auto" class="style-scope yt-live-chat-author-chip" :type="authorTypeText">{{
-          authorName
-          }}<!-- 这里是已验证勋章 -->
-          <span id="chip-badges" class="style-scope yt-live-chat-author-chip"></span>
-        </span>
-        <span id="chat-badges" class="style-scope yt-live-chat-author-chip">
-          <author-badge class="style-scope yt-live-chat-author-chip"
-            :isAdmin="authorType === 2" :privilegeType="privilegeType"
-          ></author-badge>
-        </span>
-      </yt-live-chat-author-chip>
-      <span id="message" class="style-scope yt-live-chat-text-message-renderer">{{
-        content
-        }}<el-badge :value="repeated" :max="99" v-show="repeated > 1" class="style-scope yt-live-chat-text-message-renderer"
-          :style="{'--repeated-mark-color': repeatedMarkColor}"
+      <span id="timestamp" class="style-scope yt-live-chat-text-message-renderer">{{ timeText }}</span>
+      <author-chip class="style-scope yt-live-chat-text-message-renderer"
+        :isInMemberMessage="false" :authorName="authorName" :authorType="authorType" :privilegeType="privilegeType"
+      ></author-chip>
+      <span id="message" class="style-scope yt-live-chat-text-message-renderer">
+        <template v-for="(content, index) in richContent">
+          <span :key="index" v-if="content.type === CONTENT_TYPE_TEXT">{{ content.text }}</span>
+          <img :key="index" v-else-if="content.type === CONTENT_TYPE_IMAGE"
+            class="emoji yt-formatted-string style-scope yt-live-chat-text-message-renderer"
+            :src="content.url" :alt="content.text" :shared-tooltip-text="content.text" :id="`emoji-${content.text}`"
+          >
+        </template>
+        <el-badge :value="repeated" :max="99" v-if="repeated > 1" class="style-scope yt-live-chat-text-message-renderer"
+          :style="{ '--repeated-mark-color': repeatedMarkColor }"
         ></el-badge>
       </span>
     </div>
@@ -28,8 +25,8 @@
 </template>
 
 <script>
-import ImgShadow from './ImgShadow.vue'
-import AuthorBadge from './AuthorBadge.vue'
+import ImgShadow from './ImgShadow'
+import AuthorChip from './AuthorChip'
 import * as constants from './constants'
 import * as utils from '@/utils'
 
@@ -41,17 +38,23 @@ export default {
   name: 'TextMessage',
   components: {
     ImgShadow,
-    AuthorBadge
+    AuthorChip
   },
   props: {
     avatarUrl: String,
     time: Date,
     authorName: String,
     authorType: Number,
-    content: String,
+    richContent: Array,
     privilegeType: Number,
     repeated: Number
   },
+  data() {
+    return {
+      CONTENT_TYPE_TEXT: constants.CONTENT_TYPE_TEXT,
+      CONTENT_TYPE_IMAGE: constants.CONTENT_TYPE_IMAGE
+    }
+  },
   computed: {
     timeText() {
       return utils.getTimeTextHourMin(this.time)
@@ -69,7 +72,7 @@ export default {
         color = [0, 0, 0]
         let t = (this.repeated - 2) / (10 - 2)
         for (let i = 0; i < 3; i++) {
-          color[i] = REPEATED_MARK_COLOR_START[i] + (REPEATED_MARK_COLOR_END[i] - REPEATED_MARK_COLOR_START[i]) * t
+          color[i] = REPEATED_MARK_COLOR_START[i] + ((REPEATED_MARK_COLOR_END[i] - REPEATED_MARK_COLOR_START[i]) * t)
         }
       }
       return `hsl(${color[0]}, ${color[1]}%, ${color[2]}%)`
@@ -95,4 +98,3 @@ yt-live-chat-text-message-renderer>#content>#message>.el-badge .el-badge__conten
 </style>
 
 <style src="@/assets/css/youtube/yt-live-chat-text-message-renderer.css"></style>
-<style src="@/assets/css/youtube/yt-live-chat-author-chip.css"></style>

frontend/src/components/ChatRenderer/Ticker.vue

@@ -17,7 +17,7 @@
               <img-shadow id="author-photo" height="24" width="24" class="style-scope yt-live-chat-ticker-paid-message-item-renderer"
                 :imgUrl="message.raw.avatarUrl"
               ></img-shadow>
-              <span id="text" dir="ltr" class="style-scope yt-live-chat-ticker-paid-message-item-renderer">{{message.text}}</span>
+              <span id="text" dir="ltr" class="style-scope yt-live-chat-ticker-paid-message-item-renderer">{{ message.text }}</span>
             </div>
           </div>
         </yt-live-chat-ticker-paid-message-item-renderer>
@@ -40,10 +40,10 @@
 
 <script>
 import * as chatConfig from '@/api/chatConfig'
-import {formatCurrency} from '@/utils'
-import ImgShadow from './ImgShadow.vue'
-import MembershipItem from './MembershipItem.vue'
-import PaidMessage from './PaidMessage.vue'
+import { formatCurrency } from '@/utils'
+import ImgShadow from './ImgShadow'
+import MembershipItem from './MembershipItem'
+import PaidMessage from './PaidMessage'
 import * as constants from './constants'
 
 export default {
@@ -142,7 +142,7 @@ export default {
         color2 = config.colors.headerBg
       }
       let pinTime = this.getPinTime(message)
-      let progress = (1 - (this.curTime - message.addTime) / (60 * 1000) / pinTime) * 100
+      let progress = (1 - ((this.curTime - message.addTime) / (60 * 1000) / pinTime)) * 100
       if (progress < 0) {
         progress = 0
       } else if (progress > 100) {
@@ -158,9 +158,9 @@ export default {
     },
     getText(message) {
       if (message.type === constants.MESSAGE_TYPE_MEMBER) {
-        return 'Member'
+        return this.$t('chat.tickerMembership')
       }
-      return 'CN¥' + formatCurrency(message.price)
+      return `CN¥${formatCurrency(message.price)}`
     },
     getPinTime(message) {
       if (message.type === constants.MESSAGE_TYPE_MEMBER) {
@@ -169,18 +169,25 @@ export default {
       return constants.getPriceConfig(message.price).pinTime
     },
     updateProgress() {
+      // 更新进度
       this.curTime = new Date()
-      for (let i = 0; i < this.messages.length;) {
-        let message = this.messages[i]
+
+      // 删除过期的消息
+      let filteredMessages = []
+      let messagesChanged = false
+      for (let message of this.messages) {
         let pinTime = this.getPinTime(message)
         if ((this.curTime - message.addTime) / (60 * 1000) >= pinTime) {
-          if (this.pinnedMessage == message) {
+          messagesChanged = true
+          if (this.pinnedMessage === message) {
             this.pinnedMessage = null
           }
-          this.messages.splice(i, 1)
-        } else {
-          i++
+          continue
         }
+        filteredMessages.push(message)
+      }
+      if (messagesChanged) {
+        this.$emit('update:messages', filteredMessages)
       }
     },
     onItemClick(message) {

frontend/src/components/ChatRenderer/constants.js

@@ -1,3 +1,5 @@
+import * as i18n from '@/i18n'
+
 export const AUTHRO_TYPE_NORMAL = 0
 export const AUTHRO_TYPE_MEMBER = 1
 export const AUTHRO_TYPE_ADMIN = 2
@@ -10,13 +12,21 @@ export const AUTHOR_TYPE_TO_TEXT = [
   'owner' // 主播
 ]
 
-export const GUARD_LEVEL_TO_TEXT = [
+const GUARD_LEVEL_TO_TEXT_KEY = [
   '',
-  '总督',
-  '提督',
-  '舰长'
+  'chat.guardLevel1',
+  'chat.guardLevel2',
+  'chat.guardLevel3'
 ]
 
+export function getShowGuardLevelText(guardLevel) {
+  let key = GUARD_LEVEL_TO_TEXT_KEY[guardLevel] || ''
+  if (key === '') {
+    return ''
+  }
+  return i18n.i18n.t(key)
+}
+
 export const MESSAGE_TYPE_TEXT = 0
 export const MESSAGE_TYPE_GIFT = 1
 export const MESSAGE_TYPE_MEMBER = 2
@@ -24,6 +34,9 @@ export const MESSAGE_TYPE_SUPER_CHAT = 3
 export const MESSAGE_TYPE_DEL = 4
 export const MESSAGE_TYPE_UPDATE = 5
 
+export const CONTENT_TYPE_TEXT = 0
+export const CONTENT_TYPE_IMAGE = 1
+
 // 美元 -> 人民币 汇率
 const EXCHANGE_RATE = 7
 export const PRICE_CONFIGS = [
@@ -100,7 +113,7 @@ export const PRICE_CONFIGS = [
     pinTime: 0
   },
   { // $1蓝
-    price: 1 * EXCHANGE_RATE,
+    price: EXCHANGE_RATE,
     colors: {
       contentBg: 'rgba(30,136,229,1)',
       headerBg: 'rgba(21,101,192,1)',
@@ -113,7 +126,7 @@ export const PRICE_CONFIGS = [
   }
 ]
 
-export function getPriceConfig (price) {
+export function getPriceConfig(price) {
   for (const config of PRICE_CONFIGS) {
     if (price >= config.price) {
       return config
@@ -122,21 +135,32 @@ export function getPriceConfig (price) {
   return PRICE_CONFIGS[PRICE_CONFIGS.length - 1]
 }
 
-export function getShowContent (message) {
+export function getShowContent(message) {
   if (message.translation) {
     return `${message.content}（${message.translation}）`
   }
   return message.content
 }
 
-export function getGiftShowContent (message, showGiftName) {
+export function getShowRichContent(message) {
+  let richContent = [...message.richContent]
+  if (message.translation) {
+    richContent.push({
+      type: CONTENT_TYPE_TEXT,
+      text: `（${message.translation}）`
+    })
+  }
+  return richContent
+}
+
+export function getGiftShowContent(message, showGiftName) {
   if (!showGiftName) {
     return ''
   }
-  return `Sent ${message.giftName}x${message.num}`
+  return i18n.i18n.t('chat.sendGift', { giftName: message.giftName, num: message.num })
 }
 
-export function getShowAuthorName (message) {
+export function getShowAuthorName(message) {
   if (message.authorNamePronunciation && message.authorNamePronunciation !== message.authorName) {
     return `${message.authorName}(${message.authorNamePronunciation})`
   }

frontend/src/components/ChatRenderer/index.vue

@@ -2,34 +2,47 @@
   <yt-live-chat-renderer class="style-scope yt-live-chat-app" style="--scrollbar-width:11px;" hide-timestamps
     @mousemove="refreshCantScrollStartTime"
   >
-    <ticker class="style-scope yt-live-chat-renderer" :messages="paidMessages" :showGiftName="showGiftName"></ticker>
+    <ticker class="style-scope yt-live-chat-renderer" :messages.sync="paidMessages" :showGiftName="showGiftName"></ticker>
     <yt-live-chat-item-list-renderer class="style-scope yt-live-chat-renderer" allow-scroll>
       <div ref="scroller" id="item-scroller" class="style-scope yt-live-chat-item-list-renderer animated" @scroll="onScroll">
         <div ref="itemOffset" id="item-offset" class="style-scope yt-live-chat-item-list-renderer" style="height: 0px;">
           <div ref="items" id="items" class="style-scope yt-live-chat-item-list-renderer" style="overflow: hidden"
-            :style="{transform: `translateY(${Math.floor(scrollPixelsRemaining)}px)`}"
+            :style="{ transform: `translateY(${Math.floor(scrollPixelsRemaining)}px)` }"
           >
             <template v-for="message in messages">
               <text-message :key="message.id" v-if="message.type === MESSAGE_TYPE_TEXT"
                 class="style-scope yt-live-chat-item-list-renderer"
-                :avatarUrl="message.avatarUrl" :time="message.time" :authorName="message.authorName"
-                :authorType="message.authorType" :content="getShowContent(message)" :privilegeType="message.privilegeType"
+                :time="message.time"
+                :avatarUrl="message.avatarUrl"
+                :authorName="message.authorName"
+                :authorType="message.authorType"
+                :privilegeType="message.privilegeType"
+                :richContent="getShowRichContent(message)"
                 :repeated="message.repeated"
               ></text-message>
               <paid-message :key="message.id" v-else-if="message.type === MESSAGE_TYPE_GIFT"
                 class="style-scope yt-live-chat-item-list-renderer"
-                :price="message.price" :avatarUrl="message.avatarUrl" :authorName="getShowAuthorName(message)"
-                :time="message.time" :content="getGiftShowContent(message)"
+                :time="message.time"
+                :avatarUrl="message.avatarUrl"
+                :authorName="getShowAuthorName(message)"
+                :price="message.price"
+                :content="getGiftShowContent(message)"
               ></paid-message>
               <membership-item :key="message.id" v-else-if="message.type === MESSAGE_TYPE_MEMBER"
                 class="style-scope yt-live-chat-item-list-renderer"
-                :avatarUrl="message.avatarUrl" :authorName="getShowAuthorName(message)" :privilegeType="message.privilegeType"
-                :title="message.title" :time="message.time"
+                :time="message.time"
+                :avatarUrl="message.avatarUrl"
+                :authorName="getShowAuthorName(message)"
+                :privilegeType="message.privilegeType"
+                :title="message.title"
               ></membership-item>
               <paid-message :key="message.id" v-else-if="message.type === MESSAGE_TYPE_SUPER_CHAT"
                 class="style-scope yt-live-chat-item-list-renderer"
-                :price="message.price" :avatarUrl="message.avatarUrl" :authorName="getShowAuthorName(message)"
-                :time="message.time" :content="getShowContent(message)"
+                :time="message.time"
+                :avatarUrl="message.avatarUrl"
+                :authorName="getShowAuthorName(message)"
+                :price="message.price"
+                :content="getShowContent(message)"
               ></paid-message>
             </template>
           </div>
@@ -41,10 +54,10 @@
 
 <script>
 import * as chatConfig from '@/api/chatConfig'
-import Ticker from './Ticker.vue'
-import TextMessage from './TextMessage.vue'
-import MembershipItem from './MembershipItem.vue'
-import PaidMessage from './PaidMessage.vue'
+import Ticker from './Ticker'
+import TextMessage from './TextMessage'
+import MembershipItem from './MembershipItem'
+import PaidMessage from './PaidMessage'
 import * as constants from './constants'
 
 // 只有要添加的消息需要平滑
@@ -114,7 +127,7 @@ export default {
   },
   computed: {
     canScrollToBottom() {
-      return this.atBottom/* || this.allowScroll*/
+      return this.atBottom/* || this.allowScroll */
     }
   },
   watch: {
@@ -137,6 +150,7 @@ export default {
       return constants.getGiftShowContent(message, this.showGiftName)
     },
     getShowContent: constants.getShowContent,
+    getShowRichContent: constants.getShowRichContent,
     getShowAuthorName: constants.getShowAuthorName,
 
     addMessage(message) {
@@ -211,12 +225,12 @@ export default {
       this.delMessages([id])
     },
     delMessages(ids) {
-      this.enqueueMessages(ids.map(id => {
-        return {
+      this.enqueueMessages(ids.map(
+        id => ({
           type: constants.MESSAGE_TYPE_DEL,
           id
-        }
-      }))
+        })
+      ))
     },
     clearMessages() {
       this.messages = []
@@ -250,15 +264,19 @@ export default {
       } else {
         let curTime = new Date()
         let interval = curTime - this.lastEnqueueTime
-        // 让发消息速度变化不要太频繁
-        if (interval > 1000) {
+        // 真实的进队列时间间隔模式大概是这样：2500, 300, 300, 300, 2500, 300, ...
+        // B站消息有缓冲，会一次发多条消息。这里把波峰视为发送了一次真实的WS消息，所以要过滤掉间隔太小的
+        if (interval > 1000 || this.enqueueIntervals.length < 5) {
           this.enqueueIntervals.push(interval)
           if (this.enqueueIntervals.length > 5) {
             this.enqueueIntervals.splice(0, this.enqueueIntervals.length - 5)
           }
+          // 这边估计得尽量大，只要不太早把消息缓冲发完就是平滑的。有MESSAGE_MAX_INTERVAL保底，不会让消息延迟太大
+          // 其实可以用单调队列求最大值，偷懒不写了
           this.estimatedEnqueueInterval = Math.max(...this.enqueueIntervals)
-          this.lastEnqueueTime = curTime
         }
+        // 上次入队时间还是要设置，否则会太早把消息缓冲发完，然后较长时间没有新消息
+        this.lastEnqueueTime = curTime
       }
 
       // 把messages分成messageGroup，每个组里最多有1个需要平滑的消息
@@ -288,7 +306,7 @@ export default {
         this.emitSmoothedMessageTimerId = window.setTimeout(this.emitSmoothedMessages)
       }
     },
-    messageNeedSmooth({type}) {
+    messageNeedSmooth({ type }) {
       return NEED_SMOOTH_MESSAGE_TYPES.indexOf(type) !== -1
     },
     emitSmoothedMessages() {
@@ -355,18 +373,18 @@ export default {
 
       for (let message of messageGroup) {
         switch (message.type) {
-          case constants.MESSAGE_TYPE_TEXT:
-          case constants.MESSAGE_TYPE_GIFT:
-          case constants.MESSAGE_TYPE_MEMBER:
-          case constants.MESSAGE_TYPE_SUPER_CHAT:
-            this.handleAddMessage(message)
-            break
-          case constants.MESSAGE_TYPE_DEL:
-            this.handleDelMessage(message)
-            break
-          case constants.MESSAGE_TYPE_UPDATE:
-            this.handleUpdateMessage(message)
-            break
+        case constants.MESSAGE_TYPE_TEXT:
+        case constants.MESSAGE_TYPE_GIFT:
+        case constants.MESSAGE_TYPE_MEMBER:
+        case constants.MESSAGE_TYPE_SUPER_CHAT:
+          this.handleAddMessage(message)
+          break
+        case constants.MESSAGE_TYPE_DEL:
+          this.handleDelMessage(message)
+          break
+        case constants.MESSAGE_TYPE_UPDATE:
+          this.handleUpdateMessage(message)
+          break
         }
       }
 
@@ -389,7 +407,7 @@ export default {
         }
       }
     },
-    handleDelMessage({id}) {
+    handleDelMessage({ id }) {
       for (let arr of [this.messages, this.paidMessages, this.messagesBuffer]) {
         for (let i = 0; i < arr.length; i++) {
           if (arr[i].id === id) {
@@ -400,7 +418,7 @@ export default {
         }
       }
     },
-    handleUpdateMessage({id, newValuesObj}) {
+    handleUpdateMessage({ id, newValuesObj }) {
       // 遍历滚动的消息
       this.forEachRecentMessage(999999999, message => {
         if (message.id !== id) {
@@ -468,7 +486,7 @@ export default {
         this.lastSmoothChatMessageAddMs = performance.now()
       }
       let interval = performance.now() - this.lastSmoothChatMessageAddMs
-      this.chatRateMs = 0.9 * this.chatRateMs + 0.1 * interval
+      this.chatRateMs = (0.9 * this.chatRateMs) + (0.1 * interval)
       if (this.isSmoothed) {
         if (this.chatRateMs < 400) {
           this.isSmoothed = false

frontend/src/i18n.js

@@ -0,0 +1,49 @@
+import Vue from 'vue'
+import VueI18n from 'vue-i18n'
+
+import zh from '@/lang/zh'
+
+let lastSetLocale = 'zh'
+let loadedLocales = ['zh']
+
+Vue.use(VueI18n)
+
+export async function setLocale(locale) {
+  lastSetLocale = locale
+  if (loadedLocales.indexOf(locale) === -1) {
+    // eslint-disable-next-line prefer-template
+    let langModule = await import('@/lang/' + locale)
+    i18n.setLocaleMessage(locale, langModule.default)
+    loadedLocales.push(locale)
+
+    // 加载完成之前又调用了setLocale，这次的不生效
+    if (locale !== lastSetLocale) {
+      return
+    }
+  }
+  window.localStorage.lang = i18n.locale = locale
+}
+
+export const i18n = new VueI18n({
+  locale: 'zh',
+  fallbackLocale: 'zh',
+  messages: {
+    zh
+  }
+})
+
+function getDefaultLocale() {
+  let locale = window.localStorage.lang
+  if (!locale) {
+    let lang = navigator.language
+    if (lang.startsWith('zh')) {
+      locale = 'zh'
+    } else if (lang.startsWith('ja')) {
+      locale = 'ja'
+    } else {
+      locale = 'en'
+    }
+  }
+  return locale
+}
+setLocale(getDefaultLocale())

frontend/src/lang/en.js

@@ -1,17 +1,21 @@
 export default {
   sidebar: {
     home: 'Home',
-    stylegen: 'Style generator',
+    stylegen: 'Style Generator',
     help: 'Help',
-    projectAddress: 'Project address',
-    giftRecordOfficial: 'Official Super Chat record',
+    projectAddress: 'Project Address',
+    giftRecordOfficial: 'Official Super Chat Record',
   },
   home: {
     roomIdEmpty: "Room ID can't be empty",
     roomIdInteger: 'Room ID must be positive integer',
+    authCodeEmpty: "Identity code can't be empty",
+    useAuthCodeWarning: 'Please prioritize the identity code',
 
     general: 'General',
+    room: 'Room',
     roomId: 'Room ID',
+    authCode: 'Identity code',
     showDanmaku: 'Show messages',
     showGift: 'Show Super Chats',
     showGiftName: 'Show gift name',
@@ -38,8 +42,14 @@ export default {
     pinyin: 'Pinyin',
     kana: 'Kana',
 
+    emoticon: 'Custom Emotes',
+    emoticonKeyword: 'Emote Code',
+    emoticonUrl: 'URL',
+    operation: 'Operation',
+    addEmoticon: 'Add emote',
+    emoticonFileTooLarge: 'File size is too large. Max size is 1MB',
+
     roomUrl: 'Room URL',
-    copy: 'Copy',
     enterRoom: 'Enter room',
     enterTestRoom: 'Enter test room',
     exportConfig: 'Export config',
@@ -63,7 +73,7 @@ export default {
     showAvatars: 'Show avatars',
     avatarSize: 'Avatar size',
 
-    userNames: 'User names',
+    userNames: 'User Names',
     showUserNames: 'Show user names',
     font: 'Font',
     fontSize: 'Font size',
@@ -74,6 +84,7 @@ export default {
     memberColor: 'Member color',
     showBadges: 'Show badges',
     showColon: 'Show colon after name',
+    emoticonSize: 'Emoticon size',
 
     messages: 'Messages',
     color: 'Color',
@@ -90,7 +101,7 @@ export default {
     moderatorMessageBgColor: 'Moderator background color',
     memberMessageBgColor: 'Member background color',
 
-    scAndNewMember: 'Super Chat / New member',
+    scAndNewMember: 'Super Chat / New Member',
     firstLineFont: 'First line font',
     firstLineFontSize: 'First line font size',
     firstLineLineHeight: 'First line line height (0 for default)',
@@ -123,10 +134,19 @@ export default {
   },
   help: {
     help: 'Help',
-    p1: '1. Copy the room ID from the Bilibili live room webpage',
-    p2: '2. Enter the room ID into the room ID on the home page. Copy the room URL',
+    p1: '1. Copy the identity code (身份码) from the webpage where you start streaming',
+    p2: '2. Enter the identity code into the room configuration on the home page. Copy the room URL',
     p3: '3. Generate styles with the style generator. Copy the CSS',
     p4: '4. Add browser source in OBS',
     p5: '5. Enter the previously copied room URL at URL, and enter the previously copied CSS at custom CSS'
+  },
+  chat: {
+    moderator: 'moderator',
+    guardLevel1: 'governor',
+    guardLevel2: 'admiral',
+    guardLevel3: 'captain',
+    sendGift: 'Sent {giftName}x{num}',
+    membershipTitle: 'New member',
+    tickerMembership: 'Member'
   }
 }

frontend/src/lang/ja.js

@@ -9,9 +9,13 @@ export default {
   home: {
     roomIdEmpty: 'ルームのIDを空白にすることはできません',
     roomIdInteger: 'ルームは正の整数でなければなりません',
+    authCodeEmpty: 'アイデンティティコードを空白にすることはできません',
+    useAuthCodeWarning: 'アイデンティティコードを優先的に使用してください',
 
     general: '常規',
+    room: 'ルーム',
     roomId: 'ルームID',
+    authCode: 'アイデンティティコード',
     showDanmaku: 'コメントを表示する',
     showGift: 'スーパーチャットと新メンバーを表示する',
     showGiftName: 'ギフト名を表示する',
@@ -38,8 +42,14 @@ export default {
     pinyin: 'ピンイン',
     kana: '仮名',
 
+    emoticon: 'カスタムスタンプ',
+    emoticonKeyword: '置き換えるキーワード',
+    emoticonUrl: 'URL',
+    operation: '操作',
+    addEmoticon: 'スタンプを追加',
+    emoticonFileTooLarge: 'ファイルサイズが大きすぎます。最大サイズは1MBです',
+
     roomUrl: 'ルームのURL',
-    copy: 'コピー',
     enterRoom: 'ルームに入る',
     enterTestRoom: 'テストルームに入る',
     exportConfig: 'コンフィグの導出',
@@ -74,6 +84,7 @@ export default {
     memberColor: 'メンバーの色',
     showBadges: '勲章を見せる',
     showColon: 'ユーザー名の後にコロンが表示されます',
+    emoticonSize: 'スタンプサイズ',
 
     messages: 'コメント',
     color: '色',
@@ -123,10 +134,19 @@ export default {
   },
   help: {
     help: 'ヘルプ',
-    p1: '1. ビリビリの生放送ウェブから生放送ルームIDをこぴーする',
-    p2: '2. ホームページでコピーしたIDを入力し、ルームのURLをこぴーする',
+    p1: '1. ビリビリの生放送を始めるのウェブからアイデンティティコード（身份码）をこぴーする',
+    p2: '2. ホームページに先ほどコピーしたアイデンティティコードを入力して、ルームのURLをこぴーする',
     p3: '3. スタイルジェネレータでお好みのコメント様子を選び、出力したCSSをコピーする',
     p4: '4. OBSでブラウザを新規作成する',
     p5: '5. プロパティでこぴーしたURLを入力し、カスタムCSSでスタイルジェネレータのCSSを入力する'
+  },
+  chat: {
+    moderator: 'モデレーター',
+    guardLevel1: '総督',
+    guardLevel2: '提督',
+    guardLevel3: '艦長',
+    sendGift: '{giftName}x{num} を贈りました',
+    membershipTitle: '新規メンバー',
+    tickerMembership: 'メンバー'
   }
 }

frontend/src/lang/zh.js

@@ -9,9 +9,13 @@ export default {
   home: {
     roomIdEmpty: '房间ID不能为空',
     roomIdInteger: '房间ID必须为正整数',
+    authCodeEmpty: '身份码不能为空',
+    useAuthCodeWarning: '请优先使用身份码',
 
     general: '常规',
+    room: '房间',
     roomId: '房间ID',
+    authCode: '身份码',
     showDanmaku: '显示弹幕',
     showGift: '显示打赏和新舰长',
     showGiftName: '显示礼物名',
@@ -38,8 +42,14 @@ export default {
     pinyin: '拼音',
     kana: '日文假名',
 
+    emoticon: '自定义表情',
+    emoticonKeyword: '替换关键词',
+    emoticonUrl: 'URL',
+    operation: '操作',
+    addEmoticon: '添加表情',
+    emoticonFileTooLarge: '文件尺寸太大，最大1MB',
+
     roomUrl: '房间URL',
-    copy: '复制',
     enterRoom: '进入房间',
     enterTestRoom: '进入测试房间',
     exportConfig: '导出配置',
@@ -74,6 +84,7 @@ export default {
     memberColor: '舰长颜色',
     showBadges: '显示勋章',
     showColon: '用户名后显示冒号',
+    emoticonSize: '表情大小',
 
     messages: '消息',
     color: '颜色',
@@ -123,10 +134,19 @@ export default {
   },
   help: {
     help: '帮助',
-    p1: '1. 从B站直播间网页复制房间ID',
-    p2: '2. 把房间ID输入到首页的房间ID，复制房间URL',
+    p1: '1. 从开始直播的页面复制身份码',
+    p2: '2. 把身份码输入到首页的房间配置，复制房间URL',
     p3: '3. 使用样式生成器生成样式，复制CSS',
     p4: '4. 在OBS中添加浏览器源',
     p5: '5. URL处输入之前复制的房间URL，自定义CSS处输入之前复制的CSS'
+  },
+  chat: {
+    moderator: '管理员',
+    guardLevel1: '总督',
+    guardLevel2: '提督',
+    guardLevel3: '舰长',
+    sendGift: '赠送 {giftName}x{num}',
+    membershipTitle: '新会员',
+    tickerMembership: '会员'
   }
 }

frontend/src/layout/Sidebar.vue

@@ -8,46 +8,53 @@
       :default-active="$route.path"
     >
       <el-menu-item index="/">
-        <i class="el-icon-s-home"></i>{{$t('sidebar.home')}}
+        <i class="el-icon-s-home"></i>{{ $t('sidebar.home') }}
       </el-menu-item>
-      <el-menu-item :index="$router.resolve({name: 'stylegen'}).href">
-        <i class="el-icon-brush"></i>{{$t('sidebar.stylegen')}}
+      <el-menu-item :index="$router.resolve({ name: 'stylegen' }).href">
+        <i class="el-icon-brush"></i>{{ $t('sidebar.stylegen') }}
       </el-menu-item>
-      <el-menu-item :index="$router.resolve({name: 'help'}).href">
-        <i class="el-icon-question"></i>{{$t('sidebar.help')}}
+      <el-menu-item :index="$router.resolve({ name: 'help' }).href">
+        <i class="el-icon-question"></i>{{ $t('sidebar.help') }}
       </el-menu-item>
       <a href="https://github.com/xfgryujk/blivechat" target="_blank">
         <el-menu-item>
-          <i class="el-icon-share"></i>{{$t('sidebar.projectAddress')}}
+          <i class="el-icon-share"></i>{{ $t('sidebar.projectAddress') }}
         </el-menu-item>
       </a>
       <a href="http://link.bilibili.com/ctool/vtuber" target="_blank">
         <el-menu-item>
-          <i class="el-icon-link"></i>{{$t('sidebar.giftRecordOfficial')}}
+          <i class="el-icon-link"></i>{{ $t('sidebar.giftRecordOfficial') }}
         </el-menu-item>
       </a>
       <el-submenu index="null">
         <template slot="title">
           <i class="el-icon-chat-line-square"></i>Language
         </template>
-        <el-menu-item v-for="{locale, name} in [
-            {locale: 'zh', name: '中文'},
-            {locale: 'ja', name: '日本語'},
-            {locale: 'en', name: 'English'}
-          ]" :key="locale"
-          @click="onSelectLanguage(locale)"
-        >{{name}}</el-menu-item>
+        <el-menu-item v-for="locale in LOCALES" :key="locale.locale" @click="onSelectLanguage(locale.locale)">
+          <template>{{ locale.name }}</template>
+        </el-menu-item>
       </el-submenu>
     </el-menu>
   </el-scrollbar>
 </template>
 
 <script>
+import * as i18n from '@/i18n'
+
 export default {
   name: 'Sidebar',
+  data() {
+    return {
+      LOCALES: [
+        { locale: 'zh', name: '中文' },
+        { locale: 'ja', name: '日本語' },
+        { locale: 'en', name: 'English' }
+      ]
+    }
+  },
   methods: {
     onSelectLanguage(locale) {
-      window.localStorage.lang = this.$i18n.locale = locale
+      i18n.setLocale(locale)
     }
   }
 }

frontend/src/layout/index.vue

@@ -1,16 +1,14 @@
 <template>
-  <el-container class="app-wrapper" :class="{mobile: isMobile}">
+  <el-container class="app-wrapper" :class="{ mobile: isMobile }">
     <div v-show="isMobile && !hideSidebar" class="drawer-bg" @click="hideSidebar = true"></div>
-    <el-aside width="230px" class="sidebar-container" :class="{'hide-sidebar': hideSidebar}">
+    <el-aside width="230px" class="sidebar-container" :class="{ 'hide-sidebar': hideSidebar }">
       <div class="logo-container">
-        <router-link to="/">
+        <router-link :to="{ name: 'home' }">
           <img src="@/assets/img/logo.png" class="sidebar-logo">
           <h1 class="sidebar-title">blivechat</h1>
         </router-link>
       </div>
-      <div class="version">
-        v1.5.2
-      </div>
+      <div class="version">{{ APP_VERSION }}</div>
       <sidebar></sidebar>
     </el-aside>
     <el-main>
@@ -23,7 +21,7 @@
 </template>
 
 <script>
-import Sidebar from './Sidebar.vue'
+import Sidebar from './Sidebar'
 
 export default {
   name: 'Layout',
@@ -32,6 +30,8 @@ export default {
   },
   data() {
     return {
+      APP_VERSION: process.env.APP_VERSION,
+
       isMobile: false,
       hideSidebar: true
     }

frontend/src/main.js

@@ -1,38 +1,30 @@
 import Vue from 'vue'
 import VueRouter from 'vue-router'
-import VueI18n from 'vue-i18n'
 import {
-  Aside, Autocomplete, Badge, Button, Card, Col, ColorPicker, Container, Divider, Form, FormItem, Image,
+  Aside, Autocomplete, Badge, Button, ButtonGroup, Card, Col, ColorPicker, Container, Divider, Form, FormItem, Image,
   Input, Main, Menu, MenuItem, Message, Option, OptionGroup, Radio, RadioGroup, Row, Select, Scrollbar,
-  Slider, Submenu, Switch, TabPane, Tabs, Tooltip
+  Slider, Submenu, Switch, Table, TableColumn, TabPane, Tabs, Tooltip
 } from 'element-ui'
 import axios from 'axios'
 
-import App from './App.vue'
+import * as i18n from './i18n'
+import App from './App'
 import Layout from './layout'
-import Home from './views/Home.vue'
+import Home from './views/Home'
 import StyleGenerator from './views/StyleGenerator'
 import Help from './views/Help'
-import Room from './views/Room.vue'
-import NotFound from './views/NotFound.vue'
+import Room from './views/Room'
+import NotFound from './views/NotFound'
 
-import zh from './lang/zh'
-import ja from './lang/ja'
-import en from './lang/en'
-
-if (process.env.NODE_ENV === 'development') {
-  // 开发时使用localhost:12450
-  axios.defaults.baseURL = 'http://localhost:12450'
-}
 axios.defaults.timeout = 10 * 1000
 
 Vue.use(VueRouter)
-Vue.use(VueI18n)
 // 初始化element
 Vue.use(Aside)
 Vue.use(Autocomplete)
 Vue.use(Badge)
 Vue.use(Button)
+Vue.use(ButtonGroup)
 Vue.use(Card)
 Vue.use(Col)
 Vue.use(ColorPicker)
@@ -55,6 +47,8 @@ Vue.use(Scrollbar)
 Vue.use(Slider)
 Vue.use(Submenu)
 Vue.use(Switch)
+Vue.use(Table)
+Vue.use(TableColumn)
 Vue.use(TabPane)
 Vue.use(Tabs)
 Vue.use(Tooltip)
@@ -71,49 +65,42 @@ const router = new VueRouter({
       path: '/',
       component: Layout,
       children: [
-        {path: '', component: Home},
-        {path: 'stylegen', name: 'stylegen', component: StyleGenerator},
-        {path: 'help', name: 'help', component: Help}
+        { path: '', name: 'home', component: Home },
+        { path: 'stylegen', name: 'stylegen', component: StyleGenerator },
+        { path: 'help', name: 'help', component: Help }
       ]
     },
-    {path: '/room/test', name: 'test_room', component: Room, props: route => ({strConfig: route.query})},
     {
-      path: '/room/:roomId',
+      path: '/room/test',
+      name: 'test_room',
+      component: Room,
+      props: route => ({ strConfig: route.query })
+    },
+    {
+      path: '/room/:roomKeyValue',
       name: 'room',
       component: Room,
       props(route) {
-        let roomId = parseInt(route.params.roomId)
-        if (isNaN(roomId)) {
-          roomId = null
+        let roomKeyType = parseInt(route.query.roomKeyType) || 1
+        if (roomKeyType < 1 || roomKeyType > 2) {
+          roomKeyType = 1
         }
-        return {roomId, strConfig: route.query}
+
+        let roomKeyValue = route.params.roomKeyValue
+        if (roomKeyType === 1) {
+          roomKeyValue = parseInt(roomKeyValue) || null
+        } else {
+          roomKeyValue = roomKeyValue || null
+        }
+        return { roomKeyType, roomKeyValue, strConfig: route.query }
       }
     },
-    {path: '*', component: NotFound}
+    { path: '*', component: NotFound }
   ]
 })
 
-let locale = window.localStorage.lang
-if (!locale) {
-  let lang = navigator.language
-  if (lang.startsWith('zh')) {
-    locale = 'zh'
-  } else if (lang.startsWith('ja')) {
-    locale = 'ja'
-  } else {
-    locale = 'en'
-  }
-}
-const i18n = new VueI18n({
-  locale,
-  fallbackLocale: 'en',
-  messages: {
-    zh, ja, en
-  }
-})
-
 new Vue({
   render: h => h(App),
   router,
-  i18n
+  i18n: i18n.i18n
 }).$mount('#app')

frontend/src/utils/index.js

@@ -1,4 +1,4 @@
-export function mergeConfig (config, defaultConfig) {
+export function mergeConfig(config, defaultConfig) {
   let res = {}
   for (let i in defaultConfig) {
     res[i] = i in config ? config[i] : defaultConfig[i]
@@ -6,14 +6,14 @@ export function mergeConfig (config, defaultConfig) {
   return res
 }
 
-export function toBool (val) {
+export function toBool(val) {
   if (typeof val === 'string') {
     return ['false', 'no', 'off', '0', ''].indexOf(val.toLowerCase()) === -1
   }
-  return !!val
+  return Boolean(val)
 }
 
-export function toInt (val, _default) {
+export function toInt(val, _default) {
   let res = parseInt(val)
   if (isNaN(res)) {
     res = _default
@@ -21,19 +21,19 @@ export function toInt (val, _default) {
   return res
 }
 
-export function formatCurrency (price) {
+export function formatCurrency(price) {
   return new Intl.NumberFormat('zh-CN', {
     minimumFractionDigits: price < 100 ? 2 : 0
   }).format(price)
 }
 
-export function getTimeTextHourMin (date) {
+export function getTimeTextHourMin(date) {
   let hour = date.getHours()
-  let min = ('00' + date.getMinutes()).slice(-2)
+  let min = `00${date.getMinutes()}`.slice(-2)
   return `${hour}:${min}`
 }
 
-export function getUuid4Hex () {
+export function getUuid4Hex() {
   let chars = []
   for (let i = 0; i < 32; i++) {
     let char = Math.floor(Math.random() * 16).toString(16)

frontend/src/utils/pronunciation/index.js

@@ -2,21 +2,21 @@ export const DICT_PINYIN = 'pinyin'
 export const DICT_KANA = 'kana'
 
 export class PronunciationConverter {
-  constructor () {
+  constructor() {
     this.pronunciationMap = new Map()
   }
 
-  async loadDict (dictName) {
+  async loadDict(dictName) {
     let promise
     switch (dictName) {
-      case DICT_PINYIN:
-        promise = import('./dictPinyin')
-        break
-      case DICT_KANA:
-        promise = import('./dictKana')
-        break
-      default:
-        return
+    case DICT_PINYIN:
+      promise = import('./dictPinyin')
+      break
+    case DICT_KANA:
+      promise = import('./dictKana')
+      break
+    default:
+      return
     }
 
     let dictTxt = (await promise).default
@@ -30,7 +30,7 @@ export class PronunciationConverter {
     this.pronunciationMap = pronunciationMap
   }
 
-  getPronunciation (text) {
+  getPronunciation(text) {
     let res = []
     let lastHasPronunciation = null
     for (let char of text) {

frontend/src/utils/trie.js

@@ -0,0 +1,58 @@
+export class Trie {
+  constructor() {
+    this._root = this._createNode()
+  }
+
+  _createNode() {
+    return {
+      children: {}, // char -> node
+      value: null
+    }
+  }
+
+  set(key, value) {
+    if (key === '') {
+      throw new Error('key is empty')
+    }
+    let node = this._root
+    for (let char of key) {
+      let nextNode = node.children[char]
+      if (nextNode === undefined) {
+        nextNode = node.children[char] = this._createNode()
+      }
+      node = nextNode
+    }
+    node.value = value
+  }
+
+  get(key) {
+    let node = this._root
+    for (let char of key) {
+      let nextNode = node.children[char]
+      if (nextNode === undefined) {
+        return null
+      }
+      node = nextNode
+    }
+    return node.value
+  }
+
+  has(key) {
+    return this.get(key) !== null
+  }
+
+  lazyMatch(str) {
+    let node = this._root
+    for (let char of str) {
+      let nextNode = node.children[char]
+      if (nextNode === undefined) {
+        return null
+      }
+      if (nextNode.value !== null) {
+        return nextNode.value
+      }
+      node = nextNode
+    }
+    return null
+  }
+}

frontend/src/views/Help.vue

@@ -1,18 +1,19 @@
 <template>
   <div>
-    <h1>{{$t('help.help')}}</h1>
-    <p>{{$t('help.p1')}}</p>
+    <h1>{{ $t('help.help') }}</h1>
+    <p>{{ $t('help.p1') }}</p>
     <p class="img-container"><el-image fit="scale-down" src="/static/img/tutorial/tutorial-1.png"></el-image></p>
-    <p>{{$t('help.p2')}}</p>
+    <p>{{ $t('help.p2') }}</p>
     <p class="img-container large-img"><el-image fit="scale-down" src="/static/img/tutorial/tutorial-2.png"></el-image></p>
-    <p>{{$t('help.p3')}}</p>
+    <p>{{ $t('help.p3') }}</p>
     <p class="img-container large-img"><el-image fit="scale-down" src="/static/img/tutorial/tutorial-3.png"></el-image></p>
-    <p>{{$t('help.p4')}}</p>
+    <p>{{ $t('help.p4') }}</p>
     <p class="img-container"><el-image fit="scale-down" src="/static/img/tutorial/tutorial-4.png"></el-image></p>
-    <p>{{$t('help.p5')}}</p>
+    <p>{{ $t('help.p5') }}</p>
     <p class="img-container large-img"><el-image fit="scale-down" src="/static/img/tutorial/tutorial-5.png"></el-image></p>
     <p><br><br><br><br><br><br><br><br>--------------------------------------------------------------------------------------------------------</p>
-    <p>喜欢的话可以推荐给别人，专栏求支持_(:з」∠)_  <a href="https://www.bilibili.com/read/cv4594365" target="_blank">https://www.bilibili.com/read/cv4594365</a></p>
+    <p>喜欢的话可以推荐给别人，专栏求支持_(:з」∠)_ <a href="https://www.bilibili.com/read/cv4594365" target="_blank">https://www.bilibili.com/read/cv4594365</a></p>
+    <p>配置官方翻译接口傻瓜式教程 <a href="https://www.bilibili.com/read/cv14663633" target="_blank">https://www.bilibili.com/read/cv14663633</a>。注意必须下载到本地才能改后台配置，本地使用方法看 <a href="https://github.com/xfgryujk/blivechat#%E4%BD%BF%E7%94%A8%E6%96%B9%E6%B3%95" target="_blank">项目地址中的使用方法说明</a></p>
   </div>
 </template>
 

frontend/src/views/Home.vue

@@ -1,16 +1,37 @@
 <template>
   <div>
     <p>
-      <el-form :model="form" ref="form" label-width="150px" :rules="{
-        roomId: [
-          {required: true, message: $t('home.roomIdEmpty'), trigger: 'blur'},
-          {type: 'integer', min: 1, message: $t('home.roomIdInteger'), trigger: 'blur'}
-        ]
-      }">
+      <el-form :model="form" ref="form" label-width="150px">
         <el-tabs type="border-card">
           <el-tab-pane :label="$t('home.general')">
-            <el-form-item :label="$t('home.roomId')" required prop="roomId">
-              <el-input v-model.number="form.roomId" type="number" min="1"></el-input>
+            <el-form-item :label="$t('home.room')" required :prop="form.roomKeyType === 1 ? 'roomId' : 'authCode'">
+              <template slot="label">{{ $t('home.room') }}
+                <router-link :to="{ name: 'help' }">
+                  <i class="el-icon-question"></i>
+                </router-link>
+              </template>
+              <el-row>
+                <el-col :span="6">
+                  <el-select v-model="form.roomKeyType" style="width: 100%">
+                    <el-option :label="$t('home.authCode')" :value="2"></el-option>
+                    <el-option :label="$t('home.roomId')" :value="1"></el-option>
+                  </el-select>
+                </el-col>
+                <el-col :span="18">
+                  <el-input v-if="form.roomKeyType === 1"
+                    v-model.number="form.roomId" type="number" min="1" :rules="[
+                      {required: true, message: $t('home.roomIdEmpty'), trigger: 'blur'},
+                      {type: 'integer', min: 1, message: $t('home.roomIdInteger'), trigger: 'blur'}
+                    ]"
+                  ></el-input>
+                  <el-input v-else
+                    v-model.number="form.authCode" :rules="[
+                      {required: true, message: $t('home.authCodeEmpty'), trigger: 'blur'}
+                    ]"
+                  ></el-input>
+                </el-col>
+              </el-row>
+              <el-row v-if="form.roomKeyType === 1" style="color: red">{{ $t('home.useAuthCodeWarning') }}</el-row>
             </el-form-item>
             <el-row :gutter="20">
               <el-col :xs="24" :sm="8">
@@ -114,6 +135,34 @@
               </el-radio-group>
             </el-form-item>
           </el-tab-pane>
+
+          <el-tab-pane :label="$t('home.emoticon')">
+            <el-table :data="form.emoticons">
+              <el-table-column prop="keyword" :label="$t('home.emoticonKeyword')" width="170">
+                <template slot-scope="scope">
+                  <el-input v-model="scope.row.keyword"></el-input>
+                </template>
+              </el-table-column>
+              <el-table-column prop="url" :label="$t('home.emoticonUrl')">
+                <template slot-scope="scope">
+                  <el-input v-model="scope.row.url"></el-input>
+                </template>
+              </el-table-column>
+              <el-table-column :label="$t('home.operation')" width="170">
+                <template slot-scope="scope">
+                  <el-button-group>
+                    <el-button type="primary" icon="el-icon-upload2" :disabled="!serverConfig.enableUploadFile"
+                      @click="uploadEmoticon(scope.row)"
+                    ></el-button>
+                    <el-button type="danger" icon="el-icon-minus" @click="delEmoticon(scope.$index)"></el-button>
+                  </el-button-group>
+                </template>
+              </el-table-column>
+            </el-table>
+            <p>
+              <el-button type="primary" icon="el-icon-plus" @click="addEmoticon">{{$t('home.addEmoticon')}}</el-button>
+            </p>
+          </el-tab-pane>
         </el-tabs>
       </el-form>
     </p>
@@ -123,11 +172,11 @@
         <el-form :model="form" label-width="150px">
           <el-form-item :label="$t('home.roomUrl')">
             <el-input ref="roomUrlInput" readonly :value="obsRoomUrl" style="width: calc(100% - 8em); margin-right: 1em;"></el-input>
-            <el-button type="primary" @click="copyUrl">{{$t('home.copy')}}</el-button>
+            <el-button type="primary" icon="el-icon-copy-document" @click="copyUrl"></el-button>
           </el-form-item>
           <el-form-item>
             <el-button type="primary" :disabled="!roomUrl" @click="enterRoom">{{$t('home.enterRoom')}}</el-button>
-            <el-button :disabled="!roomUrl" @click="enterTestRoom">{{$t('home.enterTestRoom')}}</el-button>
+            <el-button @click="enterTestRoom">{{$t('home.enterTestRoom')}}</el-button>
             <el-button @click="exportConfig">{{$t('home.exportConfig')}}</el-button>
             <el-button @click="importConfig">{{$t('home.importConfig')}}</el-button>
           </el-form-item>
@@ -139,10 +188,10 @@
 
 <script>
 import _ from 'lodash'
-import axios from 'axios'
 import download from 'downloadjs'
 
-import {mergeConfig} from '@/utils'
+import { mergeConfig } from '@/utils'
+import * as mainApi from '@/api/main'
 import * as chatConfig from '@/api/chatConfig'
 
 export default {
@@ -151,15 +200,25 @@ export default {
     return {
       serverConfig: {
         enableTranslate: true,
+        enableUploadFile: true,
         loaderUrl: ''
       },
       form: {
+        ...chatConfig.getLocalConfig(),
+        roomKeyType: parseInt(window.localStorage.roomKeyType || '2'),
         roomId: parseInt(window.localStorage.roomId || '1'),
-        ...chatConfig.getLocalConfig()
+        authCode: window.localStorage.authCode || '',
       }
     }
   },
   computed: {
+    roomKeyValue() {
+      if (this.form.roomKeyType === 1) {
+        return this.form.roomId
+      } else {
+        return this.form.authCode
+      }
+    },
     roomUrl() {
       return this.getRoomUrl(false)
     },
@@ -177,7 +236,9 @@ export default {
   },
   watch: {
     roomUrl: _.debounce(function() {
+      window.localStorage.roomKeyType = this.form.roomKeyType
       window.localStorage.roomId = this.form.roomId
+      window.localStorage.authCode = this.form.authCode
       chatConfig.setLocalConfig(this.form)
     }, 500)
   },
@@ -187,28 +248,69 @@ export default {
   methods: {
     async updateServerConfig() {
       try {
-        this.serverConfig = (await axios.get('/api/server_info')).data.config
+        this.serverConfig = (await mainApi.getServerInfo()).config
       } catch (e) {
-        this.$message.error('Failed to fetch server information: ' + e)
+        this.$message.error(`Failed to fetch server information: ${e}`)
+        throw e
       }
     },
+
+    addEmoticon() {
+      this.form.emoticons.push({
+        keyword: '[Kappa]',
+        url: ''
+      })
+    },
+    delEmoticon(index) {
+      this.form.emoticons.splice(index, 1)
+    },
+    uploadEmoticon(emoticon) {
+      let input = document.createElement('input')
+      input.type = 'file'
+      input.accept = 'image/png, image/jpeg, image/jpg, image/gif'
+      input.onchange = async() => {
+        let file = input.files[0]
+        if (file.size > 1024 * 1024) {
+          this.$message.error(this.$t('home.emoticonFileTooLarge'))
+          return
+        }
+
+        let res
+        try {
+          res = await mainApi.uploadEmoticon(file)
+        } catch (e) {
+          this.$message.error(`Failed to upload: ${e}`)
+          throw e
+        }
+        emoticon.url = res.url
+      }
+      input.click()
+    },
+
     enterRoom() {
-      window.open(this.roomUrl, `room ${this.form.roomId}`, 'menubar=0,location=0,scrollbars=0,toolbar=0,width=600,height=600')
+      window.open(this.roomUrl, `room ${this.roomKeyValue}`, 'menubar=0,location=0,scrollbars=0,toolbar=0,width=600,height=600')
     },
     enterTestRoom() {
       window.open(this.getRoomUrl(true), 'test room', 'menubar=0,location=0,scrollbars=0,toolbar=0,width=600,height=600')
     },
     getRoomUrl(isTestRoom) {
-      if (isTestRoom && this.form.roomId === '') {
+      if (!isTestRoom && !this.roomKeyValue) {
         return ''
       }
-      let query = {...this.form}
+
+      let query = {
+        ...this.form,
+        emoticons: JSON.stringify(this.form.emoticons),
+        lang: this.$i18n.locale
+      }
       delete query.roomId
+      delete query.authCode
+
       let resolved
       if (isTestRoom) {
-        resolved = this.$router.resolve({name: 'test_room', query})
+        resolved = this.$router.resolve({ name: 'test_room', query })
       } else {
-        resolved = this.$router.resolve({name: 'room', params: {roomId: this.form.roomId}, query})
+        resolved = this.$router.resolve({ name: 'room', params: { roomKeyValue: this.roomKeyValue }, query })
       }
       return `${window.location.protocol}//${window.location.host}${resolved.href}`
     },
@@ -234,12 +336,21 @@ export default {
             this.$message.error(this.$t('home.failedToParseConfig') + e)
             return
           }
-          cfg = mergeConfig(cfg, chatConfig.DEFAULT_CONFIG)
-          this.form = {roomId: this.form.roomId, ...cfg}
+          this.importConfigFromObj(cfg)
         }
         reader.readAsText(input.files[0])
       }
       input.click()
+    },
+    importConfigFromObj(cfg) {
+      cfg = mergeConfig(cfg, chatConfig.deepCloneDefaultConfig())
+      chatConfig.sanitizeConfig(cfg)
+      this.form = {
+        ...cfg,
+        roomKeyType: this.form.roomKeyType,
+        roomId: this.form.roomId,
+        authCode: this.form.authCode
+      }
     }
   }
 }

frontend/src/views/Room.vue

@@ -3,11 +3,15 @@
 </template>
 
 <script>
-import {mergeConfig, toBool, toInt} from '@/utils'
+import * as i18n from '@/i18n'
+import { mergeConfig, toBool, toInt } from '@/utils'
+import * as trie from '@/utils/trie'
 import * as pronunciation from '@/utils/pronunciation'
 import * as chatConfig from '@/api/chatConfig'
+import * as chat from '@/api/chat'
 import ChatClientTest from '@/api/chat/ChatClientTest'
-import ChatClientDirect from '@/api/chat/ChatClientDirect'
+import ChatClientDirectWeb from '@/api/chat/ChatClientDirectWeb'
+import ChatClientDirectOpenLive from '@/api/chat/ChatClientDirectOpenLive'
 import ChatClientRelay from '@/api/chat/ChatClientRelay'
 import ChatRenderer from '@/components/ChatRenderer'
 import * as constants from '@/components/ChatRenderer/constants'
@@ -18,8 +22,12 @@ export default {
     ChatRenderer
   },
   props: {
-    roomId: {
+    roomKeyType: {
       type: Number,
+      default: 1
+    },
+    roomKeyValue: {
+      type: [Number, String],
       default: null
     },
     strConfig: {
@@ -29,22 +37,49 @@ export default {
   },
   data() {
     return {
-      config: {...chatConfig.DEFAULT_CONFIG},
+      config: chatConfig.deepCloneDefaultConfig(),
       chatClient: null,
-      pronunciationConverter: null
+      pronunciationConverter: null,
+      textEmoticons: [], // 官方的文本表情
     }
   },
   computed: {
-    blockKeywords() {
-      return this.config.blockKeywords.split('\n').filter(val => val)
+    blockKeywordsTrie() {
+      let blockKeywords = this.config.blockKeywords.split('\n')
+      let res = new trie.Trie()
+      for (let keyword of blockKeywords) {
+        if (keyword !== '') {
+          res.set(keyword, true)
+        }
+      }
+      return res
     },
-    blockUsers() {
-      return this.config.blockUsers.split('\n').filter(val => val)
+    blockUsersTrie() {
+      let blockUsers = this.config.blockUsers.split('\n')
+      let res = new trie.Trie()
+      for (let user of blockUsers) {
+        if (user !== '') {
+          res.set(user, true)
+        }
+      }
+      return res
+    },
+    emoticonsTrie() {
+      let res = new trie.Trie()
+      for (let emoticons of [this.config.emoticons, this.textEmoticons]) {
+        for (let emoticon of emoticons) {
+          if (emoticon.keyword !== '' && emoticon.url !== '') {
+            res.set(emoticon.keyword, emoticon)
+          }
+        }
+      }
+      return res
     }
   },
   mounted() {
     this.initConfig()
     this.initChatClient()
+    this.initTextEmoticons()
     if (this.config.giftUsernamePronunciation !== '') {
       this.pronunciationConverter = new pronunciation.PronunciationConverter()
       this.pronunciationConverter.loadDict(this.config.giftUsernamePronunciation)
@@ -63,6 +98,11 @@ export default {
   },
   methods: {
     initConfig() {
+      let locale = this.strConfig.lang
+      if (locale) {
+        i18n.setLocale(locale)
+      }
+
       let cfg = {}
       // 留空的使用默认值
       for (let i in this.strConfig) {
@@ -70,7 +110,7 @@ export default {
           cfg[i] = this.strConfig[i]
         }
       }
-      cfg = mergeConfig(cfg, chatConfig.DEFAULT_CONFIG)
+      cfg = mergeConfig(cfg, chatConfig.deepCloneDefaultConfig())
 
       cfg.minGiftPrice = toInt(cfg.minGiftPrice, chatConfig.DEFAULT_CONFIG.minGiftPrice)
       cfg.showDanmaku = toBool(cfg.showDanmaku)
@@ -79,24 +119,44 @@ export default {
       cfg.mergeSimilarDanmaku = toBool(cfg.mergeSimilarDanmaku)
       cfg.mergeGift = toBool(cfg.mergeGift)
       cfg.maxNumber = toInt(cfg.maxNumber, chatConfig.DEFAULT_CONFIG.maxNumber)
+
       cfg.blockGiftDanmaku = toBool(cfg.blockGiftDanmaku)
       cfg.blockLevel = toInt(cfg.blockLevel, chatConfig.DEFAULT_CONFIG.blockLevel)
       cfg.blockNewbie = toBool(cfg.blockNewbie)
       cfg.blockNotMobileVerified = toBool(cfg.blockNotMobileVerified)
       cfg.blockMedalLevel = toInt(cfg.blockMedalLevel, chatConfig.DEFAULT_CONFIG.blockMedalLevel)
+
       cfg.relayMessagesByServer = toBool(cfg.relayMessagesByServer)
       cfg.autoTranslate = toBool(cfg.autoTranslate)
+      cfg.emoticons = this.toObjIfJson(cfg.emoticons)
 
+      chatConfig.sanitizeConfig(cfg)
       this.config = cfg
     },
+    toObjIfJson(str) {
+      if (typeof str !== 'string') {
+        return str
+      }
+      try {
+        return JSON.parse(str)
+      } catch {
+        return {}
+      }
+    },
     initChatClient() {
-      if (this.roomId === null) {
+      if (this.roomKeyValue === null) {
         this.chatClient = new ChatClientTest()
+      } else if (this.config.relayMessagesByServer) {
+        let roomKey = {
+          type: this.roomKeyType,
+          value: this.roomKeyValue
+        }
+        this.chatClient = new ChatClientRelay(roomKey, this.config.autoTranslate)
       } else {
-        if (!this.config.relayMessagesByServer) {
-          this.chatClient = new ChatClientDirect(this.roomId)
+        if (this.roomKeyType === 1) {
+          this.chatClient = new ChatClientDirectWeb(this.roomKeyValue)
         } else {
-          this.chatClient = new ChatClientRelay(this.roomId, this.config.autoTranslate)
+          this.chatClient = new ChatClientDirectOpenLive(this.roomKeyValue)
         }
       }
       this.chatClient.onAddText = this.onAddText
@@ -107,6 +167,9 @@ export default {
       this.chatClient.onUpdateTranslation = this.onUpdateTranslation
       this.chatClient.start()
     },
+    async initTextEmoticons() {
+      this.textEmoticons = await chat.getTextEmoticons()
+    },
 
     start() {
       this.chatClient.start()
@@ -127,6 +190,7 @@ export default {
         authorName: data.authorName,
         authorType: data.authorType,
         content: data.content,
+        richContent: this.getRichContent(data),
         privilegeType: data.privilegeType,
         repeated: 1,
         translation: data.translation
@@ -169,7 +233,7 @@ export default {
         authorName: data.authorName,
         authorNamePronunciation: this.getPronunciation(data.authorName),
         privilegeType: data.privilegeType,
-        title: 'New member'
+        title: this.$t('chat.membershipTitle')
       }
       this.$refs.renderer.addMessage(message)
     },
@@ -194,15 +258,13 @@ export default {
       this.$refs.renderer.addMessage(message)
     },
     onDelSuperChat(data) {
-      for (let id of data.ids) {
-        this.$refs.renderer.delMessage(id)
-      }
+      this.$refs.renderer.delMessages(data.ids)
     },
     onUpdateTranslation(data) {
       if (!this.config.autoTranslate) {
         return
       }
-      this.$refs.renderer.updateMessage(data.id, {translation: data.translation})
+      this.$refs.renderer.updateMessage(data.id, { translation: data.translation })
     },
 
     filterTextMessage(data) {
@@ -217,24 +279,27 @@ export default {
       } else if (this.config.blockMedalLevel > 0 && data.medalLevel < this.config.blockMedalLevel) {
         return false
       }
-      return this.filterSuperChatMessage(data)
+      return this.filterByContent(data.content) && this.filterByAuthorName(data.authorName)
     },
     filterSuperChatMessage(data) {
-      for (let keyword of this.blockKeywords) {
-        if (data.content.indexOf(keyword) !== -1) {
-          return false
-        }
-      }
-      return this.filterNewMemberMessage(data)
+      return this.filterByContent(data.content) && this.filterByAuthorName(data.authorName)
     },
     filterNewMemberMessage(data) {
-      for (let user of this.blockUsers) {
-        if (data.authorName === user) {
+      return this.filterByAuthorName(data.authorName)
+    },
+    filterByContent(content) {
+      let blockKeywordsTrie = this.blockKeywordsTrie
+      for (let i = 0; i < content.length; i++) {
+        let remainContent = content.substring(i)
+        if (blockKeywordsTrie.lazyMatch(remainContent) !== null) {
           return false
         }
       }
       return true
     },
+    filterByAuthorName(authorName) {
+      return !this.blockUsersTrie.has(authorName)
+    },
     mergeSimilarText(content) {
       if (!this.config.mergeSimilarDanmaku) {
         return false
@@ -252,6 +317,66 @@ export default {
         return ''
       }
       return this.pronunciationConverter.getPronunciation(text)
+    },
+    getRichContent(data) {
+      let richContent = []
+
+      // B站官方表情
+      if (data.emoticon !== null) {
+        richContent.push({
+          type: constants.CONTENT_TYPE_IMAGE,
+          text: data.content,
+          url: data.emoticon
+        })
+        return richContent
+      }
+
+      // 没有文本表情，只能是纯文本
+      if (this.config.emoticons.length === 0 && this.textEmoticons.length === 0) {
+        richContent.push({
+          type: constants.CONTENT_TYPE_TEXT,
+          text: data.content
+        })
+        return richContent
+      }
+
+      // 可能含有文本表情，需要解析
+      let emoticonsTrie = this.emoticonsTrie
+      let startPos = 0
+      let pos = 0
+      while (pos < data.content.length) {
+        let remainContent = data.content.substring(pos)
+        let matchEmoticon = emoticonsTrie.lazyMatch(remainContent)
+        if (matchEmoticon === null) {
+          pos++
+          continue
+        }
+
+        // 加入之前的文本
+        if (pos !== startPos) {
+          richContent.push({
+            type: constants.CONTENT_TYPE_TEXT,
+            text: data.content.slice(startPos, pos)
+          })
+        }
+
+        // 加入表情
+        richContent.push({
+          type: constants.CONTENT_TYPE_IMAGE,
+          text: matchEmoticon.keyword,
+          url: matchEmoticon.url
+        })
+        pos += matchEmoticon.keyword.length
+        startPos = pos
+      }
+      // 加入尾部的文本
+      if (pos !== startPos) {
+        richContent.push({
+          type: constants.CONTENT_TYPE_TEXT,
+          text: data.content.slice(startPos, pos)
+        })
+      }
+      return richContent
     }
   }
 }

frontend/src/views/StyleGenerator/Legacy.vue

@@ -1,7 +1,7 @@
 <template>
   <div>
     <el-form label-width="150px" size="mini">
-      <h3>{{$t('stylegen.outlines')}}</h3>
+      <h3>{{ $t('stylegen.outlines') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -20,7 +20,7 @@
         </el-form-item>
       </el-card>
 
-      <h3>{{$t('stylegen.avatars')}}</h3>
+      <h3>{{ $t('stylegen.avatars') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -36,7 +36,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.userNames')}}</h3>
+      <h3>{{ $t('stylegen.userNames') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -100,7 +100,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.messages')}}</h3>
+      <h3>{{ $t('stylegen.messages') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -129,9 +129,16 @@
         <el-form-item :label="$t('stylegen.onNewLine')">
           <el-switch v-model="form.messageOnNewLine"></el-switch>
         </el-form-item>
+        <el-row :gutter="20">
+          <el-col :xs="24" :sm="12">
+            <el-form-item :label="$t('stylegen.emoticonSize')">
+              <el-input v-model.number="form.emoticonSize" type="number" min="0"></el-input>
+            </el-form-item>
+          </el-col>
+        </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.time')}}</h3>
+      <h3>{{ $t('stylegen.time') }}</h3>
       <el-card shadow="never">
         <el-form-item :label="$t('stylegen.showTime')">
           <el-switch v-model="form.showTime"></el-switch>
@@ -162,7 +169,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.backgrounds')}}</h3>
+      <h3>{{ $t('stylegen.backgrounds') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -202,7 +209,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.scAndNewMember')}}</h3>
+      <h3>{{ $t('stylegen.scAndNewMember') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -299,7 +306,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.animation')}}</h3>
+      <h3>{{ $t('stylegen.animation') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -350,7 +357,7 @@ import _ from 'lodash'
 
 import FontSelect from './FontSelect'
 import * as common from './common'
-import {mergeConfig} from '@/utils'
+import { mergeConfig } from '@/utils'
 
 export const DEFAULT_CONFIG = {
   showOutlines: true,
@@ -376,6 +383,7 @@ export const DEFAULT_CONFIG = {
   messageLineHeight: 0,
   messageColor: '#ffffff',
   messageOnNewLine: false,
+  emoticonSize: 18,
 
   showTime: false,
   timeFont: 'Imprima',
@@ -476,7 +484,7 @@ yt-live-chat-renderer * {
   line-height: ${this.form.messageLineHeight || this.form.messageFontSize}px !important;
 }`
     },
-    showOutlinesStyle () {
+    showOutlinesStyle() {
       if (!this.form.showOutlines || !this.form.outlineSize) {
         return ''
       }
@@ -541,7 +549,12 @@ yt-live-chat-text-message-renderer #message * {
 ${!this.form.messageOnNewLine ? '' : `yt-live-chat-text-message-renderer #message {
   display: block !important;
   overflow: visible !important;
-}`}`
+}`}
+
+yt-live-chat-text-message-renderer #message .emoji {
+  width: auto !important;
+  height: ${this.form.emoticonSize}px !important;
+}`
     },
     timeStyle() {
       return common.getTimeStyle(this.form)
@@ -655,11 +668,11 @@ yt-live-chat-ticker-sponsor-item-renderer * {
       try {
         return mergeConfig(JSON.parse(window.localStorage.stylegenConfig), DEFAULT_CONFIG)
       } catch {
-        return {...DEFAULT_CONFIG}
+        return { ...DEFAULT_CONFIG }
       }
     },
     resetConfig() {
-      this.form = {...DEFAULT_CONFIG}
+      this.form = { ...DEFAULT_CONFIG }
     },
 
     getBgStyleForAuthorType(authorType, color) {

frontend/src/views/StyleGenerator/LineLike.vue

@@ -1,7 +1,7 @@
 <template>
   <div>
     <el-form label-width="150px" size="mini">
-      <h3>{{$t('stylegen.avatars')}}</h3>
+      <h3>{{ $t('stylegen.avatars') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -17,7 +17,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.userNames')}}</h3>
+      <h3>{{ $t('stylegen.userNames') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -76,7 +76,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.messages')}}</h3>
+      <h3>{{ $t('stylegen.messages') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -102,9 +102,16 @@
             </el-form-item>
           </el-col>
         </el-row>
+        <el-row :gutter="20">
+          <el-col :xs="24" :sm="12">
+            <el-form-item :label="$t('stylegen.emoticonSize')">
+              <el-input v-model.number="form.emoticonSize" type="number" min="0"></el-input>
+            </el-form-item>
+          </el-col>
+        </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.time')}}</h3>
+      <h3>{{ $t('stylegen.time') }}</h3>
       <el-card shadow="never">
         <el-form-item :label="$t('stylegen.showTime')">
           <el-switch v-model="form.showTime"></el-switch>
@@ -135,7 +142,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.backgrounds')}}</h3>
+      <h3>{{ $t('stylegen.backgrounds') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -170,7 +177,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.scAndNewMember')}}</h3>
+      <h3>{{ $t('stylegen.scAndNewMember') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -249,7 +256,7 @@
         </el-row>
       </el-card>
 
-      <h3>{{$t('stylegen.animation')}}</h3>
+      <h3>{{ $t('stylegen.animation') }}</h3>
       <el-card shadow="never">
         <el-row :gutter="20">
           <el-col :xs="24" :sm="12">
@@ -300,7 +307,7 @@ import _ from 'lodash'
 
 import FontSelect from './FontSelect'
 import * as common from './common'
-import {mergeConfig} from '@/utils'
+import { mergeConfig } from '@/utils'
 
 export const DEFAULT_CONFIG = {
   showAvatars: true,
@@ -320,6 +327,7 @@ export const DEFAULT_CONFIG = {
   messageFontSize: 18,
   messageLineHeight: 0,
   messageColor: '#000000',
+  emoticonSize: 18,
 
   showTime: false,
   timeFont: 'Noto Sans SC',
@@ -459,13 +467,18 @@ yt-live-chat-text-message-renderer #message {
   border-radius: 30px;
 }
 
+yt-live-chat-text-message-renderer #message .emoji {
+  width: auto !important;
+  height: ${this.form.emoticonSize}px !important;
+}
+
 /* The triangle beside dialog */
 yt-live-chat-text-message-renderer #message::before {
   content: "";
   display: inline-block;
   position: absolute;
-  top: ${this.form.showUserNames ? ((this.form.userNameLineHeight || this.form.userNameFontSize) + 10) : 20}px;
-  left: ${this.form.showAvatars ? (this.form.avatarSize + this.form.avatarSize / 4 - 8) : -8}px;
+  top: ${this.form.showUserNames ? (this.form.userNameLineHeight || this.form.userNameFontSize) + 10 : 20}px;
+  left: ${this.form.showAvatars ? this.form.avatarSize + (this.form.avatarSize / 4) - 8 : -8}px;
   border: 8px solid transparent;
   border-right: 18px solid;
   transform: rotate(35deg);
@@ -573,11 +586,11 @@ yt-live-chat-ticker-sponsor-item-renderer * {
       try {
         return mergeConfig(JSON.parse(window.localStorage.stylegenLineLikeConfig), DEFAULT_CONFIG)
       } catch {
-        return {...DEFAULT_CONFIG}
+        return { ...DEFAULT_CONFIG }
       }
     },
     resetConfig() {
-      this.form = {...DEFAULT_CONFIG}
+      this.form = { ...DEFAULT_CONFIG }
     },
 
     getBgStyleForAuthorType(authorType, color) {

frontend/src/views/StyleGenerator/common.js

@@ -53,7 +53,7 @@ yt-live-chat-membership-item-renderer a {
   text-decoration: none !important;
 }`
 
-export function getImportStyle (allFonts) {
+export function getImportStyle(allFonts) {
   let fontsNeedToImport = new Set()
   for (let font of allFonts) {
     if (fonts.NETWORK_FONTS.indexOf(font) !== -1) {
@@ -67,7 +67,7 @@ export function getImportStyle (allFonts) {
   return res.join('\n')
 }
 
-export function getAvatarStyle (config) {
+export function getAvatarStyle(config) {
   return `/* Avatars */
 yt-live-chat-text-message-renderer #author-photo,
 yt-live-chat-text-message-renderer #author-photo img,
@@ -83,7 +83,7 @@ yt-live-chat-membership-item-renderer #author-photo img {
 }`
 }
 
-export function getTimeStyle (config) {
+export function getTimeStyle(config) {
   return `/* Timestamps */
 yt-live-chat-text-message-renderer #timestamp {
   display: ${config.showTime ? 'inline' : 'none'} !important;
@@ -94,7 +94,7 @@ yt-live-chat-text-message-renderer #timestamp {
 }`
 }
 
-export function getAnimationStyle (config) {
+export function getAnimationStyle(config) {
   if (!config.animateIn && !config.animateOut) {
     return ''
   }
@@ -113,13 +113,13 @@ export function getAnimationStyle (config) {
       : ` transform: translateX(${config.reverseSlide ? 16 : -16}px);`
     } }`)
     curTime += config.fadeInTime
-    keyframes.push(`  ${(curTime / totalTime) * 100}% { opacity: 1; transform: none; }`)
+    keyframes.push(`  ${curTime / totalTime * 100}% { opacity: 1; transform: none; }`)
   }
   if (config.animateOut) {
     curTime += config.animateOutWaitTime * 1000
-    keyframes.push(`  ${(curTime / totalTime) * 100}% { opacity: 1; transform: none; }`)
+    keyframes.push(`  ${curTime / totalTime * 100}% { opacity: 1; transform: none; }`)
     curTime += config.fadeOutTime
-    keyframes.push(`  ${(curTime / totalTime) * 100}% { opacity: 0;${!config.slide ? ''
+    keyframes.push(`  ${curTime / totalTime * 100}% { opacity: 0;${!config.slide ? ''
       : ` transform: translateX(${config.reverseSlide ? -16 : 16}px);`
     } }`)
   }
@@ -136,7 +136,7 @@ yt-live-chat-paid-message-renderer {
 }`
 }
 
-export function cssEscapeStr (str) {
+export function cssEscapeStr(str) {
   let res = []
   for (let char of str) {
     res.push(cssEscapeChar(char))
@@ -144,7 +144,7 @@ export function cssEscapeStr (str) {
   return res.join('')
 }
 
-function cssEscapeChar (char) {
+function cssEscapeChar(char) {
   if (!needEscapeChar(char)) {
     return char
   }
@@ -153,7 +153,7 @@ function cssEscapeChar (char) {
   return `\\${hexCode} `
 }
 
-function needEscapeChar (char) {
+function needEscapeChar(char) {
   let code = char.codePointAt(0)
   if (0x20 <= code && code <= 0x7E) {
     return char === '"' || char === '\\'

frontend/src/views/StyleGenerator/index.vue

@@ -11,21 +11,21 @@
       </el-tabs>
 
       <el-form label-width="150px" size="mini">
-        <h3>{{$t('stylegen.result')}}</h3>
+        <h3>{{ $t('stylegen.result') }}</h3>
         <el-card shadow="never">
           <el-form-item label="CSS">
             <el-input v-model="inputResult" ref="result" type="textarea" :rows="20"></el-input>
           </el-form-item>
           <el-form-item>
-            <el-button type="primary" @click="copyResult">{{$t('stylegen.copy')}}</el-button>
-            <el-button @click="resetConfig">{{$t('stylegen.resetConfig')}}</el-button>
+            <el-button type="primary" @click="copyResult">{{ $t('stylegen.copy') }}</el-button>
+            <el-button @click="resetConfig">{{ $t('stylegen.resetConfig') }}</el-button>
           </el-form-item>
         </el-card>
       </el-form>
     </el-col>
 
     <el-col :sm="24" :md="8">
-      <div :style="{position: 'relative', top: `${exampleTop}px`}">
+      <div :style="{ position: 'relative', top: `${exampleTop}px` }">
         <el-form inline style="line-height: 40px">
           <el-form-item :label="$t('stylegen.playAnimation')" style="margin: 0">
             <el-switch v-model="playAnimation" @change="onPlayAnimationChange"></el-switch>
@@ -34,7 +34,7 @@
             <el-switch v-model="exampleBgLight" :active-text="$t('stylegen.light')" :inactive-text="$t('stylegen.dark')"></el-switch>
           </el-form-item>
         </el-form>
-        <div id="example-container" :class="{light: exampleBgLight}">
+        <div id="example-container" :class="{ light: exampleBgLight }">
           <div id="fakebody">
             <room ref="room"></room>
           </div>

frontend/vue.config.js

@@ -0,0 +1,27 @@
+// 不能用localhost，https://forum.dfinity.org/t/development-workflow-quickly-test-code-modifications/1793/21
+const API_BASE_URL = 'http://127.0.0.1:12450'
+
+module.exports = {
+  devServer: {
+    proxy: {
+      '/api': {
+        target: API_BASE_URL,
+        ws: true
+      },
+      '/emoticons': {
+        target: API_BASE_URL
+      }
+    }
+  },
+  chainWebpack: config => {
+    const APP_VERSION = `v${process.env.npm_package_version}`
+
+    config.plugin('define')
+      .tap(args => {
+        let defineMap = args[0]
+        let env = defineMap['process.env']
+        env['APP_VERSION'] = JSON.stringify(APP_VERSION)
+        return args
+      })
+  }
+}

main.py

@@ -1,68 +1,109 @@
 # -*- coding: utf-8 -*-
-
 import argparse
+import asyncio
 import logging
 import logging.handlers
 import os
+import signal
+import sys
 import webbrowser
+from typing import *
 
 import tornado.ioloop
 import tornado.web
 
 import api.chat
 import api.main
+import api.open_live
 import config
-import models.avatar
 import models.database
-import models.translate
+import services.avatar
+import services.chat
+import services.translate
 import update
+import utils.request
 
 logger = logging.getLogger(__name__)
 
-BASE_PATH = os.path.dirname(os.path.realpath(__file__))
-WEB_ROOT = os.path.join(BASE_PATH, 'frontend', 'dist')
-LOG_FILE_NAME = os.path.join(BASE_PATH, 'log', 'blivechat.log')
-
-routes = [
-    (r'/api/server_info', api.main.ServerInfoHandler),
-    (r'/api/chat', api.chat.ChatHandler),
-    (r'/api/room_info', api.chat.RoomInfoHandler),
-    (r'/api/avatar_url', api.chat.AvatarHandler),
-
-    (r'/(.*)', api.main.MainHandler, {'path': WEB_ROOT, 'default_filename': 'index.html'})
+ROUTES = [
+    *api.main.ROUTES,
+    *api.chat.ROUTES,
+    *api.open_live.ROUTES,
+    *api.main.LAST_ROUTES,
 ]
 
+server: Optional[tornado.httpserver.HTTPServer] = None
+
+shut_down_event: Optional[asyncio.Event] = None
+
+
+async def main():
+    if not init():
+        return 1
+    try:
+        await run()
+    finally:
+        await shut_down()
+    return 0
+
+
+def init():
+    init_signal_handlers()
 
-def main():
     args = parse_args()
 
     init_logging(args.debug)
+    logger.info('App started, initializing')
     config.init()
+
+    utils.request.init()
     models.database.init(args.debug)
-    models.avatar.init()
-    models.translate.init()
-    api.chat.init()
+
+    services.avatar.init()
+    services.translate.init()
+    services.chat.init()
+
     update.check_update()
 
-    run_server(args.host, args.port, args.debug)
+    init_server(args.host, args.port, args.debug)
+    return server is not None
+
+
+def init_signal_handlers():
+    global shut_down_event
+    shut_down_event = asyncio.Event()
+
+    signums = (signal.SIGINT, signal.SIGTERM)
+    try:
+        loop = asyncio.get_running_loop()
+        for signum in signums:
+            loop.add_signal_handler(signum, on_shut_down_signal)
+    except NotImplementedError:
+        # 不太安全，但Windows只能用这个
+        for signum in signums:
+            signal.signal(signum, on_shut_down_signal)
+
+
+def on_shut_down_signal(*_args):
+    shut_down_event.set()
 
 
 def parse_args():
     parser = argparse.ArgumentParser(description='用于OBS的仿YouTube风格的bilibili直播评论栏')
-    parser.add_argument('--host', help='服务器host，默认为127.0.0.1', default='127.0.0.1')
-    parser.add_argument('--port', help='服务器端口，默认为12450', type=int, default=12450)
+    parser.add_argument('--host', help='服务器host，默认和配置中的一样', default=None)
+    parser.add_argument('--port', help='服务器端口，默认和配置中的一样', type=int, default=None)
     parser.add_argument('--debug', help='调试模式', action='store_true')
     return parser.parse_args()
 
 
 def init_logging(debug):
+    filename = os.path.join(config.BASE_PATH, 'log', 'blivechat.log')
     stream_handler = logging.StreamHandler()
     file_handler = logging.handlers.TimedRotatingFileHandler(
-        LOG_FILE_NAME, encoding='utf-8', when='midnight', backupCount=7, delay=True
+        filename, encoding='utf-8', when='midnight', backupCount=7, delay=True
     )
     logging.basicConfig(
         format='{asctime} {levelname} [{name}]: {message}',
-        datefmt='%Y-%m-%d %H:%M:%S',
         style='{',
         level=logging.INFO if not debug else logging.DEBUG,
         handlers=[stream_handler, file_handler]
@@ -72,31 +113,56 @@ def init_logging(debug):
     logging.getLogger('tornado.access').setLevel(logging.WARNING)
 
 
-def run_server(host, port, debug):
+def init_server(host, port, debug):
+    cfg = config.get_config()
+    if host is None:
+        host = cfg.host
+    if port is None:
+        port = cfg.port
+
     app = tornado.web.Application(
-        routes,
+        ROUTES,
         websocket_ping_interval=10,
         debug=debug,
         autoreload=False
     )
-    cfg = config.get_config()
     try:
-        app.listen(
+        global server
+        server = app.listen(
             port,
             host,
-            xheaders=cfg.tornado_xheaders
+            xheaders=cfg.tornado_xheaders,
+            max_body_size=1024 * 1024,
+            max_buffer_size=1024 * 1024
         )
     except OSError:
         logger.warning('Address is used %s:%d', host, port)
         return
     finally:
-        url = 'http://localhost/' if port == 80 else f'http://localhost:{port}/'
-        # 防止更新版本后浏览器加载缓存
-        url += '?_v=' + update.VERSION
-        webbrowser.open(url)
+        if cfg.open_browser_at_startup:
+            url = 'http://localhost/' if port == 80 else f'http://localhost:{port}/'
+            webbrowser.open(url)
     logger.info('Server started: %s:%d', host, port)
-    tornado.ioloop.IOLoop.current().start()
+
+
+async def run():
+    logger.info('Running event loop')
+    await shut_down_event.wait()
+    logger.info('Received shutdown signal')
+
+
+async def shut_down():
+    logger.info('Closing server')
+    server.stop()
+    await server.close_all_connections()
+
+    logger.info('Closing websocket connections')
+    await services.chat.shut_down()
+
+    await utils.request.shut_down()
+
+    logger.info('App shut down')
 
 
 if __name__ == '__main__':
-    main()
+    sys.exit(asyncio.run(main()))

models/avatar.py

@@ -1,214 +0,0 @@
-# -*- coding: utf-8 -*-
-
-import asyncio
-import datetime
-import logging
-import re
-from typing import *
-
-import aiohttp
-import sqlalchemy
-import sqlalchemy.exc
-
-import config
-import models.database
-
-logger = logging.getLogger(__name__)
-
-
-DEFAULT_AVATAR_URL = '//static.hdslb.com/images/member/noface.gif'
-
-_main_event_loop = asyncio.get_event_loop()
-_http_session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=10))
-# user_id -> avatar_url
-_avatar_url_cache: Dict[int, str] = {}
-# 正在获取头像的Future，user_id -> Future
-_uid_fetch_future_map: Dict[int, asyncio.Future] = {}
-# 正在获取头像的user_id队列
-_uid_queue_to_fetch = None
-# 上次被B站ban时间
-_last_fetch_banned_time: Optional[datetime.datetime] = None
-
-
-def init():
-    cfg = config.get_config()
-    global _uid_queue_to_fetch
-    _uid_queue_to_fetch = asyncio.Queue(cfg.fetch_avatar_max_queue_size)
-    asyncio.ensure_future(_get_avatar_url_from_web_consumer())
-
-
-async def get_avatar_url(user_id):
-    avatar_url = await get_avatar_url_or_none(user_id)
-    if avatar_url is None:
-        avatar_url = DEFAULT_AVATAR_URL
-    return avatar_url
-
-
-async def get_avatar_url_or_none(user_id):
-    avatar_url = get_avatar_url_from_memory(user_id)
-    if avatar_url is not None:
-        return avatar_url
-    avatar_url = await get_avatar_url_from_database(user_id)
-    if avatar_url is not None:
-        return avatar_url
-    return await get_avatar_url_from_web(user_id)
-
-
-def get_avatar_url_from_memory(user_id):
-    return _avatar_url_cache.get(user_id, None)
-
-
-def get_avatar_url_from_database(user_id) -> Awaitable[Optional[str]]:
-    return asyncio.get_event_loop().run_in_executor(
-        None, _do_get_avatar_url_from_database, user_id
-    )
-
-
-def _do_get_avatar_url_from_database(user_id):
-    try:
-        with models.database.get_session() as session:
-            user = session.query(BilibiliUser).filter(BilibiliUser.uid == user_id).one_or_none()
-            if user is None:
-                return None
-            avatar_url = user.avatar_url
-
-            # 如果离上次更新太久就更新所有缓存
-            if (datetime.datetime.now() - user.update_time).days >= 3:
-                def refresh_cache():
-                    _avatar_url_cache.pop(user_id, None)
-                    get_avatar_url_from_web(user_id)
-
-                _main_event_loop.call_soon(refresh_cache)
-            else:
-                # 否则只更新内存缓存
-                _update_avatar_cache_in_memory(user_id, avatar_url)
-    except sqlalchemy.exc.OperationalError:
-        # SQLite会锁整个文件，忽略就行
-        return None
-    except sqlalchemy.exc.SQLAlchemyError:
-        logger.exception('_do_get_avatar_url_from_database failed:')
-        return None
-    return avatar_url
-
-
-def get_avatar_url_from_web(user_id) -> Awaitable[Optional[str]]:
-    # 如果已有正在获取的future则返回，防止重复获取同一个uid
-    future = _uid_fetch_future_map.get(user_id, None)
-    if future is not None:
-        return future
-    # 否则创建一个获取任务
-    _uid_fetch_future_map[user_id] = future = _main_event_loop.create_future()
-    future.add_done_callback(lambda _future: _uid_fetch_future_map.pop(user_id, None))
-    try:
-        _uid_queue_to_fetch.put_nowait(user_id)
-    except asyncio.QueueFull:
-        future.set_result(None)
-    return future
-
-
-async def _get_avatar_url_from_web_consumer():
-    while True:
-        try:
-            user_id = await _uid_queue_to_fetch.get()
-            future = _uid_fetch_future_map.get(user_id, None)
-            if future is None:
-                continue
-
-            # 防止在被ban的时候获取
-            global _last_fetch_banned_time
-            if _last_fetch_banned_time is not None:
-                cur_time = datetime.datetime.now()
-                if (cur_time - _last_fetch_banned_time).total_seconds() < 3 * 60 + 3:
-                    # 3分钟以内被ban，解封大约要15分钟
-                    future.set_result(None)
-                    continue
-                else:
-                    _last_fetch_banned_time = None
-
-            asyncio.ensure_future(_get_avatar_url_from_web_coroutine(user_id, future))
-
-            # 限制频率，防止被B站ban
-            cfg = config.get_config()
-            await asyncio.sleep(cfg.fetch_avatar_interval)
-        except Exception:
-            logger.exception('_get_avatar_url_from_web_consumer error:')
-
-
-async def _get_avatar_url_from_web_coroutine(user_id, future):
-    try:
-        avatar_url = await _do_get_avatar_url_from_web(user_id)
-    except BaseException as e:
-        future.set_exception(e)
-    else:
-        future.set_result(avatar_url)
-
-
-async def _do_get_avatar_url_from_web(user_id):
-    try:
-        async with _http_session.get('https://api.bilibili.com/x/space/acc/info',
-                                     params={'mid': user_id}) as r:
-            if r.status != 200:
-                logger.warning('Failed to fetch avatar: status=%d %s uid=%d', r.status, r.reason, user_id)
-                if r.status == 412:
-                    # 被B站ban了
-                    global _last_fetch_banned_time
-                    _last_fetch_banned_time = datetime.datetime.now()
-                return None
-            data = await r.json()
-    except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
-        return None
-
-    avatar_url = process_avatar_url(data['data']['face'])
-    update_avatar_cache(user_id, avatar_url)
-    return avatar_url
-
-
-def process_avatar_url(avatar_url):
-    # 去掉协议，兼容HTTP、HTTPS
-    m = re.fullmatch(r'(?:https?:)?(.*)', avatar_url)
-    if m is not None:
-        avatar_url = m[1]
-    # 缩小图片加快传输
-    if not avatar_url.endswith('noface.gif'):
-        avatar_url += '@48w_48h'
-    return avatar_url
-
-
-def update_avatar_cache(user_id, avatar_url):
-    _update_avatar_cache_in_memory(user_id, avatar_url)
-    asyncio.get_event_loop().run_in_executor(
-        None, _update_avatar_cache_in_database, user_id, avatar_url
-    )
-
-
-def _update_avatar_cache_in_memory(user_id, avatar_url):
-    _avatar_url_cache[user_id] = avatar_url
-    cfg = config.get_config()
-    while len(_avatar_url_cache) > cfg.avatar_cache_size:
-        _avatar_url_cache.pop(next(iter(_avatar_url_cache)), None)
-
-
-def _update_avatar_cache_in_database(user_id, avatar_url):
-    try:
-        with models.database.get_session() as session:
-            user = session.query(BilibiliUser).filter(BilibiliUser.uid == user_id).one_or_none()
-            if user is None:
-                user = BilibiliUser(uid=user_id, avatar_url=avatar_url,
-                                    update_time=datetime.datetime.now())
-                session.add(user)
-            else:
-                user.avatar_url = avatar_url
-                user.update_time = datetime.datetime.now()
-            session.commit()
-    except (sqlalchemy.exc.OperationalError, sqlalchemy.exc.IntegrityError):
-        # SQLite会锁整个文件，忽略就行，另外还有多线程导致ID重复的问题
-        pass
-    except sqlalchemy.exc.SQLAlchemyError:
-        logger.exception('_update_avatar_cache_in_database failed:')
-
-
-class BilibiliUser(models.database.OrmBase):
-    __tablename__ = 'bilibili_users'
-    uid = sqlalchemy.Column(sqlalchemy.Integer, primary_key=True)
-    avatar_url = sqlalchemy.Column(sqlalchemy.String(100))
-    update_time = sqlalchemy.Column(sqlalchemy.DateTime)

models/bilibili.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+import datetime
+
+import sqlalchemy
+from sqlalchemy.orm import Mapped, mapped_column
+
+import models.database
+
+
+class BilibiliUser(models.database.OrmBase):
+    __tablename__ = 'bilibili_users'
+    uid: Mapped[int] = mapped_column(sqlalchemy.BigInteger, primary_key=True)  # 创建表后最好手动改成unsigned
+    avatar_url: Mapped[str] = mapped_column(sqlalchemy.String(100))
+    update_time: Mapped[datetime.datetime]

models/database.py

@@ -1,35 +1,32 @@
 # -*- coding: utf-8 -*-
-
-import contextlib
 from typing import *
 
-import sqlalchemy.ext.declarative
 import sqlalchemy.orm
 
 import config
 
-OrmBase = sqlalchemy.ext.declarative.declarative_base()
-engine = None
-DbSession: Optional[Type[sqlalchemy.orm.Session]] = None
+_engine: Optional[sqlalchemy.Engine] = None
 
 
-def init(debug):
+class OrmBase(sqlalchemy.orm.DeclarativeBase):
+    pass
+
+
+def init(_debug):
     cfg = config.get_config()
-    global engine, DbSession
-    # engine = sqlalchemy.create_engine(cfg.database_url, echo=debug)
-    engine = sqlalchemy.create_engine(cfg.database_url)
-    DbSession = sqlalchemy.orm.sessionmaker(bind=engine)
+    global _engine
+    _engine = sqlalchemy.create_engine(
+        cfg.database_url,
+        pool_size=5,  # 保持的连接数
+        max_overflow=5,  # 临时的额外连接数
+        pool_timeout=3,  # 连接数达到最大时获取新连接的超时时间
+        # pool_pre_ping=True,  # 获取连接时先检测是否可用
+        pool_recycle=60 * 60,  # 回收超过1小时的连接，防止数据库服务器主动断开不活跃的连接
+        # echo=debug,  # 输出SQL语句
+    )
 
-    OrmBase.metadata.create_all(engine)
+    OrmBase.metadata.create_all(_engine)
 
 
-@contextlib.contextmanager
-def get_session():
-    session = DbSession()
-    try:
-        yield session
-    except:
-        session.rollback()
-        raise
-    finally:
-        session.close()
+def get_session() -> sqlalchemy.orm.Session:
+    return sqlalchemy.orm.Session(_engine)

requirements.txt

@@ -1,3 +1,5 @@
-aiohttp==3.7.4
-sqlalchemy==1.3.13
-tornado==6.0.2
+-r blivedm/requirements.txt
+cachetools==5.3.1
+pycryptodome==3.10.1
+sqlalchemy==2.0.19
+tornado==6.3.2

services/avatar.py

@@ -0,0 +1,423 @@
+# -*- coding: utf-8 -*-
+import asyncio
+import dataclasses
+import datetime
+import logging
+import re
+from typing import *
+
+import aiohttp
+import cachetools
+import sqlalchemy.exc
+
+import config
+import models.bilibili as bl_models
+import models.database
+import utils.request
+
+logger = logging.getLogger(__name__)
+
+
+DEFAULT_AVATAR_URL = '//static.hdslb.com/images/member/noface.gif'
+
+_avatar_fetchers: List['AvatarFetcher'] = []
+# user_id -> avatar_url
+_avatar_url_cache: Optional[cachetools.TTLCache] = None
+# 正在获取头像的Future，user_id -> Future
+_uid_fetch_future_map: Dict[int, asyncio.Future] = {}
+# 正在获取头像的任务队列
+_task_queue: Optional['asyncio.Queue[FetchTask]'] = None
+
+
+@dataclasses.dataclass
+class FetchTask:
+    user_id: int
+    future: 'asyncio.Future[Optional[str]]'
+
+
+def init():
+    cfg = config.get_config()
+    global _avatar_url_cache, _task_queue
+    _avatar_url_cache = cachetools.TTLCache(cfg.avatar_cache_size, 10 * 60)
+    _task_queue = asyncio.Queue(cfg.fetch_avatar_max_queue_size)
+    asyncio.get_running_loop().create_task(_do_init())
+
+
+async def _do_init():
+    fetchers = [
+        MedalAnchorAvatarFetcher(3),
+        UserCardAvatarFetcher(3),
+        GameUserCenterAvatarFetcher(3),
+    ]
+    await asyncio.gather(*(fetcher.init() for fetcher in fetchers))
+    global _avatar_fetchers
+    _avatar_fetchers = fetchers
+
+
+async def get_avatar_url(user_id) -> str:
+    avatar_url = await get_avatar_url_or_none(user_id)
+    if avatar_url is None:
+        avatar_url = DEFAULT_AVATAR_URL
+    return avatar_url
+
+
+async def get_avatar_url_or_none(user_id) -> Optional[str]:
+    if user_id == 0:
+        return None
+
+    # 查内存
+    avatar_url = _get_avatar_url_from_memory(user_id)
+    if avatar_url is not None:
+        return avatar_url
+
+    # 查数据库
+    user = await _get_avatar_url_from_database(user_id)
+    if user is not None:
+        avatar_url = user.avatar_url
+        _update_avatar_cache_in_memory(user_id, avatar_url)
+        # 如果距离数据库上次更新太久，则在后台从接口获取，并更新所有缓存
+        if (datetime.datetime.now() - user.update_time).days >= 1:
+            asyncio.create_task(_refresh_avatar_cache_from_web(user_id))
+        return avatar_url
+
+    # 从接口获取
+    avatar_url = await _get_avatar_url_from_web(user_id)
+    if avatar_url is not None:
+        update_avatar_cache(user_id, avatar_url)
+        return avatar_url
+
+    return None
+
+
+async def _refresh_avatar_cache_from_web(user_id):
+    avatar_url = await _get_avatar_url_from_web(user_id)
+    if avatar_url is None:
+        return
+    update_avatar_cache(user_id, avatar_url)
+
+
+def update_avatar_cache(user_id, avatar_url):
+    if user_id == 0:
+        return
+    _update_avatar_cache_in_memory(user_id, avatar_url)
+    _update_avatar_cache_in_database(user_id, avatar_url)
+
+
+def update_avatar_cache_if_expired(user_id, avatar_url):
+    # 内存缓存过期了才更新，减少写入数据库的频率
+    if _get_avatar_url_from_memory(user_id) is None:
+        update_avatar_cache(user_id, avatar_url)
+
+
+def process_avatar_url(avatar_url):
+    # 去掉协议，兼容HTTP、HTTPS
+    m = re.fullmatch(r'(?:https?:)?(.*)', avatar_url)
+    if m is not None:
+        avatar_url = m[1]
+    return avatar_url
+
+
+def _get_avatar_url_from_memory(user_id) -> Optional[str]:
+    return _avatar_url_cache.get(user_id, None)
+
+
+def _update_avatar_cache_in_memory(user_id, avatar_url):
+    _avatar_url_cache[user_id] = avatar_url
+
+
+def _get_avatar_url_from_database(user_id) -> Awaitable[Optional[bl_models.BilibiliUser]]:
+    loop = asyncio.get_running_loop()
+    return loop.run_in_executor(None, _do_get_avatar_url_from_database, user_id)
+
+
+def _do_get_avatar_url_from_database(user_id) -> Optional[bl_models.BilibiliUser]:
+    try:
+        with models.database.get_session() as session:
+            user: bl_models.BilibiliUser = session.scalars(
+                sqlalchemy.select(bl_models.BilibiliUser).filter(
+                    bl_models.BilibiliUser.uid == user_id
+                )
+            ).one_or_none()
+            if user is None:
+                return None
+            return user
+    except sqlalchemy.exc.OperationalError:
+        # SQLite会锁整个文件，忽略就行
+        return None
+    except sqlalchemy.exc.SQLAlchemyError:
+        logger.exception('_do_get_avatar_url_from_database failed:')
+        return None
+
+
+def _update_avatar_cache_in_database(user_id, avatar_url) -> Awaitable[None]:
+    return asyncio.get_running_loop().run_in_executor(
+        None, _do_update_avatar_cache_in_database, user_id, avatar_url
+    )
+
+
+def _do_update_avatar_cache_in_database(user_id, avatar_url):
+    try:
+        with models.database.get_session() as session:
+            user = session.scalars(
+                sqlalchemy.select(bl_models.BilibiliUser).filter(
+                    bl_models.BilibiliUser.uid == user_id
+                )
+            ).one_or_none()
+            if user is None:
+                user = bl_models.BilibiliUser(
+                    uid=user_id
+                )
+                session.add(user)
+            user.avatar_url = avatar_url
+            user.update_time = datetime.datetime.now()
+            session.commit()
+    except (sqlalchemy.exc.OperationalError, sqlalchemy.exc.IntegrityError):
+        # SQLite会锁整个文件，忽略就行。另外还有多线程导致ID重复的问题，这里对一致性要求不高就没加for update
+        pass
+    except sqlalchemy.exc.SQLAlchemyError:
+        logger.exception('_do_update_avatar_cache_in_database failed:')
+
+
+def _get_avatar_url_from_web(user_id) -> Awaitable[Optional[str]]:
+    # 如果已有正在获取的future则返回，防止重复获取同一个uid
+    future = _uid_fetch_future_map.get(user_id, None)
+    if future is not None:
+        return future
+    # 否则创建一个获取任务
+    future = asyncio.get_running_loop().create_future()
+
+    task = FetchTask(
+        user_id=user_id,
+        future=future
+    )
+    if not _push_task(task):
+        future.set_result(None)
+        return future
+
+    _uid_fetch_future_map[user_id] = future
+    future.add_done_callback(lambda _future: _uid_fetch_future_map.pop(user_id, None))
+    return future
+
+
+def _push_task(task: FetchTask):
+    if not _has_available_avatar_fetcher():
+        return False
+
+    try:
+        _task_queue.put_nowait(task)
+        return True
+    except asyncio.QueueFull:
+        return False
+
+
+def _pop_task() -> Awaitable[FetchTask]:
+    return _task_queue.get()
+
+
+def _cancel_all_tasks_if_no_available_avatar_fetcher():
+    if _has_available_avatar_fetcher():
+        return
+
+    logger.warning('No available avatar fetcher')
+    while not _task_queue.empty():
+        task = _task_queue.get_nowait()
+        task.future.set_result(None)
+
+
+def _has_available_avatar_fetcher():
+    return any(fetcher.is_available for fetcher in _avatar_fetchers)
+
+
+class AvatarFetcher:
+    def __init__(self, query_interval):
+        self._query_interval = query_interval
+        self._be_available_event = asyncio.Event()
+        self._be_available_event.set()
+
+        self._cool_down_timer_handle = None
+
+    async def init(self):
+        asyncio.create_task(self._fetch_consumer())
+        return True
+
+    @property
+    def is_available(self):
+        return self._cool_down_timer_handle is None
+
+    def _on_availability_change(self):
+        if self.is_available:
+            self._be_available_event.set()
+        else:
+            self._be_available_event.clear()
+            _cancel_all_tasks_if_no_available_avatar_fetcher()
+
+    async def _fetch_consumer(self):
+        cls_name = type(self).__name__
+        while True:
+            try:
+                if not self.is_available:
+                    logger.info('%s waiting to become available', cls_name)
+                    await self._be_available_event.wait()
+                    logger.info('%s became available', cls_name)
+
+                task = await _pop_task()
+                # 为了简化代码，约定只会在_fetch_wrapper里变成不可用，所以获取task之后这里还是可用的
+                assert self.is_available
+
+                start_time = datetime.datetime.now()
+                await self._fetch_wrapper(task)
+                cost_time = (datetime.datetime.now() - start_time).total_seconds()
+
+                # 限制频率，防止被B站ban
+                await asyncio.sleep(self._query_interval - cost_time)
+            except Exception:  # noqa
+                logger.exception('%s error:', cls_name)
+
+    async def _fetch_wrapper(self, task: FetchTask) -> Optional[str]:
+        try:
+            avatar_url = await self._do_fetch(task.user_id)
+        except BaseException as e:
+            task.future.set_exception(e)
+            return None
+
+        task.future.set_result(avatar_url)
+        return avatar_url
+
+    async def _do_fetch(self, user_id) -> Optional[str]:
+        raise NotImplementedError
+
+    def _cool_down(self, sleep_time):
+        if self._cool_down_timer_handle is not None:
+            return
+
+        self._cool_down_timer_handle = asyncio.get_running_loop().call_later(
+            sleep_time, self._on_cool_down_timeout
+        )
+        self._on_availability_change()
+
+    def _on_cool_down_timeout(self):
+        self._cool_down_timer_handle = None
+        self._on_availability_change()
+
+
+class MedalAnchorAvatarFetcher(AvatarFetcher):
+    async def _do_fetch(self, user_id) -> Optional[str]:
+        try:
+            async with utils.request.http_session.get(
+                'https://api.live.bilibili.com/xlive/web-room/v1/index/getDanmuMedalAnchorInfo',
+                headers={
+                    **utils.request.BILIBILI_COMMON_HEADERS,
+                    'Origin': 'https://live.bilibili.com',
+                    'Referer': 'https://live.bilibili.com/'
+                },
+                params={
+                    'ruid': user_id,
+                    'token': '',
+                    'platform': 'web',
+                    'jsonp': 'jsonp'
+                }
+            ) as r:
+                if r.status != 200:
+                    logger.warning(
+                        'MedalAnchorAvatarFetcher failed to fetch avatar: status=%d %s uid=%d',
+                        r.status, r.reason, user_id
+                    )
+                    if r.status == 412:
+                        # 被B站ban了
+                        self._cool_down(3 * 60)
+                    return None
+                data = await r.json()
+        except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
+            return None
+
+        code = data['code']
+        if code != 0:
+            # 这里虽然失败但不会被ban一段时间
+            logger.info(
+                'MedalAnchorAvatarFetcher failed to fetch avatar: code=%d %s uid=%d',
+                code, data['message'], user_id
+            )
+            return None
+
+        return process_avatar_url(data['data']['rface'])
+
+
+class UserCardAvatarFetcher(AvatarFetcher):
+    async def _do_fetch(self, user_id) -> Optional[str]:
+        try:
+            async with utils.request.http_session.get(
+                'https://api.bilibili.com/x/web-interface/card',
+                headers={
+                    **utils.request.BILIBILI_COMMON_HEADERS,
+                    'Origin': 'https://t.bilibili.com',
+                    'Referer': 'https://t.bilibili.com/'
+                },
+                params={
+                    'mid': user_id,
+                    'photo': 'true'
+                }
+            ) as r:
+                if r.status != 200:
+                    logger.warning(
+                        'UserCardAvatarFetcher failed to fetch avatar: status=%d %s uid=%d',
+                        r.status, r.reason, user_id
+                    )
+                    if r.status == 412:
+                        # 被B站ban了
+                        self._cool_down(3 * 60)
+                    return None
+                data = await r.json()
+        except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
+            return None
+
+        code = data['code']
+        if code != 0:
+            # 这里虽然失败但不会被ban一段时间
+            logger.info(
+                'UserCardAvatarFetcher failed to fetch avatar: code=%d %s uid=%d',
+                code, data['message'], user_id
+            )
+            return None
+
+        return process_avatar_url(data['data']['card']['face'])
+
+
+class GameUserCenterAvatarFetcher(AvatarFetcher):
+    async def _do_fetch(self, user_id) -> Optional[str]:
+        try:
+            async with utils.request.http_session.get(
+                'https://line3-h5-mobile-api.biligame.com/game/center/h5/user/space/info',
+                headers={
+                    **utils.request.BILIBILI_COMMON_HEADERS,
+                    'Origin': 'https://app.biligame.com',
+                    'Referer': 'https://app.biligame.com/'
+                },
+                params={
+                    'uid': user_id,
+                    'sdk_type': '1'
+                }
+            ) as r:
+                if r.status != 200:
+                    # 这个接口经常502
+                    logger.info(
+                        'GameUserCenterAvatarFetcher failed to fetch avatar: status=%d %s uid=%d',
+                        r.status, r.reason, user_id
+                    )
+                    if r.status == 412:
+                        # 被B站ban了
+                        self._cool_down(3 * 60)
+                    return None
+                data = await r.json()
+        except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
+            return None
+
+        code = data['code']
+        if code != 0:
+            # 这里虽然失败但不会被ban一段时间
+            logger.info(
+                'GameUserCenterAvatarFetcher failed to fetch avatar: code=%d %s uid=%d',
+                code, data['message'], user_id
+            )
+            return None
+
+        return process_avatar_url(data['data']['face'])

services/chat.py

@@ -0,0 +1,690 @@
+# -*- coding: utf-8 -*-
+import asyncio
+import enum
+import logging
+import uuid
+from typing import *
+
+import api.chat
+import api.open_live as api_open_live
+import blivedm.blivedm as blivedm
+import blivedm.blivedm.models.open_live as dm_open_models
+import blivedm.blivedm.models.web as dm_web_models
+import config
+import services.avatar
+import services.translate
+import utils.request
+
+logger = logging.getLogger(__name__)
+
+
+class RoomKeyType(enum.IntEnum):
+    ROOM_ID = 1
+    AUTH_CODE = 2
+
+
+class RoomKey(NamedTuple):
+    """内部用来标识一个房间，由客户端加入房间时传入"""
+    type: RoomKeyType
+    value: Union[int, str]
+
+    def __str__(self):
+        res = str(self.value)
+        if self.type == RoomKeyType.AUTH_CODE:
+            # 身份码要脱敏
+            res = '***' + res[-3:]
+        return res
+    __repr__ = __str__
+
+
+# 用于类型标注的类型别名
+LiveClientType = Union['WebLiveClient', 'OpenLiveClient']
+
+# 到B站的连接管理
+_live_client_manager: Optional['LiveClientManager'] = None
+# 到客户端的连接管理
+client_room_manager: Optional['ClientRoomManager'] = None
+# 直播消息处理器
+_live_msg_handler: Optional['LiveMsgHandler'] = None
+
+
+def init():
+    global _live_client_manager, client_room_manager, _live_msg_handler
+    _live_client_manager = LiveClientManager()
+    client_room_manager = ClientRoomManager()
+    _live_msg_handler = LiveMsgHandler()
+
+
+async def shut_down():
+    if client_room_manager is not None:
+        client_room_manager.shut_down()
+    if _live_client_manager is not None:
+        await _live_client_manager.shut_down()
+
+
+class LiveClientManager:
+    """管理到B站的连接"""
+    def __init__(self):
+        self._live_clients: Dict[RoomKey, LiveClientType] = {}
+        self._close_client_futures: Set[asyncio.Future] = set()
+
+    async def shut_down(self):
+        while len(self._live_clients) != 0:
+            room_key = next(iter(self._live_clients))
+            self.del_live_client(room_key)
+
+        await asyncio.gather(*self._close_client_futures, return_exceptions=True)
+
+    def add_live_client(self, room_key: RoomKey):
+        if room_key in self._live_clients:
+            return
+
+        logger.info('room=%s creating live client', room_key)
+
+        self._live_clients[room_key] = live_client = self._create_live_client(room_key)
+        live_client.set_handler(_live_msg_handler)
+        # 直接启动吧，这里不用管init_room失败的情况，万一失败了会在on_client_stopped里删除掉这个客户端
+        live_client.start()
+
+        logger.info('room=%s live client created, %d live clients', room_key, len(self._live_clients))
+
+    @staticmethod
+    def _create_live_client(room_key: RoomKey):
+        if room_key.type == RoomKeyType.ROOM_ID:
+            return WebLiveClient(room_key)
+        elif room_key.type == RoomKeyType.AUTH_CODE:
+            return OpenLiveClient(room_key)
+        raise ValueError(f'Unknown RoomKeyType={room_key.type}')
+
+    def del_live_client(self, room_key: RoomKey):
+        live_client = self._live_clients.pop(room_key, None)
+        if live_client is None:
+            return
+
+        logger.info('room=%s removing live client', room_key)
+
+        live_client.set_handler(None)
+        future = asyncio.create_task(live_client.stop_and_close())
+        self._close_client_futures.add(future)
+        future.add_done_callback(lambda _future: self._close_client_futures.discard(future))
+
+        logger.info('room=%s live client removed, %d live clients', room_key, len(self._live_clients))
+
+        client_room_manager.del_room(room_key)
+
+
+class WebLiveClient(blivedm.BLiveClient):
+    HEARTBEAT_INTERVAL = 10
+
+    def __init__(self, room_key: RoomKey):
+        assert room_key.type == RoomKeyType.ROOM_ID
+        super().__init__(
+            room_key.value,
+            uid=0,
+            session=utils.request.http_session,
+            heartbeat_interval=self.HEARTBEAT_INTERVAL,
+        )
+
+    @property
+    def room_key(self):
+        return RoomKey(RoomKeyType.ROOM_ID, self.tmp_room_id)
+
+    async def init_room(self):
+        res = await super().init_room()
+        if res:
+            logger.info('room=%s live client init succeeded, room_id=%d', self.room_key, self.room_id)
+        else:
+            logger.info('room=%s live client init with a downgrade, room_id=%d', self.room_key, self.room_id)
+        # 允许降级
+        return True
+
+
+class OpenLiveClient(blivedm.OpenLiveClient):
+    HEARTBEAT_INTERVAL = 10
+
+    def __init__(self, room_key: RoomKey):
+        assert room_key.type == RoomKeyType.AUTH_CODE
+        cfg = config.get_config()
+        super().__init__(
+            access_key_id=cfg.open_live_access_key_id,
+            access_key_secret=cfg.open_live_access_key_secret,
+            app_id=cfg.open_live_app_id,
+            room_owner_auth_code=room_key.value,
+            session=utils.request.http_session,
+            heartbeat_interval=self.HEARTBEAT_INTERVAL,
+        )
+
+    @property
+    def room_key(self):
+        return RoomKey(RoomKeyType.AUTH_CODE, self.room_owner_auth_code)
+
+    async def init_room(self):
+        res = await super().init_room()
+        if res:
+            logger.info('room=%s live client init succeeded, room_id=%d', self.room_key, self.room_id)
+        else:
+            logger.info('room=%s live client init failed', self.room_key)
+        return res
+
+    async def _start_game(self):
+        try:
+            data = await api_open_live.request_open_live_or_common_server(
+                api_open_live.START_GAME_OPEN_LIVE_URL,
+                api_open_live.START_GAME_COMMON_SERVER_URL,
+                {'code': self._room_owner_auth_code, 'app_id': self._app_id}
+            )
+        except api_open_live.TransportError:
+            logger.error('_start_game() failed')
+            return False
+        except api_open_live.BusinessError:
+            logger.warning('_start_game() failed')
+            return False
+        return self._parse_start_game(data['data'])
+
+    async def _end_game(self):
+        if self._game_id in (None, ''):
+            return True
+
+        try:
+            await api_open_live.request_open_live_or_common_server(
+                api_open_live.END_GAME_OPEN_LIVE_URL,
+                api_open_live.END_GAME_COMMON_SERVER_URL,
+                {'app_id': self._app_id, 'game_id': self._game_id}
+            )
+        except api_open_live.TransportError:
+            logger.error('room=%d _end_game() failed', self.room_id)
+            return False
+        except api_open_live.BusinessError as e:
+            if e.code in (7000, 7003):
+                # 项目已经关闭了也算成功
+                return True
+            logger.warning('room=%d _end_game() failed', self.room_id)
+            return False
+        return True
+
+    async def _send_game_heartbeat(self):
+        if self._game_id in (None, ''):
+            logger.warning('game=%d _send_game_heartbeat() failed, game_id not found', self._game_id)
+            return False
+
+        # 保存一下，防止await之后game_id改变
+        game_id = self._game_id
+        try:
+            await api_open_live.request_open_live_or_common_server(
+                api_open_live.GAME_HEARTBEAT_OPEN_LIVE_URL,
+                api_open_live.GAME_HEARTBEAT_COMMON_SERVER_URL,
+                {'game_id': game_id}
+            )
+        except api_open_live.TransportError:
+            logger.error('room=%d _send_game_heartbeat() failed', self.room_id)
+            return False
+        except api_open_live.BusinessError as e:
+            logger.warning('room=%d _send_game_heartbeat() failed', self.room_id)
+            if e.code == 7003 and self._game_id == game_id:
+                # 项目异常关闭，可能是心跳超时，需要重新开启项目
+                self._need_init_room = True
+                if self._websocket is not None and not self._websocket.closed:
+                    await self._websocket.close()
+            return False
+        return True
+
+
+class ClientRoomManager:
+    """管理到客户端的连接"""
+    # 房间没有客户端后延迟多久删除房间，不立即删除防止短时间后重连
+    DELAY_DEL_ROOM_TIMEOUT = 10
+
+    def __init__(self):
+        self._rooms: Dict[RoomKey, ClientRoom] = {}
+        self._delay_del_timer_handles: Dict[RoomKey, asyncio.TimerHandle] = {}
+
+    def shut_down(self):
+        while len(self._rooms) != 0:
+            room_key = next(iter(self._rooms))
+            self.del_room(room_key)
+
+        for timer_handle in self._delay_del_timer_handles.values():
+            timer_handle.cancel()
+        self._delay_del_timer_handles.clear()
+
+    def add_client(self, room_key: RoomKey, client: 'api.chat.ChatHandler'):
+        room = self._get_or_add_room(room_key)
+        room.add_client(client)
+
+        self._clear_delay_del_timer(room_key)
+
+    def del_client(self, room_key: RoomKey, client: 'api.chat.ChatHandler'):
+        room = self.get_room(room_key)
+        if room is None:
+            return
+
+        room.del_client(client)
+
+        if room.client_count == 0:
+            self.delay_del_room(room_key, self.DELAY_DEL_ROOM_TIMEOUT)
+
+    def get_room(self, room_key: RoomKey):
+        return self._rooms.get(room_key, None)
+
+    def _get_or_add_room(self, room_key: RoomKey):
+        room = self._rooms.get(room_key, None)
+        if room is None:
+            logger.info('room=%s creating client room', room_key)
+            self._rooms[room_key] = room = ClientRoom(room_key)
+            logger.info('room=%s client room created, %d client rooms', room_key, len(self._rooms))
+
+            _live_client_manager.add_live_client(room_key)
+        return room
+
+    def del_room(self, room_key: RoomKey):
+        self._clear_delay_del_timer(room_key)
+
+        room = self._rooms.pop(room_key, None)
+        if room is None:
+            return
+
+        logger.info('room=%s removing client room', room_key)
+        room.clear_clients()
+        logger.info('room=%s client room removed, %d client rooms', room_key, len(self._rooms))
+
+        _live_client_manager.del_live_client(room_key)
+
+    def delay_del_room(self, room_key: RoomKey, timeout):
+        self._clear_delay_del_timer(room_key)
+        self._delay_del_timer_handles[room_key] = asyncio.get_running_loop().call_later(
+            timeout, self._on_delay_del_room, room_key
+        )
+
+    def _clear_delay_del_timer(self, room_key: RoomKey):
+        timer_handle = self._delay_del_timer_handles.pop(room_key, None)
+        if timer_handle is not None:
+            timer_handle.cancel()
+
+    def _on_delay_del_room(self, room_key: RoomKey):
+        self._delay_del_timer_handles.pop(room_key, None)
+        self.del_room(room_key)
+
+
+class ClientRoom:
+    def __init__(self, room_key: RoomKey):
+        self._room_key = room_key
+        self._clients: List[api.chat.ChatHandler] = []
+        self._auto_translate_count = 0
+
+    @property
+    def room_key(self) -> RoomKey:
+        return self._room_key
+
+    @property
+    def client_count(self):
+        return len(self._clients)
+
+    @property
+    def need_translate(self):
+        return self._auto_translate_count > 0
+
+    def add_client(self, client: 'api.chat.ChatHandler'):
+        logger.info('room=%s addding client %s', self._room_key, client.request.remote_ip)
+
+        self._clients.append(client)
+        if client.auto_translate:
+            self._auto_translate_count += 1
+
+        logger.info('room=%s added client %s, %d clients', self._room_key, client.request.remote_ip,
+                    self.client_count)
+
+    def del_client(self, client: 'api.chat.ChatHandler'):
+        client.close()
+        try:
+            self._clients.remove(client)
+        except ValueError:
+            return
+        if client.auto_translate:
+            self._auto_translate_count -= 1
+
+        logger.info('room=%s removed client %s, %d clients', self._room_key, client.request.remote_ip,
+                    self.client_count)
+
+    def clear_clients(self):
+        logger.info('room=%s clearing %d clients', self._room_key, self.client_count)
+
+        for client in self._clients:
+            client.close()
+        self._clients.clear()
+        self._auto_translate_count = 0
+
+    def send_cmd_data(self, cmd, data):
+        body = api.chat.make_message_body(cmd, data)
+        for client in self._clients:
+            client.send_body_no_raise(body)
+
+    def send_cmd_data_if(self, filterer: Callable[['api.chat.ChatHandler'], bool], cmd, data):
+        body = api.chat.make_message_body(cmd, data)
+        for client in filter(filterer, self._clients):
+            client.send_body_no_raise(body)
+
+
+class LiveMsgHandler(blivedm.BaseHandler):
+    def on_client_stopped(self, client: LiveClientType, exception: Optional[Exception]):
+        _live_client_manager.del_live_client(client.room_key)
+
+    def _on_danmaku(self, client: WebLiveClient, message: dm_web_models.DanmakuMessage):
+        asyncio.create_task(self.__on_danmaku(client, message))
+
+    async def __on_danmaku(self, client: WebLiveClient, message: dm_web_models.DanmakuMessage):
+        avatar_url = message.face
+        if avatar_url != '':
+            services.avatar.update_avatar_cache_if_expired(message.uid, avatar_url)
+        else:
+            # 先异步调用再获取房间，因为返回时房间可能已经不存在了
+            avatar_url = await services.avatar.get_avatar_url(message.uid)
+
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        if message.uid == client.room_owner_uid:
+            author_type = 3  # 主播
+        elif message.admin:
+            author_type = 2  # 房管
+        elif message.privilege_type != 0:  # 1总督，2提督，3舰长
+            author_type = 1  # 舰队
+        else:
+            author_type = 0
+
+        if message.dm_type == 1:
+            content_type = api.chat.ContentType.EMOTICON
+            content_type_params = api.chat.make_emoticon_params(
+                message.emoticon_options_dict['url'],
+            )
+        else:
+            content_type = api.chat.ContentType.TEXT
+            content_type_params = None
+
+        need_translate = (
+            content_type != api.chat.ContentType.EMOTICON and self._need_translate(message.msg, room, client)
+        )
+        if need_translate:
+            translation = services.translate.get_translation_from_cache(message.msg)
+            if translation is None:
+                # 没有缓存，需要后面异步翻译后通知
+                translation = ''
+            else:
+                need_translate = False
+        else:
+            translation = ''
+
+        msg_id = uuid.uuid4().hex
+        room.send_cmd_data(api.chat.Command.ADD_TEXT, api.chat.make_text_message_data(
+            avatar_url=avatar_url,
+            timestamp=int(message.timestamp / 1000),
+            author_name=message.uname,
+            author_type=author_type,
+            content=message.msg,
+            privilege_type=message.privilege_type,
+            is_gift_danmaku=bool(message.msg_type),
+            author_level=message.user_level,
+            is_newbie=message.urank < 10000,
+            is_mobile_verified=bool(message.mobile_verify),
+            medal_level=0 if message.medal_room_id != client.room_id else message.medal_level,
+            id_=msg_id,
+            translation=translation,
+            content_type=content_type,
+            content_type_params=content_type_params,
+        ))
+
+        if need_translate:
+            await self._translate_and_response(message.msg, room.room_key, msg_id)
+
+    def _on_gift(self, client: WebLiveClient, message: dm_web_models.GiftMessage):
+        avatar_url = services.avatar.process_avatar_url(message.face)
+        services.avatar.update_avatar_cache_if_expired(message.uid, avatar_url)
+
+        # 丢人
+        if message.coin_type != 'gold':
+            return
+
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        room.send_cmd_data(api.chat.Command.ADD_GIFT, {
+            'id': uuid.uuid4().hex,
+            'avatarUrl': avatar_url,
+            'timestamp': message.timestamp,
+            'authorName': message.uname,
+            'totalCoin': message.total_coin,
+            'giftName': message.gift_name,
+            'num': message.num
+        })
+
+    def _on_buy_guard(self, client: WebLiveClient, message: dm_web_models.GuardBuyMessage):
+        asyncio.create_task(self.__on_buy_guard(client, message))
+
+    @staticmethod
+    async def __on_buy_guard(client: WebLiveClient, message: dm_web_models.GuardBuyMessage):
+        # 先异步调用再获取房间，因为返回时房间可能已经不存在了
+        avatar_url = await services.avatar.get_avatar_url(message.uid)
+
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        room.send_cmd_data(api.chat.Command.ADD_MEMBER, {
+            'id': uuid.uuid4().hex,
+            'avatarUrl': avatar_url,
+            'timestamp': message.start_time,
+            'authorName': message.username,
+            'privilegeType': message.guard_level
+        })
+
+    def _on_super_chat(self, client: WebLiveClient, message: dm_web_models.SuperChatMessage):
+        avatar_url = services.avatar.process_avatar_url(message.face)
+        services.avatar.update_avatar_cache_if_expired(message.uid, avatar_url)
+
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        need_translate = self._need_translate(message.message, room, client)
+        if need_translate:
+            translation = services.translate.get_translation_from_cache(message.message)
+            if translation is None:
+                # 没有缓存，需要后面异步翻译后通知
+                translation = ''
+            else:
+                need_translate = False
+        else:
+            translation = ''
+
+        msg_id = str(message.id)
+        room.send_cmd_data(api.chat.Command.ADD_SUPER_CHAT, {
+            'id': msg_id,
+            'avatarUrl': avatar_url,
+            'timestamp': message.start_time,
+            'authorName': message.uname,
+            'price': message.price,
+            'content': message.message,
+            'translation': translation
+        })
+
+        if need_translate:
+            asyncio.create_task(self._translate_and_response(
+                message.message, room.room_key, msg_id, services.translate.Priority.HIGH
+            ))
+
+    def _on_super_chat_delete(self, client: WebLiveClient, message: dm_web_models.SuperChatDeleteMessage):
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        room.send_cmd_data(api.chat.Command.DEL_SUPER_CHAT, {
+            'ids': list(map(str, message.ids))
+        })
+
+    @staticmethod
+    def _need_translate(text, room: ClientRoom, client: LiveClientType):
+        cfg = config.get_config()
+        return (
+            cfg.enable_translate
+            and room.need_translate
+            and (not cfg.allow_translate_rooms or client.room_id in cfg.allow_translate_rooms)
+            and services.translate.need_translate(text)
+        )
+
+    @staticmethod
+    async def _translate_and_response(text, room_key: RoomKey, msg_id, priority=services.translate.Priority.NORMAL):
+        translation = await services.translate.translate(text, priority)
+        if translation is None:
+            return
+
+        room = client_room_manager.get_room(room_key)
+        if room is None:
+            return
+
+        room.send_cmd_data_if(
+            lambda client: client.auto_translate,
+            api.chat.Command.UPDATE_TRANSLATION,
+            api.chat.make_translation_message_data(
+                msg_id,
+                translation
+            )
+        )
+
+    #
+    # 开放平台消息
+    #
+
+    def _on_open_live_danmaku(self, client: OpenLiveClient, message: dm_open_models.DanmakuMessage):
+        avatar_url = message.uface
+        services.avatar.update_avatar_cache_if_expired(message.uid, avatar_url)
+
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        if message.uid == client.room_owner_uid:
+            author_type = 3  # 主播
+        elif message.guard_level != 0:  # 1总督，2提督，3舰长
+            author_type = 1  # 舰队
+        else:
+            author_type = 0
+
+        if message.dm_type == 1:
+            content_type = api.chat.ContentType.EMOTICON
+            content_type_params = api.chat.make_emoticon_params(message.emoji_img_url)
+        else:
+            content_type = api.chat.ContentType.TEXT
+            content_type_params = None
+
+        need_translate = (
+            content_type != api.chat.ContentType.EMOTICON and self._need_translate(message.msg, room, client)
+        )
+        if need_translate:
+            translation = services.translate.get_translation_from_cache(message.msg)
+            if translation is None:
+                # 没有缓存，需要后面异步翻译后通知
+                translation = ''
+            else:
+                need_translate = False
+        else:
+            translation = ''
+
+        room.send_cmd_data(api.chat.Command.ADD_TEXT, api.chat.make_text_message_data(
+            avatar_url=avatar_url,
+            timestamp=message.timestamp,
+            author_name=message.uname,
+            author_type=author_type,
+            content=message.msg,
+            privilege_type=message.guard_level,
+            medal_level=0 if not message.fans_medal_wearing_status else message.fans_medal_level,
+            id_=message.msg_id,
+            translation=translation,
+            content_type=content_type,
+            content_type_params=content_type_params,
+        ))
+
+        if need_translate:
+            asyncio.create_task(self._translate_and_response(message.msg, room.room_key, message.msg_id))
+
+    def _on_open_live_gift(self, client: OpenLiveClient, message: dm_open_models.GiftMessage):
+        avatar_url = services.avatar.process_avatar_url(message.uface)
+        services.avatar.update_avatar_cache_if_expired(message.uid, avatar_url)
+
+        # 丢人
+        if not message.paid:
+            return
+
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        room.send_cmd_data(api.chat.Command.ADD_GIFT, {
+            'id': message.msg_id,
+            'avatarUrl': avatar_url,
+            'timestamp': message.timestamp,
+            'authorName': message.uname,
+            'totalCoin': message.price,
+            'giftName': message.gift_name,
+            'num': message.gift_num
+        })
+
+    def _on_open_live_buy_guard(self, client: OpenLiveClient, message: dm_open_models.GuardBuyMessage):
+        avatar_url = message.user_info.uface
+        services.avatar.update_avatar_cache_if_expired(message.user_info.uid, avatar_url)
+
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        room.send_cmd_data(api.chat.Command.ADD_MEMBER, {
+            'id': message.msg_id,
+            'avatarUrl': avatar_url,
+            'timestamp': message.timestamp,
+            'authorName': message.user_info.uname,
+            'privilegeType': message.guard_level
+        })
+
+    def _on_open_live_super_chat(self, client: OpenLiveClient, message: dm_open_models.SuperChatMessage):
+        avatar_url = services.avatar.process_avatar_url(message.uface)
+        services.avatar.update_avatar_cache_if_expired(message.uid, avatar_url)
+
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        need_translate = self._need_translate(message.message, room, client)
+        if need_translate:
+            translation = services.translate.get_translation_from_cache(message.message)
+            if translation is None:
+                # 没有缓存，需要后面异步翻译后通知
+                translation = ''
+            else:
+                need_translate = False
+        else:
+            translation = ''
+
+        msg_id = str(message.message_id)
+        room.send_cmd_data(api.chat.Command.ADD_SUPER_CHAT, {
+            'id': msg_id,
+            'avatarUrl': avatar_url,
+            'timestamp': message.start_time,
+            'authorName': message.uname,
+            'price': message.rmb,
+            'content': message.message,
+            'translation': translation
+        })
+
+        if need_translate:
+            asyncio.create_task(self._translate_and_response(
+                message.message, room.room_key, msg_id, services.translate.Priority.HIGH
+            ))
+
+    def _on_open_live_super_chat_delete(self, client: OpenLiveClient, message: dm_open_models.SuperChatDeleteMessage):
+        room = client_room_manager.get_room(client.room_key)
+        if room is None:
+            return
+
+        room.send_cmd_data(api.chat.Command.DEL_SUPER_CHAT, {
+            'ids': list(map(str, message.message_ids))
+        })

services/translate.py

@@ -1,7 +1,9 @@
 # -*- coding: utf-8 -*-
-
 import asyncio
+import base64
+import dataclasses
 import datetime
+import enum
 import functools
 import hashlib
 import hmac
@@ -11,9 +13,13 @@ import random
 import re
 from typing import *
 
+import Crypto.Cipher.AES as cry_aes  # noqa
+import Crypto.Util.Padding as cry_pad  # noqa
 import aiohttp
+import cachetools
 
 import config
+import utils.request
 
 logger = logging.getLogger(__name__)
 
@@ -22,23 +28,38 @@ NO_TRANSLATE_TEXTS = {
     '强', '余裕', '余裕余裕', '大丈夫', '再放送', '放送事故', '清楚', '清楚清楚'
 }
 
-_main_event_loop = asyncio.get_event_loop()
-_http_session = None
 _translate_providers: List['TranslateProvider'] = []
 # text -> res
-_translate_cache: Dict[str, str] = {}
+_translate_cache: Optional[cachetools.LRUCache] = None
 # 正在翻译的Future，text -> Future
 _text_future_map: Dict[str, asyncio.Future] = {}
+# 正在翻译的任务队列，索引是优先级
+_task_queues: List['asyncio.Queue[TranslateTask]'] = []
+
+
+class Priority(enum.IntEnum):
+    HIGH = 0
+    NORMAL = 1
+
+
+@dataclasses.dataclass
+class TranslateTask:
+    priority: Priority
+    text: str
+    future: 'asyncio.Future[Optional[str]]'
+    remain_retry_count: int
 
 
 def init():
-    asyncio.ensure_future(_do_init())
+    cfg = config.get_config()
+    global _translate_cache, _task_queues
+    _translate_cache = cachetools.LRUCache(cfg.translation_cache_size)
+    # 总队列长度会超过translate_max_queue_size，不用这么严格
+    _task_queues = [asyncio.Queue(cfg.translate_max_queue_size) for _ in range(len(Priority))]
+    asyncio.get_running_loop().create_task(_do_init())
 
 
 async def _do_init():
-    global _http_session
-    _http_session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=10))
-
     cfg = config.get_config()
     if not cfg.enable_translate:
         return
@@ -56,21 +77,17 @@ def create_translate_provider(cfg):
     type_ = cfg['type']
     if type_ == 'TencentTranslateFree':
         return TencentTranslateFree(
-            cfg['query_interval'], cfg['max_queue_size'], cfg['source_language'],
-            cfg['target_language']
+            cfg['query_interval'], cfg['source_language'], cfg['target_language']
         )
-    elif type_ == 'BilibiliTranslateFree':
-        return BilibiliTranslateFree(cfg['query_interval'], cfg['max_queue_size'])
     elif type_ == 'TencentTranslate':
         return TencentTranslate(
-            cfg['query_interval'], cfg['max_queue_size'], cfg['source_language'],
-            cfg['target_language'], cfg['secret_id'], cfg['secret_key'],
-            cfg['region']
+            cfg['query_interval'], cfg['source_language'], cfg['target_language'],
+            cfg['secret_id'], cfg['secret_key'], cfg['region']
         )
     elif type_ == 'BaiduTranslate':
         return BaiduTranslate(
-            cfg['query_interval'], cfg['max_queue_size'], cfg['source_language'],
-            cfg['target_language'], cfg['app_id'], cfg['secret']
+            cfg['query_interval'], cfg['source_language'], cfg['target_language'],
+            cfg['app_id'], cfg['secret']
         )
     return None
 
@@ -97,14 +114,14 @@ def get_translation_from_cache(text):
     return _translate_cache.get(key, None)
 
 
-def translate(text) -> Awaitable[Optional[str]]:
+def translate(text, priority=Priority.NORMAL) -> Awaitable[Optional[str]]:
     key = text.strip().lower()
     # 如果已有正在翻译的future则返回，防止重复翻译
     future = _text_future_map.get(key, None)
     if future is not None:
         return future
     # 否则创建一个翻译任务
-    future = _main_event_loop.create_future()
+    future = asyncio.get_running_loop().create_future()
 
     # 查缓存
     res = _translate_cache.get(key, None)
@@ -112,25 +129,18 @@ def translate(text) -> Awaitable[Optional[str]]:
         future.set_result(res)
         return future
 
-    # 负载均衡，找等待时间最少的provider
-    min_wait_time = None
-    min_wait_time_provider = None
-    for provider in _translate_providers:
-        if not provider.is_available:
-            continue
-        wait_time = provider.wait_time
-        if min_wait_time is None or wait_time < min_wait_time:
-            min_wait_time = wait_time
-            min_wait_time_provider = provider
-
-    # 没有可用的
-    if min_wait_time_provider is None:
+    task = TranslateTask(
+        priority=priority,
+        text=text,
+        future=future,
+        remain_retry_count=3 if priority == Priority.HIGH else 1
+    )
+    if not _push_task(task):
         future.set_result(None)
         return future
 
     _text_future_map[key] = future
     future.add_done_callback(functools.partial(_on_translate_done, key))
-    min_wait_time_provider.translate(text, future)
     return future
 
 
@@ -139,119 +149,222 @@ def _on_translate_done(key, future):
     # 缓存
     try:
         res = future.result()
-    except Exception:
+    except Exception:  # noqa
         return
     if res is None:
         return
     _translate_cache[key] = res
-    cfg = config.get_config()
-    while len(_translate_cache) > cfg.translation_cache_size:
-        _translate_cache.pop(next(iter(_translate_cache)), None)
+
+
+def _push_task(task: TranslateTask):
+    if not _has_available_translate_provider():
+        return False
+
+    queue = _task_queues[task.priority]
+    if not queue.full():
+        queue.put_nowait(task)
+        return True
+
+    if task.priority != Priority.HIGH:
+        return False
+
+    # 高优先级的尝试降级，挤掉低优先级的任务
+    queue = _task_queues[Priority.NORMAL]
+    if queue.full():
+        lower_task = queue.get_nowait()
+        lower_task.future.set_result(None)
+    queue.put_nowait(task)
+    return True
+
+
+async def _pop_task() -> TranslateTask:
+    # 按优先级遍历，看是否已经有任务
+    for queue in _task_queues:
+        if not queue.empty():
+            return queue.get_nowait()
+
+    done_future_set, pending_future_set = await asyncio.wait(
+        [asyncio.create_task(queue.get()) for queue in _task_queues],
+        return_when=asyncio.FIRST_COMPLETED
+    )
+    for future in pending_future_set:
+        future.cancel()
+
+    # 如果有多个队列都取到任务了，只返回优先级最高的那一个，剩下的放回队列
+    assert len(done_future_set) != 0
+    tasks = [await future for future in done_future_set]
+    if len(tasks) > 1:
+        tasks.sort(key=lambda task_: task_.priority)
+
+    res = None
+    for task in tasks:
+        if res is None:
+            res = task
+            continue
+
+        if not _push_task(task):
+            task.future.set_result(None)
+    return res
+
+
+def _cancel_all_tasks_if_no_available_translate_provider():
+    if _has_available_translate_provider():
+        return
+
+    logger.warning('No available translate provider')
+    for queue in _task_queues:
+        while not queue.empty():
+            task = queue.get_nowait()
+            task.future.set_result(None)
+
+
+def _has_available_translate_provider():
+    return any(provider.is_available for provider in _translate_providers)
 
 
 class TranslateProvider:
-    async def init(self):
-        return True
-
-    @property
-    def is_available(self):
-        return True
-
-    @property
-    def wait_time(self):
-        return 0
-
-    def translate(self, text, future):
-        raise NotImplementedError
-
-
-class FlowControlTranslateProvider(TranslateProvider):
-    def __init__(self, query_interval, max_queue_size):
+    def __init__(self, query_interval):
         self._query_interval = query_interval
-        # (text, future)
-        self._text_queue = asyncio.Queue(max_queue_size)
+        self._be_available_event = asyncio.Event()
+        self._be_available_event.set()
 
     async def init(self):
-        asyncio.ensure_future(self._translate_consumer())
+        asyncio.create_task(self._translate_consumer())
         return True
 
     @property
     def is_available(self):
-        return not self._text_queue.full()
+        return True
 
-    @property
-    def wait_time(self):
-        return self._text_queue.qsize() * self._query_interval
-
-    def translate(self, text, future):
-        try:
-            self._text_queue.put_nowait((text, future))
-        except asyncio.QueueFull:
-            future.set_result(None)
+    def _on_availability_change(self):
+        if self.is_available:
+            self._be_available_event.set()
+        else:
+            self._be_available_event.clear()
+            _cancel_all_tasks_if_no_available_translate_provider()
 
     async def _translate_consumer(self):
+        cls_name = type(self).__name__
         while True:
             try:
-                text, future = await self._text_queue.get()
-                asyncio.ensure_future(self._translate_coroutine(text, future))
+                if not self.is_available:
+                    logger.info('%s waiting to become available', cls_name)
+                    await self._be_available_event.wait()
+                    logger.info('%s became available', cls_name)
+
+                task = await _pop_task()
+                # 为了简化代码，约定只会在_translate_wrapper里变成不可用，所以获取task之后这里还是可用的
+                assert self.is_available
+
+                start_time = datetime.datetime.now()
+                await self._translate_wrapper(task)
+                cost_time = (datetime.datetime.now() - start_time).total_seconds()
+
                 # 频率限制
-                await asyncio.sleep(self._query_interval)
-            except Exception:
-                logger.exception('FlowControlTranslateProvider error:')
+                await asyncio.sleep(self._query_interval - cost_time)
+            except Exception:  # noqa
+                logger.exception('%s error:', cls_name)
 
-    async def _translate_coroutine(self, text, future):
+    async def _translate_wrapper(self, task: TranslateTask) -> Optional[str]:
         try:
-            res = await self._do_translate(text)
+            exc = None
+            task.remain_retry_count -= 1
+            res = await self._do_translate(task.text)
         except BaseException as e:
-            future.set_exception(e)
-        else:
-            future.set_result(res)
+            exc = e
+            res = None
+        if res is not None:
+            task.future.set_result(res)
+            return res
 
-    async def _do_translate(self, text):
+        if task.remain_retry_count > 0:
+            # 还可以重试则放回队列
+            if not _push_task(task):
+                task.future.set_result(None)
+        else:
+            # 否则设置异常或None结果
+            if exc is not None:
+                task.future.set_exception(exc)
+            else:
+                task.future.set_result(None)
+        return None
+
+    async def _do_translate(self, text) -> Optional[str]:
         raise NotImplementedError
 
 
-class TencentTranslateFree(FlowControlTranslateProvider):
-    def __init__(self, query_interval, max_queue_size, source_language, target_language):
-        super().__init__(query_interval, max_queue_size)
+class TencentTranslateFree(TranslateProvider):
+    def __init__(self, query_interval, source_language, target_language):
+        super().__init__(query_interval)
+        self._be_available_event.clear()  # _do_init之后才可用
         self._source_language = source_language
         self._target_language = target_language
 
-        self._qtv = ''
-        self._qtk = ''
+        self._server_time_delta = 0
+        self._uc_key = self._uc_iv = ''
+        self._qtv = self._qtk = ''
         self._reinit_future = None
+
         # 连续失败的次数
         self._fail_count = 0
 
     async def init(self):
         if not await super().init():
             return False
-        if not await self._do_init():
-            return False
-        self._reinit_future = asyncio.ensure_future(self._reinit_coroutine())
+        self._reinit_future = asyncio.create_task(self._reinit_coroutine())
         return True
 
     async def _do_init(self):
         try:
-            async with _http_session.get('https://fanyi.qq.com/') as r:
+            async with utils.request.http_session.get('https://fanyi.qq.com/') as r:
                 if r.status != 200:
                     logger.warning('TencentTranslateFree init request failed: status=%d %s', r.status, r.reason)
                     return False
                 html = await r.text()
 
-            m = re.search(r"""\breauthuri\s*=\s*['"](.+?)['"]""", html)
-            if m is None:
-                logger.exception('TencentTranslateFree init failed: reauthuri not found')
-                return False
-            reauthuri = m[1]
+                try:
+                    server_time = r.headers['Date']
+                    server_time = datetime.datetime.strptime(server_time, '%a, %d %b %Y %H:%M:%S GMT')
+                    server_time = server_time.replace(tzinfo=datetime.timezone.utc).timestamp()
+                    self._server_time_delta = int((datetime.datetime.now().timestamp() - server_time) * 1000)
+                except (KeyError, ValueError):
+                    self._server_time_delta = 0
+        except (aiohttp.ClientError, asyncio.TimeoutError):
+            logger.exception('TencentTranslateFree init error:')
+            return False
 
-            async with _http_session.post('https://fanyi.qq.com/api/' + reauthuri) as r:
+        # 获取token URL
+        m = re.search(r"""\breauthuri\s*=\s*['"](.+?)['"]""", html)
+        if m is None:
+            logger.exception('TencentTranslateFree init failed: reauthuri not found')
+            return False
+        reauthuri = m[1]
+
+        # 获取验证用的key、iv
+        m = re.search(r"""\s*=\s*['"]((?:\w+\|\w+-)+\w+\|\w+)['"]""", html)
+        if m is None:
+            logger.exception('TencentTranslateFree init failed: initial global variables not found')
+            return False
+        uc_key = None
+        uc_iv = None
+        for item in m[1].split('-'):
+            key, _, value = item.partition('|')
+            if key == 'a137':
+                uc_key = value
+            elif key == 'E74':
+                uc_iv = value
+            if uc_key is not None and uc_iv is not None:
+                break
+
+        # 获取token
+        try:
+            async with utils.request.http_session.post('https://fanyi.qq.com/api/' + reauthuri) as r:
                 if r.status != 200:
                     logger.warning('TencentTranslateFree init request failed: reauthuri=%s, status=%d %s',
                                    reauthuri, r.status, r.reason)
                     return False
                 data = await r.json()
-        except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
+        except (aiohttp.ClientError, asyncio.TimeoutError):
             logger.exception('TencentTranslateFree init error:')
             return False
 
@@ -264,42 +377,45 @@ class TencentTranslateFree(FlowControlTranslateProvider):
             logger.warning('TencentTranslateFree init failed: qtk not found')
             return False
 
+        self._uc_key = uc_key
+        self._uc_iv = uc_iv
         self._qtv = qtv
         self._qtk = qtk
+
+        self._on_availability_change()
         return True
 
     async def _reinit_coroutine(self):
-        try:
-            while True:
-                await asyncio.sleep(30)
-                logger.debug('TencentTranslateFree reinit')
-                asyncio.ensure_future(self._do_init())
-        except asyncio.CancelledError:
-            pass
+        while True:
+            logger.debug('TencentTranslateFree reinit')
+            start_time = datetime.datetime.now()
+            try:
+                await self._do_init()
+            except Exception:  # noqa
+                pass
+            cost_time = (datetime.datetime.now() - start_time).total_seconds()
+
+            await asyncio.sleep(30 - cost_time)
 
     @property
     def is_available(self):
-        return self._qtv != '' and self._qtk != '' and super().is_available
+        return '' not in (self._uc_key, self._uc_iv, self._qtv, self._qtk) and super().is_available
 
-    async def _translate_coroutine(self, text, future):
-        try:
-            res = await self._do_translate(text)
-        except BaseException as e:
-            future.set_exception(e)
-            self._on_fail()
-            return
-        future.set_result(res)
-        if res is None:
-            self._on_fail()
-        else:
+    async def _translate_wrapper(self, task: TranslateTask) -> Optional[str]:
+        res = await super()._translate_wrapper(task)
+        if res is not None:
             self._fail_count = 0
+        else:
+            self._on_fail()
+        return res
 
-    async def _do_translate(self, text):
+    async def _do_translate(self, text) -> Optional[str]:
         try:
-            async with _http_session.post(
+            async with utils.request.http_session.post(
                 'https://fanyi.qq.com/api/translate',
                 headers={
-                    'Referer': 'https://fanyi.qq.com/'
+                    'Referer': 'https://fanyi.qq.com/',
+                    'uc': self._get_uc()
                 },
                 data={
                     'source': self._source_language,
@@ -312,8 +428,9 @@ class TencentTranslateFree(FlowControlTranslateProvider):
                 if r.status != 200:
                     logger.warning('TencentTranslateFree request failed: status=%d %s', r.status, r.reason)
                     return None
+                self._update_uc_key(r)
                 data = await r.json()
-        except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
+        except (aiohttp.ClientError, asyncio.TimeoutError):
             return None
         if data['errCode'] != 0:
             logger.warning('TencentTranslateFree failed: %d %s', data['errCode'], data['errMsg'])
@@ -321,55 +438,73 @@ class TencentTranslateFree(FlowControlTranslateProvider):
         res = ''.join(record['targetText'] for record in data['translate']['records'])
         if res == '' and text.strip() != '':
             # qtv、qtk过期
-            logger.warning('TencentTranslateFree result is empty %s', data)
+            logger.info('TencentTranslateFree result is empty %s', data)
             return None
         return res
 
+    def _get_uc(self):
+        user_actions = self._gen_user_actions()
+        cur_timestamp = str(int(datetime.datetime.now().timestamp() * 1000))
+        server_time_delta = str(self._server_time_delta)
+        uc = '|'.join([user_actions, cur_timestamp, server_time_delta])
+
+        aes = cry_aes.new(self._uc_key.encode('utf-8'), cry_aes.MODE_CBC, self._uc_iv.encode('utf-8'))
+        uc = cry_pad.pad(uc.encode('utf-8'), aes.block_size, 'pkcs7')
+        uc = aes.encrypt(uc)
+        uc = base64.b64encode(uc).decode('utf-8')
+        return uc
+
+    @staticmethod
+    def _gen_user_actions():
+        # 1：点击翻译；2：源输入框聚焦或失去焦点；3：点击源语言列表；4：点击交换语言；5：点击目标语言列表；6：源输入框输入、粘贴
+        user_actions = []
+        if random.randint(1, 5) == 1:
+            for i in range(random.randint(1, 2)):
+                user_actions.append('2')
+        user_actions.append('6')
+        for i in range(random.randint(0, 6)):
+            user_actions.append(random.choice('26'))
+        if random.randint(1, 5) == 1:
+            user_actions.append('1')
+        return ''.join(user_actions)
+
+    def _update_uc_key(self, r):
+        try:
+            hf_f = r.headers['f']
+            hf_ts = int(r.headers['ts'])
+        except (KeyError, ValueError):
+            return
+
+        cur_timestamp = int(datetime.datetime.now().timestamp() * 1000)
+        hf_f = base64.b64decode(hf_f.encode('utf-8')).decode('utf-8')
+        pos = int(hf_f[72: 72 + 4])
+        uc_key = hf_f[pos: pos + 16]
+        uc_iv = hf_f[pos + 16: pos + 16 + 16]
+
+        self._server_time_delta = cur_timestamp - hf_ts
+        self._uc_key = uc_key
+        self._uc_iv = uc_iv
+
     def _on_fail(self):
         self._fail_count += 1
-        # 目前没有测试出被ban的情况，为了可靠性，连续失败20次时冷却直到下次重新init
-        if self._fail_count >= 20:
+        # 为了可靠性，连续失败5次时冷却直到下次重新init
+        if self._fail_count >= 5:
             self._cool_down()
 
     def _cool_down(self):
         logger.info('TencentTranslateFree is cooling down')
         # 下次_do_init后恢复
+        self._uc_key = self._uc_iv = ''
         self._qtv = self._qtk = ''
         self._fail_count = 0
 
-
-class BilibiliTranslateFree(FlowControlTranslateProvider):
-    def __init__(self, query_interval, max_queue_size):
-        super().__init__(query_interval, max_queue_size)
-
-    async def _do_translate(self, text):
-        try:
-            async with _http_session.get(
-                'https://api.live.bilibili.com/av/v1/SuperChat/messageTranslate',
-                params={
-                    'room_id': '21396545',
-                    'ruid': '407106379',
-                    'parent_area_id': '9',
-                    'area_id': '371',
-                    'msg': text
-                }
-            ) as r:
-                if r.status != 200:
-                    logger.warning('BilibiliTranslateFree request failed: status=%d %s', r.status, r.reason)
-                    return None
-                data = await r.json()
-        except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
-            return None
-        if data['code'] != 0:
-            logger.warning('BilibiliTranslateFree failed: %d %s', data['code'], data['msg'])
-            return None
-        return data['data']['message_trans']
+        self._on_availability_change()
 
 
-class TencentTranslate(FlowControlTranslateProvider):
-    def __init__(self, query_interval, max_queue_size, source_language, target_language,
+class TencentTranslate(TranslateProvider):
+    def __init__(self, query_interval, source_language, target_language,
                  secret_id, secret_key, region):
-        super().__init__(query_interval, max_queue_size)
+        super().__init__(query_interval)
         self._source_language = source_language
         self._target_language = target_language
         self._secret_id = secret_id
@@ -382,7 +517,7 @@ class TencentTranslate(FlowControlTranslateProvider):
     def is_available(self):
         return self._cool_down_timer_handle is None and super().is_available
 
-    async def _do_translate(self, text):
+    async def _do_translate(self, text) -> Optional[str]:
         try:
             async with self._request_tencent_cloud(
                 'TextTranslate',
@@ -444,7 +579,7 @@ class TencentTranslate(FlowControlTranslateProvider):
             'X-TC-Region': self._region
         }
 
-        return _http_session.post('https://tmt.tencentcloudapi.com/', headers=headers, data=body_bytes)
+        return utils.request.http_session.post('https://tmt.tencentcloudapi.com/', headers=headers, data=body_bytes)
 
     def _on_fail(self, code):
         if self._cool_down_timer_handle is not None:
@@ -467,18 +602,20 @@ class TencentTranslate(FlowControlTranslateProvider):
             # 需要手动处理，等5分钟
             sleep_time = 5 * 60
         if sleep_time != 0:
-            self._cool_down_timer_handle = asyncio.get_event_loop().call_later(
+            self._cool_down_timer_handle = asyncio.get_running_loop().call_later(
                 sleep_time, self._on_cool_down_timeout
             )
+            self._on_availability_change()
 
     def _on_cool_down_timeout(self):
         self._cool_down_timer_handle = None
+        self._on_availability_change()
 
 
-class BaiduTranslate(FlowControlTranslateProvider):
-    def __init__(self, query_interval, max_queue_size, source_language, target_language,
+class BaiduTranslate(TranslateProvider):
+    def __init__(self, query_interval, source_language, target_language,
                  app_id, secret):
-        super().__init__(query_interval, max_queue_size)
+        super().__init__(query_interval)
         self._source_language = source_language
         self._target_language = target_language
         self._app_id = app_id
@@ -490,9 +627,9 @@ class BaiduTranslate(FlowControlTranslateProvider):
     def is_available(self):
         return self._cool_down_timer_handle is None and super().is_available
 
-    async def _do_translate(self, text):
+    async def _do_translate(self, text) -> Optional[str]:
         try:
-            async with _http_session.post(
+            async with utils.request.http_session.post(
                 'https://fanyi-api.baidu.com/api/trans/vip/translate',
                 data=self._add_sign({
                     'q': text,
@@ -529,9 +666,11 @@ class BaiduTranslate(FlowControlTranslateProvider):
             # 账户余额不足，需要手动处理，等5分钟
             sleep_time = 5 * 60
         if sleep_time != 0:
-            self._cool_down_timer_handle = asyncio.get_event_loop().call_later(
+            self._cool_down_timer_handle = asyncio.get_running_loop().call_later(
                 sleep_time, self._on_cool_down_timeout
             )
+            self._on_availability_change()
 
     def _on_cool_down_timeout(self):
         self._cool_down_timer_handle = None
+        self._on_availability_change()

update.py

@@ -1,27 +1,29 @@
 # -*- coding: utf-8 -*-
-
 import asyncio
 
 import aiohttp
 
-VERSION = 'v1.5.2'
+import utils.request
+
+VERSION = 'v1.8.0'
 
 
 def check_update():
-    asyncio.ensure_future(_do_check_update())
+    asyncio.get_running_loop().create_task(_do_check_update())
 
 
 async def _do_check_update():
     try:
-        async with aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=10)) as session:
-            async with session.get('https://api.github.com/repos/xfgryujk/blivechat/releases/latest') as r:
-                data = await r.json()
-                if data['name'] != VERSION:
-                    print('---------------------------------------------')
-                    print('New version available:', data['name'])
-                    print(data['body'])
-                    print('Download:', data['html_url'])
-                    print('---------------------------------------------')
+        async with utils.request.http_session.get(
+            'https://api.github.com/repos/xfgryujk/blivechat/releases/latest'
+        ) as r:
+            data = await r.json()
+            if data['name'] != VERSION:
+                print('---------------------------------------------')
+                print('New version available:', data['name'])
+                print(data['body'])
+                print('Download:', data['html_url'])
+                print('---------------------------------------------')
     except aiohttp.ClientConnectionError:
         print('Failed to check update: connection failed')
     except asyncio.TimeoutError:

utils/request.py

@@ -0,0 +1,24 @@
+# -*- coding: utf-8 -*-
+from typing import *
+
+import aiohttp
+
+# 不带这堆头部有时候也能成功请求，但是带上后成功的概率更高
+BILIBILI_COMMON_HEADERS = {
+    'Origin': 'https://www.bilibili.com',
+    'Referer': 'https://www.bilibili.com/',
+    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko)'
+                  ' Chrome/114.0.0.0 Safari/537.36'
+}
+
+http_session: Optional[aiohttp.ClientSession] = None
+
+
+def init():
+    global http_session
+    http_session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=10))
+
+
+async def shut_down():
+    if http_session is not None:
+        await http_session.close()