import json

import db

from ..rendering.render_element_html import render_element_html


def list_elements(slug, screen_id):
    """Renvoie TOUS les éléments de l'écran, à plat (parent_id indique
    l'imbrication) — pratique pour construire l'arbre côté appelant. Seuls
    les éléments de premier niveau (parent_id NULL) reçoivent un
    "rendered_html" : celui d'un conteneur/répéteur/groupe de champs inclut
    déjà, en le générant récursivement, le HTML de tout ce qui est posé à
    l'intérieur — les enfants n'ont donc pas besoin d'être dessinés une
    deuxième fois séparément."""
    conn = db.connect(slug)
    rows = conn.execute(
        "SELECT * FROM _screen_elements WHERE screen_id = ? ORDER BY z_index, id", (screen_id,)
    ).fetchall()
    conn.close()
    result = []
    for r in rows:
        d = dict(r)
        d["style"] = json.loads(d["style"] or "{}")
        d["attributes"] = json.loads(d["attributes"] or "{}")
        result.append(d)
    children_map = {}
    for d in result:
        if d.get("parent_id"):
            children_map.setdefault(d["parent_id"], []).append(d)
    for d in result:
        if not d.get("parent_id"):
            d["rendered_html"] = render_element_html(d, slug, children_map)
    return result