diff --git a/surfsense_backend/main.py b/surfsense_backend/main.py index 81ef520..8d3f356 100644 --- a/surfsense_backend/main.py +++ b/surfsense_backend/main.py @@ -11,6 +11,7 @@ logging.basicConfig( if __name__ == "__main__": parser = argparse.ArgumentParser(description='Run the SurfSense application') parser.add_argument('--reload', action='store_true', help='Enable hot reloading') + parser.add_argument('--proxy', nargs='?', const='0.0.0.0/0', default=None, help='Enable proxy headers with allowed IPs (e.g., --proxy 0.0.0.0/0)') args = parser.parse_args() uvicorn.run( @@ -18,5 +19,7 @@ if __name__ == "__main__": host="0.0.0.0", log_level="info", reload=args.reload, - reload_dirs=["app"] + reload_dirs=["app"], + proxy_headers=bool(args.proxy), + forwarded_allow_ips=args.proxy if args.proxy else "", )