Update app/renderer.py

app/renderer.py

@@ -6,24 +6,43 @@ from typing import Any, Dict
 from jinja2 import Environment, FileSystemLoader, select_autoescape
 from weasyprint import HTML
 
-from .config import BASE_DIR  # si lo usas para rutas
+from .config import BASE_DIR
 
+
+# =========================
+#  RUTAS
+# =========================
 TEMPLATES_DIR = BASE_DIR / "templates"
 
+
+# =========================
+#  MAPA PRIMARIO DE TEMPLATES
+#  (coincidencias exactas)
+# =========================
 TEMPLATE_MAP: Dict[str, str] = {
     "RESTlet": "RESTlet.html.j2",
+
     "ClientScript": "ClientScript.html.j2",
+
     "Map/Reduce": "MapReduce.html.j2",
     "MapReduce": "MapReduce.html.j2",
+
     "Scheduled": "ScheduledScript.html.j2",
     "ScheduledScript": "ScheduledScript.html.j2",
+
     "Suitelet": "Suitelet.html.j2",
+
     "UserEvent": "UserEventScript.html.j2",
     "UserEventScript": "UserEventScript.html.j2",
+
     "UserModule.js": "InternalModule.html.j2",
     "InternalModule": "InternalModule.html.j2",
 }
 
+
+# =========================
+#  ENTORNO DE TEMPLATES
+# =========================
 env = Environment(
     loader=FileSystemLoader(str(TEMPLATES_DIR)),
     autoescape=select_autoescape(["html", "xml"]),
@@ -32,31 +51,85 @@ env = Environment(
 )
 
 
+# =========================
+#  RESOLVER DE TEMPLATES
+# =========================
 def _resolve_template(doc: Dict[str, Any]) -> str:
     metadata = doc.get("metadata") or {}
-    script_type = (
-        metadata.get("script_type") or metadata.get("type") or "InternalModule"
+    raw_type = metadata.get("script_type") or metadata.get("type") or "InternalModule"
+
+    # ---- Normalización del tipo ----
+    normalized = (
+        str(raw_type)
+        .strip()
+        .lower()
+        .replace(" ", "")
+        .replace("-", "")
+        .replace("_", "")
+        .replace("/", "")
     )
-    return TEMPLATE_MAP.get(script_type, "InternalModule.html.j2")
 
+    # ---- Mapa normalizado ----
+    normalized_map = {
+        "restlet": "RESTlet.html.j2",
+
+        "clientscript": "ClientScript.html.j2",
+        "clients": "ClientScript.html.j2",
+        "cs": "ClientScript.html.j2",
 
+        "mapreduce": "MapReduce.html.j2",
+        "mr": "MapReduce.html.j2",
+
+        "scheduledscript": "ScheduledScript.html.j2",
+        "scheduled": "ScheduledScript.html.j2",
+        "ss": "ScheduledScript.html.j2",
+
+        "suitelet": "Suitelet.html.j2",
+        "sl": "Suitelet.html.j2",
+
+        "userevent": "UserEventScript.html.j2",
+        "usereventscript": "UserEventScript.html.j2",
+        "ue": "UserEventScript.html.j2",
+
+        "internalmodule": "InternalModule.html.j2",
+        "module": "InternalModule.html.j2",
+        "usermodulejs": "InternalModule.html.j2",
+    }
+
+    # ---- Primera prueba: coincidencia exacta ----
+    if raw_type in TEMPLATE_MAP:
+        return TEMPLATE_MAP[raw_type]
+
+    # ---- Segunda prueba: coincidencia normalizada ----
+    if normalized in normalized_map:
+        return normalized_map[normalized]
+
+    # ---- Fallback ----
+    return "InternalModule.html.j2"
+
+
+# =========================
+#  RENDER HTML
+# =========================
 def render_html(doc: Dict[str, Any]) -> str:
     template_name = _resolve_template(doc)
     template = env.get_template(template_name)
-    html = template.render(
+
+    return template.render(
         doc=doc,
         generated_at=datetime.utcnow().strftime("%Y-%m-%d %H:%M UTC"),
     )
-    return html
 
 
+# =========================
+#  RENDER PDF
+# =========================
 def render_pdf_bytes(doc: Dict[str, Any]) -> bytes:
-    """
-    Genera un PDF vectorial a partir del HTML usando WeasyPrint.
-    El texto es seleccionable.
-    """
     html = render_html(doc)
 
-    # base_url permite que WeasyPrint resuelva rutas relativas (si tuvieras CSS/imagenes locales)
-    pdf_bytes: bytes = HTML(string=html, base_url=str(TEMPLATES_DIR)).write_pdf()
+    pdf_bytes: bytes = HTML(
+        string=html,
+        base_url=str(TEMPLATES_DIR)
+    ).write_pdf()
+
     return pdf_bytes