voidx-cli 3.5.2__py3-none-any.whl

tests/test_paste_handling.py

@@ -0,0 +1,364 @@
+from tui_helpers import *  # noqa: F403
+
+import asyncio
+import re
+
+import pytest
+from rich.console import Console
+
+from voidx.ui.tools.clipboard_image import ClipboardImageResult
+from voidx.ui.tools.clipboard_text import ClipboardTextResult
+
+def test_paste_clipboard_image_inserts_image_token(tmp_path, monkeypatch):
+    def fake_paste(_workspace: str) -> ClipboardImageResult:
+        return ClipboardImageResult(
+            status="ok",
+            message="Pasted image",
+            rel_path=".voidx/attachments/clip.png",
+            size=123,
+        )
+
+    monkeypatch.setattr("voidx_cli.app.paste_clipboard_image_from_system", fake_paste)
+    tui = _tui(tmp_path)
+
+    result = tui.paste_clipboard_image()
+
+    assert result.ok
+    assert tui._get_input_text() == "[Pasted image #1 123B] "
+    assert tui._notice == "Pasted image"
+
+
+def test_paste_clipboard_image_submit_expands_to_image_token(tmp_path, monkeypatch):
+    def fake_paste(_workspace: str) -> ClipboardImageResult:
+        return ClipboardImageResult(
+            status="ok",
+            message="Pasted image",
+            rel_path=".voidx/attachments/clip.png",
+            size=2048,
+        )
+
+    monkeypatch.setattr("voidx_cli.app.paste_clipboard_image_from_system", fake_paste)
+    tui = _tui(tmp_path)
+
+    tui.paste_clipboard_image()
+    tui._process_input(b"describe")
+    tui._process_input(b"\r")
+
+    assert tui._queue.get_nowait() == "[image-clip] describe"
+
+
+def test_ctrl_v_pastes_clipboard_image_when_available(tmp_path, monkeypatch):
+    def fake_paste_image(_workspace: str) -> ClipboardImageResult:
+        return ClipboardImageResult(
+            status="ok",
+            message="Pasted image",
+            rel_path=".voidx/attachments/clip.png",
+            size=123,
+        )
+
+    def fail_text_paste() -> ClipboardTextResult:
+        raise AssertionError("text fallback should not run when image paste succeeds")
+
+    monkeypatch.setattr("voidx_cli.app.paste_clipboard_image_from_system", fake_paste_image)
+    monkeypatch.setattr("voidx_cli.app.paste_clipboard_text_from_system", fail_text_paste)
+    tui = _tui(tmp_path)
+
+    tui._process_input(b"\x16")
+
+    assert tui._get_input_text() == "[Pasted image #1 123B] "
+    assert tui._paste_entries[0]["expanded"] == "[image-clip]"
+
+
+def test_ctrl_v_falls_back_to_clipboard_text_when_no_image(tmp_path, monkeypatch):
+    def fake_paste_image(_workspace: str) -> ClipboardImageResult:
+        return ClipboardImageResult(status="no_image", message="Clipboard does not contain an image.")
+
+    def fake_paste_text() -> ClipboardTextResult:
+        return ClipboardTextResult(status="ok", message="Pasted text", text="hello\nworld")
+
+    monkeypatch.setattr("voidx_cli.app.paste_clipboard_image_from_system", fake_paste_image)
+    monkeypatch.setattr("voidx_cli.app.paste_clipboard_text_from_system", fake_paste_text)
+    tui = _tui(tmp_path)
+
+    tui._process_input(b"\x16")
+
+    assert tui._get_input_text() == "[Pasted text #1 +1 lines]"
+    assert tui._paste_entries[0]["expanded"] == "hello\nworld"
+    assert tui._queue.empty()
+
+
+def test_bracketed_paste_multiline_text_collapses_to_token(tmp_path):
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = [""]
+    tui._cursor_col = 0
+
+    paste_data = b"\x1b[200~line1\r\nline2\r\nline3\x1b[201~"
+    tui._process_input(paste_data)
+
+    assert tui._get_input_text() == "[Pasted text #1 +2 lines]"
+    assert tui._paste_entries[0]["expanded"] == "line1\nline2\nline3"
+    assert tui._queue.empty()
+
+
+def test_bracketed_paste_large_text_collapses_to_token(tmp_path):
+    tui = _tui(tmp_path)
+    tui._tty = True
+    pasted = "line1\nline2\nline3\nline4"
+
+    tui._process_input(b"\x1b[200~" + pasted.encode() + b"\x1b[201~")
+
+    assert tui._get_input_text() == "[Pasted text #1 +3 lines]"
+    assert tui._paste_entries[0]["expanded"] == pasted
+    assert tui._queue.empty()
+
+
+def test_empty_bracketed_paste_falls_back_to_clipboard_image(tmp_path, monkeypatch):
+    def fake_paste(_workspace: str) -> ClipboardImageResult:
+        return ClipboardImageResult(
+            status="ok",
+            message="Pasted image",
+            rel_path=".voidx/attachments/clip.png",
+            size=123,
+        )
+
+    monkeypatch.setattr("voidx_cli.app.paste_clipboard_image_from_system", fake_paste)
+    tui = _tui(tmp_path)
+    tui._tty = True
+
+    tui._process_input(b"\x1b[200~\x1b[201~")
+
+    assert tui._get_input_text() == "[Pasted image #1 123B] "
+    assert tui._paste_entries[0]["expanded"] == "[image-clip]"
+    assert tui._notice == "Pasted image"
+    assert tui._queue.empty()
+
+
+def test_collapsed_paste_submit_expands_to_full_text(tmp_path):
+    tui = _tui(tmp_path)
+    tui._tty = True
+    pasted = "line1\nline2\nline3\nline4"
+    token = "[Pasted text #1 +3 lines]"
+
+    tui._process_input(b"\x1b[200~" + pasted.encode() + b"\x1b[201~")
+    tui._process_input(b"\r")
+
+    assert tui._queue.get_nowait() == f"<pasted>\n{pasted}\n</pasted>"
+    assert tui._input_history == [token]
+    assert tui._get_input_text() == ""
+
+
+def test_collapsed_paste_history_restores_registry(tmp_path):
+    tui = _tui(tmp_path)
+    tui._tty = True
+    pasted = "line1\nline2\nline3\nline4"
+    token = "[Pasted text #1 +3 lines]"
+
+    tui._process_input(b"\x1b[200~" + pasted.encode() + b"\x1b[201~")
+    tui._process_input(b"\r")
+    tui._process_input(b"\x1b[A")
+    tui._process_input(b"\r")
+
+    expected = f"<pasted>\n{pasted}\n</pasted>"
+    assert tui._queue.get_nowait() == expected
+    assert tui._queue.get_nowait() == expected
+    assert tui._input_history == [token]
+
+
+@pytest.mark.asyncio
+async def test_interrupted_submit_restores_collapsed_paste_token(tmp_path):
+    tui = _tui(tmp_path)
+    tui._tty = True
+    pasted = "line1\nline2\nline3\nline4"
+    started = asyncio.Event()
+    cancelled = asyncio.Event()
+
+    async def on_submit(text: str) -> bool:
+        assert text == f"<pasted>\n{pasted}\n</pasted>"
+        started.set()
+        try:
+            await asyncio.Event().wait()
+        except asyncio.CancelledError:
+            cancelled.set()
+            raise
+
+    tui._process_input(b"\x1b[200~" + pasted.encode() + b"\x1b[201~")
+    tui._process_input(b"\r")
+
+    consumer = asyncio.create_task(tui._consume(on_submit))
+    await started.wait()
+
+    tui._handle_interrupt()
+    await asyncio.wait_for(cancelled.wait(), timeout=1)
+    await asyncio.sleep(0)
+    consumer.cancel()
+
+    assert tui._get_input_text() == "[Pasted text #1 +3 lines]"
+    assert tui._paste_entries[0]["expanded"] == pasted
+    assert tui._notice == "Interrupted. Restored last message for editing."
+
+
+def test_user_typed_paste_lookalike_does_not_expand(tmp_path):
+    tui = _tui(tmp_path)
+    token = "[Pasted text #1 +3 lines]"
+    tui._input_lines = [token]
+    tui._cursor_col = len(token)
+
+    tui._process_input(b"\r")
+
+    assert tui._queue.get_nowait() == token
+
+
+def test_registered_paste_tokens_render_dim_cyan(tmp_path):
+    tui = _tui(tmp_path)
+    display = tui._register_text_paste("line1\nline2\nline3\nline4")
+    tui._input_lines = [display]
+    tui._cursor_row = 0
+    tui._cursor_col = 0
+    tui._console = Console(file=None, force_terminal=True, width=80, height=24, _environ={})
+
+    ansi = tui._capture_renderable(tui._render_bottom_impl(), tui._frame_width())
+    plain = re.sub(r"\x1b\[[0-9;]*[a-zA-Z]", "", ansi)
+
+    assert display in plain
+    assert "\x1b[2;36m" in ansi or "\x1b[36;2m" in ansi or "\x1b[2m" in ansi
+
+
+def test_bracketed_paste_single_line_does_not_submit(tmp_path):
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = [""]
+    tui._cursor_col = 0
+
+    paste_data = b"\x1b[200~hello world\x1b[201~"
+    tui._process_input(paste_data)
+
+    # Single-line paste is inserted as plain text, not a token
+    assert tui._get_input_text() == "hello world"
+    assert tui._paste_entries == []
+    assert tui._queue.empty()
+
+
+def test_bracketed_paste_with_trailing_key(tmp_path):
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = [""]
+    tui._cursor_col = 0
+
+    # Paste followed by a regular keypress
+    paste_data = b"\x1b[200~text\x1b[201~x"
+    tui._process_input(paste_data)
+
+    # Single-line paste is inserted as plain text, followed by the key
+    assert tui._get_input_text() == "textx"
+    assert tui._paste_entries == []
+    assert tui._queue.empty()
+
+
+def test_bracketed_paste_split_across_reads(tmp_path):
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = [""]
+    tui._cursor_col = 0
+
+    # First read: paste start + partial content
+    tui._process_input(b"\x1b[200~line1\r\n")
+    assert tui._paste_buffer is not None
+    assert tui._queue.empty()
+
+    # Second read: rest of content + paste end
+    tui._process_input(b"line2\x1b[201~")
+    assert tui._paste_buffer is None
+    assert tui._get_input_text() == "[Pasted text #1 +1 lines]"
+    assert tui._paste_entries[0]["expanded"] == "line1\nline2"
+    assert tui._queue.empty()
+
+
+def test_bracketed_paste_cr_only_normalised_to_newline(tmp_path):
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = [""]
+    tui._cursor_col = 0
+
+    # Bare CR (no LF) should also become a newline
+    paste_data = b"\x1b[200~line1\rline2\x1b[201~"
+    tui._process_input(paste_data)
+
+    assert tui._get_input_text() == "[Pasted text #1 +1 lines]"
+    assert tui._paste_entries[0]["expanded"] == "line1\nline2"
+    assert tui._queue.empty()
+
+
+def test_bracketed_paste_single_line_short_inserts_as_plain_text(tmp_path):
+    """Short single-line paste (e.g. IME confirmation) is inserted as plain
+    text, not collapsed into a [Pasted text] token."""
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = [""]
+    tui._cursor_col = 0
+
+    paste_data = b"\x1b[200~hello\x1b[201~"
+    tui._process_input(paste_data)
+
+    assert tui._get_input_text() == "hello"
+    assert tui._paste_entries == []
+    assert tui._queue.empty()
+
+
+def test_bracketed_paste_single_line_long_inserts_as_plain_text(tmp_path):
+    """Even a longer single-line paste is inserted as plain text — only
+    multiline content becomes a paste token."""
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = [""]
+    tui._cursor_col = 0
+
+    pasted = "a" * 200
+    paste_data = b"\x1b[200~" + pasted.encode() + b"\x1b[201~"
+    tui._process_input(paste_data)
+
+    assert tui._get_input_text() == pasted
+    assert tui._paste_entries == []
+    assert tui._queue.empty()
+
+
+def test_bracketed_paste_multiline_still_becomes_token(tmp_path):
+    """Multiline paste content is still collapsed into a paste token."""
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = [""]
+    tui._cursor_col = 0
+
+    paste_data = b"\x1b[200~line1\nline2\x1b[201~"
+    tui._process_input(paste_data)
+
+    assert tui._get_input_text() == "[Pasted text #1 +1 lines]"
+    assert tui._paste_entries[0]["expanded"] == "line1\nline2"
+    assert tui._queue.empty()
+
+
+def test_bracketed_paste_short_then_submit_no_pasted_tag(tmp_path):
+    """Short paste submitted immediately should not be wrapped in <pasted> tags."""
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = [""]
+    tui._cursor_col = 0
+
+    tui._process_input(b"\x1b[200~hello\x1b[201~")
+    tui._process_input(b"\r")
+
+    assert tui._queue.get_nowait() == "hello"
+    assert tui._get_input_text() == ""
+
+
+def test_bracketed_paste_short_with_existing_text(tmp_path):
+    """Short paste appended to existing text stays as plain text."""
+    tui = _tui(tmp_path)
+    tui._tty = True
+    tui._input_lines = ["say "]
+    tui._cursor_col = 4
+
+    tui._process_input(b"\x1b[200~hi\x1b[201~")
+
+    assert tui._get_input_text() == "say hi"
+    assert tui._paste_entries == []

