From 40a82b807ce0f94aa51a6c3adff3848b9326fe79 Mon Sep 17 00:00:00 2001 From: guochongling <2940397985@qq.com> Date: Fri, 3 Jan 2025 10:21:06 +0800 Subject: [PATCH] =?UTF-8?q?1.=E9=A3=9E=E7=89=9B=E8=87=AA=E5=8A=A8=E4=B8=8B?= =?UTF-8?q?=E8=BD=BD=E8=B5=84=E6=BA=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- plugins/README.md | 9 +- plugins/_priority.json | 1 + plugins/fnos.py | 221 +++++++++++++++++++++++++++++++++++++++++ 3 files changed, 227 insertions(+), 4 deletions(-) create mode 100644 plugins/fnos.py diff --git a/plugins/README.md b/plugins/README.md index 53b13e1..b576ec1 100644 --- a/plugins/README.md +++ b/plugins/README.md @@ -88,7 +88,8 @@ docker run -d \ ## 🤝 贡献者 -| 插件 | 说明 | 贡献者 | -| ------- | -------------------- | --------------------------------------- | -| plex.py | 自动刷新 Plex 媒体库 | [zhazhayu](https://github.com/zhazhayu) | -| alist_strm_gen.py | 自动生成strm | [xiaoQQya](https://github.com/xiaoQQya) | \ No newline at end of file +| 插件 | 说明 | 贡献者 | +|-------------------|---------------| --------------------------------------- | +| plex.py | 自动刷新 Plex 媒体库 | [zhazhayu](https://github.com/zhazhayu) | +| alist_strm_gen.py | 自动生成strm | [xiaoQQya](https://github.com/xiaoQQya) | +| fnos.py | 飞牛自动下载转存资源 | [key762](https://github.com/key762) | \ No newline at end of file diff --git a/plugins/_priority.json b/plugins/_priority.json index dbf851e..e7627a7 100644 --- a/plugins/_priority.json +++ b/plugins/_priority.json @@ -4,5 +4,6 @@ "alist_strm_gen", "aria2", "emby", + "fnos", "plex" ] \ No newline at end of file diff --git a/plugins/fnos.py b/plugins/fnos.py new file mode 100644 index 0000000..71d8782 --- /dev/null +++ b/plugins/fnos.py @@ -0,0 +1,221 @@ +import ssl +import json +import sys +import certifi +import asyncio +import websockets +import base64 +import secrets +from Crypto.PublicKey import RSA +from Crypto.Cipher import PKCS1_v1_5 as Cipher_pkcs1_v1_5 +from Crypto.Cipher import AES +from Crypto.Hash import HMAC, SHA256 + +""" + 配合 飞牛系统的Alist 项目，转存后自动下载 +""" + +async def create_websocket(url): + if 'wss' in url: + ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT) + ssl_context.minimum_version = ssl.TLSVersion.TLSv1_2 + ssl_context.maximum_version = ssl.TLSVersion.TLSv1_3 + ssl_context.load_verify_locations(certifi.where()) + return await websockets.connect(url, ssl=ssl_context, ping_interval=None) + else: + return await websockets.connect(url, ping_interval=None) + +async def wss_connect(websocket): + response = await websocket.recv() + return response + +async def close_websocket(websocket): + await websocket.close() + +async def send_ping(websocket): + while True: + await asyncio.sleep(5) # 每10秒发送一次Ping消息 + await websocket.send('{"req":"ping"}') + +def rsa_encrypt(message, public_key): + public_key = RSA.import_key(public_key) + cipher = Cipher_pkcs1_v1_5.new(public_key) + cipher_text = base64.b64encode(cipher.encrypt(message.encode('utf-8'))) + return cipher_text.decode('utf-8') + +def encrypt(text, key, iv): + cipher = AES.new(key, AES.MODE_CBC, iv) + pad = lambda s: s + (16 - len(s) % 16) * chr(16 - len(s) % 16) + encrypted = base64.b64encode(cipher.encrypt(pad(text).encode())) + return encrypted.decode() + +def unpad(data): + pad = data[-1] + if type(pad) is int: + pad = chr(pad) + return data[:-ord(pad)] + +def decrypt(text, key, iv): + # 将加密数据转换位bytes类型数据 + encodebytes = base64.decodebytes(text.encode()) + # 解密 + cipher = AES.new(key, AES.MODE_CBC, iv) + text_decrypted = cipher.decrypt(encodebytes) + text_decrypted = unpad(text_decrypted) + return base64.b64encode(text_decrypted).decode() + +oneMark = True +def print_progress_bar(iteration, total, prefix='', suffix='', length=35): + global oneMark + percent = (iteration / total) * 100 + filled_length = int(length * iteration // total) + bar = '#' * filled_length + ' ' * (length - filled_length) + percent_str = str(int(percent)).zfill(2) + if percent < 100: + percent_str = " " + percent_str + if oneMark: + print(f'{prefix} {bar} {percent_str}% {suffix}', end='') + oneMark = False + else: + print(f'\r{prefix} {bar} {percent_str}% {suffix}', end='') + sys.stdout.flush() + +def seconds_to_hms(seconds): + hours = seconds // 3600 + remainder = seconds % 3600 + minutes = remainder // 60 + seconds = remainder % 60 + hours_str = str(int(hours)).zfill(2) + minutes_str = str(int(minutes)).zfill(2) + seconds_str = str(int(seconds)).zfill(2) + return f'{hours_str}:{minutes_str}:{seconds_str}' + +def format_byte_repr(byte_num): + KB = 1024 + MB = KB * KB + GB = MB * KB + TB = GB * KB + try: + if isinstance(byte_num, str): + byte_num = int(byte_num) + if byte_num > TB: + result = '%sTB' % round(byte_num / TB, 2) + elif byte_num > GB: + result = '%sGB' % round(byte_num / GB, 2) + elif byte_num > MB: + result = '%sMB' % round(byte_num / MB, 2) + elif byte_num > KB: + result = '%sKB' % round(byte_num / KB, 2) + else: + result = '%sB' % byte_num + return result + except Exception as e: + print(e.args) + return byte_num + +class Fnos: + + default_config = { + "websocket": "", # 飞牛的websocket地址 + "user": "", # 飞牛的用户账号 + "password": "", # 飞牛的用户密码 + "mount_path": "", # Alist挂载的地址 + "download_wait": "true", # 是否等待下载完成 + } + default_task_config = { + "download_path": "", # 下载路径 + } + is_active = True + + def __init__(self, **kwargs): + self.plugin_name = self.__class__.__name__.lower() + if kwargs: + for key, _ in self.default_config.items(): + if key in kwargs: + setattr(self, key, kwargs[key]) + else: + print(f"{self.__class__.__name__} 模块缺少必要参数: {key}") + if self.websocket and self.user and self.password and self.mount_path and self.download_wait: + self.is_active = True + + def run(self, task, **kwargs): + dramaList = [] + if kwargs['tree'] is not None: + for node in kwargs['tree'].all_nodes(): + if node.data['is_dir'] is False: + dramaList.append(f'"{self.mount_path}{node.data['path']}"') + if len(dramaList) < 0: + print(f"飞牛:😄 此次转存无需下载文件!") + else: + task_config = task.get("addition", {}).get(self.plugin_name, self.default_task_config) + print(f"飞牛:🎞️ 转存有需下载文件️") + loop = asyncio.new_event_loop() + asyncio.set_event_loop(loop) + websocket = loop.run_until_complete(create_websocket(self.websocket)) + loop.run_until_complete(websocket.send('{"reqid":"676cf70d00000000000000000001","req":"util.crypto.getRSAPub"}')) + try: + aesKeyByte = None + aesIvByte = None + num = 0 + asyncio.ensure_future(send_ping(websocket)) + while True: + response = loop.run_until_complete(wss_connect(websocket)) + if "-----BEGIN PUBLIC KEY-----" in response: + pub = json.loads(response).get("pub") + si = json.loads(response).get("si") + userData = '{"reqid":"676cf70d00000000000000000002","user":"'+self.user+'","password":"'+self.password+'","deviceType":"Browser","deviceName":"Mac OS-Google Chrome","stay":true,"req":"user.login","si":"' + si + '"}' + aesKeyStr = "lUfJn1XJ9akUvmmwQplpVIy1XNC2jJ3q" + aesIv = secrets.token_bytes(16) + aesIvBase64 = base64.b64encode(aesIv).decode('utf-8') + iv = aesIvBase64 + rsa = rsa_encrypt(aesKeyStr, pub) + aes = encrypt(userData, aesKeyStr.encode(), aesIv) + aesKeyByte = aesKeyStr.encode() + aesIvByte = aesIv + sendMsg = '{"rsa":"' + rsa + '","iv":"' + iv + '","aes":"' + aes + '","req":"encrypted"}' + loop.run_until_complete(websocket.send(sendMsg)) + elif "676cf70d00000000000000000002" in response: + print(f"飞牛:👨 用户认证成功🏅") + secret = json.loads(response).get('secret') + keys = decrypt(secret, aesKeyByte, aesIvByte) + Secret = base64.b64decode(keys) + a = '{"reqid":"676cf70d00000000000000000003","files":['+','.join(dramaList)+'],"pathTo":"'+task_config.get("download_path")+'","overwrite":1,"description":"剧集自动下载","req":"file.cp"}' + mark = base64.b64encode(HMAC.new(Secret, a.encode(), digestmod=SHA256).digest()).decode() + loop.run_until_complete(websocket.send(mark + a)) + elif "pong" in response: + pass + elif "676cf70d00000000000000000003" in response and '"sysNotify":"taskId"' in response: + print(f"飞牛:💼 收到资源下载任务") + pass + elif "676cf70d00000000000000000003" in response and 'percent' in response: + data = json.loads(response) + if 'true' in self.download_wait.lower(): + if num != 0 or num < int(data.get('percent')): + time = seconds_to_hms(data.get('time')) + du = format_byte_repr(data.get('size')) + '/' + format_byte_repr(data.get('sizeTotal')) + speed = format_byte_repr(data.get('speed')) + '/S' + suffix = f'{time} {du} {speed}' + print_progress_bar(data.get('percent'), 100, prefix='⌛飞牛: ️', suffix=suffix) + num = data.get('percent') + else: + print(f"飞牛:🎞️ 下载任务后台执行") + break + elif '"taskInfo":{"reqid":"676cf70d00000000000000000003"' in response: + pass + elif "676cf70d00000000000000000003" in response and '"result":"succ"' in response: + print() + print(f"飞牛: 下载任务完成✅") + break + elif "676cf70d00000000000000000003" in response and '"result":"fail"' in response: + print() + print(f"飞牛: 下载任务异常❌,检查您配置") + break + elif "676cf70d00000000000000000003" in response and '"result":"cancel"' in response: + print() + print(f"飞牛: 下载任务被取消❌") + break + else: + print(f"{response}") + except Exception as e: + print(f"飞牛: 下载任务异常❌ {e}") + loop.run_until_complete(close_websocket(websocket)) \ No newline at end of file