main.py

from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from storage import create_room, join_room, rooms

app = FastAPI(title="Raja-Mantri-Chor-Sipahi Backend")
from uuid import uuid4

app = FastAPI()


# ----------- Request Models -----------

class CreateRoomRequest(BaseModel):
    name: str


class JoinRoomRequest(BaseModel):
    room_id: str
    name: str


# ----------- APIs -----------

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


@app.post("/room/create")
def room_create(payload: CreateRoomRequest):
    room_id, player_id = create_room(payload.name)
    return {
        "room_id": room_id,
        "player_id": player_id,
        "message": "Room created successfully"
    }


rooms = {}

class JoinRoomRequest(BaseModel):
    room_id: str
    name: str

@app.post("/room/join")
def join_room(data: JoinRoomRequest):
    if data.room_id not in rooms:
        raise HTTPException(status_code=404, detail="Room not found")

    if len(rooms[data.room_id]["players"]) >= 4:
        raise HTTPException(status_code=400, detail="Room is full")

    player_id = str(uuid4())
    rooms[data.room_id]["players"][player_id] = data.name

    return {
        "room_id": data.room_id,
        "player_id": player_id,
        "name": data.name
    }


@app.get("/room/players/{room_id}")
def room_players(room_id: str):
    if room_id not in rooms:
        raise HTTPException(status_code=404, detail="Room not found")

    players = rooms[room_id]["players"]
    return {
        "room_id": room_id,
        "players": [
            {
                "player_id": p["player_id"],
                "name": p["name"]
            } for p in players
        ]
    }

from game_logic import assign_roles, get_player_role

@app.post("/room/assign/{room_id}")
def room_assign(room_id: str):
    if room_id not in rooms:
        raise HTTPException(status_code=404, detail="Room not found")

    room = rooms[room_id]

    success = assign_roles(room)
    if not success:
        raise HTTPException(
            status_code=400,
            detail="Cannot assign roles (need 4 players or already assigned)"
        )

    return {"message": "Roles assigned successfully"}


@app.get("/role/me/{room_id}/{player_id}")
def role_me(room_id: str, player_id: str):
    if room_id not in rooms:
        raise HTTPException(status_code=404, detail="Room not found")

    room = rooms[room_id]
    role = get_player_role(room, player_id)

    if role is None:
        raise HTTPException(status_code=404, detail="Player not found")

    return {"role": role}