wehub-resource-sync
142 lines
4.2 KiB
Python
142 lines
4.2 KiB
Python
"""Lightweight syntax highlighting for Kami HTML templates.
|
|
|
|
Scans HTML for <pre><code class="language-*"> blocks and applies
|
|
Pygments-based inline-style highlighting using Kami design tokens.
|
|
Blocks without a language- class pass through unchanged.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import html as html_mod
|
|
import re
|
|
import sys
|
|
|
|
from shared import token_value
|
|
|
|
CODE_BLOCK_RE = re.compile(
|
|
r'(<pre[^>]*>\s*<code\s+class="language-([\w+-]+)"[^>]*>)'
|
|
r'(.*?)'
|
|
r'(</code>\s*</pre>)',
|
|
re.DOTALL,
|
|
)
|
|
|
|
_KAMI_PALETTE: dict[str, str] | None = None
|
|
|
|
|
|
def _kami_palette() -> dict[str, str]:
|
|
"""Resolve design-token colors lazily.
|
|
|
|
Kept out of module scope so importing this module (e.g. by build.py for a
|
|
command that never highlights code) does not read tokens.json. That keeps
|
|
the import resilient on a half-installed checkout, matching shared.py's
|
|
baked-in fallbacks.
|
|
"""
|
|
global _KAMI_PALETTE
|
|
if _KAMI_PALETTE is None:
|
|
_KAMI_PALETTE = {
|
|
"brand": token_value("brand"),
|
|
"stone": token_value("stone"),
|
|
"olive": token_value("olive"),
|
|
"dark_warm": token_value("dark-warm"),
|
|
"near_black": token_value("near-black"),
|
|
}
|
|
return _KAMI_PALETTE
|
|
|
|
|
|
_WARNED_MISSING_PYGMENTS = False
|
|
|
|
|
|
def _warn_missing_pygments() -> None:
|
|
global _WARNED_MISSING_PYGMENTS
|
|
if _WARNED_MISSING_PYGMENTS:
|
|
return
|
|
print(
|
|
"WARN: Pygments is not installed; language-tagged code blocks will render monochrome. "
|
|
"Install with `python3 -m pip install Pygments` to enable syntax highlighting.",
|
|
file=sys.stderr,
|
|
)
|
|
_WARNED_MISSING_PYGMENTS = True
|
|
|
|
|
|
def _build_kami_style():
|
|
from pygments.style import Style
|
|
from pygments.token import (
|
|
Comment, Keyword, Literal, Name, Number, Operator,
|
|
Punctuation, String, Token,
|
|
)
|
|
|
|
palette = _kami_palette()
|
|
|
|
class KamiStyle(Style):
|
|
background_color = ""
|
|
default_style = ""
|
|
styles = {
|
|
Token: "",
|
|
Comment: palette["stone"],
|
|
Comment.Single: palette["stone"],
|
|
Comment.Multiline: palette["stone"],
|
|
Comment.Preproc: palette["stone"],
|
|
Keyword: palette["brand"],
|
|
Keyword.Constant: palette["brand"],
|
|
Keyword.Namespace: palette["brand"],
|
|
Keyword.Type: palette["brand"],
|
|
Name.Builtin: palette["brand"],
|
|
Name.Function: palette["near_black"],
|
|
Name.Class: palette["near_black"],
|
|
Name.Decorator: palette["olive"],
|
|
String: palette["olive"],
|
|
String.Doc: palette["stone"],
|
|
Number: palette["dark_warm"],
|
|
Number.Float: palette["dark_warm"],
|
|
Number.Integer: palette["dark_warm"],
|
|
Literal: palette["dark_warm"],
|
|
Operator: "",
|
|
Punctuation: "",
|
|
}
|
|
|
|
return KamiStyle
|
|
|
|
|
|
def _highlight_block(match: re.Match[str]) -> str:
|
|
from pygments import highlight as pyg_highlight
|
|
from pygments.formatters import HtmlFormatter
|
|
from pygments.lexers import get_lexer_by_name
|
|
|
|
open_tag = match.group(1)
|
|
language = match.group(2)
|
|
code = match.group(3)
|
|
close_tag = match.group(4)
|
|
|
|
code_text = html_mod.unescape(code)
|
|
|
|
try:
|
|
lexer = get_lexer_by_name(language, stripall=False)
|
|
except Exception:
|
|
return match.group(0)
|
|
|
|
formatter = HtmlFormatter(
|
|
style=_build_kami_style(),
|
|
noclasses=True,
|
|
nowrap=True,
|
|
)
|
|
|
|
highlighted = pyg_highlight(code_text, lexer, formatter)
|
|
return f'{open_tag}{highlighted}{close_tag}'
|
|
|
|
|
|
def highlight_code_blocks(html_text: str) -> str:
|
|
"""Apply syntax highlighting to language-tagged code blocks.
|
|
|
|
Returns HTML unchanged if Pygments is not installed or no
|
|
language-tagged blocks are found.
|
|
"""
|
|
if not CODE_BLOCK_RE.search(html_text):
|
|
return html_text
|
|
|
|
try:
|
|
import pygments # noqa: F401
|
|
except ImportError:
|
|
_warn_missing_pygments()
|
|
return html_text
|
|
|
|
return CODE_BLOCK_RE.sub(_highlight_block, html_text)
|