mirror of
https://github.com/rommapp/romm.git
synced 2026-02-19 07:50:57 +01:00
124 lines
3.6 KiB
Python
124 lines
3.6 KiB
Python
import uvicorn
|
|
import alembic.config
|
|
import re
|
|
import sys
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi_pagination import add_pagination
|
|
from starlette.middleware.authentication import AuthenticationMiddleware
|
|
from starlette.middleware.sessions import SessionMiddleware
|
|
|
|
from config import (
|
|
DEV_PORT,
|
|
DEV_HOST,
|
|
ROMM_AUTH_SECRET_KEY,
|
|
ROMM_AUTH_ENABLED,
|
|
ENABLE_RESCAN_ON_FILESYSTEM_CHANGE,
|
|
RESCAN_ON_FILESYSTEM_CHANGE_DELAY,
|
|
ENABLE_SCHEDULED_RESCAN,
|
|
SCHEDULED_RESCAN_CRON,
|
|
ENABLE_SCHEDULED_UPDATE_SWITCH_TITLEDB,
|
|
SCHEDULED_UPDATE_SWITCH_TITLEDB_CRON,
|
|
ENABLE_SCHEDULED_UPDATE_MAME_XML,
|
|
SCHEDULED_UPDATE_MAME_XML_CRON,
|
|
)
|
|
from endpoints import search, platform, rom, identity, oauth, scan # noqa
|
|
from handler import dbh
|
|
from utils.socket import socket_app
|
|
from utils.auth import (
|
|
HybridAuthBackend,
|
|
CustomCSRFMiddleware,
|
|
create_default_admin_user,
|
|
)
|
|
|
|
app = FastAPI(title="RomM API", version="0.1.0")
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
if ROMM_AUTH_ENABLED and "pytest" not in sys.modules:
|
|
# CSRF protection (except endpoints listed in exempt_urls)
|
|
app.add_middleware(
|
|
CustomCSRFMiddleware,
|
|
secret=ROMM_AUTH_SECRET_KEY,
|
|
exempt_urls=[re.compile(r"^/token.*"), re.compile(r"^/ws")],
|
|
)
|
|
|
|
# Handles both basic and oauth authentication
|
|
app.add_middleware(
|
|
AuthenticationMiddleware,
|
|
backend=HybridAuthBackend(),
|
|
)
|
|
|
|
# Enables support for sessions on requests
|
|
app.add_middleware(
|
|
SessionMiddleware,
|
|
secret_key=ROMM_AUTH_SECRET_KEY,
|
|
same_site="strict",
|
|
https_only=False,
|
|
)
|
|
|
|
app.include_router(oauth.router)
|
|
app.include_router(identity.router)
|
|
app.include_router(platform.router)
|
|
app.include_router(rom.router)
|
|
app.include_router(search.router)
|
|
|
|
add_pagination(app)
|
|
app.mount("/ws", socket_app)
|
|
|
|
|
|
# Endpoint to set the CSRF token in cache
|
|
@app.get("/heartbeat")
|
|
def heartbeat():
|
|
return {
|
|
"ROMM_AUTH_ENABLED": ROMM_AUTH_ENABLED,
|
|
"WATCHER": {
|
|
"ENABLED": ENABLE_RESCAN_ON_FILESYSTEM_CHANGE,
|
|
"TITLE": "Rescan on filesystem change",
|
|
"MESSAGE": f"Runs a scan when a change is detected in the library path, with a {RESCAN_ON_FILESYSTEM_CHANGE_DELAY} minutes delay",
|
|
},
|
|
"SCHEDULER": {
|
|
"RESCAN": {
|
|
"ENABLED": ENABLE_SCHEDULED_RESCAN,
|
|
"CRON": SCHEDULED_RESCAN_CRON,
|
|
"TITLE": "Scheduled rescan",
|
|
"MESSAGE": "Rescans the entire library",
|
|
},
|
|
"SWITCH_TITLEDB": {
|
|
"ENABLED": ENABLE_SCHEDULED_UPDATE_SWITCH_TITLEDB, # noqa
|
|
"CRON": SCHEDULED_UPDATE_SWITCH_TITLEDB_CRON,
|
|
"TITLE": "Scheduled Switch TitleDB update",
|
|
"MESSAGE": "Updates the Nintedo Switch TitleDB file",
|
|
},
|
|
"MAME_XML": {
|
|
"ENABLED": ENABLE_SCHEDULED_UPDATE_MAME_XML,
|
|
"CRON": SCHEDULED_UPDATE_MAME_XML_CRON,
|
|
"TITLE": "Scheduled MAME XML update",
|
|
"MESSAGE": "Updates the Nintedo MAME XML file",
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
@app.on_event("startup")
|
|
def startup() -> None:
|
|
"""Startup application."""
|
|
|
|
# Create default admin user if no admin user exists
|
|
if len(dbh.get_admin_users()) == 0 and "pytest" not in sys.modules:
|
|
create_default_admin_user()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Run migrations
|
|
alembic.config.main(argv=["upgrade", "head"])
|
|
|
|
# Run application
|
|
uvicorn.run("main:app", host=DEV_HOST, port=DEV_PORT, reload=True)
|