"""Menu discovery from screen text."""
from __future__ import annotations
import re
from typing import TYPE_CHECKING, Any
if TYPE_CHECKING:
from collections.abc import Iterable
_OPTION_PATTERNS = [
re.compile(r"^\s*<(?P<key>[A-Za-z0-9])>\s*(?P<label>.+?)\s*$"),
re.compile(r"^\s*\[(?P<key>[A-Za-z0-9])\]\s*(?P<label>.+?)\s*$"),
re.compile(r"^\s*(?P<key>[A-Za-z0-9])\)\s*(?P<label>.+?)\s*$"),
re.compile(r"^\s*(?P<key>[A-Za-z0-9])\.\s*(?P<label>.+?)\s*$"),
re.compile(r"^\s*(?P<key>[A-Za-z0-9])\s+-\s+(?P<label>.+?)\s*$"),
]
def _first_nonblank(lines: Iterable[str]) -> str:
"""Get first non-blank line."""
for line in lines:
if line.strip():
return line.strip()
return ""
def _last_nonblank(lines: Iterable[str]) -> str:
"""Get last non-blank line."""
for line in reversed(list(lines)):
if line.strip():
return line.strip()
return ""
def discover_menu(screen: str) -> dict[str, Any]:
"""Discover menu options from screen text.
Args:
screen: Screen text to analyze
Returns:
Dictionary containing:
- title: Menu title (first non-blank line)
- prompt: Menu prompt (last non-blank line if contains : or ?)
- options: List of discovered options with key, label, line number
"""
lines = screen.splitlines()
title = _first_nonblank(lines)
prompt = ""
tail = _last_nonblank(lines)
if ":" in tail or "?" in tail:
prompt = tail
options = []
for idx, line in enumerate(lines):
for pattern in _OPTION_PATTERNS:
match = pattern.match(line)
if match:
options.append(
{
"key": match.group("key"),
"label": match.group("label").strip(),
"line": idx + 1,
}
)
break
return {"title": title, "prompt": prompt, "options": options}
