transfer 插件 (#117)

transfer 插件，可以上传和下载文件。
1.
-transfer upload <file_path 1>
<file_path 2>
...
上传文件，每个路径换行分隔
2. -transfer download <save_path> ( 需要回复包含文件的信息 )

README.md

@@ -88,3 +88,4 @@
     - `neteaseshuffle` : 随机网抑热歌。
     - `neteasemusic` : 网易云搜歌/随机热歌/点歌。
     - `videodl` : 下载YTB/B站视频。
+    - `transfer` : 上传和下载文件。

list.json

@@ -379,6 +379,16 @@
             "supported": true,
             "des-short": "每天定时发送消息",
             "des": "命令: msgst, msgset。"
+        },
+        {
+            "name": "transfer",
+            "version": "1.0",
+            "section": "daily",
+            "maintainer": "c3b2a",
+            "size": "2.7 kb",
+            "supported": true,
+            "des-short": "上传和下载文件。",
+            "des": "命令: transfer"
         }
     ]
 }

transfer.py

@@ -0,0 +1,71 @@
+import asyncio, zipfile, os
+from io import BytesIO
+from uuid import uuid4
+from os.path import exists, isfile
+from pagermaid import bot
+from pagermaid.listener import listener
+
+async def make_zip(source_dir, output_filename):
+    zipf = zipfile.ZipFile(output_filename, "w")    
+    pre_len = len(os.path.dirname(source_dir))
+    for parent, dirnames, filenames in os.walk(source_dir):
+        for filename in filenames:
+            pathfile = os.path.join(parent, filename)
+            arcname = pathfile[pre_len:].strip(os.path.sep)
+            zipf.write(pathfile, arcname)
+    zipf.close()
+
+async def del_msg(context, t_lim):
+    await asyncio.sleep(t_lim)
+    try:
+        await context.delete()
+    except:
+        pass
+
+@listener(is_plugin=True, outgoing=True, command="transfer",
+          description="上传 / 下载文件",
+          parameters="upload <filepath>` 或 `download <filepath>")
+async def transfer(context):
+    params = context.parameter
+    if len(params) < 2:
+        await context.edit("参数缺失")
+        await del_msg(context, 3)
+        return
+    params[1] = " ".join(params[1:])
+    file_list = params[1].split("\n")
+    chat_id = context.chat_id
+    if params[0] == "upload":
+        index = 1
+        for file_path in file_list:
+            await context.edit(f"正在上传第 {index} 个文件")
+            if exists(file_path):
+                if isfile(file_path):
+                    await bot.send_file(chat_id, file_path, force_document=True)
+                else:
+                    token = file_path.split("/")
+                    token = token[len(token) - 1]
+                    await make_zip(file_path, f"/tmp/{token}.zip")
+                    await bot.send_file(chat_id, f"/tmp/{token}.zip", force_document=True)
+                    os.remove(f"/tmp/{token}.zip")
+            index += 1
+        await context.edit("上传完毕")
+        await del_msg(context, 3)
+    elif params[0] == "download":
+        message = await context.get_reply_message()
+        if message and message.media:
+            _file = BytesIO()
+            await bot.download_file(message.media.document, _file)
+            if not exists(file_list[0]):
+                with open(file_list[0], "wb") as f:
+                    f.write(_file.getvalue())
+                await context.edit(f"保存成功, 保存路径 {file_list[0]}")
+                await del_msg(context, 5)
+            else:
+                await context.edit("路径已存在文件")
+                await del_msg(context, 3)
+        else:
+            await context.edit("未回复消息或回复消息中不包含文件")
+            await del_msg(context, 3)
+    else:
+        await context.edit("未知命令")
+        await del_msg(context, 3)