feat(mcp-docugen): nuovo tool MCP document_to_pdf via Playwright/Chromium
Aggiunge la possibilità di convertire un documento Markdown in PDF
direttamente lato server, senza richiedere al chiamante di avere
md-to-pdf, pandoc o altri tool sull'host. Il PDF è restituito come
stringa base64 nella risposta JSON-RPC, pronto a essere salvato,
allegato o spedito al cliente.
- pdf_renderer.py: nuovo modulo che parsea il frontmatter YAML del
Markdown (incluso il blocco pdf_options stile Puppeteer/md-to-pdf),
rende il body in HTML via markdown-it-py (supporta tabelle e
HTML inline) e produce il PDF tramite Chromium headless gestito
da Playwright. Le pdf_options camelCase (printBackground,
displayHeaderFooter, headerTemplate, ...) vengono tradotte negli
argomenti snake_case di page.pdf().
- mcp_tools.py: nuovo tool `document_to_pdf(markdown)` che ritorna
`{pdf_b64, size_bytes}`; `document_generate` esteso con il
parametro `output_format ∈ {md, pdf, both}` per emettere il PDF
contestualmente alla generazione del Markdown.
- pyproject.toml + uv.lock: aggiunte le dipendenze playwright>=1.48
e markdown-it-py[plugins]>=3.0.
- mcp-docugen.Dockerfile: nuova fase di runtime che installa le
librerie native richieste da Chromium (libnss3, libgbm1, ecc.) e
scarica il binario Chromium di Playwright in /opt/ms-playwright.
- 7 nuovi test unit (83 totali) che coprono lo split del frontmatter,
il rendering Markdown→HTML con tabelle, la traduzione delle
pdf_options camelCase→snake_case e l'errore su YAML invalido. Il
test E2E che richiede Chromium è marcato skip in unit; lo smoke
via MCP conferma generazione PDF da 134 KB / 4 pagine.
README aggiornato con le tre strade di conversione (server-side,
client-side, bundling) e la stima del nuovo costo immagine.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,86 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import pytest
|
||||
|
||||
from mcp_docugen.pdf_renderer import (
|
||||
PdfRenderError,
|
||||
_md_to_html,
|
||||
_normalize_pdf_options,
|
||||
_split_frontmatter,
|
||||
)
|
||||
|
||||
|
||||
def test_split_frontmatter_extracts_yaml_dict():
|
||||
md = "---\ntitle: Foo\npdf_options:\n format: A4\n---\n\n# Body\n"
|
||||
fm, body = _split_frontmatter(md)
|
||||
assert fm == {"title": "Foo", "pdf_options": {"format": "A4"}}
|
||||
assert body.lstrip().startswith("# Body")
|
||||
|
||||
|
||||
def test_split_frontmatter_no_frontmatter_returns_empty_dict():
|
||||
md = "# Just body\nNo YAML here.\n"
|
||||
fm, body = _split_frontmatter(md)
|
||||
assert fm == {}
|
||||
assert body == md
|
||||
|
||||
|
||||
def test_split_frontmatter_invalid_yaml_raises():
|
||||
md = "---\ntitle: : : broken\n---\nbody"
|
||||
with pytest.raises(PdfRenderError):
|
||||
_split_frontmatter(md)
|
||||
|
||||
|
||||
def test_md_to_html_renders_table_and_html_passthrough():
|
||||
body = "| A | B |\n|---|---|\n| 1 | 2 |\n\n<div class=\"x\">raw</div>\n"
|
||||
html = _md_to_html(body)
|
||||
assert "<table>" in html
|
||||
assert "<th>A</th>" in html
|
||||
assert '<div class="x">raw</div>' in html
|
||||
|
||||
|
||||
def test_normalize_pdf_options_translates_camelcase_to_snake():
|
||||
src = {
|
||||
"format": "A4",
|
||||
"printBackground": True,
|
||||
"displayHeaderFooter": True,
|
||||
"headerTemplate": "<div>H</div>",
|
||||
"footerTemplate": "<div>F</div>",
|
||||
"margin": {"top": "18mm", "bottom": "18mm"},
|
||||
"landscape": False,
|
||||
"scale": 0.95,
|
||||
}
|
||||
out = _normalize_pdf_options(src)
|
||||
assert out["format"] == "A4"
|
||||
assert out["print_background"] is True
|
||||
assert out["display_header_footer"] is True
|
||||
assert out["header_template"] == "<div>H</div>"
|
||||
assert out["footer_template"] == "<div>F</div>"
|
||||
assert out["margin"]["top"] == "18mm"
|
||||
assert out["landscape"] is False
|
||||
assert out["scale"] == pytest.approx(0.95)
|
||||
|
||||
|
||||
def test_normalize_pdf_options_drops_unknown_keys():
|
||||
src = {"format": "A4", "weird": 123, "viewportWidth": 1200}
|
||||
out = _normalize_pdf_options(src)
|
||||
assert out == {"format": "A4"}
|
||||
|
||||
|
||||
def test_normalize_pdf_options_handles_non_dict():
|
||||
assert _normalize_pdf_options(None) == {}
|
||||
assert _normalize_pdf_options("not a dict") == {}
|
||||
|
||||
|
||||
@pytest.mark.skipif(
|
||||
True, reason="requires Chromium binary; runs only in container/CI with Playwright installed"
|
||||
)
|
||||
async def test_render_markdown_to_pdf_smoke():
|
||||
from mcp_docugen.pdf_renderer import render_markdown_to_pdf
|
||||
|
||||
md = (
|
||||
"---\npdf_options:\n format: A4\n---\n\n"
|
||||
"<style>h1 { color: red; }</style>\n\n# Hello\n\nBody text.\n"
|
||||
)
|
||||
result = await render_markdown_to_pdf(md)
|
||||
assert result.pdf_bytes.startswith(b"%PDF-")
|
||||
assert result.size_bytes > 1000
|
||||
Reference in New Issue
Block a user