open-edison 0.1.43__py3-none-any.whl → 0.1.64__py3-none-any.whl

src/setup_tui/main.py

@@ -0,0 +1,294 @@
+import argparse
+import asyncio
+import contextlib
+import sys
+from collections.abc import Generator
+
+import questionary
+from loguru import logger as log
+
+import src.oauth_manager as oauth_mod
+from src.config import MCPServerConfig, get_config_dir
+from src.mcp_importer.api import (
+    CLIENT,
+    authorize_server_oauth,
+    detect_clients,
+    export_edison_to,
+    has_oauth_tokens,
+    import_from,
+    save_imported_servers,
+    verify_mcp_server,
+)
+from src.mcp_importer.parsers import deduplicate_by_name
+from src.oauth_manager import OAuthStatus, get_oauth_manager
+
+
+def show_welcome_screen(*, dry_run: bool = False) -> None:
+    """Display the welcome screen for open-edison setup."""
+    welcome_text = """
+    ╔══════════════════════════════════════════════════════════════╗
+    ║                                                              ║
+    ║                    Welcome to open-edison                    ║
+    ║                                                              ║
+    ║     This setup wizard will help you configure open-edison    ║
+    ║     for your development environment.                        ║
+    ║                                                              ║
+    ╚══════════════════════════════════════════════════════════════╝
+    """
+
+    print(welcome_text)
+
+    # Prompt to continue
+    questionary.confirm("Ready to begin the setup process?", default=True).ask()
+
+
+def handle_mcp_source(  # noqa: C901
+    source: CLIENT, *, dry_run: bool = False, skip_oauth: bool = False
+) -> list[MCPServerConfig]:
+    """Handle the MCP source."""
+    if not questionary.confirm(
+        f"We have found {source.name} installed. Would you like to import its MCP servers to open-edison?",
+        default=True,
+    ).ask():
+        return []
+
+    configs = import_from(source)
+
+    # Filter out any "open-edison" configs
+    if "open-edison" in [config.name for config in configs]:
+        print(
+            "Found an 'open-edison' config. This is not allowed, so it will be excluded from the import."
+        )
+
+    configs = [config for config in configs if config.name != "open-edison"]
+
+    print(f"Loaded {len(configs)} MCP server configuration from {source.name}!")
+
+    verified_configs: list[MCPServerConfig] = []
+
+    for config in configs:
+        print(f"Verifying the configuration for {config.name}... ")
+        result = verify_mcp_server(config)
+        if result:
+            # For remote servers, only prompt if OAuth is actually required
+            if config.is_remote_server():
+                # Heuristic: if inline headers are present (e.g., API key), treat as not requiring OAuth
+                has_inline_headers: bool = any(
+                    (a == "--header" or a.startswith("--header")) for a in config.args
+                )
+                if not has_inline_headers:
+                    # Prefer cached result from verification; only check if missing
+                    oauth_mgr = get_oauth_manager()
+                    info = oauth_mgr.get_server_info(config.name)
+                    if info is None:
+                        info = asyncio.run(
+                            oauth_mgr.check_oauth_requirement(config.name, config.get_remote_url())
+                        )
+
+                    if info.status == OAuthStatus.NEEDS_AUTH:
+                        tokens_present: bool = has_oauth_tokens(config)
+                        if not tokens_present:
+                            if skip_oauth:
+                                print(
+                                    f"Skipping OAuth for {config.name} due to --skip-oauth (OAuth required, no tokens). This server will not be imported."
+                                )
+                                continue
+
+                            if questionary.confirm(
+                                f"{config.name} requires OAuth and no credentials were found. Obtain credentials now?",
+                                default=True,
+                            ).ask():
+                                success = authorize_server_oauth(config)
+                                if not success:
+                                    print(
+                                        f"Failed to obtain OAuth credentials for {config.name}. Skipping this server."
+                                    )
+                                    continue
+                            else:
+                                print(f"Skipping {config.name} per user choice.")
+                                continue
+
+            print(f"Verification successful for {config.name}.")
+            verified_configs.append(config)
+        else:
+            print(
+                f"Verification failed for the configuration of {config.name}. Please check the configuration and try again."
+            )
+
+    return verified_configs
+
+
+def confirm_configs(configs: list[MCPServerConfig], *, dry_run: bool = False) -> bool:
+    """Confirm the MCP configs."""
+    print("These are the servers you have selected:")
+
+    for config in configs:
+        print(f"○ {config.name}")
+
+    return questionary.confirm(
+        "Are you sure you want to use these servers with open-edison?", default=True
+    ).ask()
+
+
+def confirm_apply_configs(client: CLIENT, *, dry_run: bool = False) -> None:
+    if not questionary.confirm(
+        f"We have detected that you have {client.name} installed. Would you like to connect it to open-edison?",
+        default=True,
+    ).ask():
+        return
+
+    export_edison_to(client, dry_run=dry_run)
+    if dry_run:
+        print(f"[dry-run] Export prepared for {client.name}; no changes written.")
+    else:
+        print(f"Successfully set up Open Edison for {client.name}!")
+
+
+def show_manual_setup_screen() -> None:
+    """Display manual setup instructions for open-edison."""
+    manual_setup_text = """
+    ╔══════════════════════════════════════════════════════════════╗
+    ║                                                              ║
+    ║                     Manual Setup Instructions                ║
+    ║                                                              ║
+    ╚══════════════════════════════════════════════════════════════╝
+
+    To set up open-edison manually in other clients, find your client's MCP config
+    JSON file and add the following configuration:
+    """
+
+    json_snippet = """\t{
+      "mcpServers": {
+        "open-edison": {
+          "command": "npx",
+          "args": [
+            "-y",
+            "mcp-remote",
+            "http://localhost:3000/mcp/",
+            "--http-only",
+            "--header",
+            "Authorization: Bearer dev-api-key-change-me"
+          ]
+        }
+      }
+    }"""
+
+    after_text = """
+    Make sure to replace 'dev-api-key-change-me' with your actual API key.
+    """
+
+    print(manual_setup_text)
+    # Use questionary's print with style for color
+    questionary.print(json_snippet, style="bold fg:ansigreen")
+    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) -> bool:  # noqa: C901
+    """Run the complete setup process.
+    Returns whether the setup was successful."""
+    show_welcome_screen(dry_run=dry_run)
+    # Additional setup steps will be added here
+
+    mcp_clients = sorted(detect_clients(), key=lambda x: x.value)
+
+    configs: list[MCPServerConfig] = []
+
+    for client in mcp_clients:
+        configs.extend(handle_mcp_source(client, dry_run=dry_run, skip_oauth=skip_oauth))
+
+    if len(configs) == 0:
+        if not questionary.confirm(
+            "No MCP servers found. Would you like to continue without them?", default=False
+        ).ask():
+            print("Setup aborted. Please configure an MCP client and try again.")
+            return False
+        return True
+
+    # Deduplicate configs
+    configs = deduplicate_by_name(configs)
+
+    if not confirm_configs(configs, dry_run=dry_run):
+        return False
+
+    for client in mcp_clients:
+        confirm_apply_configs(client, dry_run=dry_run)
+
+    # Persist imported servers into config.json
+    if len(configs) > 0:
+        save_imported_servers(configs, dry_run=dry_run)
+
+    show_manual_setup_screen()
+
+    return True
+
+
+# Triggered from cli.py
+def run_import_tui(args: argparse.Namespace, force: bool = False) -> bool:
+    """Run the import TUI, if necessary."""
+    # Find config dir, check if ".setup_tui_ran" exists
+    config_dir = get_config_dir()
+    config_dir.mkdir(parents=True, exist_ok=True)
+
+    setup_tui_ran_file = config_dir / ".setup_tui_ran"
+    success = True
+    if not setup_tui_ran_file.exists() or force:
+        success = run(dry_run=args.wizard_dry_run, skip_oauth=args.wizard_skip_oauth)
+
+    setup_tui_ran_file.touch()
+
+    return success
+
+
+def main(argv: list[str] | None = None) -> int:
+    parser = argparse.ArgumentParser(description="Open Edison Setup TUI")
+    parser.add_argument("--dry-run", action="store_true", help="Preview actions without writing")
+    parser.add_argument(
+        "--skip-oauth",
+        action="store_true",
+        help="Skip OAuth for remote servers (they will be omitted from import)",
+    )
+    args = parser.parse_args(argv)
+
+    run(dry_run=args.dry_run, skip_oauth=args.skip_oauth)
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

