py-mcpdock-cli 1.0.13__py3-none-any.whl

cli/utils/runtime.py

@@ -0,0 +1,163 @@
+import subprocess
+import sys
+import shutil
+import os
+from typing import Dict, Optional, List
+import questionary
+from rich import print as rprint
+
+from ..types.registry import ConnectionDetails, RegistryServer # Assuming RegistryServer includes connections
+from .logger import verbose
+
+
+def check_command_installed(command: str) -> bool:
+    """Checks if a command is available in the system's PATH."""
+    if not shutil.which(command):
+        verbose(f"Command '{command}' not found in PATH.")
+        return False
+    # Optionally, run a version command to be more certain
+    try:
+        # Use shell=True if command might be an alias or shell function, but be cautious
+        # Using the direct path from shutil.which is safer if possible
+        cmd_path = shutil.which(command)
+        if not cmd_path:
+             return False # Should not happen if which found it, but safety check
+        result = subprocess.run([cmd_path, "--version"], capture_output=True, text=True, timeout=5, check=False)
+        verbose(f"'{command} --version' exited with code {result.returncode}")
+        # Check return code - 0 usually means success
+        return result.returncode == 0
+    except (subprocess.TimeoutExpired, OSError, FileNotFoundError) as e:
+        verbose(f"Error checking version for '{command}': {e}")
+        # If version check fails, assume it might still work or is problematic
+        return False # Treat version check failure as potentially not installed correctly
+    except Exception as e:
+        verbose(f"Unexpected error checking '{command}': {e}")
+        return False
+
+async def prompt_for_install(command: str, install_url: str, command_name: str) -> bool:
+    """Asks the user if they want to install a missing command."""
+    should_install = await questionary.confirm(
+        f"{command_name} is required for this operation. Would you like to attempt installation?",
+        default=True
+    ).ask_async()
+
+    if not should_install:
+        rprint(f"[yellow]{command_name} installation declined. You can install it manually from {install_url}[/yellow]")
+        return False
+
+    return await install_command(command, install_url, command_name)
+
+async def install_command(command: str, install_url: str, command_name: str) -> bool:
+    """Attempts to install a command using provided curl/powershell scripts."""
+    rprint(f"Attempting to install {command_name}...")
+    try:
+        if sys.platform == "win32":
+            ps_command = f"powershell -ExecutionPolicy ByPass -NoProfile -Command \"irm '{install_url}/install.ps1' | iex\""
+            verbose(f"Running install command (Windows): {ps_command}")
+            # Using shell=True for powershell pipeline
+            result = subprocess.run(ps_command, shell=True, capture_output=True, text=True, check=True)
+        else:
+            sh_command = f"curl -LsSf '{install_url}/install.sh' | sh"
+            verbose(f"Running install command (Unix): {sh_command}")
+            # Using shell=True for curl pipeline
+            result = subprocess.run(sh_command, shell=True, capture_output=True, text=True, check=True)
+
+        verbose(f"Installation stdout:\n{result.stdout}")
+        if result.stderr:
+             verbose(f"Installation stderr:\n{result.stderr}")
+        rprint(f"[green]✓ {command_name} installed successfully.[/green]")
+        return True
+    except subprocess.CalledProcessError as e:
+        rprint(f"[red]Failed to install {command_name}.[/red]")
+        rprint(f"Stderr:\n{e.stderr}")
+        rprint(f"You can try installing it manually from {install_url}")
+        return False
+    except Exception as e:
+        rprint(f"[red]An unexpected error occurred during installation: {e}[/red]")
+        rprint(f"You can try installing it manually from {install_url}")
+        return False
+
+def is_command_required(connection: ConnectionDetails, command: str) -> bool:
+    """Checks if a command is listed in the stdioFunction for a stdio connection."""
+    return (
+        connection.type == "stdio" and
+        isinstance(connection.stdioFunction, list) and
+        command in connection.stdioFunction
+    )
+
+async def ensure_command_installed(
+    connection: ConnectionDetails,
+    command: str,
+    command_name: str,
+    install_url: str,
+) -> bool:
+    """Checks if a required command is installed, prompts to install if not."""
+    if is_command_required(connection, command):
+        verbose(f"{command_name} installation check required for connection type {connection.type}.")
+        if not check_command_installed(command):
+            rprint(f"[yellow]Required command '{command_name}' ('{command}') not found or not working.[/yellow]")
+            installed = await prompt_for_install(command, install_url, command_name)
+            if not installed:
+                rprint(f"[yellow]Warning: {command_name} is not installed. The server might fail to launch.[/yellow]")
+                return False
+            # Re-check after installation attempt
+            if not check_command_installed(command):
+                 rprint(f"[red]Error: {command_name} installation seemed to succeed, but '{command}' is still not working correctly.[/red]")
+                 return False
+            return True # Installed successfully
+        else:
+            verbose(f"Command '{command_name}' ('{command}') is installed.")
+            return True # Already installed
+    else:
+        verbose(f"Command '{command_name}' not required for this connection.")
+        return True # Not required for this connection
+
+async def ensure_uv_installed(connection: ConnectionDetails) -> bool:
+    """Ensures UV (uvx command) is installed if required by the connection."""
+    # Assuming uvx is the command to check for uv
+    return await ensure_command_installed(connection, "uvx", "UV", "https://astral.sh/uv")
+
+async def ensure_bun_installed(connection: ConnectionDetails) -> bool:
+    """Ensures Bun (bunx command) is installed if required by the connection."""
+    # Assuming bunx is the command to check for bun
+    return await ensure_command_installed(connection, "bunx", "Bun", "https://bun.sh")
+
+def get_runtime_environment(base_env: Optional[Dict[str, str]] = None) -> Dict[str, str]:
+    """Gets runtime environment variables.
+
+    Placeholder: This function needs to replicate the behavior of
+    `@modelcontextprotocol/sdk/client/stdio.js#getDefaultEnvironment`
+    if specific environment variables are required by the MCP servers.
+    Currently, it merges provided base_env with the current process environment.
+    """
+    default_env = os.environ.copy()
+    verbose("Using current process environment as default runtime environment.")
+    # --- Placeholder Start ---
+    # TODO: Determine if specific MCP environment variables are needed
+    # and set them here if possible.
+    # Example: default_env["MCP_PYTHON_CLIENT"] = "1"
+    # --- Placeholder End ---
+
+    if base_env:
+        return {**default_env, **base_env}
+    else:
+        return default_env
+
+def check_and_notify_remote_server(server: RegistryServer) -> bool:
+    """Checks if the server is remote and prints a security notice if it is."""
+    # Determine if remote based on connections or explicit flag
+    is_remote = (
+        any(conn.type == "ws" and conn.deploymentUrl for conn in server.connections)
+        and server.remote is not False # Explicitly false overrides ws check
+    )
+
+    if is_remote:
+        verbose("Remote server detected, showing security notice")
+        rprint(
+            "[blue]Installing remote server. Please ensure you trust the server author, "
+            "especially when sharing sensitive data.[/blue]"
+        )
+        # Add link if available
+        # rprint(f"[blue]For information on data policy, please visit: [underline]https://example.com/data-policy[/underline][/blue]")
+
+    return is_remote

