Merge branch 'dev' of https://github.com/MoFox-Studio/MoFox_Bot into dev

src/common/remote.py

@@ -0,0 +1,165 @@
+import asyncio
+
+import aiohttp
+import platform
+
+from datetime import datetime, timezone
+from src.common.logger import get_logger
+from src.common.tcp_connector import get_tcp_connector
+from src.config.config import global_config
+from src.manager.async_task_manager import AsyncTask
+from src.manager.local_store_manager import local_storage
+
+logger = get_logger("remote")
+
+TELEMETRY_SERVER_URL = "http://124.248.67.228:10058"
+"""遥测服务地址"""
+
+
+class TelemetryHeartBeatTask(AsyncTask):
+    HEARTBEAT_INTERVAL = 300
+
+    def __init__(self):
+        super().__init__(task_name="Telemetry Heart Beat Task", run_interval=self.HEARTBEAT_INTERVAL)
+        self.server_url = TELEMETRY_SERVER_URL
+        """遥测服务地址"""
+
+        self.client_uuid: str | None = local_storage["mmc_uuid"] if "mmc_uuid" in local_storage else None  # type: ignore
+        """客户端UUID"""
+
+        self.info_dict = self._get_sys_info()
+        """系统信息字典"""
+
+    @staticmethod
+    def _get_sys_info() -> dict[str, str]:
+        """获取系统信息"""
+        info_dict = {
+            "os_type": "Unknown",
+            "py_version": platform.python_version(),
+            "mmc_version": global_config.MMC_VERSION,
+        }
+
+        match platform.system():
+            case "Windows":
+                info_dict["os_type"] = "Windows"
+            case "Linux":
+                info_dict["os_type"] = "Linux"
+            case "Darwin":
+                info_dict["os_type"] = "macOS"
+            case _:
+                info_dict["os_type"] = "Unknown"
+
+        return info_dict
+
+    async def _req_uuid(self) -> bool:
+        """
+        向服务端请求UUID（不应在已存在UUID的情况下调用，会覆盖原有的UUID）
+        """
+
+        if "deploy_time" not in local_storage:
+            logger.error("本地存储中缺少部署时间，无法请求UUID")
+            return False
+
+        try_count: int = 0
+        while True:
+            # 如果不存在，则向服务端请求一个新的UUID（注册客户端）
+            logger.info("正在向遥测服务端请求UUID...")
+
+            try:
+                async with aiohttp.ClientSession(connector=await get_tcp_connector()) as session:
+                    async with session.post(
+                        f"{TELEMETRY_SERVER_URL}/stat/reg_client",
+                        json={"deploy_time": datetime.fromtimestamp(local_storage["deploy_time"], tz=timezone.utc).isoformat()},
+                        timeout=aiohttp.ClientTimeout(total=5),  # 设置超时时间为5秒
+                    ) as response:
+                        logger.debug(f"{TELEMETRY_SERVER_URL}/stat/reg_client")
+                        logger.debug(local_storage["deploy_time"])  # type: ignore
+                        logger.debug(f"Response status: {response.status}")
+
+                        if response.status == 200:
+                            data = await response.json()
+                            if client_id := data.get("mmc_uuid"):
+                                # 将UUID存储到本地
+                                local_storage["mmc_uuid"] = client_id
+                                self.client_uuid = client_id
+                                logger.info(f"成功获取UUID: {self.client_uuid}")
+                                return True  # 成功获取UUID，返回True
+                            else:
+                                logger.error("无效的服务端响应")
+                        else:
+                            response_text = await response.text()
+                            logger.error(
+                                f"请求UUID失败，不过你还是可以正常使用墨狐，状态码: {response.status}, 响应内容: {response_text}"
+                            )
+            except Exception as e:
+                import traceback
+
+                error_msg = str(e) or "未知错误"
+                logger.warning(
+                    f"请求UUID出错，不过你还是可以正常使用墨狐: {type(e).__name__}: {error_msg}"
+                )  # 可能是网络问题
+                logger.debug(f"完整错误信息: {traceback.format_exc()}")
+
+            # 请求失败，重试次数+1
+            try_count += 1
+            if try_count > 3:
+                # 如果超过3次仍然失败，则退出
+                logger.error("获取UUID失败，请检查网络连接或服务端状态")
+                return False
+            else:
+                # 如果可以重试，等待后继续（指数退避）
+                logger.info(f"获取UUID失败，将于 {4**try_count} 秒后重试...")
+                await asyncio.sleep(4**try_count)
+
+    async def _send_heartbeat(self):
+        """向服务器发送心跳"""
+        headers = {
+            "Client-UUID": self.client_uuid,
+            "User-Agent": f"HeartbeatClient/{self.client_uuid[:8]}",  # type: ignore
+        }
+
+        logger.debug(f"正在发送心跳到服务器: {self.server_url}")
+        logger.debug(str(headers))
+
+        try:
+            async with aiohttp.ClientSession(connector=await get_tcp_connector()) as session:
+                async with session.post(
+                    f"{self.server_url}/stat/client_heartbeat",
+                    headers=headers,
+                    json=self.info_dict,
+                    timeout=aiohttp.ClientTimeout(total=5),  # 设置超时时间为5秒
+                ) as response:
+                    logger.debug(f"Response status: {response.status}")
+
+                    # 处理响应
+                    if 200 <= response.status < 300:
+                        # 成功
+                        logger.debug(f"心跳发送成功，状态码: {response.status}")
+                    elif response.status == 403:
+                        # 403 Forbidden
+                        logger.warning(
+                            "（此消息不会影响正常使用）心跳发送失败，403 Forbidden: 可能是UUID无效或未注册。"
+                            "处理措施：重置UUID，下次发送心跳时将尝试重新注册。"
+                        )
+                        self.client_uuid = None
+                        del local_storage["mmc_uuid"]  # 删除本地存储的UUID
+                    else:
+                        # 其他错误
+                        response_text = await response.text()
+                        logger.warning(
+                            f"（此消息不会影响正常使用）状态未发送，状态码: {response.status}, 响应内容: {response_text}"
+                        )
+        except Exception as e:
+            import traceback
+
+            error_msg = str(e) or "未知错误"
+            logger.warning(f"（此消息不会影响正常使用）状态未发生: {type(e).__name__}: {error_msg}")
+            logger.debug(f"完整错误信息: {traceback.format_exc()}")
+
+    async def run(self):
+        # 发送心跳
+        if self.client_uuid is None and not await self._req_uuid():
+            logger.warning("获取UUID失败，跳过此次心跳")
+            return
+
+        await self._send_heartbeat()

src/main.py

@@ -7,6 +7,7 @@ from maim_message import MessageServer
 
 from src.manager.async_task_manager import async_task_manager
 from src.chat.utils.statistic import OnlineTimeRecordTask, StatisticOutputTask
+from src.common.remote import TelemetryHeartBeatTask
 from src.chat.emoji_system.emoji_manager import get_emoji_manager
 from src.chat.message_receive.chat_stream import get_chat_manager
 from src.config.config import global_config
@@ -202,6 +203,9 @@ MoFox_Bot(第三方修改版)
 
         # 添加统计信息输出任务
         await async_task_manager.add_task(StatisticOutputTask())
+        
+        # 添加遥测心跳任务
+        await async_task_manager.add_task(TelemetryHeartBeatTask())
 
         # 注册默认事件
         event_manager.init_default_events()