From 18d16e2542997d5eb39e6b8b94ab8bb130be4e40 Mon Sep 17 00:00:00 2001
From: chenjintang-shrimp <jimmy163mail@163.com>
Date: Thu, 7 Aug 2025 12:00:19 +0000
Subject: [PATCH] feat(rooms): add router PUT /rooms/{room}/users/{user}

---
 app/router/room.py | 39 ++++++++++++++++++++++++++++++++++++---
 1 file changed, 36 insertions(+), 3 deletions(-)

diff --git a/app/router/room.py b/app/router/room.py
index b992e5b..1c51753 100644
--- a/app/router/room.py
+++ b/app/router/room.py
@@ -5,21 +5,27 @@ from time import timezone
 from typing import Literal
 
 from app.database.lazer_user import User
-from app.database.playlists import Playlist
+from app.database.playlists import Playlist, PlaylistResp
 from app.database.room import Room, RoomBase, RoomResp
 from app.dependencies.database import get_db, get_redis
 from app.dependencies.fetcher import get_fetcher
 from app.dependencies.user import get_current_user
 from app.fetcher import Fetcher
-from app.models.multiplayer_hub import MultiplayerRoom, ServerMultiplayerRoom
+from app.models.multiplayer_hub import (
+    MultiplayerRoom,
+    MultiplayerRoomUser,
+    ServerMultiplayerRoom,
+)
 from app.models.room import RoomStatus
 from app.signalr.hub import MultiplayerHubs
 
 from .api_router import router
 
-from fastapi import Depends, Query
+from fastapi import Depends, HTTPException, Query
 from redis.asyncio import Redis
+from sqlmodel import select
 from sqlmodel.ext.asyncio.session import AsyncSession
+from starlette.status import HTTP_417_EXPECTATION_FAILED
 
 
 @router.get("/rooms", tags=["rooms"], response_model=list[RoomResp])
@@ -111,3 +117,30 @@ async def get_room(
 ):
     server_room = MultiplayerHubs.rooms[room]
     return await RoomResp.from_hub(server_room)
+
+
+@router.delete("/rooms/{room}", tags=["room"])
+async def delete_room(room: int, db: AsyncSession = Depends(get_db)):
+    db_room = (await db.exec(select(Room).where(Room.id == room))).first()
+    if db_room is None:
+        raise HTTPException(404, "Room not found")
+    else:
+        await db.delete(db_room)
+        return None
+
+
+@router.put("/rooms/{room}/users/{user}", tags=["room"])
+async def add_user_to_room(room: int, user: int, db: AsyncSession = Depends(get_db)):
+    server_room = MultiplayerHubs.rooms[room]
+    server_room.room.users.append(MultiplayerRoomUser(user_id=user))
+    db_room = (await db.exec(select(Room).where(Room.id == room))).first()
+    if db_room is not None:
+        db_room.participant_count += 1
+        await db.commit()
+        resp = await RoomResp.from_hub(server_room)
+        await db.refresh(db_room)
+        for item in db_room.playlist:
+            resp.playlist.append(await PlaylistResp.from_db(item))
+        return resp
+    else:
+        raise HTTPException(404, "room not found0")