src/single_user_mcp.py

@@ -9,6 +9,7 @@ from typing import Any, TypedDict
 
 from fastmcp import Client as FastMCPClient
 from fastmcp import Context, FastMCP
+from fastmcp.server.dependencies import get_context
 from loguru import logger as log
 
 from src.config import Config, MCPServerConfig
@@ -281,9 +282,6 @@ class SingleUserMCP(FastMCP[Any]):
     async def _send_list_changed_notifications(self) -> None:
         """Send notifications to clients about changed component lists."""
         try:
-            # Import here to avoid circular imports
-            from fastmcp.server.dependencies import get_context
-
             try:
                 context = get_context()
                 # Queue notifications for all component types since we don't know
@@ -323,12 +321,18 @@ class SingleUserMCP(FastMCP[Any]):
         log.info("✅ Single User MCP server initialized with composite proxy")
 
         # Invalidate and warm lists to ensure reload
+        log.debug("Reloading tool list...")
         _ = await self._tool_manager.list_tools()
+        log.debug("Reloading resource list...")
         _ = await self._resource_manager.list_resources()
+        log.debug("Reloading prompt list...")
         _ = await self._prompt_manager.list_prompts()
+        log.debug("Reloading complete")
 
         # Send notifications to clients about changed component lists
+        log.debug("Sending list changed notifications...")
         await self._send_list_changed_notifications()