py_mcpdock_cli-1.0.13.dist-info/METADATA

@@ -0,0 +1,28 @@
+Metadata-Version: 2.4
+Name: py-mcpdock-cli
+Version: 1.0.13
+Summary: Python CLI for managing MCP servers
+Author-email: dw <qindongwoxin@gmail.com>
+Project-URL: Homepage, https://github.com/yourusername/py-mcpdock-cli
+Project-URL: Bug Tracker, https://github.com/yourusername/py-mcpdock-cli/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: click>=8.0.0
+Requires-Dist: questionary>=1.10.0
+Requires-Dist: rich>=12.0.0
+Requires-Dist: asyncio>=3.4.3
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: mcp
+Requires-Dist: aiohttp>=3.8.0
+
+# MCP CLI Python Implementation
+
+This is a Python implementation of the Model Context Protocol (MCP) command-line tool.
+

py_mcpdock_cli-1.0.13.dist-info/RECORD

@@ -0,0 +1,27 @@
+cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cli/main.py,sha256=EmP_Om2c0ltcu58B-beGaQXIwnDzme1hO-r4g-tij5w,317
+cli/mock_servers.json,sha256=12t6b6xlVL25G5nc1CIUxJRCBGWBqhqq6wpWfRYkfq4,6154
+cli/registry.py,sha256=4-kXmj_NJNgErGvnes0L9H1UcS0F36fW-odVBzQqlos,4479
+cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cli/commands/install.py,sha256=VHk3uTwqYfoW0h_NOAsyBOXMl0a-J3WXTfxm1qwuFEQ,7431
+cli/commands/run.py,sha256=PMzyyiQKINY9ED0cB0aynTtWuMHyK53bvzvT_LEM3pQ,5444
+cli/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cli/config/app_config.py,sha256=nvAkTEX9qgNx1QIK8ITskAmYtBX32Ihdb-d5bDr8MQQ,253
+cli/config/client_config.py,sha256=ct_BGjbKyBSvHKE5iTICKR7NmJDp4g44_o7h3koyngs,10473
+cli/runners/__init__.py,sha256=wyvNbZA3TnWv10gM5dmVFPs8FZgAJhi7z9i1mNSOXAQ,734
+cli/runners/command_runner.py,sha256=J5f7qaHtbOr1tOwo_QAQYMEp_nQUl6_ex6mCzpLFNpk,5119
+cli/runners/stdio_runner.py,sha256=WDUf-loiJHUt44AEvzJeAtxgqrUofGUlFwGbfYhzAPc,20493
+cli/runners/stream_http_runner.py,sha256=rf3rMcMqgqcIz0YaC91g6KnQKL1DqkvX-T3eRP_DCWQ,5998
+cli/runners/ws_runner.py,sha256=2hePDQGFThkihYZR6_Zo0B1COL1Geymevy6GrdcQRpA,1393
+cli/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cli/types/registry.py,sha256=ivIpSTAKsT_vj3eQBn7rBeQvonUQcCMEZXB0-uBZ-No,2024
+cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cli/utils/client.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cli/utils/config.py,sha256=vTWnr8brTyuuwTkGWn3C7GM32xVRagXHjn2P3LxDtK4,16317
+cli/utils/logger.py,sha256=wfP2KJIORFTj8KqTHzhqRJDmDsbjKst8t4_y9yuD4p0,2280
+cli/utils/runtime.py,sha256=m5_8leYccmSJeNb9XI__951HnrHZNo0Bn40rFAbHtgU,7829
+py_mcpdock_cli-1.0.13.dist-info/METADATA,sha256=t6GpTTsDbIjAfs5GZ5F35kFVBRhWUeyC2cOrz-oqyfM,1035
+py_mcpdock_cli-1.0.13.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
+py_mcpdock_cli-1.0.13.dist-info/entry_points.txt,sha256=LJzmhLLM_PEyc3ALv8YV_dPsdXY7hj3Pd9vJ6fHQly8,38
+py_mcpdock_cli-1.0.13.dist-info/top_level.txt,sha256=2ImG917oaVHlm0nP9oJE-Qrgs-fq_fGWgba2H1f8fpE,4
+py_mcpdock_cli-1.0.13.dist-info/RECORD,,

py_mcpdock_cli-1.0.13.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (78.1.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

py_mcpdock_cli-1.0.13.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mcpy = cli.main:cli

py_mcpdock_cli-1.0.13.dist-info/top_level.txt

@@ -0,0 +1 @@
+cli