48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
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.
|
|
|
|
Args:
|
|
module: The dotted path of the primary module to document.
|
|
project_name: Optional override for the project name.
|
|
out_dir: Directory where the MCP JSON resources and nav will be written.
|
|
"""
|
|
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 from a pre-built bundle.
|
|
|
|
Args:
|
|
mcp_root: Path to the directory containing index.json, nav.json, and modules/.
|
|
"""
|
|
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()
|