+        log.debug("List changed notifications sent")
 
     def _calculate_risk_level(self, trifecta: dict[str, bool]) -> str:
         """

src/tools/io.py

@@ -0,0 +1,35 @@
+import os
+from collections.abc import Iterator
+from contextlib import contextmanager
+
+
+@contextmanager
+def suppress_fds(*, suppress_stdout: bool = False, suppress_stderr: bool = True) -> Iterator[None]:
+    """Temporarily redirect process-level stdout/stderr to os.devnull.
+
+    Args:
+        suppress_stdout: If True, redirect fd 1 to devnull
+        suppress_stderr: If True, redirect fd 2 to devnull
+
+    Yields:
+        None
+    """
+    saved: list[tuple[int, int]] = []
+    try:
+        if suppress_stdout:
+            saved.append((1, os.dup(1)))
+            devnull_out = os.open(os.devnull, os.O_WRONLY)
+            os.dup2(devnull_out, 1)
+            os.close(devnull_out)
+        if suppress_stderr:
+            saved.append((2, os.dup(2)))
+            devnull_err = os.open(os.devnull, os.O_WRONLY)
+            os.dup2(devnull_err, 2)
+            os.close(devnull_err)
+        yield
+    finally:
+        for fd, backup in saved:
+            try:
+                os.dup2(backup, fd)
+            finally:
+                os.close(backup)

src/vulture_whitelist.py

@@ -0,0 +1,3 @@
+from src.oauth_override import OpenEdisonOAuth
+
+OpenEdisonOAuth.redirect_handler  # noqa: B018 unused method (src/oauth_override.py:7)

open_edison-0.1.43.dist-info/RECORD

@@ -1,35 +0,0 @@
-src/__init__.py,sha256=bEYMwBiuW9jzF07iWhas4Vb30EcpnqfpNfz_Q6yO1jU,209
-src/__main__.py,sha256=kQsaVyzRa_ESC57JpKDSQJAHExuXme0rM5beJsYxFeA,161
-src/cli.py,sha256=fqX-HuRDePRasexpnURQ_pVYeycJuWxllMcwfqDxMQw,8490
-src/config.py,sha256=RSsAYzl8cj6eaDN1RORMcfKKWBcp4bKTQp2BdhAL9mg,10258
-src/config.pyi,sha256=FgehEGli8ZXSjGlANBgMGv5497q4XskQciOc1fUcxqM,2033
-src/events.py,sha256=aFQrVXDIZwt55Dz6OtyoXu2yi9evqo-8jZzo3CR2Tto,4965
-src/oauth_manager.py,sha256=W9QSo0vfGDQ_i-QWCngkv7YLSL3Rk5jfPmqjU1J2rnU,9911
-src/permissions.py,sha256=NGAnlG_z59HEiVA-k3cYvwmmiuHzxuNb5Tbd5umbL00,10483
-src/server.py,sha256=cnO5bgxT-lrfuwk9AIvB_HBV8SWOtFClfGUn5_zFWyo,45652
-src/single_user_mcp.py,sha256=rJrlqHcIubGkos_24ux5rb3OoKYDzvagCHghhfDeXTI,18535
-src/telemetry.py,sha256=-RZPIjpI53zbsKmp-63REeZ1JirWHV5WvpSRa2nqZEk,11321
-src/frontend_dist/index.html,sha256=s95FMkH8VLisvawLH7bZxbLzRUFvMhHkH6ZMzpVBngs,673
-src/frontend_dist/sw.js,sha256=rihX1es-vWwjmtnXyaksJjs2dio6MVAOTAWwQPeJUYw,2164
-src/frontend_dist/assets/index-BUUcUfTt.js,sha256=awoyPI6u0v6ao2iarZdSkrSDUvyU8aNkMLqHMvgVgyY,257666
-src/frontend_dist/assets/index-o6_8mdM8.css,sha256=nwmX_6q55mB9463XN2JM8BdeihjkALpQK83Fc3_iGvE,15936
-src/mcp_importer/__init__.py,sha256=Mk59pVr7OMGfYGWeSYk8-URfhIcrs3SPLYS7fmJbMII,275
-src/mcp_importer/__main__.py,sha256=0jVfxKzyr6koVu1ghhWseah5ilKIoGovE6zkEZ-u-Og,515
-src/mcp_importer/api.py,sha256=47tur0xgl1NBI1Vnh3cpScEmDS64bKMYcWjZDuqx7HQ,6644
-src/mcp_importer/cli.py,sha256=Pe0GLWm1nMd1VuNXOSkxIrFZuGNFc9dNvfBsvf-bdBI,3487
-src/mcp_importer/export_cli.py,sha256=daEadB6nL8P4OpEGFx0GshuN1a091L7BhiitpV1bPqA,6294
-src/mcp_importer/exporters.py,sha256=fSgl6seduoXFp7YnKH26UEaC1sFBnd4whSut7CJLBQs,11348
-src/mcp_importer/import_api.py,sha256=xWaKoE3vibSWpA5roVL7qEMS73vcmAC0tcHP6CsZw6E,95
-src/mcp_importer/importers.py,sha256=zGN8lT7qQJ95jDTd-ck09j_w5PSvH-uj33TILoHfHbs,2191
-src/mcp_importer/merge.py,sha256=KIGT7UgbAm07-LdyoUXEJ7ABSIiPTFlj_qjz669yFxg,1569
-src/mcp_importer/parsers.py,sha256=JRE7y_Gg-QmlAARvZdrI9CmUyy-ODvDPbS695pb3Aw8,4856
-src/mcp_importer/paths.py,sha256=4L-cPr7KCM9X9gAUP7Da6ictLNrPWuQ_IM419zqY-2I,2700
-src/mcp_importer/quick_cli.py,sha256=4mJe10q_lZCYLm75QBt1rYy2j8mGEsRZoAqA0agjfSM,1834
-src/mcp_importer/types.py,sha256=h03TbAnJbap6OWWd0dT0QcFWNvSaiVFWH9V9PD6x4s0,138
-src/middleware/data_access_tracker.py,sha256=bArBffWgYmvxOx9z_pgXQhogvnWQcc1m6WvEblDD4gw,15039
-src/middleware/session_tracking.py,sha256=5W1VH9HNqIZeX0HNxDEm41U4GY6SqKSXtApDEeZK2qo,23084
-open_edison-0.1.43.dist-info/METADATA,sha256=OO5PDNk7pByRgIShryHaCtp2_7ua9XOwC8eBVdU-b5o,13188
-open_edison-0.1.43.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-open_edison-0.1.43.dist-info/entry_points.txt,sha256=YiGNm9x2I00hgT10HDyB4gxC1LcaV_mu8bXFjolu0Yw,171
-open_edison-0.1.43.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-open_edison-0.1.43.dist-info/RECORD,,