Merge branch 'master' of github.com:discord-python/site

1 files changed, 37 insertions, 0 deletions

diff --git a/pysite/rst/roles.py b/pysite/rst/roles.py
index bff00533..414832df 100644
--- a/pysite/rst/roles.py
+++ b/pysite/rst/roles.py
@@ -2,6 +2,7 @@
 from docutils import nodes
 from docutils.parsers.rst.roles import set_classes
 from docutils.parsers.rst.states import Inliner
+from flask import url_for
 from jinja2 import escape
 
 
@@ -44,3 +45,39 @@ def icon_role(_role: str, rawtext: str, text: str, lineno: int, inliner: Inliner
 
         node = nodes.raw(html, html, format="html", **options)
         return [node], []
+
+
+def url_for_role(_role: str, rawtext: str, text: str, lineno: int, inliner: Inliner,
+                 options: dict = None, _content: dict = None):
+    if options is None:
+        options = {}
+
+    set_classes(options)
+
+    if "/" in text:
+        parts = [escape(x) for x in text.split("/")]
+    else:
+        msg = inliner.reporter.error("URL specification must be in the form <page.name>/<text>", line=lineno)
+        prb = inliner.problematic(text, rawtext, msg)
+
+        return [prb], [msg]
+
+    if len(parts) != 2:
+        msg = inliner.reporter.error("URL specification must be in the form <page.name>/<text>", line=lineno)
+        prb = inliner.problematic(text, rawtext, msg)
+
+        return [prb], [msg]
+    else:
+        try:
+            url = url_for(parts[0])
+            name = parts[1]
+
+            html = f"""<a href="{url}">{name}</a>"""
+
+            node = nodes.raw(html, html, format="html", **options)
+            return [node], []
+        except Exception as e:
+            msg = inliner.reporter.error(str(e), line=lineno)
+            prb = inliner.problematic(text, rawtext, msg)
+
+            return [prb], [msg]