wiki: Implement embedding images with {{...}} syntax

[ipfire.org.git] / src / backend / wiki.py

diff --git a/src/backend/wiki.py b/src/backend/wiki.py
index e3275c2a3472b8a1e5b724bd9451d066f33a2cc4..92d84c301ac4b4a10b69b509d32b7978046e20a8 100644 (file)
--- a/src/backend/wiki.py
+++ b/src/backend/wiki.py
@@ -30,6 +30,23 @@ class Wiki(misc.Object):
                if res:
                        return Page(self.backend, res.id, data=res)
 
+       def make_path(self, page, path):
+               # Nothing to do for absolute links
+               if path.startswith("/"):
+                       pass
+
+               # Relative links (one-level down)
+               elif path.startswith("./"):
+                       path = os.path.join(page, path)
+
+               # All other relative links
+               else:
+                       p = os.path.dirname(page)
+                       path = os.path.join(p, path)
+
+               # Normalise links
+               return os.path.normpath(path)
+
        def get_page_title(self, page, default=None):
                # Try to retrieve title from cache
                title = self.memcache.get("wiki:title:%s" % page)
@@ -206,13 +223,6 @@ class Wiki(misc.Object):
                        mimetype, blob_id, size) VALUES(%s,  %s, %s, %s, %s, %s, %s) RETURNING *", path,
                        filename, author.uid, address, mimetype, blob.id, len(data))
 
-       def find_image(self, path, filename):
-               for p in (path, os.path.dirname(path)):
-                       file = self.get_file_by_path(os.path.join(p, filename))
-
-                       if file and file.is_image():
-                               return file
-
        def render(self, path, text):
                r = WikiRenderer(self.backend, path)
 
@@ -494,18 +504,16 @@ class WikiRenderer(misc.Object):
        # Mail link
        email_link = re.compile(r"\[\[([a-zA-Z0-9_.+-]+@[a-zA-Z0-9-]+\.[a-zA-Z0-9-.]+)(?:\|(.+?))?\]\]")
 
+       # Images
+       images = re.compile(r"{{([\w\d\/\-\.]+)(?:\|(.+?))?}}")
+
        def init(self, path):
                self.path = path
 
        def _render_wiki_link(self, m):
                path, alias = m.groups()
 
-               # Allow relative links
-               if not path.startswith("/"):
-                       path = os.path.join(self.path, path)
-
-               # Normalise links
-               path = os.path.normpath(path)
+               path = self.backend.wiki.make_path(self.path, path)
 
                return """<a href="%s">%s</a>""" % (
                        path,
@@ -560,41 +568,36 @@ class WikiRenderer(misc.Object):
                return """<a class="link-external" href="mailto:%s">%s</a>""" \
                        % (address, alias or address)
 
-       def render(self, text):
-               logging.debug("Rendering %s" % self.path)
+       def _render_image(self, m):
+               url, text = m.groups()
 
-               # Link images
-               replacements = []
-               for match in re.finditer(r"!\[(.*?)\]\((.*?)\)", text):
-                       alt_text, url = match.groups()
+               # Skip any absolute and external URLs
+               if url.startswith("/") or url.startswith("https://") or url.startswith("http://"):
+                       return """<img src="%s" alt="%s">""" % (url, text or "")
 
-                       # Skip any absolute and external URLs
-                       if url.startswith("/") or url.startswith("https://") or url.startswith("http://"):
-                               continue
+               # Try to split query string
+               url, delimiter, qs = url.partition("?")
 
-                       # Try to split query string
-                       url, delimiter, qs = url.partition("?")
+               # Parse query arguments
+               args = urllib.parse.parse_qs(qs)
 
-                       # Parse query arguments
-                       args = urllib.parse.parse_qs(qs)
+               # Build absolute path
+               url = self.backend.wiki.make_path(self.path, url)
 
-                       # Find image
-                       file = self.backend.wiki.find_image(self.path, url)
-                       if not file:
-                               continue
+               # Find image
+               file = self.backend.wiki.get_file_by_path(url)
+               if not file or not file.is_image():
+                       return "<!-- Could not find image %s in %s -->" % (url, self.path)
 
-                       # Scale down the image if not already done
-                       if not "s" in args:
-                               args["s"] = "768"
+               # Scale down the image if not already done
+               if not "s" in args:
+                       args["s"] = "768"
 
-                       # Format URL
-                       url = "%s?%s" % (file.url, urllib.parse.urlencode(args))
+               return """<a href="%s?action=detail"><img src="%s?%s" alt="%s"></a>""" \
+                       % (url, url, urllib.parse.urlencode(args), text or "")
 
-                       replacements.append((match.span(), file, alt_text, url))
-
-               # Apply all replacements
-               for (start, end), file, alt_text, url in reversed(replacements):
-                       text = text[:start] + "[![%s](%s)](%s?action=detail)" % (alt_text, url, file.url) + text[end:]
+       def render(self, text):
+               logging.debug("Rendering %s" % self.path)
 
                # Handle wiki links
                text = self.wiki_link.sub(self._render_wiki_link, text)
@@ -608,5 +611,8 @@ class WikiRenderer(misc.Object):
                # Handle email links
                text = self.email_link.sub(self._render_email_link, text)
 
+               # Handle images
+               text = self.images.sub(self._render_image, text)
+
                # Borrow this from the blog
                return self.backend.blog._render_text(text, lang="markdown")