ObsiGate/backend/tools/documents.py

"""Document-production tools (phase 2 #92) — WRITE, confirmation required.

The assistant can generate real files inside a vault:

* ``create_xlsx`` — spreadsheet (openpyxl);
* ``create_docx`` — Word document (python-docx);
* ``create_csv``  — CSV (stdlib);
* ``create_pdf``  — PDF (reportlab, from markdown-ish content).

Every tool is ``WRITE`` (two-step confirm in the UI / propose-apply over MCP),
vault-scoped through ``requires_vault`` and saved via the shared mutation
service (path safety, read-only check, backup on overwrite).
"""

from __future__ import annotations

import csv as csv_lib
import io
import logging
import re
from typing import Any
from xml.sax import saxutils

from backend.services.errors import ServiceError
from backend.services.mutations import save_raw_file
from backend.tools.context import ToolContext, ToolError, ToolRisk
from backend.tools.registry import tool
from backend.tools.schemas import CsvInput, DocxInput, PdfInput, SpreadsheetInput

logger = logging.getLogger("obsigate.tools.documents")

MAX_PDF_CHARS = 200_000
MAX_ROWS = 5_000


def _save(vault: str, path: str, content: bytes, overwrite: bool) -> dict[str, Any]:
    """Shared save helper (maps ServiceError to ToolError)."""
    try:
        return save_raw_file(vault, path, content, overwrite=overwrite, allow_docs=True)
    except ServiceError as e:
        raise ToolError(e.message, code=e.code, details=e.details) from e


def _check_rows(rows: list[list[Any]]) -> None:
    if not rows:
        raise ToolError("Aucune ligne fournie", code="invalid_arguments")
    if len(rows) > MAX_ROWS:
        raise ToolError(
            f"Trop de lignes ({len(rows)} > {MAX_ROWS})", code="invalid_arguments"
        )


def _check_extension(path: str, expected: str) -> str:
    """Enforce the document extension; return the normalized path."""
    path = (path or "").strip()
    if not path.lower().endswith(expected):
        raise ToolError(
            f"Extension attendue : {expected}", code="invalid_arguments"
        )
    return path


@tool(
    name="create_xlsx",
    description=(
        "Create an .xlsx spreadsheet in a vault from rows of cell values "
        "(first row = header). Use for tables, budgets, checklists the user "
        "asked to turn into an Excel file."
    ),
    input_model=SpreadsheetInput,
    risk=ToolRisk.WRITE,
    requires_vault=True,
)
def create_xlsx(ctx: ToolContext, params: SpreadsheetInput) -> dict[str, Any]:
    """Build the workbook with openpyxl and save it into the vault."""
    from openpyxl import Workbook

    _check_rows(params.rows)
    path = _check_extension(params.path, ".xlsx")
    wb = Workbook()
    ws = wb.active
    ws.title = params.sheet_name[:31] or "Feuille1"
    for row in params.rows:
        ws.append(list(row))
    buffer = io.BytesIO()
    wb.save(buffer)
    return _save(params.vault, path, buffer.getvalue(), params.overwrite)


@tool(
    name="create_docx",
    description=(
        "Create a .docx Word document in a vault from an optional title and "
        "ordered paragraphs. Use for letters, reports, structured drafts."
    ),
    input_model=DocxInput,
    risk=ToolRisk.WRITE,
    requires_vault=True,
)
def create_docx(ctx: ToolContext, params: DocxInput) -> dict[str, Any]:
    """Build the document with python-docx and save it into the vault."""
    from docx import Document

    if not params.paragraphs:
        raise ToolError("Aucun paragraphe fourni", code="invalid_arguments")
    path = _check_extension(params.path, ".docx")
    doc = Document()
    if params.title.strip():
        doc.add_heading(params.title.strip(), level=1)
    for paragraph in params.paragraphs:
        doc.add_paragraph(paragraph)
    buffer = io.BytesIO()
    doc.save(buffer)
    return _save(params.vault, path, buffer.getvalue(), params.overwrite)


@tool(
    name="create_csv",
    description=(
        "Create a .csv file in a vault from rows of cell values (first row = "
        "header). Use for flat data exports, simple tables."
    ),
    input_model=CsvInput,
    risk=ToolRisk.WRITE,
    requires_vault=True,
)
def create_csv(ctx: ToolContext, params: CsvInput) -> dict[str, Any]:
    """Serialize the rows and save the CSV into the vault."""
    _check_rows(params.rows)
    path = _check_extension(params.path, ".csv")
    delimiter = params.delimiter if params.delimiter in (",", ";", "\t") else ","
    buffer = io.StringIO()
    writer = csv_lib.writer(buffer, delimiter=delimiter, lineterminator="\n")
    writer.writerows(params.rows)
    return _save(params.vault, path, buffer.getvalue().encode("utf-8"), params.overwrite)


_HEADING_RE = re.compile(r"^(#{1,6})\s+(.*)$")


def _markdown_to_flowables(content: str) -> list[tuple[str, str]]:
    """Split markdown-ish content into (style, text) blocks for reportlab."""
    blocks: list[tuple[str, str]] = []
    for raw_line in content.splitlines():
        line = raw_line.rstrip()
        if not line.strip():
            continue
        heading = _HEADING_RE.match(line)
        if heading:
            blocks.append((f"H{min(3, len(heading.group(1)))}", heading.group(2).strip()))
        else:
            blocks.append(("P", line.strip()))
    return blocks


@tool(
    name="create_pdf",
    description=(
        "Create a .pdf document in a vault from markdown-ish content "
        "(# headings + paragraphs). Use for printable deliverables."
    ),
    input_model=PdfInput,
    risk=ToolRisk.WRITE,
    requires_vault=True,
)
def create_pdf(ctx: ToolContext, params: PdfInput) -> dict[str, Any]:
    """Render the content with reportlab and save the PDF into the vault."""
    from reportlab.lib.pagesizes import A4
    from reportlab.lib.styles import getSampleStyleSheet
    from reportlab.platypus import Paragraph, SimpleDocTemplate, Spacer

    path = _check_extension(params.path, ".pdf")
    text = params.content[:MAX_PDF_CHARS]
    styles = getSampleStyleSheet()
    style_map = {
        "P": styles["BodyText"],
        "H1": styles["Heading1"],
        "H2": styles["Heading2"],
        "H3": styles["Heading3"],
    }
    buffer = io.BytesIO()
    doc = SimpleDocTemplate(buffer, pagesize=A4, title=params.title[:200])
    story: list[Any] = [Paragraph(saxutils.escape(params.title[:300]), styles["Title"])]
    for style, line in _markdown_to_flowables(text):
        story.append(Spacer(1, 4))
        story.append(Paragraph(saxutils.escape(line), style_map[style]))
    doc.build(story)
    return _save(params.vault, path, buffer.getvalue(), params.overwrite)