added mcp_renderer
This commit is contained in:
@@ -16,7 +16,8 @@ groups:
|
|||||||
Renderers:
|
Renderers:
|
||||||
- docforge/renderers/index.md
|
- docforge/renderers/index.md
|
||||||
- docforge/renderers/base.md
|
- docforge/renderers/base.md
|
||||||
- docforge/renderers/mkdocs.md
|
- docforge/renderers/mkdocs_renderer.md
|
||||||
|
- docforge/renderers/mcp_renderer.md
|
||||||
CLI:
|
CLI:
|
||||||
- docforge/cli/index.md
|
- docforge/cli/index.md
|
||||||
- docforge/cli/main.md
|
- docforge/cli/main.md
|
||||||
|
|||||||
@@ -52,7 +52,7 @@ pip install doc-forge
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
from .loaders import GriffeLoader, discover_module_paths
|
from .loaders import GriffeLoader, discover_module_paths
|
||||||
from .renderers import MkDocsRenderer
|
from .renderers import MkDocsRenderer, MCPRenderer
|
||||||
from .cli import main
|
from .cli import main
|
||||||
from . import models
|
from . import models
|
||||||
|
|
||||||
@@ -60,6 +60,7 @@ __all__ = [
|
|||||||
"GriffeLoader",
|
"GriffeLoader",
|
||||||
"discover_module_paths",
|
"discover_module_paths",
|
||||||
"MkDocsRenderer",
|
"MkDocsRenderer",
|
||||||
|
"MCPRenderer",
|
||||||
"models",
|
"models",
|
||||||
"main",
|
"main",
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
from .loaders import GriffeLoader, discover_module_paths
|
from .loaders import GriffeLoader, discover_module_paths
|
||||||
from .renderers import MkDocsRenderer
|
from .renderers import MkDocsRenderer, MCPRenderer
|
||||||
from .cli import main
|
from .cli import main
|
||||||
from . import models
|
from . import models
|
||||||
|
|
||||||
@@ -7,6 +7,7 @@ __all__ = [
|
|||||||
"GriffeLoader",
|
"GriffeLoader",
|
||||||
"discover_module_paths",
|
"discover_module_paths",
|
||||||
"MkDocsRenderer",
|
"MkDocsRenderer",
|
||||||
|
"MCPRenderer",
|
||||||
"models",
|
"models",
|
||||||
"main",
|
"main",
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -18,7 +18,9 @@ To add a new renderer, implement the `DocRenderer` protocol defined in
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
from .mkdocs_renderer import MkDocsRenderer
|
from .mkdocs_renderer import MkDocsRenderer
|
||||||
|
from .mcp_renderer import MCPRenderer
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
"MkDocsRenderer",
|
"MkDocsRenderer",
|
||||||
|
"MCPRenderer",
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
from .mkdocs_renderer import MkDocsRenderer
|
from .mkdocs_renderer import MkDocsRenderer
|
||||||
|
from .mcp_renderer import MCPRenderer
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
"MkDocsRenderer",
|
"MkDocsRenderer",
|
||||||
|
"MCPRenderer",
|
||||||
]
|
]
|
||||||
|
|||||||
78
docforge/renderers/mcp_renderer.py
Normal file
78
docforge/renderers/mcp_renderer.py
Normal file
@@ -0,0 +1,78 @@
|
|||||||
|
from pathlib import Path
|
||||||
|
from typing import Iterable
|
||||||
|
|
||||||
|
from docforge.models import Project, Module, DocObject
|
||||||
|
|
||||||
|
|
||||||
|
class MCPRenderer:
|
||||||
|
"""
|
||||||
|
Renderer that emits documentation as MCP resources.
|
||||||
|
"""
|
||||||
|
|
||||||
|
name = "mcp"
|
||||||
|
|
||||||
|
def generate_sources(self, project: Project, out_dir: Path) -> None:
|
||||||
|
"""
|
||||||
|
Generate MCP-compatible resources for the project.
|
||||||
|
|
||||||
|
Each module is rendered as a standalone MCP document.
|
||||||
|
"""
|
||||||
|
out_dir.mkdir(parents=True, exist_ok=True)
|
||||||
|
|
||||||
|
for module in project.get_all_modules():
|
||||||
|
self._write_module(module, out_dir)
|
||||||
|
|
||||||
|
def _write_module(self, module: Module, out_dir: Path) -> None:
|
||||||
|
"""
|
||||||
|
Render a module and all contained objects.
|
||||||
|
"""
|
||||||
|
resource_path = self._module_resource_path(module)
|
||||||
|
content = self._render_module(module)
|
||||||
|
|
||||||
|
file_path = out_dir / resource_path
|
||||||
|
file_path.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
file_path.write_text(content, encoding="utf-8")
|
||||||
|
|
||||||
|
def _render_module(self, module: Module) -> str:
|
||||||
|
"""
|
||||||
|
Render a module into MCP-friendly Markdown.
|
||||||
|
"""
|
||||||
|
lines: list[str] = []
|
||||||
|
|
||||||
|
lines.append(f"# Module `{module.path}`\n")
|
||||||
|
|
||||||
|
if module.docstring:
|
||||||
|
lines.append(module.docstring.strip() + "\n")
|
||||||
|
|
||||||
|
for obj in module.get_all_objects():
|
||||||
|
lines.extend(self._render_object(obj, level=2))
|
||||||
|
|
||||||
|
return "\n".join(lines).strip() + "\n"
|
||||||
|
|
||||||
|
def _render_object(self, obj: DocObject, level: int) -> Iterable[str]:
|
||||||
|
"""
|
||||||
|
Recursively render DocObjects.
|
||||||
|
"""
|
||||||
|
prefix = "#" * level
|
||||||
|
lines: list[str] = []
|
||||||
|
|
||||||
|
lines.append(f"{prefix} {obj.kind} `{obj.name}`")
|
||||||
|
|
||||||
|
if obj.signature:
|
||||||
|
lines.append(f"```python\n{obj.signature}\n```")
|
||||||
|
|
||||||
|
if obj.docstring:
|
||||||
|
lines.append(obj.docstring.strip())
|
||||||
|
|
||||||
|
for member in obj.get_all_members():
|
||||||
|
lines.extend(self._render_object(member, level + 1))
|
||||||
|
|
||||||
|
return lines
|
||||||
|
|
||||||
|
def _module_resource_path(self, module: Module) -> Path:
|
||||||
|
"""
|
||||||
|
Convert a module path into an MCP resource path.
|
||||||
|
Example:
|
||||||
|
docforge.models.module -> docforge/models/module.md
|
||||||
|
"""
|
||||||
|
return Path(module.path.replace(".", "/") + ".md")
|
||||||
31
docforge/renderers/mcp_renderer.pyi
Normal file
31
docforge/renderers/mcp_renderer.pyi
Normal file
@@ -0,0 +1,31 @@
|
|||||||
|
from pathlib import Path
|
||||||
|
from typing import Iterable
|
||||||
|
|
||||||
|
from docforge.models import Project, Module, DocObject
|
||||||
|
|
||||||
|
|
||||||
|
class MCPRenderer:
|
||||||
|
"""
|
||||||
|
Renderer that emits MCP-compatible documentation resources.
|
||||||
|
"""
|
||||||
|
|
||||||
|
name: str
|
||||||
|
|
||||||
|
def generate_sources(self, project: Project, out_dir: Path) -> None:
|
||||||
|
"""Generate MCP resources for the given project."""
|
||||||
|
|
||||||
|
def _write_module(self, module: Module, out_dir: Path) -> None:
|
||||||
|
"""Write a single module as an MCP resource."""
|
||||||
|
|
||||||
|
def _render_module(self, module: Module) -> str:
|
||||||
|
"""Render a module and its contents into a text document."""
|
||||||
|
|
||||||
|
def _render_object(
|
||||||
|
self,
|
||||||
|
obj: DocObject,
|
||||||
|
level: int,
|
||||||
|
) -> Iterable[str]:
|
||||||
|
"""Recursively render a documented object and its members."""
|
||||||
|
|
||||||
|
def _module_resource_path(self, module: Module) -> Path:
|
||||||
|
"""Compute the MCP resource path for a module."""
|
||||||
3
docs/docforge/renderers/mcp_renderer.md
Normal file
3
docs/docforge/renderers/mcp_renderer.md
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
# Mcp Renderer
|
||||||
|
|
||||||
|
::: docforge.renderers.mcp_renderer
|
||||||
@@ -1,3 +0,0 @@
|
|||||||
# Mkdocs
|
|
||||||
|
|
||||||
::: docforge.renderers.mkdocs
|
|
||||||
3
docs/docforge/renderers/mkdocs_renderer.md
Normal file
3
docs/docforge/renderers/mkdocs_renderer.md
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
# Mkdocs Renderer
|
||||||
|
|
||||||
|
::: docforge.renderers.mkdocs_renderer
|
||||||
@@ -52,7 +52,8 @@ nav:
|
|||||||
- Renderers:
|
- Renderers:
|
||||||
- docforge/renderers/index.md
|
- docforge/renderers/index.md
|
||||||
- docforge/renderers/base.md
|
- docforge/renderers/base.md
|
||||||
- docforge/renderers/mkdocs.md
|
- docforge/renderers/mkdocs_renderer.md
|
||||||
|
- docforge/renderers/mcp_renderer.md
|
||||||
- CLI:
|
- CLI:
|
||||||
- docforge/cli/index.md
|
- docforge/cli/index.md
|
||||||
- docforge/cli/main.md
|
- docforge/cli/main.md
|
||||||
|
|||||||
0
tests/renderers/mcp/__init__.py
Normal file
0
tests/renderers/mcp/__init__.py
Normal file
18
tests/renderers/mcp/test_mcp_content.py
Normal file
18
tests/renderers/mcp/test_mcp_content.py
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from docforge import MCPRenderer
|
||||||
|
from docforge.models import Project, Module
|
||||||
|
|
||||||
|
|
||||||
|
def test_mcp_file_content(tmp_path: Path):
|
||||||
|
project = Project("testpkg")
|
||||||
|
project.add_module(Module("testpkg.mod"))
|
||||||
|
|
||||||
|
out_dir = tmp_path / "mcp"
|
||||||
|
renderer = MCPRenderer()
|
||||||
|
|
||||||
|
renderer.generate_sources(project, out_dir)
|
||||||
|
|
||||||
|
content = (out_dir / "testpkg" / "mod.md").read_text()
|
||||||
|
|
||||||
|
assert "# Module `testpkg.mod`" in content
|
||||||
20
tests/renderers/mcp/test_mcp_idempotency.py
Normal file
20
tests/renderers/mcp/test_mcp_idempotency.py
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from docforge import MCPRenderer
|
||||||
|
from docforge.models import Project, Module
|
||||||
|
|
||||||
|
|
||||||
|
def test_mcp_idempotent(tmp_path: Path):
|
||||||
|
project = Project("testpkg")
|
||||||
|
project.add_module(Module("testpkg.mod"))
|
||||||
|
|
||||||
|
out_dir = tmp_path / "mcp"
|
||||||
|
renderer = MCPRenderer()
|
||||||
|
|
||||||
|
renderer.generate_sources(project, out_dir)
|
||||||
|
first = (out_dir / "testpkg" / "mod.md").read_text()
|
||||||
|
|
||||||
|
renderer.generate_sources(project, out_dir)
|
||||||
|
second = (out_dir / "testpkg" / "mod.md").read_text()
|
||||||
|
|
||||||
|
assert first == second
|
||||||
29
tests/renderers/mcp/test_mcp_module_coverage.py
Normal file
29
tests/renderers/mcp/test_mcp_module_coverage.py
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from docforge.loaders import GriffeLoader, discover_module_paths
|
||||||
|
from docforge import MCPRenderer
|
||||||
|
|
||||||
|
|
||||||
|
def test_mcp_emits_all_modules(tmp_path: Path) -> None:
|
||||||
|
loader = GriffeLoader()
|
||||||
|
discovered_paths = discover_module_paths(
|
||||||
|
"docforge",
|
||||||
|
Path(r"C:\Users\vishe\WorkSpace\code\aetos\doc-forge"),
|
||||||
|
)
|
||||||
|
project = loader.load_project(discovered_paths)
|
||||||
|
|
||||||
|
renderer = MCPRenderer()
|
||||||
|
renderer.generate_sources(project, tmp_path)
|
||||||
|
|
||||||
|
emitted = {
|
||||||
|
p.relative_to(tmp_path).as_posix()
|
||||||
|
for p in tmp_path.rglob("*.md")
|
||||||
|
}
|
||||||
|
|
||||||
|
expected = {
|
||||||
|
m.path.replace(".", "/") + ".md"
|
||||||
|
for m in project.get_all_modules()
|
||||||
|
}
|
||||||
|
|
||||||
|
missing = expected - emitted
|
||||||
|
assert not missing, f"Missing MCP resources for modules: {missing}"
|
||||||
18
tests/renderers/mcp/test_mcp_structure.py
Normal file
18
tests/renderers/mcp/test_mcp_structure.py
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from docforge import MCPRenderer
|
||||||
|
from docforge.models import Project, Module
|
||||||
|
|
||||||
|
|
||||||
|
def test_mcp_directory_structure(tmp_path: Path):
|
||||||
|
project = Project("testpkg")
|
||||||
|
project.add_module(Module("testpkg"))
|
||||||
|
project.add_module(Module("testpkg.sub"))
|
||||||
|
|
||||||
|
out_dir = tmp_path / "mcp"
|
||||||
|
renderer = MCPRenderer()
|
||||||
|
|
||||||
|
renderer.generate_sources(project, out_dir)
|
||||||
|
|
||||||
|
assert (out_dir / "testpkg.md").exists()
|
||||||
|
assert (out_dir / "testpkg" / "sub.md").exists()
|
||||||
0
tests/renderers/mkdocs/__init__.py
Normal file
0
tests/renderers/mkdocs/__init__.py
Normal file
Reference in New Issue
Block a user