tests/test_startup.py

@@ -0,0 +1,60 @@
+import re
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
+
+from rich.cells import cell_len
+
+from voidx.ui.output.console import VoidConsole
+from voidx.ui.output.dock import ANSI_LINE_PREFIX, BottomInputDock, set_dock
+from voidx.ui.session import show_startup
+
+
+_ANSI_RE = re.compile(r"\x1b\[[0-9;?]*[ -/]*[@-~]")
+
+
+def _plain(line: str) -> str:
+    return _ANSI_RE.sub("", line.replace(ANSI_LINE_PREFIX, "")).rstrip()
+
+
+def test_startup_banner_has_no_internal_blank_rows():
+    test_dock = BottomInputDock()
+    set_dock(test_dock)
+    test_dock.begin_capture()
+    try:
+        show_startup(
+            VoidConsole(),
+            model="mimo-v2.5",
+            provider="mimo",
+            workspace="/Users/chikham/workspace/voidx",
+            session_title="你好",
+            is_new=False,
+        )
+
+        lines = [_plain(line) for line in test_dock.tree.render(80)]
+
+        assert lines
+        assert all(line.strip() for line in lines)
+        assert len(lines) <= 9
+        assert max(cell_len(line) for line in lines) <= 80
+
+        text = "\n".join(lines)
+        assert "/\\________/\\    ╭╮" in text
+        assert "◒      ◒" in text
+        assert "╭╮" in text
+        assert "o     O" in text
+        assert "Workspace" in text
+        assert "Ask anything" in text
+        assert "/model switch" in text
+        assert "@ attach" in text
+        if sys.platform != "win32":
+            assert "Ctrl+" in text  # truncated at narrow width
+        assert "PgUp/PgDn scroll transcript" in text
+        assert "Panels:" in text
+        assert "Enter accept" in text
+        assert "•_•" not in text
+    finally:
+        test_dock.deactivate()
+        test_dock.reset()
+        set_dock(None)