mcp_devtools_cli.py

import uvicorn
import os
import argparse
import tomllib # Added tomllib import

def get_project_version() -> str:
    """
    Reads the project version from pyproject.toml.
    Assumes pyproject.toml is in the same directory as the script.
    """
    script_dir = os.path.dirname(__file__)
    # Construct path to pyproject.toml, assuming it's in the same directory
    pyproject_path = os.path.join(script_dir, 'pyproject.toml')

    if not os.path.exists(pyproject_path):
        return "unknown"

    try:
        with open(pyproject_path, "rb") as f:
            pyproject_data = tomllib.load(f)
        version = pyproject_data.get("project", {}).get("version")
        if version:
            return str(version)
    except Exception:
        # Catch any errors during file reading or TOML parsing
        pass
    return "unknown"

def main():
    """
    CLI entrypoint for the MCP DevTools server.
    Conditionally enables reload based on an environment variable.
    """
    parser = argparse.ArgumentParser(description="Run the MCP DevTools Server.")
    parser.add_argument('-p', '--port', type=int, help='Port to run the server on.')
    parser.add_argument('-v', '--version', action='version', 
                       version=f'mcp-devtools v{get_project_version()}')
    args = parser.parse_args()

    # Use port from args, then .env, then default
    # The server scripts create a .env file.
    from dotenv import load_dotenv
    load_dotenv()
    port = args.port or int(os.getenv("MCP_PORT", 1337))
    host = os.getenv("MCP_HOST", "127.0.0.1")

    # Check for the reload environment variable.
    # This will be 'false' by default when running via 'uvx'.
    reload_enabled = os.getenv('MCP_DEVTOOLS_RELOAD', 'false').lower() in ('true', '1', 't')

    # Get and print the server version
    server_version = get_project_version()
    print(f"MCP DevTools Server v{server_version}")
    print(f"DevTools MCP server listening at http://{host}:{port}/sse")
    if reload_enabled:
        print("Auto-reloading is enabled.")
    else:
        print("Auto-reloading is disabled.")

    uvicorn.run(
        "server:app",
        host=host,
        port=port,
        reload=reload_enabled, # Set reload conditionally
        log_level="info"
    )

if __name__ == "__main__":
    main()