feat: 新增maim_message配置，支持wss，tcp模式，支持token

src/common/message/api.py

@@ -3,25 +3,53 @@ import os
 import importlib.metadata
 from maim_message import MessageServer
 from src.common.logger_manager import get_logger
+from src.config.config import global_config
 
 
 # 检查maim_message版本
 try:
     maim_message_version = importlib.metadata.version("maim_message")
-    version_compatible = [int(x) for x in maim_message_version.split(".")] >= [0, 3, 0]
+    version_compatible = [int(x) for x in maim_message_version.split(".")] >= [0, 3, 3]
 except (importlib.metadata.PackageNotFoundError, ValueError):
     version_compatible = False
 
-# 根据版本决定是否使用自定义logger
+# 读取配置项
+maim_message_config = global_config.maim_message
+
+# 设置基本参数
 kwargs = {
     "host": os.environ["HOST"],
     "port": int(os.environ["PORT"]),
     "app": global_server.get_app(),
 }
 
-# 只有在版本 >= 0.3.0 时才使用自定义logger
+# 只有在版本 >= 0.3.0 时才使用高级特性
 if version_compatible:
+    # 添加自定义logger
     maim_message_logger = get_logger("maim_message")
     kwargs["custom_logger"] = maim_message_logger
 
+    # 添加token认证
+    if maim_message_config.auth_token:
+        if len(maim_message_config.auth_token) > 0:
+            kwargs["enable_token"] = True
+
+    if maim_message_config.use_custom:
+        # 添加WSS模式支持
+        del kwargs["app"]
+        kwargs["host"] = maim_message_config.host
+        kwargs["port"] = maim_message_config.port
+        kwargs["mode"] = maim_message_config.mode
+        if maim_message_config.use_wss:
+            if maim_message_config.cert_file:
+                kwargs["ssl_certfile"] = maim_message_config.cert_file
+            if maim_message_config.key_file:
+                kwargs["ssl_keyfile"] = maim_message_config.key_file
+        kwargs["enable_custom_uvicorn_logger"] = False
+
+
 global_api = MessageServer(**kwargs)
+
+if version_compatible and maim_message_config.auth_token:
+    for token in maim_message_config.auth_token:
+        global_api.add_valid_token(token)