doc-forge/docforge/cli/mcp/logic.py

from pathlib import Path
import click
from docforge.loaders import GriffeLoader, discover_module_paths
from docforge.renderers import MCPRenderer
from docforge.servers import MCPServer

def generate_resources(module: str, project_name: str | None, out_dir: Path) -> None:
    """
    Generate MCP-compatible documentation resources.
    """
    loader = GriffeLoader()
    discovered_paths = discover_module_paths(module)
    project = loader.load_project(discovered_paths, project_name)

    renderer = MCPRenderer()
    renderer.generate_sources(project, out_dir)

def serve(mcp_root: Path) -> None:
    """
    Serve MCP documentation.
    """
    if not mcp_root.exists():
        raise click.ClickException(f"mcp_docs directory not found: {mcp_root}")

    required = [
        mcp_root / "index.json",
        mcp_root / "nav.json",
        mcp_root / "modules",
    ]

    for path in required:
        if not path.exists():
            raise click.ClickException(f"Invalid MCP bundle, missing: {path.name}")

    server = MCPServer(
        mcp_root=mcp_root,
        name="doc-forge-mcp",
    )
    server.run()