add feature sending chat

blivedm/clients/web.py

@@ -9,12 +9,15 @@ import yarl
 from . import ws_base
 from .. import utils
 
+import time
+
 __all__ = (
     'BLiveClient',
 )
 
 logger = logging.getLogger('blivedm')
 
+SEND_CHAT_URL = 'https://api.live.bilibili.com/msg/send'
 UID_INIT_URL = 'https://api.bilibili.com/x/web-interface/nav'
 BUVID_INIT_URL = 'https://data.bilibili.com/v/'
 ROOM_INIT_URL = 'https://api.live.bilibili.com/xlive/web-room/v1/index/getInfoByRoom'
@@ -264,3 +267,45 @@ class BLiveClient(ws_base.WebSocketClientBase):
         if self._host_server_token is not None:
             auth_params['key'] = self._host_server_token
         await self._websocket.send_bytes(self._make_packet(auth_params, ws_base.Operation.AUTH))
+
+    async def send_bulletchat(self, bulletchat: ws_base.BulletChat = None):
+        """
+        :param bulletchat:BulletChat对象。
+        """
+        
+        cookies = self._session.cookie_jar.filter_cookies(yarl.URL(SEND_CHAT_URL))
+        sessdata_cookie = cookies.get('SESSDATA', None)
+        if sessdata_cookie is None or sessdata_cookie.value == '':
+            return False
+        
+        headers = { 'referer': 'https://live.bilibili.com/',
+                    'User-Agent': utils.USER_AGENT}
+        dict_data = {"bubble": bulletchat.bubble,
+                     "msg": bulletchat.msg,
+                     "color": bulletchat.color,
+                     "mode": bulletchat.mode,
+                     "fontsize": bulletchat.fontsize,
+                     "rnd": str(int(time.time())),
+                     "roomid": self.room_id,
+                     "csrf": cookies['bili_jct'].value,
+                     "csrf_token": cookies['bili_jct'].value}
+        dict_cookies = {"SESSDATA": cookies['SESSDATA'].value,
+                        "buvid3": cookies['buvid3'].value,
+                        "bili_jct": cookies['bili_jct'].value,}
+        
+        try:
+            async with self._session.request(
+                method="POST",
+                url=SEND_CHAT_URL,
+                data=dict_data,
+                headers=headers,
+                cookies=dict_cookies,
+            ) as res:
+                if res.status != 200:
+                    logger.warning('send_bulletchat() failed, status=%d, reason=%s', res.status, res.reason)
+                    return False
+                # print(res)
+                return True
+        except (aiohttp.ClientConnectionError, asyncio.TimeoutError):
+            logger.exception('send_bulletchat() failed:')
+            return False

blivedm/clients/ws_base.py

@@ -492,3 +492,24 @@ class WebSocketClientBase:
             self._handler.handle(self, command)
         except Exception as e:
             logger.exception('room=%d _handle_command() failed, command=%s', self.room_id, command, exc_info=e)
+
+class BulletChat:
+    def __init__(self,
+                 bubble: int = 0,
+                 msg: str = None,
+                 color: int = int("FFFFFF", 16),
+                 mode: int = 1,
+                 fontsize: int = 25):
+        """
+        :param bubble:功能未知，默认0。
+        :param msg:弹幕内容，长度需不大于30。
+        :param color:字体颜色。
+        :param mode:弹幕模式。
+        :param fontsize:字体大小。
+        :param roomid:房间号。
+        """
+        self.bubble = bubble
+        self.msg = msg
+        self.color = color
+        self.mode = mode
+        self.fontsize = fontsize

blivedm/requirements.txt

@@ -0,0 +1,3 @@
+aiohttp~=3.9.0
+Brotli~=1.1.0
+yarl~=1.9.3

chatbot.py

@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*-
+
+import time
+
+import asyncio
+import http.cookies
+import random
+from typing import *
+
+import aiohttp
+
+import blivedm
+import blivedm.models.web as web_models
+from blivedm.clients.ws_base import BulletChat
+
+import USER_INFO
+
+# 直播间ID的取值看直播间URL
+TEST_ROOM_ID = 9792252
+
+# 这里填一个已登录账号的cookie的SESSDATA字段的值。不填也可以连接，但是收到弹幕的用户名会打码，UID会变成0
+cookies = http.cookies.SimpleCookie()
+cookies['SESSDATA'] = USER_INFO.SESSDATA
+cookies['SESSDATA']['domain'] = 'bilibili.com'
+cookies['bili_jct'] = USER_INFO.bili_jct
+cookies['buvid3'] = USER_INFO.buvid3
+
+session: Optional[aiohttp.ClientSession] = None
+
+async def dm_monitor():
+    init_session()
+    try:
+        await run_single_client()
+    finally:
+        await session.close()
+
+def init_session():
+    global session
+    session = aiohttp.ClientSession()
+    session.cookie_jar.update_cookies(cookies)
+
+async def run_single_client():
+    room_id = TEST_ROOM_ID
+
+    global client
+    client = blivedm.BLiveClient(room_id, session=session)
+    handler = MyHandler()
+    client.set_handler(handler)
+
+    client.start()
+    try:
+        await client.join()
+    finally:
+        await client.stop_and_close()
+
+class MyHandler(blivedm.BaseHandler):
+    _CMD_CALLBACK_DICT = blivedm.BaseHandler._CMD_CALLBACK_DICT.copy()
+    
+    def __interact_word_callback(self, client: blivedm.BLiveClient, command: dict):
+        uname = command['data']['uname']
+        msg_type = command['data']['msg_type']
+        if msg_type == 2:
+            asyncio.create_task(client.send_bulletchat(BulletChat(msg=f'谢谢{uname}的关注喵')))
+
+    _CMD_CALLBACK_DICT['INTERACT_WORD'] = __interact_word_callback
+
+    def _on_heartbeat(self, client: blivedm.BLiveClient, message: web_models.HeartbeatMessage):
+        pass
+
+    def _on_danmaku(self, client: blivedm.BLiveClient, message: web_models.DanmakuMessage):
+        pass
+
+    def _on_gift(self, client: blivedm.BLiveClient, message: web_models.GiftMessage):
+        pass
+
+    def _on_buy_guard(self, client: blivedm.BLiveClient, message: web_models.GuardBuyMessage):
+        pass
+
+    def _on_super_chat(self, client: blivedm.BLiveClient, message: web_models.SuperChatMessage):
+        pass
+
+
+if __name__ == '__main__':
+    asyncio.run(dm_monitor())