try a context processor

1 files changed, 12 insertions, 0 deletions

diff --git a/newspipe/bootstrap.py b/newspipe/bootstrap.py
index 9160c857..1af61d2f 100644
--- a/newspipe/bootstrap.py
+++ b/newspipe/bootstrap.py
@@ -5,6 +5,7 @@ import os
 
 from flask import Flask
 from flask import request
+from flask import url_for as f_url_for
 from flask_babel import Babel
 from flask_babel import format_datetime
 from flask_migrate import Migrate
@@ -111,3 +112,14 @@ application.jinja_env.filters["datetime"] = format_datetime
 application.jinja_env.filters["datetimeformat"] = datetimeformat
 # inject application in Jinja env
 application.jinja_env.globals["application"] = application
+
+@application.context_processor
+def utility_processor():
+    def url_for(endpoint, **values):
+        """ Custom reverse-proxy support for url_for """
+        print(f"DEBUG: generating url_for {endpoint}")
+        _uf = f_url_for(endpoint, **values)
+        if application.config and "PREFIX" in application.config:
+            _uf = str(application.config("PREFIX")) + _uf
+        return _uf
+    return dict(url_for=url_for)