From 5a78131509cf4ea5a75c026c68a7c3e6d6d9eaa1 Mon Sep 17 00:00:00 2001
From: BukeLy <bukely0119@foxmail.com>
Date: Tue, 26 May 2026 15:04:24 +0800
Subject: [PATCH] fix(filesystem): suppress chat input echo while streaming

---
 pageindex/filesystem/cli.py | 51 +++++++++++++++++++++++++++++++++++--
 tests/test_pifs_cli.py      | 22 ++++++++++++++++
 2 files changed, 71 insertions(+), 2 deletions(-)

diff --git a/pageindex/filesystem/cli.py b/pageindex/filesystem/cli.py
index 9625fe8..ab749e2 100644
--- a/pageindex/filesystem/cli.py
+++ b/pageindex/filesystem/cli.py
@@ -1,10 +1,13 @@
 from __future__ import annotations
 
 import argparse
+import contextlib
 import os
+import re
 import shlex
 import sys
 from pathlib import Path
+from typing import Iterator, TextIO
 
 from .agent import REASONING_EFFORT_CHOICES, REASONING_SUMMARY_CHOICES, run_pifs_agent
 from .commands import PIFSCommandError, PIFSCommandExecutor
@@ -14,6 +17,7 @@ from .core import PageIndexFileSystem
 AGENT_STREAM_MODE_CHOICES = ("off", "tools", "model", "all")
 DEFAULT_AGENT_MODEL = "gpt-5.4-mini"
 EXIT_COMMANDS = {"exit", "quit", ":q"}
+ANSI_ESCAPE_RE = re.compile(r"\x1b(?:\[[0-?]*[ -/]*[@-~]|.)")
 
 
 def _load_env_file(path: str | None = None, *, workspace: str | None = None) -> Path | None:
@@ -127,6 +131,48 @@ def _agent_kwargs(args: argparse.Namespace) -> dict[str, object]:
     }
 
 
+def _sanitize_chat_question(raw: str) -> str:
+    text = ANSI_ESCAPE_RE.sub("", raw)
+    chars: list[str] = []
+    for char in text:
+        if char in {"\b", "\x7f"}:
+            if chars:
+                chars.pop()
+            continue
+        if char in {"\r", "\n"}:
+            continue
+        if ord(char) < 32 or ord(char) == 127:
+            continue
+        chars.append(char)
+    return "".join(chars).strip()
+
+
+@contextlib.contextmanager
+def _suppress_tty_input_echo(stdin: TextIO | None = None) -> Iterator[None]:
+    stream = sys.stdin if stdin is None else stdin
+    if not hasattr(stream, "isatty") or not stream.isatty():
+        yield
+        return
+    try:
+        import termios
+
+        fd = stream.fileno()
+        original = termios.tcgetattr(fd)
+        muted = original[:]
+        muted[3] = muted[3] & ~termios.ECHO
+        termios.tcsetattr(fd, termios.TCSADRAIN, muted)
+    except Exception:
+        yield
+        return
+    try:
+        yield
+    finally:
+        with contextlib.suppress(Exception):
+            termios.tcflush(fd, termios.TCIFLUSH)
+        with contextlib.suppress(Exception):
+            termios.tcsetattr(fd, termios.TCSADRAIN, original)
+
+
 def _run_ask(argv: list[str], *, workspace_default: str | None) -> int:
     args = _parse_agent_command(
         "ask",
@@ -156,7 +202,7 @@ def _run_chat(argv: list[str], *, workspace_default: str | None) -> int:
     filesystem = _filesystem_from_workspace(args.workspace)
     while True:
         try:
-            question = input("pifs> ").strip()
+            question = _sanitize_chat_question(input("pifs> "))
         except EOFError:
             break
         except KeyboardInterrupt:
@@ -166,7 +212,8 @@ def _run_chat(argv: list[str], *, workspace_default: str | None) -> int:
             continue
         if question.lower() in EXIT_COMMANDS:
             break
-        answer = run_pifs_agent(filesystem, question, **_agent_kwargs(args))
+        with _suppress_tty_input_echo():
+            answer = run_pifs_agent(filesystem, question, **_agent_kwargs(args))
         if args.stream_mode == "off":
             print(answer)
     return 0
diff --git a/tests/test_pifs_cli.py b/tests/test_pifs_cli.py
index 74e95b4..85e2b6d 100644
--- a/tests/test_pifs_cli.py
+++ b/tests/test_pifs_cli.py
@@ -149,6 +149,28 @@ def test_cli_chat_runs_one_question_and_exits(monkeypatch, capsys, tmp_path):
     assert kwargs["stream_mode"] == "all"
 
 
+def test_cli_chat_sanitizes_control_input(monkeypatch, capsys, tmp_path):
+    from pageindex.filesystem import cli
+
+    workspace = tmp_path / "workspace"
+    inputs = iter(["\x12", "he\x7fllo\x1b[A", "exit"])
+    agent_calls = []
+
+    def fake_run_pifs_agent(filesystem, question, **kwargs):
+        agent_calls.append(question)
+        return f"answer:{question}"
+
+    monkeypatch.setattr(cli, "PageIndexFileSystem", FakeFileSystem)
+    monkeypatch.setattr(cli, "run_pifs_agent", fake_run_pifs_agent)
+    monkeypatch.setattr("builtins.input", lambda prompt="": next(inputs))
+
+    status = cli.main(["chat", "--workspace", str(workspace), "--stream-mode", "off"])
+
+    assert status == 0
+    assert agent_calls == ["hllo"]
+    assert capsys.readouterr().out == "answer:hllo\n"
+
+
 def test_cli_ask_does_not_reprint_streamed_agent_output(monkeypatch, capsys, tmp_path):
     from pageindex.filesystem import cli