endstone-qqsync-lite-plugin 0.0.1__py3-none-any.whl

endstone_qqsync_lite_plugin/__init__.py

@@ -0,0 +1,3 @@
+from endstone_qqsync_lite_plugin.qqsync_lite_plugin import qqsync
+
+__all__ = ["qqsync"]

endstone_qqsync_lite_plugin/client.py

@@ -0,0 +1,124 @@
+import asyncio
+import json
+from typing import Optional
+from .utils.imports import import_websockets
+
+websockets = import_websockets()
+
+class WebSocketClient:
+    """WebSocket 客户端，用于与 OneBot WS 交互"""
+    
+    def __init__(self, plugin):
+        self.plugin = plugin
+        self.logger = plugin.logger
+        self.ws = None
+        self._running = False
+
+    async def connect_forever(self):
+        """保持与 OneBot WS 服务的连接"""
+        if self._running:
+            return
+        self._running = True
+
+        napcat_ws = self.plugin.config_manager.get("napcat_ws")
+        access_token = self.plugin.config_manager.get("access_token")
+        headers = {"Authorization": f"Bearer {access_token}"} if access_token else {}
+
+        self.logger.info(f"正在连接 NapCat WS 地址: {napcat_ws}")
+        delay = 2
+
+        while self._running:
+            try:
+                async with websockets.connect(
+                    napcat_ws,
+                    additional_headers=headers,
+                    ping_interval=20,
+                    ping_timeout=10,
+                    close_timeout=10
+                ) as websocket:
+                    self.ws = websocket
+                    self.plugin._current_ws = websocket
+                    delay = 2  # 连接成功后重置重连延迟
+                    self.logger.info("已成功连接 NapCat WS")
+
+                    # 发送服务器启动通知
+                    if self.plugin._send_startup_message:
+                        if self.plugin.config_manager.get("enable_game_to_qq", True):
+                            await self.send_group_msg_to_all("[QQSync-Lite] 服务器已启动！")
+                        self.plugin._send_startup_message = False
+
+                    # 消息接收主循环
+                    async for message in websocket:
+                        try:
+                            data = json.loads(message)
+                            await self._handle_message(data)
+                        except Exception as e:
+                            self.logger.error(f"处理收到的消息时出错: {e}")
+            except Exception as e:
+                self.ws = None
+                self.plugin._current_ws = None
+                if self._running:
+                    self.logger.warning(f"NapCat WS 连接异常或断开: {e}，将在 {delay} 秒后重试...")
+                    await asyncio.sleep(delay)
+                    delay = min(30, delay * 1.5)
+                else:
+                    break
+
+        self.logger.info("NapCat WS 客户端已停止运行")
+
+    async def _handle_message(self, data: dict):
+        """处理来自 OneBot 协议的消息"""
+        post_type = data.get("post_type")
+        if post_type != "message":
+            return
+        
+        message_type = data.get("message_type")
+        if message_type != "group":
+            return
+
+        group_id = data.get("group_id")
+        target_groups = self.plugin.config_manager.get("target_groups", [])
+        target_groups = [int(gid) for gid in target_groups]
+        if group_id not in target_groups:
+            return
+
+        raw_message = data.get("raw_message", "")
+        sender = data.get("sender", {})
+        nickname = sender.get("nickname", "未知")
+        card = sender.get("card", "")
+        display_name = card if card else nickname
+
+        # 转发到游戏内
+        if self.plugin.config_manager.get("enable_qq_to_game", True):
+            from .handlers import handle_qq_message
+            await handle_qq_message(self.plugin, group_id, display_name, raw_message)
+
+    async def send_group_msg_to_all(self, text: str):
+        """广播群消息到所有配置的 QQ 群"""
+        if not self.ws or self.ws.state != 1:
+            return
+        
+        target_groups = self.plugin.config_manager.get("target_groups", [])
+        target_groups = [int(gid) for gid in target_groups]
+        
+        for group_id in target_groups:
+            try:
+                payload = {
+                    "action": "send_group_msg",
+                    "params": {
+                        "group_id": group_id,
+                        "message": text
+                    }
+                }
+                await self.ws.send(json.dumps(payload))
+            except Exception as e:
+                self.logger.error(f"发送群消息到群 {group_id} 失败: {e}")
+
+    def stop(self):
+        """停止客户端并关闭连接"""
+        self._running = False
+        if self.ws:
+            if self.plugin._loop and self.plugin._loop.is_running():
+                asyncio.run_coroutine_threadsafe(self.ws.close(), self.plugin._loop)
+        self.ws = None
+        self.plugin._current_ws = None

endstone_qqsync_lite_plugin/config_manager.py

@@ -0,0 +1,64 @@
+import json
+from pathlib import Path
+from typing import Any, Dict
+
+class ConfigManager:
+    """配置管理器"""
+    def __init__(self, data_folder: Path, logger):
+        self.data_folder = data_folder
+        self.logger = logger
+        self.config_file = data_folder / "config.json"
+        self._config: Dict[str, Any] = {}
+        
+        self.default_config = {
+            "napcat_ws": "ws://127.0.0.1:3001",
+            "access_token": "",
+            "target_groups": [12345678],
+            "group_names": {},
+            "enable_qq_to_game": True,
+            "enable_game_to_qq": True,
+            "enable_join_quit_msg": True,
+            "enable_death_msg": True,
+            "api_qq_enable": False
+        }
+        self._load_config()
+
+    def _load_config(self):
+        """加载配置文件，若不存在则写入默认配置"""
+        if not self.config_file.exists():
+            self.config_file.parent.mkdir(parents=True, exist_ok=True)
+            try:
+                with open(self.config_file, "w", encoding="utf-8") as f:
+                    json.dump(self.default_config, f, indent=2, ensure_ascii=False)
+                self.logger.info("已创建默认配置文件 config.json")
+            except Exception as e:
+                self.logger.error(f"创建默认配置文件失败: {e}")
+
+        try:
+            with open(self.config_file, "r", encoding="utf-8") as f:
+                self._config = json.load(f)
+        except Exception as e:
+            self.logger.error(f"读取配置文件失败: {e}")
+            self._config = self.default_config.copy()
+
+        # 合并缺少的默认配置项
+        updated = False
+        for k, v in self.default_config.items():
+            if k not in self._config:
+                self._config[k] = v
+                updated = True
+
+        if updated:
+            self.save_config()
+
+    def get(self, key: str, default: Any = None) -> Any:
+        """获取配置项"""
+        return self._config.get(key, default)
+
+    def save_config(self):
+        """保存当前配置到文件"""
+        try:
+            with open(self.config_file, "w", encoding="utf-8") as f:
+                json.dump(self._config, f, indent=2, ensure_ascii=False)
+        except Exception as e:
+            self.logger.error(f"保存配置文件失败: {e}")

endstone_qqsync_lite_plugin/handlers.py

