初始化

2025-10-28 13:00:35 +08:00
commit ff3d9cc343
37 changed files with 10368 additions and 0 deletions
--- a/utils/parser.py
+++ b/utils/parser.py
@@ -0,0 +1,92 @@
+"""指令解析器"""
+import re
+import logging
+from typing import Optional, Tuple
+
+logger = logging.getLogger(__name__)
+
+
+class CommandParser:
+    """指令解析器"""
+    
+    # 指令映射表
+    COMMAND_MAP = {
+        # 骰娘
+        '.r': 'dice',
+        '.roll': 'dice',
+        
+        # 石头剪刀布
+        '.rps': 'rps',
+        
+        # 运势占卜
+        '.fortune': 'fortune',
+        '.运势': 'fortune',
+        
+        # 猜数字
+        '.guess': 'guess',
+        '.猜数字': 'guess',
+        
+        # 问答
+        '.quiz': 'quiz',
+        '.问答': 'quiz',
+        
+        # 帮助
+        '.help': 'help',
+        '.帮助': 'help',
+        
+        # 统计
+        '.stats': 'stats',
+        '.统计': 'stats',
+    }
+    
+    # 机器人名称模式（用于从@消息中提取）
+    AT_PATTERN = re.compile(r'@\s*\S+\s+(.+)', re.DOTALL)
+    
+    @classmethod
+    def parse(cls, content: str) -> Optional[Tuple[str, str]]:
+        """解析消息内容，提取游戏类型和指令
+        
+        Args:
+            content: 消息内容
+            
+        Returns:
+            (游戏类型, 完整指令) 或 None
+        """
+        # 去除首尾空格
+        content = content.strip()
+        
+        # 尝试提取@后的内容
+        at_match = cls.AT_PATTERN.search(content)
+        if at_match:
+            content = at_match.group(1).strip()
+        
+        # 检查是否以指令开头
+        for cmd_prefix, game_type in cls.COMMAND_MAP.items():
+            if content.startswith(cmd_prefix):
+                # 返回游戏类型和完整指令
+                return game_type, content
+        
+        # 没有匹配的指令
+        logger.debug(f"未识别的指令: {content}")
+        return None
+    
+    @classmethod
+    def extract_command_args(cls, command: str) -> Tuple[str, str]:
+        """提取指令和参数
+        
+        Args:
+            command: 完整指令，如 ".r 1d20" 或 ".guess 50"
+            
+        Returns:
+            (指令前缀, 参数部分)
+        """
+        parts = command.split(maxsplit=1)
+        cmd = parts[0] if parts else ""
+        args = parts[1] if len(parts) > 1 else ""
+        return cmd, args
+    
+    @classmethod
+    def is_help_command(cls, command: str) -> bool:
+        """判断是否为帮助指令"""
+        return command.strip() in ['.help', '.帮助', 'help', '帮助']
+