添加对niconico支持 (#37)

main.py

@@ -5,6 +5,7 @@
 import time
 import urllib
 import uuid
+from datetime import datetime
 from http.cookies import SimpleCookie
 from pathlib import Path
 from typing import Dict, Tuple, Union
@@ -370,6 +371,22 @@ async def run(self):
                 await asyncio.to_thread(self.run_record, stream, url, title, 'ts')
 
 
+class NicoNico(LiveRecoder):
+    async def run(self):
+        url = f'https://live.nicovideo.jp/watch/{self.id}'
+        if url not in recording:
+            response = (await self.request(
+                method='GET',
+                url=url,
+            )).text
+            live_json = json.loads(re.search('<script type="application/ld\+json">(.*?)</script>', response).group(1))
+            end_ts = datetime.strptime(live_json['publication']['endDate'], '%Y-%m-%dT%H:%M:%S%z').timestamp()
+            now_ts = datetime.now().timestamp()
+            if end_ts > now_ts:
+                stream = self.get_streamlink().streams(url).get('best')  # HLSStream[mpegts]
+                await asyncio.to_thread(self.run_record, stream, url, live_json['name'], 'flv')
+
+
 async def run():
     with open('config.json', 'r', encoding='utf-8') as f:
         config = json.load(f)