devplacepy/devplacepy/templating.py

# retoor <retoor@molodetz.nl>

import time
from datetime import datetime, timezone
import jinja2
from fastapi.templating import Jinja2Templates
from markupsafe import Markup, escape
from devplacepy.cache import TTLCache
from devplacepy.config import STATIC_VERSION, TEMPLATES_DIR, TEMPLATE_AUTO_RELOAD
from devplacepy.constants import TOPICS, REACTION_EMOJI
from devplacepy.database import get_int_setting, get_setting, get_table
from devplacepy.avatar import avatar_url, avatar_seed
from devplacepy.utils import format_date as _format_date
from devplacepy.utils import time_ago as _time_ago
from devplacepy.utils import get_badge, is_admin, is_primary_admin, pretty_json
from devplacepy.attachments import format_file_size, file_icon_emoji
from devplacepy.content import is_owner as _owns
from devplacepy.customization import custom_css_tag, custom_js_tag, page_type_for

EM_DASH_FORMS = ("\u2014", "&mdash;", "&#8212;", "&#x2014;", "&#X2014;")


def normalize_em_dash(text: str) -> str:
    for form in EM_DASH_FORMS:
        text = text.replace(form, "-")
    return text


class _Template(jinja2.Template):
    def render(self, *args, **kwargs) -> str:
        return normalize_em_dash(super().render(*args, **kwargs))

    async def render_async(self, *args, **kwargs) -> str:
        return normalize_em_dash(await super().render_async(*args, **kwargs))


templates = Jinja2Templates(directory=str(TEMPLATES_DIR))
templates.env.template_class = _Template
templates.env.auto_reload = TEMPLATE_AUTO_RELOAD


def is_self(user, uid) -> bool:
    return bool(user) and user["uid"] == uid


def guest_disabled(user) -> Markup:
    if user:
        return Markup("")
    return Markup(' disabled aria-disabled="true"')


def static_url(path) -> str:
    if not isinstance(path, str):
        return path
    if not path.startswith("/static/") or path.startswith("/static/uploads/"):
        return path
    return f"/static/v{STATIC_VERSION}{path[len('/static'):]}"


templates.env.globals["static_url"] = static_url
templates.env.globals["static_version"] = STATIC_VERSION
templates.env.globals["is_admin"] = is_admin
templates.env.globals["is_primary_admin"] = is_primary_admin
templates.env.globals["owns"] = _owns
templates.env.globals["is_self"] = is_self
templates.env.globals["guest_disabled"] = guest_disabled

from devplacepy.docs_devrant import devrant_endpoints

templates.env.globals["devrant_endpoints"] = devrant_endpoints

_unread_cache = TTLCache(ttl=10, max_size=1000)
_messages_cache = TTLCache(ttl=10, max_size=1000)


def clear_unread_cache(user_uid: str) -> None:
    _unread_cache.pop(user_uid)


def clear_messages_cache(user_uid: str) -> None:
    _messages_cache.pop(user_uid)


def _cached_count(cache: TTLCache, table_name: str, cache_key: str, **filters) -> int:
    cached = cache.get(cache_key)
    if cached is not None:
        return cached
    table = get_table(table_name)
    count = table.count(**filters)
    cache.set(cache_key, count)
    return count


def jinja_unread_count(user_uid: str) -> int:
    return _cached_count(_unread_cache, "notifications", user_uid, user_uid=user_uid, read=False)


def jinja_unread_messages(user_uid: str) -> int:
    return _cached_count(_messages_cache, "messages", user_uid, receiver_uid=user_uid, read=False)


def jinja_user_projects(user_uid: str) -> list[dict]:
    projects = get_table("projects")
    return list(projects.find(user_uid=user_uid))


templates.env.globals["get_unread_count"] = jinja_unread_count
templates.env.globals["get_unread_messages"] = jinja_unread_messages
templates.env.globals["get_user_projects"] = jinja_user_projects
templates.env.globals["avatar_url"] = avatar_url
templates.env.globals["avatar_seed"] = avatar_seed
templates.env.globals["format_date"] = _format_date


