storage.py

import uuid


rooms = {}

def generate_id():
    return str(uuid.uuid4())

def create_player(name):
    return {
        "player_id": generate_id(),
        "name": name,
        "role": None,
        "score": 0
    }

def create_room(player_name):
    room_id = generate_id()
    player = create_player(player_name)

    room = {
        "room_id": room_id,
        "players": [player],
        "state": "waiting",         
        "roles_assigned": False,
        "mantri_guess": None
    }

    rooms[room_id] = room
    return room_id, player["player_id"]

def join_room(room_id, player_name):
    if room_id not in rooms:
        return None, "Room not found"

    room = rooms[room_id]

    if len(room["players"]) >= 4:
        return None, "Room is full"

    player = create_player(player_name)
    room["players"].append(player)

    return player["player_id"], None