"""Standalone single-file HTML export (single or multi-session with sidebar); no remote deps.""" import datetime import secrets from typing import Any, Dict, List from urllib.parse import quote from hermes_cli.timefmt import coerce_epoch # --- Icons (Lucide-style SVGs) --- ICON_USER = '' ICON_BOT = '' ICON_TERMINAL = '' ICON_WRENCH = '' ICON_SPARKLES = '' ICON_CHEVRON_RIGHT = '' ICON_SEARCH = '' ICON_SHIELD = '' ICON_HERMES = '' HTML_TEMPLATE = """ {page_title}

{sidebar_html}

{sessions_html}

""" def _escape_html(text: Any) -> str: return ( str(text).replace("&", "&").replace("<", "<").replace(">", ">") .replace('"', """).replace("'", "'") ) def _format_timestamp(ts: Any) -> str: # A corrupt cell renders as N/A; never raw text (a TEXT timestamp would otherwise reach an HTML sink). return datetime.datetime.fromtimestamp(ts).strftime("%Y-%m-%d %H:%M:%S") if ts and (ts := coerce_epoch(ts)) else "N/A" _ROLE_ICONS = {"user": ICON_USER, "assistant": ICON_BOT, "system": ICON_SHIELD} _CHEVRON_HTML = ICON_CHEVRON_RIGHT.replace('class="', 'class="chevron ') def _content_text(content: Any) -> str: """Flatten multimodal/complex content parts into display text.""" if not isinstance(content, list): return content parts = [] for part in content: if not isinstance(part, dict): parts.append(str(part)) elif part.get("type") == "text": parts.append(part.get("text", "")) elif part.get("type") == "image_url": parts.append("[Image Attachment]") return "\n".join(parts) def _collapsible(kind: str, icon: str, title: str, inner: str, indent: str, outer_class: str = "") -> str: """A chevron-headed collapsible block (tool-call / reasoning / system-prompt).""" return f""" {indent}

{indent} {indent}

{indent} {inner} {indent}

{indent}

{indent}""" def _generate_messages_html(messages: List[Dict[str, Any]]) -> str: html_list = [] for i, msg in enumerate(messages): role = msg.get("role", "unknown") if role == "session_meta": # internal metadata, never rendered continue content = _content_text(msg.get("content") or "") # The role feeds two sinks and is externally influenced for tool/MCP messages: HTML-escape # the display text; reduce the class token to alnum/-/_ so a crafted role can neither break # out of the attribute nor split into extra classes (real roles keep matching `.message-`). role_class = "".join(c if c.isalnum() or c in "-_" else "-" for c in str(role).lower()) delay_style = f' style="animation-delay: {min(i * 0.05, 1.0)}s"' if i < 10 else "" # stagger first items html = ( f'

' f'

{_CHEVRON_HTML} {_ROLE_ICONS.get(role, ICON_TERMINAL)} {_escape_html(role)}

' f'

{_format_timestamp(msg.get("timestamp", 0))}

' '

' ) for tc in msg.get("tool_calls") or (): fn = tc.get("function", {}) html += _collapsible( "tool-call", ICON_WRENCH, f"Tool Call: {_escape_html(fn.get('name', 'unknown'))}", f"

{_escape_html(fn.get('arguments', '{}'))}

", " " * 16, ) if content: escaped = _escape_html(content) html += f'

{f"

{escaped}

" if role == "tool" else escaped}

' if reasoning := msg.get("reasoning") or msg.get("reasoning_content"): html += _collapsible( "reasoning", ICON_SPARKLES, "Reasoning", f'

{_escape_html(reasoning)}

', " " * 12, ) html_list.append(html + "

") return "\n".join(html_list) def _sidebar_item_html(s: Dict[str, Any]) -> str: sid = str(s.get("id", "N/A")) title = s.get("title") or s.get("preview") or "Untitled Session" title = title[:47] + "..." if len(title) > 50 else title return f"""

{_escape_html(title)}

{_escape_html(sid[:8])} {_format_timestamp(s.get("started_at", 0)).split(" ")[0]}

""" def _sidebar_html(sessions: List[Dict[str, Any]]) -> str: return f""" """ def _session_view_html(s: Dict[str, Any], is_multi: bool) -> str: escaped_sid = _escape_html(str(s.get("id", "N/A"))) system_html = _collapsible( "system-prompt", ICON_SHIELD, "System Prompt (Persona)", f'

{_escape_html(s["system_prompt"])}

', " " * 12, outer_class="system-prompt-section active", ) if s.get("system_prompt") else "" return f"""

{_escape_html(s.get("title") or "Hermes Session")}

ID: {escaped_sid}

Model: {_escape_html(s.get("model") or "Unknown")}

Started: {_format_timestamp(s.get("started_at", 0))}

{system_html}

{_generate_messages_html(s.get("messages", []))}

""" def generate_multi_session_html_export(sessions: List[Dict[str, Any]]) -> str: if not sessions: return "

No sessions to export.

" is_multi = len(sessions) > 1 return HTML_TEMPLATE.format( page_title="Hermes Session Export" if is_multi else _escape_html(sessions[0].get("title") or "Hermes Session"), sidebar_html=_sidebar_html(sessions) if is_multi else "", sessions_html="\n".join(_session_view_html(s, is_multi) for s in sessions), main_margin="var(--sidebar-width)" if is_multi else "0", layout_class="layout-multi" if is_multi else "layout-single", generated_at=datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), script_nonce=secrets.token_urlsafe(16), ) def generate_html_export(session_data: Dict[str, Any]) -> str: """Single-session export (legacy entry point).""" return generate_multi_session_html_export([session_data])