html_generator.py

# -*- coding: utf-8 -*-
"""
html_generator.py — Assemble la page web finale d'un export à partir des templates
statiques (templates/index.html, templates/style.css, templates/app.js) et des
réglages choisis dans le dialogue (titre, thème, fond de plan, logo, outils...).
"""

import base64
import os
import shutil

from .leaflet_generator import build_extra_libs, render_app_js
from .web_i18n import get_strings

TEMPLATES_DIR = os.path.join(os.path.dirname(__file__), "templates")

FOND_URLS = {
    "OpenStreetMap": "https://{s}.tile.openstreetmap.org/{z}/{x}/{y}.png",
    "Google Satellite": "https://mt1.google.com/vt/lyrs=s&x={x}&y={y}&z={z}",
    "Google Hybrid": "https://mt1.google.com/vt/lyrs=y&x={x}&y={y}&z={z}",
}

THEMES = {
    "Clair": {
        "bg_body": "#f4f5f7",
        "bg_sidebar": "rgba(255,255,255,0.97)",
        "bg_panel": "rgba(0,0,0,0.03)",
        "bg_select": "#ffffff",
        "text_main": "#1a1a2e",
        "text_muted": "#666666",
        "accent": "#1a8a7f",
        "border": "rgba(0,0,0,0.1)",
        "popup_bg": "#ffffff",
        "popup_text": "#1a1a2e",
    },
    "Sombre": {
        "bg_body": "#f4f5f7",
        "bg_sidebar": "rgba(255,255,255,0.04)",
        "bg_panel": "rgba(255,255,255,0.08)",
        "bg_select": "#252538",
        "text_main": "#ffffff",
        "text_muted": "#bbbbbb",
        "accent": "#4ecdc4",
        "border": "rgba(255,255,255,0.1)",
        "popup_bg": "#1a1a2e",
        "popup_text": "#ffffff",
    },
    "Professionnel": {
        "bg_body": "#f4f5f7",
        "bg_sidebar": "rgba(255,255,255,0.98)",
        "bg_panel": "rgba(15,40,75,0.06)",
        "bg_select": "#ffffff",
        "text_main": "#0f284b",
        "text_muted": "#5a6b80",
        "accent": "#0f6e9e",
        "border": "rgba(15,40,75,0.15)",
        "popup_bg": "#ffffff",
        "popup_text": "#0f284b",
    },
    "Coloré": {
        "bg_body": "#221530",
        "bg_sidebar": "rgba(255,255,255,0.04)",
        "bg_panel": "rgba(255,255,255,0.09)",
        "bg_select": "#3d2455",
        "text_main": "#ffffff",
        "text_muted": "#d8c4ee",
        "accent": "#ff6b9d",
        "border": "rgba(255,255,255,0.15)",
        "popup_bg": "#3d2455",
        "popup_text": "#ffffff",
    },
}


def _opt(dialog, nom, defaut=False):
    w = getattr(dialog, nom, None)
    return w.isChecked() if w else defaut


def _lire_options(dialog):
    return {
        "geoloc": _opt(dialog, "chkGeoloc"),
        "mesure": _opt(dialog, "chkMesure"),
        "pleinecran": _opt(dialog, "chkPleinEcran"),
        "echelle": _opt(dialog, "chkScale", True),
        "recherche": _opt(dialog, "chkRecherche"),
        "minimap": _opt(dialog, "chkMiniMap"),
        "mousepos": _opt(dialog, "chkMousePosition"),
        "imprimer": _opt(dialog, "chkImprimer"),
        "attribution": _opt(dialog, "chkAttribution", True),
    }


def _lire_logo_base64(dialog):
    w_logo = getattr(dialog, "chkAfficherLogo", None)
    w_path = getattr(dialog, "lblLogoPath", None)

    if not w_logo or not w_logo.isChecked() or not w_path:
        return None

    logo_path = w_path.toolTip()
    if (
        not logo_path
        or logo_path in ("Aucun logo", "")
        or not os.path.exists(logo_path)
    ):
        return None

    try:
        with open(logo_path, "rb") as f:
            ext = os.path.splitext(logo_path)[1][1:].lower()
            if ext in ("jpg", "jpeg"):
                ext = "jpeg"
            elif ext == "svg":
                ext = "svg+xml"
            encoded = base64.b64encode(f.read()).decode("utf-8")
            return f"data:image/{ext};base64,{encoded}"
    except Exception:
        return None


