kaas

backend/rooms.py

@@ -1,10 +1,29 @@
-from fastapi import FastAPI
-from pydantic import BaseModel
+import re
+import secrets
+import string
+
 from database import execute, fetch_one, fetch_all
 
 DEFAULT_ROLE = "member"
 
-def create_room(user_id, room_name, invite_code):
+
+def validate_invite_code(invite_code: str):
+    if not 6 <= len(invite_code) <= 32:
+        return False
+
+    if not re.fullmatch(r"[A-Za-z0-9]+", invite_code):
+        return False
+
+    return True
+
+
+def create_room(user_id: int, room_name: str, invite_code: str):
+
+    if not validate_invite_code(invite_code):
+        return {
+            "status": "failed",
+            "reason": "invalid_invite_code",
+        }
 
     room_id = execute(
         """
@@ -14,7 +33,7 @@ def create_room(user_id, room_name, invite_code):
         (
             room_name,
             user_id,
-            invite_code
+            invite_code,
         )
     )
 
@@ -26,314 +45,11 @@ def create_room(user_id, room_name, invite_code):
         (
             room_id,
             user_id,
-            "owner"
+            "owner",
         )
     )
 
     return {
         "room_id": room_id,
-        "member_id": member_id
-    }
-
-def search_room(
-    invite_code
-):
-    query = """
-    SELECT name
-    FROM rooms
-    WHERE invite_code = ?
-    """
-    room = fetch_one(
-        query,
-        (
-            invite_code,
-        )
-    )
-
-    if room is None:
-        return None
-
-    return room["name"]
-
-
-def accept_room(user_id, invite_code):
-
-    query = """
-    SELECT id
-    FROM rooms
-    WHERE invite_code = ?
-    """
-
-    room = fetch_one(
-        query,
-        (invite_code,)
-    )
-
-    if room is None:
-        return None
-
-    room_id = room["id"]
-
-    existing = fetch_one(
-        """
-        SELECT *
-        FROM room_members
-        WHERE room_id = ?
-        AND user_id = ?
-        """,
-        (
-            room_id,
-            user_id
-        )
-    )
-
-    if existing:
-        return None
-
-
-    query = """
-    INSERT INTO room_members
-    (room_id, user_id, role)
-    VALUES (?, ?, ?)
-    """
-
-    member_id = execute(
-        query,
-        (
-            room_id,
-            user_id,
-            "member"
-        )
-    )
-
-    return {
-        "room_id": room_id,
-        "member_id": member_id
-    }
-
-def get_rooms(user_id):
-    query = """
-    SELECT
-        rooms.id,
-        rooms.name,
-        rooms.owner_id,
-        rooms.invite_code,
-        room_members.role
-    FROM rooms
-    JOIN room_members
-        ON rooms.id = room_members.room_id
-    WHERE room_members.user_id = ?
-    """
-
-    rooms = fetch_all(
-        query,
-        (
-            user_id,
-        )
-    )
-
-    return rooms
-
-def delete_room(room_id):
-    room = execute(
-        """
-        SELECT id
-        FROM rooms
-        WHERE id = ?
-        """,
-        (room_id,)
-    )
-
-    if not room:
-        return {"error": "room not found"}
-
-    execute(
-        """
-        DELETE FROM rooms
-        WHERE id = ?
-        """,
-        (room_id,)
-    )
-
-    return {"status": "deleted"}
-
-def get_room(room_id, user_id):
-
-    room = fetch_one(
-        """
-        SELECT 
-            rooms.id,
-            rooms.name,
-            rooms.owner_id,
-            room_members.role
-        FROM rooms
-        JOIN room_members
-            ON rooms.id = room_members.room_id
-        WHERE rooms.id = ?
-        AND room_members.user_id = ?
-        """,
-        (
-            room_id,
-            user_id
-        )
-    )
-
-    if room is None:
-        return None
-
-
-    members = fetch_all(
-        """
-        SELECT
-            users.id,
-            users.username,
-            room_members.role
-        FROM room_members
-        JOIN users
-            ON users.id = room_members.user_id
-        WHERE room_members.room_id = ?
-        """,
-        (room_id,)
-    )
-
-
-    events = fetch_all(
-        """
-        SELECT *
-        FROM events
-        WHERE room_id = ?
-        ORDER BY start_time
-        """,
-        (room_id,)
-    )
-
-
-    return {
-        "room": dict(room),
-        "members": [dict(member) for member in members],
-        "events": [dict(event) for event in events]
-    }
-
-def get_room_members(room_id):
-
-    return fetch_all(
-        """
-        SELECT 
-            users.id,
-            users.username
-        FROM room_members
-        JOIN users
-        ON users.id = room_members.user_id
-        WHERE room_members.room_id = ?
-        """,
-        (room_id,)
-    )
-
-def leave_room(room_id, user_id):
-
-    from database import execute
-
-    execute(
-        """
-        DELETE FROM room_members
-        WHERE room_id = ?
-        AND user_id = ?
-        """,
-        (
-            room_id,
-            user_id
-        )
-    )
-
-def change_room_name(room_id, new_name, user_id):
-    room = fetch_one(
-        """
-        SELECT id
-        FROM rooms
-        WHERE id = ?
-        """,
-        (room_id,)
-    )
-
-    if not room:
-        return {
-            "status": "error",
-            "reason": "room_not_found"
-        }
-
-    execute(
-        """
-        UPDATE rooms
-        SET name = ?
-        WHERE id = ?
-        """,
-        (
-            new_name,
-            room_id
-        )
-    )
-
-    return {
-        "status": "updated",
-        "new_name": new_name,
-        "room_id": room_id,
-        "updated_by": user_id
-    }
-
-import secrets
-import string
-
-def regenerate_invite_code(room_id):
-    new_code = ''.join(
-        secrets.choice(string.ascii_uppercase + string.digits)
-        for _ in range(8)
-    )
-
-    execute(
-        """
-        UPDATE rooms
-        SET invite_code = ?
-        WHERE id = ?
-        """,
-        (new_code, room_id)
-    )
-
-    return {
-        "status": "updated",
-        "room_id": room_id,
-        "invite_code": new_code
-    }
-
-def remove_room_member(room_id, user_id):
-    member = fetch_one(
-        """
-        SELECT 1
-        FROM room_members
-        WHERE room_id = ? AND user_id = ?
-        """,
-        (room_id, user_id)
-    )
-
-    if member is None:
-        return False
-
-    execute(
-        """
-        DELETE FROM room_members
-        WHERE room_id = ? AND user_id = ?
-        """,
-        (room_id, user_id)
-    )
-
-    return True
-
-def update_member_role(room_id, user_id, role):
-    execute(
-        """
-        UPDATE room_members
-        SET role = ?
-        WHERE room_id = ?
-        AND user_id = ?
-        """,
-        (role, room_id, user_id),
-    )
+        "member_id": member_id,
+    }