From 8d0e1d1c4166b0a42695582bb222cea590ad8784 Mon Sep 17 00:00:00 2001
From: Piotr Dobrowolski <admin@tastycode.pl>
Date: Fri, 26 Jan 2018 20:07:28 +0100
Subject: [PATCH] Add PROXY_FIX config option to enable werkzeug ProxyFix

---
 at.cfg.dist | 2 ++
 at.py       | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/at.cfg.dist b/at.cfg.dist
index c37efdf..b2e0720 100644
--- a/at.cfg.dist
+++ b/at.cfg.dist
@@ -17,3 +17,5 @@ SPECIAL_DEVICES = {
 }
 
 SECRET_KEY = 'adaba'
+
+PROXY_FIX = False
diff --git a/at.py b/at.py
index 01347bb..8248bd7 100644
--- a/at.py
+++ b/at.py
@@ -23,6 +23,10 @@ app.jinja_env.add_extension('jinja2.ext.i18n')
 app.jinja_env.install_null_translations()
 app.updater = None
 
+if app.config.get('PROXY_FIX'):
+    from werkzeug.contrib.fixers import ProxyFix
+    app.wsgi_app = ProxyFix(app.wsgi_app)
+
 auth = SpaceAuth(app)