import secrets from files.helpers.const import * from files.helpers.settings import get_setting from files.helpers.cloudflare import CLOUDFLARE_AVAILABLE from files.routes.wrappers import * from files.__main__ import app, limiter def session_init(): if not session.get("session_id"): session.permanent = True session["session_id"] = secrets.token_hex(49) @app.before_request def before_request(): g.desires_auth = False if not IS_LOCALHOST: app.config["COOKIE_DOMAIN"] = f".{request.host}" app.config["SESSION_COOKIE_DOMAIN"] = app.config["COOKIE_DOMAIN"] if SITE == 'marsey.world' and request.path != '/kofi': abort(404) g.agent = request.headers.get("User-Agent", "") if not g.agent and request.path != '/kofi': return 'Please use a "User-Agent" header!', 403 ua = g.agent.lower() if request.host != SITE: return {"error": "Unauthorized host provided"}, 403 if request.headers.get("CF-Worker"): return {"error": "Cloudflare workers are not allowed to access this website."}, 403 if not get_setting('bots') and request.headers.get("Authorization"): abort(403) if '; wv) ' in ua: g.browser = 'webview' elif ' firefox/' in ua: g.browser = 'firefox' elif 'iphone' in ua or 'ipad' in ua or 'ipod' in ua or 'mac os' in ua: g.browser = 'apple' else: g.browser = 'chromium' g.is_tor = request.headers.get("cf-ipcountry") == "T1" request.path = request.path.rstrip('/') if not request.path: request.path = '/' request.full_path = request.full_path.rstrip('?').rstrip('/') if not request.full_path: request.full_path = '/' session_init() limiter.check() g.db = db_session() @app.after_request def after_request(response:Response): if response.status_code < 400: _set_cloudflare_cookie(response) _commit_and_close_db() return response @app.teardown_appcontext def teardown_request(error): _rollback_and_close_db() stdout.flush() def _set_cloudflare_cookie(response:Response) -> None: ''' Sets a cookie that can be used by an upstream DDoS protection and caching provider ''' if not g.desires_auth: return if not CLOUDFLARE_AVAILABLE or not CLOUDFLARE_COOKIE_VALUE: return logged_in = bool(getattr(g, 'v', None)) if not logged_in and request.cookies.get("lo"): response.delete_cookie("lo", domain=app.config["COOKIE_DOMAIN"], samesite="Lax") elif logged_in and not request.cookies.get("lo"): response.set_cookie("lo", CLOUDFLARE_COOKIE_VALUE if logged_in else '', max_age=SESSION_LIFETIME, samesite="Lax", domain=app.config["COOKIE_DOMAIN"]) def _commit_and_close_db() -> bool: if not getattr(g, 'db', None): return False g.db.commit() g.db.close() del g.db return True def _rollback_and_close_db() -> bool: if not getattr(g, 'db', None): return False g.db.rollback() g.db.close() del g.db return True