from typing import Any

from flask import jsonify, request
from werkzeug.wrappers import Response

import db
import document_engine
from core.flask_app import app


@app.route("/document/<slug>/elements/<int:element_id>/update", methods=["POST"])
def document_element_update(slug: str, element_id: int) -> Response | tuple[Response, int]:
    """Remplace les attributs d'un élément — un formulaire du panneau
    Propriétés (voir static/document/js/document-editor.js) envoie
    systématiquement l'état complet de ses champs."""
    if document_engine.get_document_element(slug, element_id) is None:
        return jsonify({"error": "élément introuvable"}), 404
    attributes: dict[str, Any] = request.get_json(force=True) or {}
    document_engine.update_document_element_attributes(slug, element_id, attributes)
    element = db.assert_not_none(
        document_engine.get_document_element(slug, element_id),
        "element_id verifie present juste au-dessus, aucune suppression concurrente possible entre-temps ici",
    )
    elements_by_parent: dict[int | None, list[dict[str, Any]]] = {}
    for el in document_engine.list_document_elements(slug):
        elements_by_parent.setdefault(el["parent_id"], []).append(el)
    return jsonify(
        {
            "ok": True,
            "rendered_html": document_engine.render_document_element(element, elements_by_parent),
        }
    )