Skip to content

Commit

Permalink
added proxy-fix middleware
Browse files Browse the repository at this point in the history
  • Loading branch information
Barakudum committed Jan 27, 2024
1 parent b1340bf commit 000fdec
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion src/jarklin/_commands/web/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,9 @@
def run() -> None:
import flask
import secrets
from werkzeug.middleware.dispatcher import DispatcherMiddleware
from werkzeug.wrappers import Response
from werkzeug.middleware.dispatcher import DispatcherMiddleware
from werkzeug.middleware.proxy_fix import ProxyFix
from .._get_config import get_config
from ...web import app

Expand Down Expand Up @@ -47,6 +48,16 @@ def make_session_permanent() -> None:
# pip install flask-kaccel for nginx
# app.config['USE_X_SENDFILE'] = config.getboolean('web', 'x_sendfile', fallback=False)

if config.getbool('web', 'proxy_fix', fallback=None) is not None:
app.wsgi_app = ProxyFix(
app=app,
x_for=config.getint('web', 'proxy_fix', 'x_forwarded_for', fallback=1),
x_proto=config.getint('web', 'proxy_fix', 'x_forwarded_proto', fallback=1),
x_host=config.getint('web', 'proxy_fix', 'x_forwarded_host', fallback=0),
x_port=config.getint('web', 'proxy_fix', 'x_forwarded_port', fallback=0),
x_prefix=config.getint('web', 'proxy_fix', 'x_forwarded_prefix', fallback=0),
)

app.run(
host=config.getstr('web', 'host', fallback=None),
port=config.getint('web', 'port', fallback=None),
Expand Down

0 comments on commit 000fdec

Please sign in to comment.