from fastapi import FastAPI, WebSocket, WebSocketDisconnect
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
import os

app = FastAPI(title="Music Platform API")

app.add_middleware(
    CORSMiddleware,
    allow_origins=["http://localhost:5173"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

os.makedirs("uploads/music", exist_ok=True)
os.makedirs("uploads/covers", exist_ok=True)

app.mount("/uploads", StaticFiles(directory="uploads"), name="uploads")

from app.routes import auth, music, playlists, rooms, admin

app.include_router(auth.router, prefix="/api/auth", tags=["auth"])
app.include_router(music.router, prefix="/api/music", tags=["music"])
app.include_router(playlists.router, prefix="/api/playlists", tags=["playlists"])
app.include_router(rooms.router, prefix="/api/rooms", tags=["rooms"])
app.include_router(admin.router)

@app.get("/")
def root():
    return {"message": "Music Platform API"}