From 119efb80e421faa29b0ffa3a3e7f32e6a284c889 Mon Sep 17 00:00:00 2001
From: Gareth Coles <gdude2002@gmail.com>
Date: Tue, 10 Apr 2018 13:26:46 +0100
Subject: [Wiki] Add url_for text role

---
 pysite/rst/roles.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

(limited to 'pysite/rst/roles.py')

diff --git a/pysite/rst/roles.py b/pysite/rst/roles.py
index bff00533..09c4d373 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]
-- 
cgit v1.2.3