From dbc49fab11cd9a1b900fcba6c5f4d52fda3839ff Mon Sep 17 00:00:00 2001
From: wbt5 <wubangtu@qq.com>
Date: Thu, 18 Jun 2020 10:23:04 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=97=E9=B1=BC=E7=9B=B4=E6=92=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 danmu/danmaku/douyu.py | 48 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 danmu/danmaku/douyu.py

diff --git a/danmu/danmaku/douyu.py b/danmu/danmaku/douyu.py
new file mode 100644
index 0000000..4e916af
--- /dev/null
+++ b/danmu/danmaku/douyu.py
@@ -0,0 +1,48 @@
+import re
+import json
+import aiohttp
+from struct import pack
+
+
+class Douyu:
+    wss_url = 'wss://danmuproxy.douyu.com:8503/'
+    heartbeat = b'\x14\x00\x00\x00\x14\x00\x00\x00\xb1\x02\x00\x00\x74\x79\x70\x65\x40\x3d\x6d\x72\x6b\x6c' \
+                b'\x2f\x00 '
+    heartbeatInterval = 60
+
+    @staticmethod
+    async def get_ws_info(url):
+        room_id = url.split('/')[-1]
+        async with aiohttp.ClientSession() as session:
+            async with session.get('https://m.douyu.com/' + str(room_id)) as resp:
+                room_page = await resp.text()
+                room_id = re.findall(r'"rid":(\d{1,7})', room_page)[0]
+        reg_datas = []
+        data = f'type@=loginreq/roomid@={room_id}/'
+        s = pack('i', 9 + len(data)) * 2
+        s += b'\xb1\x02\x00\x00'  # 689
+        s += data.encode('ascii') + b'\x00'
+        reg_datas.append(s)
+        data = f'type@=joingroup/rid@={room_id}/gid@=-9999/'
+        s = pack('i', 9 + len(data)) * 2
+        s += b'\xb1\x02\x00\x00'  # 689
+        s += data.encode('ascii') + b'\x00'
+        reg_datas.append(s)
+        return Douyu.wss_url, reg_datas
+
+    @staticmethod
+    def decode_msg(data):
+        msgs = []
+        for msg in re.findall(b'(type@=.*?)\x00', data):
+            try:
+                msg = msg.replace(b'@=', b'":"').replace(b'/', b'","')
+                msg = msg.replace(b'@A', b'@').replace(b'@S', b'/')
+                msg = json.loads((b'{"' + msg[:-2] + b'}').decode('utf8', 'ignore'))
+                msg['name'] = msg.get('nn', '')
+                msg['content'] = msg.get('txt', '')
+                msg['msg_type'] = {'dgb': 'gift', 'chatmsg': 'danmaku',
+                                   'uenter': 'enter'}.get(msg['type'], 'other')
+                msgs.append(msg)
+            except Exception as e:
+                pass
+        return msgs