Redirect to the configured host before OIDC if incorrect

Author: galenguyer

packet/utils.py

@@ -4,9 +4,10 @@
 from datetime import datetime, time, timedelta, date
 from functools import wraps, lru_cache
 from typing import Any, Callable, TypeVar, cast
+from urllib.parse import urlparse
 
 import requests
-from flask import session, redirect
+from flask import session, redirect, request
 
 from packet import auth, app, db, ldap
 from packet.mail import send_start_packet_mail
@@ -64,6 +65,16 @@ def is_freshman_on_floor(rit_username: str) -> bool:
         return False
 
 
+@app.before_request
+def before_reqest_callback() -> Any:
+    """
+    Pre-request function to ensure we're on the right URL before OIDC sees anything
+    """
+    if urlparse(request.base_url).hostname != app.config['SERVER_NAME']:
+        return redirect(request.base_url.replace(urlparse(request.base_url).hostname,
+            app.config['SERVER_NAME']), code=302)
+    return None
+
 def packet_auth(func: WrappedFunc) -> WrappedFunc:
     """
     Decorator for easily configuring oidc