@@ -0,0 +1,118 @@
+import re
+from endstone.event import event_handler, PlayerChatEvent, PlayerJoinEvent, PlayerQuitEvent, PlayerDeathEvent
+from endstone import ColorFormat
+
+# 常用 emoji 字典映射
+EMOJI_MAP = {
+    '😀': '[笑脸]', '😁': '[开心]', '😂': '[笑哭]', '🤣': '[大笑]', '😃': '[微笑]',
+    '😄': '[开心]', '😅': '[汗笑]', '👍': '[赞]', '👎': '[踩]', '👌': '[OK]',
+    '🔥': '[火]', '💯': '[100分]', '❤': '[红心]', '💔': '[心碎]'
+}
+
+def clean_and_parse_cq(text: str) -> str:
+    """清理和解析 QQ 消息中的 CQ 码和 Emoji"""
+    if not text:
+        return ""
+    
+    # 替换已知 emoji
+    for emoji, desc in EMOJI_MAP.items():
+        text = text.replace(emoji, desc)
+
+    # 过滤其他未映射 emoji 区段
+    emoji_pattern = re.compile(
+        r'[\U0001F600-\U0001F64F\U0001F300-\U0001F5FF\U0001F680-\U0001F6FF\U00002600-\U000026FF\U00002700-\U000027BF]+'
+    )
+    text = emoji_pattern.sub('[表情]', text)
+
+    # 解析 CQ 码
+    def replace_cq(match):
+        cq_type = match.group(1)
+        cq_map = {
+            "image": "[图片]", "video": "[视频]", "record": "[语音]", 
+            "face": "[表情]", "reply": "[回复]", "forward": "[转发]",
+            "file": "[文件]", "share": "[分享]", "location": "[位置]"
+        }
+        if cq_type in cq_map:
+            return cq_map[cq_type]
+        elif cq_type == "at":
+            params = match.group(2) or ""
+            if "qq=all" in params:
+                return "@全体成员"
+            else:
+                qq_match = re.search(r'qq=(\d+)', params)
+                return f"@{qq_match.group(1)}" if qq_match else "@某人"
+        return "[非文本]"
+
+    cq_pattern = r'\[CQ:([^,\]]+)(?:,([^\]]*))?\]'
+    text = re.sub(cq_pattern, replace_cq, text)
+    
+    # 清理控制字符
+    text = re.sub(r'[\x00-\x1f\x7f-\x9f]', '', text)
+    return text.strip()
+
+async def handle_qq_message(plugin, group_id: int, sender_name: str, raw_message: str):
+    """处理并格式化 QQ 消息转发到游戏"""
+    parsed_msg = clean_and_parse_cq(raw_message)
+    if not parsed_msg:
+        return
+
+    group_names = plugin.config_manager.get("group_names", {})
+    group_name = group_names.get(str(group_id), "")
+    
+    if group_name:
+        game_message = f"{ColorFormat.GREEN}[QQ群] {ColorFormat.AQUA}[{group_name}] {sender_name}: {parsed_msg}{ColorFormat.RESET}"
+    else:
+        game_message = f"{ColorFormat.GREEN}[QQ群] {ColorFormat.AQUA}{sender_name}: {parsed_msg}{ColorFormat.RESET}"
+
+    plugin.logger.info(f"[QQ群->游戏] {sender_name}: {parsed_msg}")
+
+    # 主线程安全广播
+    def send():
+        for player in plugin.server.online_players:
+            player.send_message(game_message)
+
+    plugin.server.scheduler.run_task(plugin, send, delay=1)
+
+
+class EventHandlers:
+    """游戏事件监听器"""
+    def __init__(self, plugin):
+        self.plugin = plugin
+
+    @event_handler
+    def on_player_chat(self, event: PlayerChatEvent):
+        """同步游戏内聊天消息到 QQ 群"""
+        if event.message.startswith('/'):
+            return
+
+        if self.plugin.config_manager.get("enable_game_to_qq", True):
+            chat_msg = f"{event.player.name}: {event.message}"
+            self.plugin.send_to_qq(chat_msg)
+
+    @event_handler
+    def on_player_join(self, event: PlayerJoinEvent):
+        """同步玩家加入提示到 QQ 群"""
+        if (self.plugin.config_manager.get("enable_game_to_qq", True) and 
+            self.plugin.config_manager.get("enable_join_quit_msg", True)):
+            join_msg = f"[+] 玩家 {event.player.name} 加入了游戏"
+            self.plugin.send_to_qq(join_msg)
+
+    @event_handler
+    def on_player_quit(self, event: PlayerQuitEvent):
+        """同步玩家离开提示到 QQ 群"""
+        if (self.plugin.config_manager.get("enable_game_to_qq", True) and 
+            self.plugin.config_manager.get("enable_join_quit_msg", True)):
+            quit_msg = f"[-] 玩家 {event.player.name} 离开了游戏"
+            self.plugin.send_to_qq(quit_msg)
+
+    @event_handler
+    def on_player_death(self, event: PlayerDeathEvent):
+        """同步玩家死亡提示到 QQ 群"""
+        if (self.plugin.config_manager.get("enable_game_to_qq", True) and 
+            self.plugin.config_manager.get("enable_death_msg", True)):
+            try:
+                language = event.player.server.language
+                death_msg = language.translate(event.death_message, locale="zh_CN")
+                self.plugin.send_to_qq(death_msg)
+            except Exception as e:
+                self.plugin.logger.error(f"处理玩家死亡消息发送失败: {e}")

