from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from api.cookies_controller import router as cookiesRouter
from api.profile_controller import router as profilesRouter
from api.video_controller import router as videosRouter
from database.database import SessionLocal, createTables
from model.profile import Profile
from notify.notify_clients import registerWebsocket

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(videosRouter)
app.include_router(profilesRouter)
app.include_router(cookiesRouter)
registerWebsocket(app)


@app.on_event("startup")
def startup():
    createTables()
    db = SessionLocal()
    if db.query(Profile).count() == 0:
        db.add(Profile(name="Standard"))
        db.commit()
    db.close()


@app.get("/")
def root():
    return {"status": "running"}