86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from flask import Flask, request, url_for
|
|
from flask_socketio import SocketIO
|
|
from werkzeug.middleware.proxy_fix import ProxyFix
|
|
from .config import (
|
|
SECRET_KEY,
|
|
SESSION_COOKIE_SECURE,
|
|
PROXY_FIX_ENABLE,
|
|
PROXY_FIX_X_FOR,
|
|
PROXY_FIX_X_PROTO,
|
|
PROXY_FIX_X_HOST,
|
|
PROXY_FIX_X_PORT,
|
|
PROXY_FIX_X_PREFIX,
|
|
SOCKETIO_CORS_ALLOWED_ORIGINS,
|
|
)
|
|
from .db import init_db
|
|
from .utils import file_md5
|
|
|
|
socketio = SocketIO(cors_allowed_origins=SOCKETIO_CORS_ALLOWED_ORIGINS, ping_timeout=30, async_mode="threading")
|
|
_static_md5_cache: dict[tuple, str] = {}
|
|
|
|
|
|
def create_app() -> Flask:
|
|
app = Flask(__name__)
|
|
if PROXY_FIX_ENABLE:
|
|
app.wsgi_app = ProxyFix(
|
|
app.wsgi_app,
|
|
x_for=PROXY_FIX_X_FOR,
|
|
x_proto=PROXY_FIX_X_PROTO,
|
|
x_host=PROXY_FIX_X_HOST,
|
|
x_port=PROXY_FIX_X_PORT,
|
|
x_prefix=PROXY_FIX_X_PREFIX,
|
|
)
|
|
app.secret_key = SECRET_KEY
|
|
app.config.update(
|
|
SESSION_COOKIE_HTTPONLY=True,
|
|
SESSION_COOKIE_SAMESITE="Lax",
|
|
SESSION_COOKIE_SECURE=SESSION_COOKIE_SECURE,
|
|
)
|
|
|
|
@app.context_processor
|
|
def static_helpers():
|
|
def static_url(filename: str) -> str:
|
|
path = Path(app.static_folder or "") / filename
|
|
try:
|
|
stat = path.stat()
|
|
key = (filename, stat.st_mtime_ns, stat.st_size)
|
|
version = _static_md5_cache.get(key)
|
|
if not version:
|
|
_static_md5_cache.clear()
|
|
version = file_md5(path)
|
|
_static_md5_cache[key] = version
|
|
return url_for("static", filename=filename, v=version)
|
|
except OSError:
|
|
return url_for("static", filename=filename)
|
|
return {"static_url": static_url}
|
|
|
|
@app.after_request
|
|
def cache_headers(response):
|
|
response.headers.pop('Content-Disposition', None)
|
|
|
|
if request.endpoint == "static":
|
|
response.headers["Cache-Control"] = "public, max-age=31536000, immutable"
|
|
else:
|
|
response.headers["Cache-Control"] = "no-store, private"
|
|
return response
|
|
|
|
from .routes.main import bp as main_bp
|
|
from .routes.api import bp as api_bp
|
|
app.register_blueprint(main_bp)
|
|
app.register_blueprint(api_bp)
|
|
init_db()
|
|
from .services.auth import install_guards
|
|
install_guards(app)
|
|
|
|
socketio.init_app(app)
|
|
from .services.workers import set_socketio
|
|
set_socketio(socketio)
|
|
from .services.websocket import register_socketio_handlers
|
|
register_socketio_handlers(socketio)
|
|
from .services.startup_config import schedule_startup_config_apply
|
|
schedule_startup_config_apply(socketio)
|
|
return app
|