aboutsummaryrefslogtreecommitdiffstats
path: root/backend/__init__.py
blob: adf304db36fdeb65f932172a1d5aab74765c0813 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import os

from starlette.applications import Starlette
from starlette.middleware import Middleware
from starlette.middleware.authentication import AuthenticationMiddleware
from starlette.middleware.cors import CORSMiddleware

from backend.authentication import JWTAuthenticationBackend
from backend.route_manager import create_route_map
from backend.middleware import DatabaseMiddleware

HOSTS = r"https://(?:(?:(?:.*--)?pydis-forms\.netlify\.app)|forms\.pythondiscord\.com)"
HAS_CUSTOM_HOST = os.getenv("ALLOWED_URL") is not None

middleware = [
    Middleware(
        CORSMiddleware,
        allow_origins=[os.getenv("ALLOWED_URL")] if HAS_CUSTOM_HOST else None,
        allow_origin_regex=HOSTS if not HAS_CUSTOM_HOST else None,
        allow_headers=[
            "Authorization",
            "Content-Type"
        ],
        allow_methods=["*"]
    ),
    Middleware(DatabaseMiddleware),
    Middleware(AuthenticationMiddleware, backend=JWTAuthenticationBackend())
]

app = Starlette(routes=create_route_map(), middleware=middleware)