from database import fetch_one


def get_room_role(room_id, user_id):
    member = fetch_one(
        """
        SELECT role
        FROM room_members
        WHERE room_id = ?
        AND user_id = ?
        """,
        (
            room_id,
            user_id
        )
    )

    if member is None:
        return None

    return member["role"]


def has_role(room_id, user_id, allowed_roles):

    role = get_room_role(
        room_id,
        user_id
    )

    if role is None:
        return False

    return role in allowed_roles