svg handling

Author: Redbot

src/md2bbcode/image_rewrite.py

@@ -0,0 +1,67 @@
+from typing import Optional
+from urllib.parse import parse_qs, quote, urlencode, urlparse
+
+_RASTER_SHIELDS_BASE = "https://raster.shields.io"
+_WESERV_BASE = "https://images.weserv.nl/"
+
+
+def rewrite_svg_url(url: str) -> Optional[str]:
+    if not url:
+        return url
+
+    parsed = urlparse(url)
+    if _is_github_actions_badge(parsed):
+        return _rewrite_github_actions_badge(parsed)
+
+    if _should_rasterize(parsed):
+        if parsed.scheme not in ("http", "https"):
+            return None
+        return _wrap_weserv(url)
+
+    return url
+
+
+def _is_github_actions_badge(parsed) -> bool:
+    if parsed.scheme not in ("http", "https"):
+        return False
+    if parsed.netloc.lower() != "github.com":
+        return False
+
+    parts = parsed.path.strip("/").split("/")
+    return (
+        len(parts) >= 6
+        and parts[2] == "actions"
+        and parts[3] == "workflows"
+        and parts[5].lower() == "badge.svg"
+    )
+
+
+def _rewrite_github_actions_badge(parsed) -> str:
+    parts = parsed.path.strip("/").split("/")
+    owner = parts[0]
+    repo = parts[1]
+    workflow = parts[4]
+
+    query = parse_qs(parsed.query)
+    params = {}
+    if query.get("branch"):
+        params["branch"] = query["branch"][0]
+    if query.get("event"):
+        params["event"] = query["event"][0]
+
+    base = f"{_RASTER_SHIELDS_BASE}/github/actions/workflow/status/{owner}/{repo}/{workflow}.png"
+    if params:
+        base = f"{base}?{urlencode(params)}"
+    return base
+
+
+def _should_rasterize(parsed) -> bool:
+    path = parsed.path.lower()
+    if path.endswith(".svg"):
+        return True
+    return False
+
+
+def _wrap_weserv(url: str) -> str:
+    encoded = quote(url, safe="")
+    return f"{_WESERV_BASE}?url={encoded}&output=png"

src/md2bbcode/renderers/bbcode.py

@@ -1,7 +1,10 @@
 from mistune.core import BaseRenderer
 from mistune.util import escape as escape_text, striptags, safe_entity
+import re
 from urllib.parse import urljoin, urlparse
 
+from md2bbcode.image_rewrite import rewrite_svg_url
+
 
 class BBCodeRenderer(BaseRenderer):
     """A renderer for converting Markdown to BBCode."""
@@ -56,7 +59,12 @@ def link(self, text: str, url: str, title=None) -> str:
 
     def image(self, text: str, url: str, title=None) -> str:
         alt_text = f' alt="{text}"' if text else ''
-        img_tag = f'[img{alt_text}]' + self.safe_url(url) + '[/img]'
+        safe_url = self.safe_url(url)
+        rewritten_url = rewrite_svg_url(safe_url)
+        if rewritten_url is None:
+            link_text = text or safe_url
+            return f"[url={safe_url}]{link_text}[/url]"
+        img_tag = f'[img{alt_text}]' + rewritten_url + '[/img]'
         # Check if alt text starts with 'pixel' and treat it as pixel art
         if text and text.lower().startswith('pixel'):
             return f'[pixelate]{img_tag}[/pixelate]'
@@ -115,6 +123,16 @@ def block_code(self, code: str, **attrs) -> str:
             return f"[CODE]{escape_text(code)}[/CODE]\n"
 
     def block_quote(self, text: str) -> str:
+        # GFMD "alerts"/admonitions are expressed as a blockquote
+        # Render these into a dedicated XenForo custom BBCode, rather than a normal QUOTE.
+        m = re.match(r"^\s*\[!(NOTE|TIP|IMPORTANT|WARNING|CAUTION)\]\s*", text, flags=re.IGNORECASE)
+        if m:
+            kind = m.group(1).lower()
+            body = text[m.end():].strip()
+            if body:
+                return f"[admonition={kind}]{body}[/admonition]\n"
+            return f"[admonition={kind}][/admonition]\n"
+
         return '[QUOTE]\n' + text + '[/QUOTE]\n'
 
     def block_html(self, html: str) -> str:

tests/test_svg_rewrite.py

@@ -29,18 +29,6 @@ def test_html_svg_wraps_weserv():
     assert '[img alt="alt text"]' in lowered
 
 
-def test_redguides_sparkline_adds_png_format():
-    sparkline_url = (
-        "https://www.redguides.com/community/resources/ghoul-resource.1623/watchers-sparkline"
-        "?months=24&w=500&h=180"
-    )
-    markdown = f'<img src="{sparkline_url}" alt="Watchers">'
-    result = process_readme(markdown, domain="")
-    lowered = result.lower()
-
-    assert "watchers-sparkline?months=24&w=500&h=180&format=png" in lowered
-
-
 def test_non_svg_image_unchanged():
     png_url = "https://example.com/x.png"
     markdown = f"![alt]({png_url})"