Add url helpers

4 files changed, 19 insertions, 7 deletions

diff --git a/gitsrht/app.py b/gitsrht/app.py
index 0cf23f6..491801b 100644
--- a/gitsrht/app.py
+++ b/gitsrht/app.py
@@ -14,6 +14,7 @@ from gitsrht.types import User
 db.init()
 
 import gitsrht.oauth
+from gitsrht import urls
 from gitsrht.git import commit_time, trim_commit
 
 def lookup_user(email):
@@ -78,3 +79,6 @@ class GitApp(SrhtFlask):
         return user
 
 app = GitApp()
+
+app.add_template_filter(urls.log_rss_url)
+app.add_template_filter(urls.refs_rss_url)
diff --git a/gitsrht/templates/log.html b/gitsrht/templates/log.html
index 90740d3..0f1e3fc 100644
--- a/gitsrht/templates/log.html
+++ b/gitsrht/templates/log.html
@@ -8,10 +8,7 @@
   <link rel="alternate"
     title="{{ repo.owner.canonical_name }}/{{ repo.name }}: {{ ref }} log"
     type="application/rss+xml"
-    href="{{ root }}{{ url_for('repo.log_rss',
-      owner=repo.owner.canonical_name,
-      repo=repo.name,
-      ref=ref if ref != "master" else None) }}">
+    href="{{ root }}{{ repo|log_rss_url(ref=ref) }}">
 {% endblock %}
 
 {% block content %}
diff --git a/gitsrht/templates/refs.html b/gitsrht/templates/refs.html
index f63f025..9fbea6d 100644
--- a/gitsrht/templates/refs.html
+++ b/gitsrht/templates/refs.html
@@ -8,9 +8,7 @@
   <link rel="alternate"
     title="{{ repo.owner.canonical_name }}/{{ repo.name }} refs"
     type="application/rss+xml"
-    href="{{ root }}{{ url_for('repo.refs_rss',
-      owner=repo.owner.canonical_name,
-      repo=repo.name) }}">
+    href="{{ root }}{{ repo|refs_rss_url }}">
 {% endblock %}
 
 {% block content %}
diff --git a/gitsrht/urls.py b/gitsrht/urls.py
new file mode 100644
index 0000000..24b4df8
--- /dev/null
+++ b/gitsrht/urls.py
@@ -0,0 +1,13 @@
+from flask import url_for
+
+def log_rss_url(repo, ref=None):
+    ref = ref if ref != "master" else None
+    return url_for("repo.log_rss",
+        owner=repo.owner.canonical_name,
+        repo=repo.name,
+        ref=ref)
+
+def refs_rss_url(repo):
+    return url_for("repo.refs_rss",
+        owner=repo.owner.canonical_name,
+        repo=repo.name)