Always supress and disable suppression at end

Author: Drummersbrother

README.md

@@ -47,13 +47,7 @@ curl -fsSL https://raw.githubusercontent.com/Edison-Watch/open-edison/main/curl_
 ```
 
 Run locally with uvx: `uvx open-edison`
-
-Optionally, run the setup wizard to import/configure MCP:
-
-```bash
-uv run python -m src.setup_tui.main
-# add --dry-run to preview without writing
-```
+That will run the setup wizard if necessary.
 
 <details>
 <summary>⬇️ Install Node.js/npm (optional for MCP tools)</summary>
@@ -104,19 +98,6 @@ OPEN_EDISON_CONFIG_DIR=~/edison-config open-edison run
 
 </details>
 
-<details>
-<summary>🔄 Import from Cursor/VS Code/Claude Code</summary>
-
-Run the interactive setup wizard to detect clients, import servers, and configure your editor:
-
-```bash
-uv run python -m src.setup_tui.main
-```
-
-Use `--dry-run` to preview without writing.
-
-</details>
-
 <details>
 <summary><img src="https://img.shields.io/badge/Docker-2CA5E0?style=for-the-badge&logo=docker&logoColor=white" alt="Docker"> Run with Docker</summary>
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "open-edison"
-version = "0.1.58"
+version = "0.1.59"
 description = "Open-source MCP security, aggregation, and monitoring. Single-user, self-hosted MCP proxy."
 readme = "README.md"
 authors = [

src/mcp_importer/api.py

@@ -238,7 +238,8 @@ async def _list_tools_only() -> Any:
                     # NOT_REQUIRED: quick unauthenticated ping
                     # TODO: In debug mode, do not suppress child process output.
                     questionary.print(
-                        f"Testing connection to '{server.name}'...", style="bold fg:ansigreen"
+                        f"Testing connection to '{server.name}'... (timeout: {connection_timeout}s)",
+                        style="bold fg:ansigreen",
                     )
                     log.debug(f"Establishing contact with remote server '{server.name}'")
                     async with asyncio.timeout(connection_timeout):

src/setup_tui/main.py

@@ -173,42 +173,45 @@ def show_manual_setup_screen() -> None:
     print(after_text)
 
 
+class _TuiLogger:
+    def _fmt(self, msg: object, *args: object) -> str:
+        try:
+            if isinstance(msg, str) and args:
+                return msg.format(*args)
+        except Exception:
+            pass
+        return str(msg)
+
+    def info(self, msg: object, *args: object, **kwargs: object) -> None:
+        questionary.print(self._fmt(msg, *args), style="fg:ansiblue")
+
+    def debug(self, msg: object, *args: object, **kwargs: object) -> None:
+        questionary.print(self._fmt(msg, *args), style="fg:ansiblack")
+
+    def warning(self, msg: object, *args: object, **kwargs: object) -> None:
+        questionary.print(self._fmt(msg, *args), style="bold fg:ansiyellow")
+
+    def error(self, msg: object, *args: object, **kwargs: object) -> None:
+        questionary.print(self._fmt(msg, *args), style="bold fg:ansired")
+
+
 @contextlib.contextmanager
 def suppress_loguru_output() -> Generator[None, None, None]:
     """Suppress loguru output."""
     with contextlib.suppress(Exception):
         log.remove()
+
+    old_logger = oauth_mod.log
+    # Route oauth_manager's log calls to questionary for TUI output
+    oauth_mod.log = _TuiLogger()  # type: ignore[attr-defined]
     yield
+    oauth_mod.log = old_logger
     log.add(sys.stdout, level="INFO")
 
 
 @suppress_loguru_output()
 def run(*, dry_run: bool = False, skip_oauth: bool = False) -> None:  # noqa: C901
     """Run the complete setup process."""
-
-    # Route oauth_manager's log calls to questionary for TUI output
-    class _TuiLogger:
-        def _fmt(self, msg: object, *args: object) -> str:
-            try:
-                if isinstance(msg, str) and args:
-                    return msg.format(*args)
-            except Exception:
-                pass
-            return str(msg)
-
-        def info(self, msg: object, *args: object, **kwargs: object) -> None:
-            questionary.print(self._fmt(msg, *args), style="fg:ansiblue")
-
-        def debug(self, msg: object, *args: object, **kwargs: object) -> None:
-            questionary.print(self._fmt(msg, *args), style="fg:ansiblack")
-
-        def warning(self, msg: object, *args: object, **kwargs: object) -> None:
-            questionary.print(self._fmt(msg, *args), style="bold fg:ansiyellow")
-
-        def error(self, msg: object, *args: object, **kwargs: object) -> None:
-            questionary.print(self._fmt(msg, *args), style="bold fg:ansired")
-
-    oauth_mod.log = _TuiLogger()  # type: ignore[attr-defined]
     show_welcome_screen(dry_run=dry_run)
     # Additional setup steps will be added here
 
@@ -244,15 +247,15 @@ def error(self, msg: object, *args: object, **kwargs: object) -> None:
 # Triggered from cli.py
 def run_import_tui(args: argparse.Namespace, force: bool = False) -> None:
     """Run the import TUI, if necessary."""
-    # Find config dir, check if ".setup_tui_run" exists
+    # Find config dir, check if ".setup_tui_ran" exists
     config_dir = get_config_dir()
     config_dir.mkdir(parents=True, exist_ok=True)
 
-    setup_tui_run_file = config_dir / ".setup_tui_run"
-    if not setup_tui_run_file.exists() or force:
+    setup_tui_ran_file = config_dir / ".setup_tui_ran"
+    if not setup_tui_ran_file.exists() or force:
         run(dry_run=args.wizard_dry_run, skip_oauth=args.wizard_skip_oauth)
 
-    setup_tui_run_file.touch()
+    setup_tui_ran_file.touch()
 
 
 def main(argv: list[str] | None = None) -> int: