fix(cli): pause spinner cleanly before printing progress output

nanobot/cli/commands.py

@@ -1,7 +1,7 @@
 """CLI commands for nanobot."""
 
 import asyncio
-from contextlib import contextmanager
+from contextlib import contextmanager, nullcontext
 import os
 import select
 import signal
@@ -170,6 +170,51 @@ async def _print_interactive_response(response: str, render_markdown: bool) -> N
     await run_in_terminal(_write)
 
 
+class _ThinkingSpinner:
+    """Spinner wrapper with pause support for clean progress output."""
+
+    def __init__(self, enabled: bool):
+        self._spinner = console.status(
+            "[dim]nanobot is thinking...[/dim]", spinner="dots"
+        ) if enabled else None
+        self._active = False
+
+    def __enter__(self):
+        if self._spinner:
+            self._spinner.start()
+        self._active = True
+        return self
+
+    def __exit__(self, *exc):
+        self._active = False
+        if self._spinner:
+            self._spinner.stop()
+        return False
+
+    @contextmanager
+    def pause(self):
+        """Temporarily stop spinner while printing progress."""
+        if self._spinner and self._active:
+            self._spinner.stop()
+        try:
+            yield
+        finally:
+            if self._spinner and self._active:
+                self._spinner.start()
+
+
+def _print_cli_progress_line(text: str, thinking: _ThinkingSpinner | None) -> None:
+    """Print a CLI progress line, pausing the spinner if needed."""
+    with thinking.pause() if thinking else nullcontext():
+        console.print(f"  [dim]↳ {text}[/dim]")
+
+
+async def _print_interactive_progress_line(text: str, thinking: _ThinkingSpinner | None) -> None:
+    """Print an interactive progress line, pausing the spinner if needed."""
+    with thinking.pause() if thinking else nullcontext():
+        await _print_interactive_line(text)
+
+
 def _is_exit_command(command: str) -> bool:
     """Return True when input should end interactive chat."""
     return command.lower() in EXIT_COMMANDS
@@ -635,39 +680,6 @@ def agent(
         channels_config=config.channels,
     )
 
-    # Show spinner when logs are off (no output to miss); skip when logs are on
-    class _ThinkingSpinner:
-        """Context manager that owns spinner lifecycle with pause support."""
-
-        def __init__(self):
-            self._spinner = None if logs else console.status(
-                "[dim]nanobot is thinking...[/dim]", spinner="dots"
-            )
-            self._active = False
-
-        def __enter__(self):
-            if self._spinner:
-                self._spinner.start()
-            self._active = True
-            return self
-
-        def __exit__(self, *exc):
-            self._active = False
-            if self._spinner:
-                self._spinner.stop()
-            return False
-
-        @contextmanager
-        def pause(self):
-            """Temporarily stop spinner for clean console output."""
-            if self._spinner and self._active:
-                self._spinner.stop()
-            try:
-                yield
-            finally:
-                if self._spinner and self._active:
-                    self._spinner.start()
-
     # Shared reference for progress callbacks
     _thinking: _ThinkingSpinner | None = None
 
@@ -677,17 +689,13 @@ def agent(
             return
         if ch and not tool_hint and not ch.send_progress:
             return
-        if _thinking:
-            with _thinking.pause():
-                console.print(f"  [dim]↳ {content}[/dim]")
-        else:
-            console.print(f"  [dim]↳ {content}[/dim]")
+        _print_cli_progress_line(content, _thinking)
 
     if message:
         # Single message mode — direct call, no bus needed
         async def run_once():
             nonlocal _thinking
-            _thinking = _ThinkingSpinner()
+            _thinking = _ThinkingSpinner(enabled=not logs)
             with _thinking:
                 response = await agent_loop.process_direct(message, session_id, on_progress=_cli_progress)
             _thinking = None
@@ -739,11 +747,8 @@ def agent(
                                 pass
                             elif ch and not is_tool_hint and not ch.send_progress:
                                 pass
-                            elif _thinking:
-                                with _thinking.pause():
-                                    await _print_interactive_line(msg.content)
                             else:
-                                await _print_interactive_line(msg.content)
+                                await _print_interactive_progress_line(msg.content, _thinking)
 
                         elif not turn_done.is_set():
                             if msg.content:
@@ -784,7 +789,7 @@ def agent(
                         ))
 
                         nonlocal _thinking
-                        _thinking = _ThinkingSpinner()
+                        _thinking = _ThinkingSpinner(enabled=not logs)
                         with _thinking:
                             await turn_done.wait()
                         _thinking = None