Merge pull request #59 from JustOneSummer/master

以http服务支持minimap_renderer小地图渲染

Author: benx1n

.env

@@ -26,5 +26,9 @@ proxy = http://localhost:7890   #代理地址，如果上面选项开启，这
 ocr_on = true                   #是否开启ocr(识图指令)
 ocr_offline = false             #是否只使用hash验证，即设置为true后只能识别服务器已记录的图片，如果群较多(>300)导致响应延迟较高可以开启
 ocr_url = http://ocr.benx1n.com:23338           #默认ocr地址，一般不用动
+minimap_renderer_on = false # 是否启用
+minimap_renderer_url = http://localhost:9876 # minimap_renderer地址
+minimap_renderer_user_name = yuyuko # 鉴权用户名 参考minimap_renderer的token.json文件
+minimap_renderer_password = yuyuko # 鉴权用户密码 参考minimap_renderer的token.json文件
 http2 = true                    #是否开启http2，默认开启，如果有问题请设置为false
 battle_listen_time = 1          #监控刷新时间，分钟级，如果性能或流量不够可以适当延长，推荐在5分钟以内

.env.prod-example

@@ -26,5 +26,9 @@ proxy = http://localhost:7890   #代理地址，如果上面选项开启，这
 ocr_on = true                   #是否开启ocr(识图指令)
 ocr_offline = false             #是否只使用hash验证，即设置为true后只能识别服务器已记录的图片，如果群较多(>300)导致响应延迟较高可以开启
 ocr_url = http://ocr.benx1n.com:23338           #默认ocr地址，一般不用动
+minimap_renderer_on = false # 是否启用
+minimap_renderer_url = http://localhost:9876 # minimap_renderer地址
+minimap_renderer_user_name = yuyuko # 鉴权用户名 参考minimap_renderer的token.json文件
+minimap_renderer_password = yuyuko # 鉴权用户密码 参考minimap_renderer的token.json文件
 http2 = true                    #是否开启http2，默认开启，如果有问题请设置为false
 battle_listen_time = 1          #监控刷新时间，分钟级，如果性能或流量不够可以适当延长，推荐在5分钟以内

poetry.lock

@@ -19,15 +19,18 @@
     GroupMessageEvent,
     Message,
     MessageEvent,
+    NoticeEvent,
     MessageSegment,
     PrivateMessageEvent,
 )
 from nonebot.log import logger
 from nonebot.params import CommandArg
 from nonebot.permission import SUPERUSER
+from nonebot.plugin.on import on_notice
 from nonebot_plugin_guild_patch import GuildMessageEvent
 
 from .data_source import dir_path, nb2_file, template_path
+from .game.minimap_renderer import get_rep
 from .game.ocr import (
     downlod_OcrResult,
     get_Random_Ocr_Pic,
@@ -52,6 +55,7 @@
 bot_pupu = on_fullmatch('噗噗', block=False, priority=5)
 bot_listen = on_message(priority=5, block=False)
 ocr_listen = on_message(priority=6, block=False)
+group_file_listen = on_notice(priority=6, block=False)
 driver = get_driver()
 
 _proxy = None
@@ -74,7 +78,8 @@
 async def main(bot: Bot, ev: MessageEvent, matchmsg: Message = CommandArg()):  # noqa: B008, PLR0915
     try:
         server_type = None
-        if isinstance(ev, PrivateMessageEvent) and (driver.config.private or str(ev.user_id) in driver.config.superusers):  # 私聊事件,superusers默认不受影响
+        if isinstance(ev, PrivateMessageEvent) and (
+                driver.config.private or str(ev.user_id) in driver.config.superusers):  # 私聊事件,superusers默认不受影响
             server_type = 'QQ'
         elif isinstance(ev, GroupMessageEvent) and driver.config.group and ev.group_id not in driver.config.ban_group_list:  # 群聊事件
             server_type = 'QQ'
@@ -173,6 +178,26 @@ async def wait_to_select(hikari):
         return hikari.error('已超时退出')
 
 
+@group_file_listen.handle()
+async def GROUP_FILE_listen(bot: Bot, ev: NoticeEvent):
+    try:
+        if not driver.config.minimap_renderer_on:
+            return
+        noticeType = str(ev.notice_type)
+        if not noticeType == 'group_upload':
+            return
+        if not str(ev.file.name).endswith('.wowsreplay'):
+            return
+        #
+        base64_file = await bot.get_image(file=ev.file.id)
+        # 调用接口转换
+        await get_rep(base64_file['base64'], bot, ev)
+    except Exception:
+        logger.error(traceback.format_exc())
+        await bot.send(ev, MessageSegment.text("请求minimap_renderer服务异常"))
+        return
+
+
 @ocr_listen.handle()
 async def OCR_listen(bot: Bot, ev: MessageEvent):
     try:

src/plugins/hikari_bot/__init__.py

@@ -0,0 +1,37 @@
+import base64
+import hashlib
+import os
+
+import requests
+from nonebot import get_driver
+from nonebot.adapters.onebot.v11 import Bot, MessageSegment, NoticeEvent
+from requests.auth import HTTPBasicAuth
+
+driver = get_driver()
+minimap_renderer_temp = "minimap_renderer_temp"
+
+
+async def get_rep(wows_rep_file_base64: str, bot: Bot, ev: NoticeEvent):
+    file_hex = hashlib.sha256(wows_rep_file_base64.encode('utf-8')).hexdigest()
+    file_bytes = base64.b64decode(wows_rep_file_base64)
+
+    file_path_temp = os.getcwd() + os.sep + minimap_renderer_temp + os.sep + file_hex
+    wowsrepla_file = file_path_temp + ".wowsreplay"
+    if not os.path.exists(wowsrepla_file):
+        with open(wowsrepla_file, 'wb') as f:
+            f.write(file_bytes)
+    await bot.send(ev, MessageSegment.text("正在处理replays文件.预计耗时1分钟"))
+    upload_url = driver.config.minimap_renderer_url + "/upload_replays_video_url"
+    with open(wowsrepla_file, 'rb') as file:
+        files = {'file': file}
+        response = requests.post(upload_url, files=files, auth=HTTPBasicAuth(driver.config.minimap_renderer_user_name, driver.config.minimap_renderer_password), timeout=600)
+        if response.status_code == 200:
+            await send_video(bot, ev, response.text)
+        else:
+            await bot.send(ev, MessageSegment.text("生成视频文件异常！请检查 minimap_renderer 是否要更新."))
+
+
+async def send_video(bot: Bot, ev: NoticeEvent, url: str):
+    # 构造视频文件消息
+    data = str(driver.config.minimap_renderer_url + "/video_url?file_name=" + url.replace("\"", ""))
+    await  bot.send(ev, MessageSegment.video(data))