d0132b5821
Widens #90327's line_input() to the whole bug class: all 46 bare input() free-text prompt sites across the setup wizards (model_setup_flows, setup, config, gateway, auth, auth_commands, plugins_cmd, skills_hub, bundles, setup_whatsapp_cloud, main) now route through line_input(), and the shared cli_output.prompt() / setup.prompt() helpers do too — so every CLI wizard gets cursor editing, not just the custom model prompt. Redirected stdin and missing prompt_toolkit keep builtin input() behavior. E2E: real PTY with raw escape bytes through line_input, cli_output.prompt, and setup.prompt (arrows + Ctrl+A/E edit correctly); redirected-stdin fallback verified.
100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
"""Shared CLI output helpers for Hermes CLI modules.
|
|
|
|
Extracts the identical ``print_info/success/warning/error`` and ``prompt()``
|
|
functions previously duplicated across setup.py, tools_config.py,
|
|
mcp_config.py, and memory_setup.py.
|
|
"""
|
|
|
|
import sys
|
|
|
|
from hermes_cli.colors import Colors, color
|
|
from hermes_cli.secret_prompt import masked_secret_prompt
|
|
|
|
|
|
# ─── Print Helpers ────────────────────────────────────────────────────────────
|
|
|
|
|
|
def print_info(text: str) -> None:
|
|
"""Print a dim informational message."""
|
|
print(color(f" {text}", Colors.DIM))
|
|
|
|
|
|
def print_success(text: str) -> None:
|
|
"""Print a green success message with ✓ prefix."""
|
|
print(color(f"✓ {text}", Colors.GREEN))
|
|
|
|
|
|
def print_warning(text: str) -> None:
|
|
"""Print a yellow warning message with ⚠ prefix."""
|
|
print(color(f"⚠ {text}", Colors.YELLOW))
|
|
|
|
|
|
def print_error(text: str) -> None:
|
|
"""Print a red error message with ✗ prefix."""
|
|
print(color(f"✗ {text}", Colors.RED))
|
|
|
|
|
|
def print_header(text: str) -> None:
|
|
"""Print a bold yellow header."""
|
|
print(color(f"\n {text}", Colors.YELLOW))
|
|
|
|
|
|
# ─── Input Prompts ────────────────────────────────────────────────────────────
|
|
|
|
|
|
def line_input(prompt_text: str) -> str:
|
|
"""Read non-secret text with normal cursor-editing keys on a real TTY.
|
|
|
|
Setup and model-selection commands run outside the interactive chat's
|
|
prompt-toolkit application, so they can safely use a short-lived prompt
|
|
here. Redirected input and output retain the built-in ``input`` behavior
|
|
used by scripts, tests, and numbered fallbacks.
|
|
"""
|
|
if not (sys.stdin.isatty() and sys.stdout.isatty()):
|
|
return input(prompt_text)
|
|
|
|
try:
|
|
from prompt_toolkit import prompt as prompt_toolkit_prompt
|
|
from prompt_toolkit.formatted_text import ANSI
|
|
except ImportError:
|
|
return input(prompt_text)
|
|
|
|
return prompt_toolkit_prompt(ANSI(prompt_text))
|
|
|
|
|
|
def prompt(
|
|
question: str,
|
|
default: str | None = None,
|
|
password: bool = False,
|
|
) -> str:
|
|
"""Prompt the user for input with optional default and password masking.
|
|
|
|
Replaces the four independent ``_prompt()`` / ``prompt()`` implementations
|
|
in setup.py, tools_config.py, mcp_config.py, and memory_setup.py.
|
|
|
|
Returns the user's input (stripped), or *default* if the user presses Enter.
|
|
Returns empty string on Ctrl-C or EOF.
|
|
"""
|
|
suffix = f" [{default}]" if default else ""
|
|
display = color(f" {question}{suffix}: ", Colors.YELLOW)
|
|
|
|
try:
|
|
if password:
|
|
value = masked_secret_prompt(display)
|
|
else:
|
|
value = line_input(display)
|
|
value = value.strip()
|
|
return value if value else (default or "")
|
|
except (KeyboardInterrupt, EOFError):
|
|
print()
|
|
return ""
|
|
|
|
|
|
def prompt_yes_no(question: str, default: bool = True) -> bool:
|
|
"""Prompt for a yes/no answer. Returns bool."""
|
|
hint = "Y/n" if default else "y/N"
|
|
answer = prompt(f"{question} ({hint})")
|
|
if not answer:
|
|
return default
|
|
return answer.lower().startswith("y")
|