def generer_export(dialog, export_data, output_dir, locale=None):
    """Génère index.html, style.css et app.js dans output_dir à partir des templates
    statiques et des réglages du dialogue. `export_data` est le dict couche -> métadonnées
    produit par l'export (fichier geojson, styles de légende, champs popup, étiquettes...).
    """

    titre = (
        dialog.txtTitreCarte.text()
        if hasattr(dialog, "txtTitreCarte")
        else "Ma Carte Web"
    )
    couleur_entete = (
        dialog.txtCouleurEntete.text() if hasattr(dialog, "txtCouleurEntete") else None
    ) or "#1a1a2e"

    s = get_strings(locale)

    logo_base64 = _lire_logo_base64(dialog)
    html_logo = (
        f'<div class="logo-container"><img src="{logo_base64}" alt="Logo"/></div>'
        if logo_base64
        else ""
    )

    fond_plan = (
        dialog.comboFondPlan.currentText() if hasattr(dialog, "comboFondPlan") else ""
    )
    # Nettoyage pour correspondance exacte (supprime les émojis s'ils existent dans la combo)
    fond_cle = next((k for k in FOND_URLS if k in fond_plan), "OpenStreetMap")
    url_fond = FOND_URLS.get(fond_cle, FOND_URLS["OpenStreetMap"])

    theme_nom = "Sombre"
    if hasattr(dialog, "comboTheme"):
        txt = dialog.comboTheme.currentText()
        if txt in THEMES:
            theme_nom = txt

    th = THEMES.get(theme_nom, THEMES["Sombre"])

    options = _lire_options(dialog)
    extra_css, extra_js = build_extra_libs(options)

    mousepos_html = (
        '<div id="mouse-pos" style="position:absolute;bottom:28px;left:5px;'
        "background:rgba(0,0,0,0.65);color:#fff;padding:3px 8px;border-radius:4px;"
        'font-size:11px;z-index:999;pointer-events:none;"></div>'
        if options.get("mousepos")
        else ""
    )

    # ── index.html ──────────────────────────────────────────────
    with open(os.path.join(TEMPLATES_DIR, "index.html"), "r", encoding="utf-8") as f:
        index_html = f.read()

    remplacements = {
        "__TITRE__": titre,
        "__EXTRA_CSS__": extra_css,
        "__EXTRA_JS__": extra_js,
        "__HTML_LOGO__": html_logo,
        "__MOUSEPOS_HTML__": mousepos_html,
        "__HEADER_COLOR__": couleur_entete,
        "__BG_BODY__": th["bg_body"],
        "__BG_SIDEBAR__": th["bg_sidebar"],
        "__BG_PANEL__": th["bg_panel"],
        "__BG_SELECT__": th["bg_select"],
        "__TEXT_MAIN__": th["text_main"],
        "__TEXT_MUTED__": th["text_muted"],
        "__ACCENT__": th["accent"],
        "__BORDER__": th["border"],
        "__POPUP_BG__": th["popup_bg"],
        "__POPUP_TEXT__": th["popup_text"],
        # ── Langue de l'interface de la carte exportée ──
        "__LANG_ATTR__": s["html_lang"],
        "__I18N_TOGGLE_SIDEBAR__": s["toggle_sidebar_title"],
        "__I18N_BASE_CONFIGURED__": s["base_configured"],
        "__I18N_LEGEND_TITLE__": s["legend_title"],
        "__I18N_EXPAND_ALL__": s["expand_all"],
        "__I18N_COLLAPSE_ALL__": s["collapse_all"],
        "__I18N_FILTER_TITLE__": s["filter_title"],
        "__I18N_FILTER_LAYER_LABEL__": s["filter_layer_label"],
        "__I18N_FILTER_CHOOSE_LAYER__": s["filter_choose_layer"],
        "__I18N_FILTER_FIELD_LABEL__": s["filter_field_label"],
        "__I18N_FILTER_CHOOSE_FIELD__": s["filter_choose_field"],
        "__I18N_FILTER_OPERATOR_LABEL__": s["filter_operator_label"],
        "__I18N_FILTER_VALUE_LABEL__": s["filter_value_label"],
        "__I18N_FILTER_VALUE_PLACEHOLDER__": s["filter_value_placeholder"],
        "__I18N_OP_EQ__": s["op_eq"],
        "__I18N_OP_NEQ__": s["op_neq"],
        "__I18N_OP_CONTAINS__": s["op_contains"],
        "__I18N_OP_STARTS__": s["op_starts"],
        "__I18N_OP_GT__": s["op_gt"],
        "__I18N_OP_LT__": s["op_lt"],
        "__I18N_OP_GTE__": s["op_gte"],
        "__I18N_OP_LTE__": s["op_lte"],
        "__I18N_APPLY__": s["apply"],
        "__I18N_RESET__": s["reset"],
    }
    for marqueur, valeur in remplacements.items():
        index_html = index_html.replace(marqueur, valeur)

    with open(os.path.join(output_dir, "index.html"), "w", encoding="utf-8") as f:
        f.write(index_html)

    # ── style.css ──────────────────────────────────────────────
    shutil.copyfile(
        os.path.join(TEMPLATES_DIR, "style.css"),
        os.path.join(output_dir, "style.css"),
    )

    # ── app.js ─────────────────────────────────────────────────
    render_app_js(
        export_data,
        url_fond,
        options,
        os.path.join(output_dir, "app.js"),
        locale=locale,
    )