summaryrefslogtreecommitdiffstats
path: root/aurweb
diff options
context:
space:
mode:
Diffstat (limited to 'aurweb')
-rwxr-xr-xaurweb/scripts/rendercomment.py59
1 files changed, 54 insertions, 5 deletions
diff --git a/aurweb/scripts/rendercomment.py b/aurweb/scripts/rendercomment.py
index c8921f8f..659e18ad 100755
--- a/aurweb/scripts/rendercomment.py
+++ b/aurweb/scripts/rendercomment.py
@@ -1,12 +1,17 @@
#!/usr/bin/python3
import re
+import pygit2
import sys
import bleach
import markdown
+import aurweb.config
import aurweb.db
+repo_path = aurweb.config.get('serve', 'repo-path')
+commit_uri = aurweb.config.get('options', 'commit_uri')
+
class LinkifyPreprocessor(markdown.preprocessors.Preprocessor):
_urlre = re.compile(r'(\b(?:https?|ftp):\/\/[\w\/\#~:.?+=&%@!\-;,]+?'
@@ -21,10 +26,53 @@ class LinkifyExtension(markdown.extensions.Extension):
md.preprocessors.add('linkify', LinkifyPreprocessor(md), '_end')
+class GitCommitsPreprocessor(markdown.preprocessors.Preprocessor):
+ _oidre = re.compile(r'(\b)([0-9a-f]{7,40})(\b)')
+ _repo = pygit2.Repository(repo_path)
+ _head = None
+
+ def __init__(self, md, head):
+ self._head = head
+ super(markdown.preprocessors.Preprocessor, self).__init__(md)
+
+ def handleMatch(self, m):
+ oid = m.group(2)
+ if oid not in self._repo:
+ return oid
+
+ prefixlen = 12
+ while prefixlen < 40:
+ if oid[:prefixlen] in self._repo:
+ break
+ prefixlen += 1
+
+ html = '[`' + oid[:prefixlen] + '`]'
+ html += '(' + commit_uri % (self._head, oid[:prefixlen]) + ')'
+
+ return html
+
+ def run(self, lines):
+ return [self._oidre.sub(self.handleMatch, line) for line in lines]
+
+
+class GitCommitsExtension(markdown.extensions.Extension):
+ _head = None
+
+ def __init__(self, head):
+ self._head = head
+ super(markdown.extensions.Extension, self).__init__()
+
+ def extendMarkdown(self, md, md_globals):
+ preprocessor = GitCommitsPreprocessor(md, self._head)
+ md.preprocessors.add('git-commits', preprocessor, '_end')
+
+
def get_comment(conn, commentid):
- cur = conn.execute('SELECT Comments FROM PackageComments WHERE ID = ?',
- [commentid])
- return cur.fetchone()[0]
+ cur = conn.execute('SELECT PackageComments.Comments, PackageBases.Name '
+ 'FROM PackageComments INNER JOIN PackageBases '
+ 'ON PackageBases.ID = PackageComments.PackageBaseID '
+ 'WHERE PackageComments.ID = ?', [commentid])
+ return cur.fetchone()
def save_rendered_comment(conn, commentid, html):
@@ -37,8 +85,9 @@ def main():
conn = aurweb.db.Connection()
- text = get_comment(conn, commentid)
- html = markdown.markdown(text, extensions=['nl2br', LinkifyExtension()])
+ text, pkgbase = get_comment(conn, commentid)
+ html = markdown.markdown(text, extensions=['nl2br', LinkifyExtension(),
+ GitCommitsExtension(pkgbase)])
allowed_tags = bleach.sanitizer.ALLOWED_TAGS + ['p', 'br']
html = bleach.clean(html, tags=allowed_tags)
save_rendered_comment(conn, commentid, html)