def _normalize_iso(dt_str: str) -> str | None:
    try:
        parsed = datetime.fromisoformat(dt_str)
    except (ValueError, TypeError):
        return None
    if parsed.tzinfo is None:
        parsed = parsed.replace(tzinfo=timezone.utc)
    return parsed.isoformat()


def local_dt(dt_str: str, mode: str = "datetime") -> Markup:
    if not dt_str:
        return Markup("")
    iso = _normalize_iso(dt_str)
    if iso is None:
        return Markup(escape(dt_str))
    if mode == "ago":
        fallback = _time_ago(dt_str)
    elif mode == "date":
        fallback = _format_date(dt_str)
    else:
        fallback = _format_date(dt_str, include_time=True)
    return Markup(
        f'<time datetime="{escape(iso)}" data-dt data-dt-mode="{escape(mode)}">'
        f"{escape(fallback)}</time>"
    )


templates.env.globals["local_dt"] = local_dt
templates.env.globals["dt_ago"] = lambda dt_str: local_dt(dt_str, "ago")
templates.env.globals["badge_info"] = get_badge
templates.env.globals["TOPICS"] = TOPICS
templates.env.globals["REACTION_EMOJI"] = REACTION_EMOJI


def jinja_max_upload_size_mb() -> int:
    return get_int_setting("max_upload_size_mb", 10)


def jinja_max_attachments() -> int:
    return get_int_setting("max_attachments_per_resource", 10)


def jinja_allowed_file_types() -> str:
    return get_setting("allowed_file_types", "")


templates.env.globals["max_upload_size_mb"] = jinja_max_upload_size_mb
templates.env.globals["max_attachments_per_resource"] = jinja_max_attachments
templates.env.globals["allowed_file_types"] = jinja_allowed_file_types

_LANGUAGE_NAMES = {
    "python": "Python",
    "javascript": "JavaScript",
    "typescript": "TypeScript",
    "html": "HTML",
    "css": "CSS",
    "c": "C",
    "cpp": "C++",
    "java": "Java",
    "go": "Go",
    "rust": "Rust",
    "sql": "SQL",
    "bash": "Bash",
    "yaml": "YAML",
    "json": "JSON",
    "markdown": "Markdown",
    "swift": "Swift",
    "php": "PHP",
    "ruby": "Ruby",
    "kotlin": "Kotlin",
    "lua": "Lua",
    "perl": "Perl",
    "haskell": "Haskell",
    "elixir": "Elixir",
    "r": "R",
    "dart": "Dart",
    "scala": "Scala",
    "plaintext": "Plain Text",
}


def jinja_language_name(code: str) -> str:
    return _LANGUAGE_NAMES.get(code, code)


templates.env.globals["language_name"] = jinja_language_name
templates.env.globals["pretty_json"] = pretty_json

templates.env.globals["format_file_size"] = format_file_size
templates.env.globals["file_icon_emoji"] = file_icon_emoji

templates.env.globals["custom_css_tag"] = custom_css_tag
templates.env.globals["custom_js_tag"] = custom_js_tag
templates.env.globals["page_type_for"] = page_type_for


def extra_head_tag() -> Markup:
    try:
        code = get_setting("extra_head", "")
        if not code.strip():
            return Markup("")
        return Markup(code)
    except Exception:
        return Markup("")


templates.env.globals["extra_head_tag"] = extra_head_tag

from devplacepy.rendering import content_preview, render_content, render_title

templates.env.globals["render_content"] = render_content
templates.env.globals["render_title"] = render_title
templates.env.globals["content_preview"] = content_preview


def nav_active(request, *prefixes: str, exact: bool = False, css_class: str = "active") -> str:
    path = request.url.path
    for prefix in prefixes:
        if path == prefix:
            return css_class
        if exact:
            continue
        base = prefix.rstrip("/")
        if base and path.startswith(base + "/"):
            return css_class
    return ""


templates.env.globals["nav_active"] = nav_active


def response_time_ms(request=None) -> str:
    start = getattr(getattr(request, "state", None), "request_start", None)
    if start is None:
        return ""
    return f"{(time.perf_counter() - start) * 1000:.1f}"


templates.env.globals["response_time_ms"] = response_time_ms