Create command_parser.py

src/command_parser.py

@@ -0,0 +1,82 @@
+from dataclasses import dataclass
+from typing import Optional
+
+@dataclass
+class ParsedCommand:
+    kind: str
+    arg1: Optional[str] = None
+    arg2: Optional[str] = None
+    arg3: Optional[str] = None
+    rest: Optional[str] = None
+
+    def to_dict(self):
+        return {
+            "kind": self.kind,
+            "arg1": self.arg1,
+            "arg2": self.arg2,
+            "arg3": self.arg3,
+            "rest": self.rest,
+        }
+
+class CommandParser:
+    """
+    Minimal deterministic command parser.
+    Supported:
+      MODE <STATE>
+      RESET | STATE | HELP
+      CARD <TYPE>
+      SPLITS
+      FIRST <split> <limit>
+      ROWS <split> <offset> <length>
+      SEARCH <split> <query...>
+      FILTER <split> <where...>
+    """
+    def parse(self, raw: str) -> ParsedCommand:
+        raw = (raw or "").strip()
+        if not raw:
+            return ParsedCommand(kind="NOOP")
+
+        parts = raw.split()
+        head = parts[0].upper()
+
+        if head == "MODE":
+            return ParsedCommand(kind="MODE", arg1=parts[1] if len(parts) > 1 else None)
+
+        if head == "RESET":
+            return ParsedCommand(kind="RESET")
+
+        if head == "STATE":
+            return ParsedCommand(kind="STATE")
+
+        if head == "HELP":
+            return ParsedCommand(kind="HELP")
+
+        if head == "CARD":
+            return ParsedCommand(kind="CARD", arg1=parts[1].upper() if len(parts) > 1 else None)
+
+        if head == "SPLITS":
+            return ParsedCommand(kind="SPLITS")
+
+        if head == "FIRST":
+            split = parts[1] if len(parts) > 1 else "train"
+            limit = parts[2] if len(parts) > 2 else "100"
+            return ParsedCommand(kind="FIRST", arg1=split, arg2=limit)
+
+        if head == "ROWS":
+            split = parts[1] if len(parts) > 1 else "train"
+            offset = parts[2] if len(parts) > 2 else "0"
+            length = parts[3] if len(parts) > 3 else "25"
+            return ParsedCommand(kind="ROWS", arg1=split, arg2=offset, arg3=length)
+
+        if head == "SEARCH":
+            split = parts[1] if len(parts) > 1 else "train"
+            rest = " ".join(parts[2:]) if len(parts) > 2 else ""
+            return ParsedCommand(kind="SEARCH", arg1=split, rest=rest)
+
+        if head == "FILTER":
+            split = parts[1] if len(parts) > 1 else "train"
+            rest = " ".join(parts[2:]) if len(parts) > 2 else ""
+            return ParsedCommand(kind="FILTER", arg1=split, rest=rest)
+
+        # fallback: treat unknown as SEARCH on train
+        return ParsedCommand(kind="SEARCH", arg1="train", rest=raw)