endstone_qqsync_lite_plugin/lib/websockets/__init__.py

@@ -0,0 +1,236 @@
+from __future__ import annotations
+
+# Importing the typing module would conflict with websockets.typing.
+from typing import TYPE_CHECKING
+
+from .imports import lazy_import
+from .version import version as __version__  # noqa: F401
+
+
+__all__ = [
+    # .asyncio.client
+    "connect",
+    "unix_connect",
+    "ClientConnection",
+    # .asyncio.router
+    "route",
+    "unix_route",
+    "Router",
+    # .asyncio.server
+    "basic_auth",
+    "broadcast",
+    "serve",
+    "unix_serve",
+    "ServerConnection",
+    "Server",
+    # .client
+    "ClientProtocol",
+    # .datastructures
+    "Headers",
+    "HeadersLike",
+    "MultipleValuesError",
+    # .exceptions
+    "ConcurrencyError",
+    "ConnectionClosed",
+    "ConnectionClosedError",
+    "ConnectionClosedOK",
+    "DuplicateParameter",
+    "InvalidHandshake",
+    "InvalidHeader",
+    "InvalidHeaderFormat",
+    "InvalidHeaderValue",
+    "InvalidMessage",
+    "InvalidOrigin",
+    "InvalidParameterName",
+    "InvalidParameterValue",
+    "InvalidProxy",
+    "InvalidProxyMessage",
+    "InvalidProxyStatus",
+    "InvalidState",
+    "InvalidStatus",
+    "InvalidUpgrade",
+    "InvalidURI",
+    "NegotiationError",
+    "PayloadTooBig",
+    "ProtocolError",
+    "ProxyError",
+    "SecurityError",
+    "WebSocketException",
+    # .frames
+    "Close",
+    "CloseCode",
+    "Frame",
+    "Opcode",
+    # .http11
+    "Request",
+    "Response",
+    # .protocol
+    "Protocol",
+    "Side",
+    "State",
+    # .server
+    "ServerProtocol",
+    # .typing
+    "Data",
+    "ExtensionName",
+    "ExtensionParameter",
+    "LoggerLike",
+    "StatusLike",
+    "Origin",
+    "Subprotocol",
+]
+
+# When type checking, import non-deprecated aliases eagerly. Else, import on demand.
+if TYPE_CHECKING:
+    from .asyncio.client import ClientConnection, connect, unix_connect
+    from .asyncio.router import Router, route, unix_route
+    from .asyncio.server import (
+        Server,
+        ServerConnection,
+        basic_auth,
+        broadcast,
+        serve,
+        unix_serve,
+    )
+    from .client import ClientProtocol
+    from .datastructures import Headers, HeadersLike, MultipleValuesError
+    from .exceptions import (
+        ConcurrencyError,
+        ConnectionClosed,
+        ConnectionClosedError,
+        ConnectionClosedOK,
+        DuplicateParameter,
+        InvalidHandshake,
+        InvalidHeader,
+        InvalidHeaderFormat,
+        InvalidHeaderValue,
+        InvalidMessage,
+        InvalidOrigin,
+        InvalidParameterName,
+        InvalidParameterValue,
+        InvalidProxy,
+        InvalidProxyMessage,
+        InvalidProxyStatus,
+        InvalidState,
+        InvalidStatus,
+        InvalidUpgrade,
+        InvalidURI,
+        NegotiationError,
+        PayloadTooBig,
+        ProtocolError,
+        ProxyError,
+        SecurityError,
+        WebSocketException,
+    )
+    from .frames import Close, CloseCode, Frame, Opcode
+    from .http11 import Request, Response
+    from .protocol import Protocol, Side, State
+    from .server import ServerProtocol
+    from .typing import (
+        Data,
+        ExtensionName,
+        ExtensionParameter,
+        LoggerLike,
+        Origin,
+        StatusLike,
+        Subprotocol,
+    )
+else:
+    lazy_import(
+        globals(),
+        aliases={
+            # .asyncio.client
+            "connect": ".asyncio.client",
+            "unix_connect": ".asyncio.client",
+            "ClientConnection": ".asyncio.client",
+            # .asyncio.router
+            "route": ".asyncio.router",
+            "unix_route": ".asyncio.router",
+            "Router": ".asyncio.router",
+            # .asyncio.server
+            "basic_auth": ".asyncio.server",
+            "broadcast": ".asyncio.server",
+            "serve": ".asyncio.server",
+            "unix_serve": ".asyncio.server",
+            "ServerConnection": ".asyncio.server",
+            "Server": ".asyncio.server",
+            # .client
+            "ClientProtocol": ".client",
+            # .datastructures
+            "Headers": ".datastructures",
+            "HeadersLike": ".datastructures",
+            "MultipleValuesError": ".datastructures",
+            # .exceptions
+            "ConcurrencyError": ".exceptions",
+            "ConnectionClosed": ".exceptions",
+            "ConnectionClosedError": ".exceptions",
+            "ConnectionClosedOK": ".exceptions",
+            "DuplicateParameter": ".exceptions",
+            "InvalidHandshake": ".exceptions",
+            "InvalidHeader": ".exceptions",
+            "InvalidHeaderFormat": ".exceptions",
+            "InvalidHeaderValue": ".exceptions",
+            "InvalidMessage": ".exceptions",
+            "InvalidOrigin": ".exceptions",
+            "InvalidParameterName": ".exceptions",
+            "InvalidParameterValue": ".exceptions",
+            "InvalidProxy": ".exceptions",
+            "InvalidProxyMessage": ".exceptions",
+            "InvalidProxyStatus": ".exceptions",
+            "InvalidState": ".exceptions",
+            "InvalidStatus": ".exceptions",
+            "InvalidUpgrade": ".exceptions",
+            "InvalidURI": ".exceptions",
+            "NegotiationError": ".exceptions",
+            "PayloadTooBig": ".exceptions",
+            "ProtocolError": ".exceptions",
+            "ProxyError": ".exceptions",
+            "SecurityError": ".exceptions",
+            "WebSocketException": ".exceptions",
+            # .frames
+            "Close": ".frames",
+            "CloseCode": ".frames",
+            "Frame": ".frames",
+            "Opcode": ".frames",
+            # .http11
+            "Request": ".http11",
+            "Response": ".http11",
+            # .protocol
+            "Protocol": ".protocol",
+            "Side": ".protocol",
+            "State": ".protocol",
+            # .server
+            "ServerProtocol": ".server",
+            # .typing
+            "Data": ".typing",
+            "ExtensionName": ".typing",
+            "ExtensionParameter": ".typing",
+            "LoggerLike": ".typing",
+            "Origin": ".typing",
+            "StatusLike": ".typing",
+            "Subprotocol": ".typing",
+        },
+        deprecated_aliases={
+            # deprecated in 9.0 - 2021-09-01
+            "framing": ".legacy",
+            "handshake": ".legacy",
+            "parse_uri": ".uri",
+            "WebSocketURI": ".uri",
+            # deprecated in 14.0 - 2024-11-09
+            # .legacy.auth
+            "BasicAuthWebSocketServerProtocol": ".legacy.auth",
+            "basic_auth_protocol_factory": ".legacy.auth",
+            # .legacy.client
+            "WebSocketClientProtocol": ".legacy.client",
+            # .legacy.exceptions
+            "AbortHandshake": ".legacy.exceptions",
+            "InvalidStatusCode": ".legacy.exceptions",
+            "RedirectHandshake": ".legacy.exceptions",
+            "WebSocketProtocolError": ".legacy.exceptions",
+            # .legacy.protocol
+            "WebSocketCommonProtocol": ".legacy.protocol",
+            # .legacy.server
+            "WebSocketServer": ".legacy.server",
+            "WebSocketServerProtocol": ".legacy.server",
+        },
+    )

endstone_qqsync_lite_plugin/lib/websockets/__main__.py

@@ -0,0 +1,5 @@
+from .cli import main
+
+
+if __name__ == "__main__":
+    main()