Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import asyncio
6from fastapi import WebSocket, WebSocketDisconnect
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12rooms = {}
13messages = {}
14next_user_id = 1
15next_room_id = 1
16next_message_id = 1
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class CreateRoomRequest(BaseModel):
27    name: str
28    subject: str
29    max_capacity: int
30 
31class CoHostRequest(BaseModel):
32    user_id: int
33 
34class MessageRequest(BaseModel):
35    content: str
36 
37def get_user_from_token(authorization: str = Header(None)):
38    if not authorization:
39        raise HTTPException(status_code=401, detail="No auth header")
40    token = authorization.replace("Bearer ", "")
41    user_id = tokens.get(token)
42    if not user_id:
43        raise HTTPException(status_code=401, detail="Invalid token")
44    return users[user_id]
45 
46@app.post("/signup")
47def signup(req: SignupRequest):
48    global next_user_id
49    user_id = next_user_id
50    next_user_id += 1
51    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
52    return {"id": user_id, "username": req.username}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for user in users.values():
57        if user["username"] == req.username and user["password"] == req.password:
58            token = secrets.token_hex(16)
59            tokens[token] = user["id"]
60            return {"token": token}
61    raise HTTPException(status_code=401, detail="Invalid credentials")
62 
63@app.get("/users/{user_id}")
64def get_user(user_id: int):
65    user = users.get(user_id)
66    if not user:
67        raise HTTPException(status_code=404, detail="User not found")
68    return user
69 
70@app.post("/rooms")
71def create_room(req: CreateRoomRequest, authorization: str = Header(None)):
72    global next_room_id
73    user = get_user_from_token(authorization)
74    room_id = next_room_id
75    next_room_id += 1
76    rooms[room_id] = {
77        "id": room_id,
78        "name": req.name,
79        "subject": req.subject,
80        "max_capacity": req.max_capacity,
81        "creator_id": user["id"],
82        "co_hosts": [],
83        "members": [user["id"]]
84    }
85    messages[room_id] = []
86    return rooms[room_id]
87 
88@app.get("/rooms/{room_id}")
89def get_room(room_id: int):
90    room = rooms.get(room_id)
91    if not room:
92        raise HTTPException(status_code=404, detail="Room not found")
93    return room
94 
95@app.post("/rooms/{room_id}/co-host")
96def promote_co_host(room_id: int, req: CoHostRequest, authorization: str = Header(None)):
97    user = get_user_from_token(authorization)
98    room = rooms.get(room_id)
99    if not room:
100        raise HTTPException(status_code=404, detail="Room not found")
101    if room["creator_id"] != user["id"]:
102        raise HTTPException(status_code=403, detail="Only creator can promote co-hosts")
103    if req.user_id not in room["members"]:
104        raise HTTPException(status_code=400, detail="User not in room")
105    if req.user_id not in room["co_hosts"]:
106        room["co_hosts"].append(req.user_id)
107    return room
108 
109@app.get("/rooms/{room_id}/messages")
110def get_messages(room_id: int):
111    msgs = messages.get(room_id)
112    if msgs is None:
113        raise HTTPException(status_code=404, detail="Room not found")
114    return msgs
115 
116@app.websocket("/ws/{room_id}")
117async def websocket_endpoint(websocket: WebSocket, room_id: int):
118    await websocket.accept()
119    token = websocket.headers.get("authorization", "").replace("Bearer ", "")
120    user_id = tokens.get(token)
121    if not user_id:
122        await websocket.send_json({"error": "unauthorized"})
123        await websocket.close()
124        return
125    user = users[user_id]
126    room = rooms.get(room_id)
127    if not room:
128        await websocket.send_json({"error": "room not found"})
129        await websocket.close()
130        return
131    if user_id not in room["members"]:
132        await websocket.send_json({"error": "not a member"})
133        await websocket.close()
134        return
135    try:
136        while True:
137            data = await websocket.receive_json()
138            if data.get("type") == "message":
139                global next_message_id
140                msg_id = next_message_id
141                next_message_id += 1
142                msg = {
143                    "id": msg_id,
144                    "user_id": user_id,
145                    "username": user["username"],
146                    "content": data["content"],
147                    "room_id": room_id
148                }
149                messages[room_id].append(msg)
150                for member_id in room["members"]:
151                    pass  # would broadcast in real impl
152                await websocket.send_json({"type": "ack", "message_id": msg_id})
153    except WebSocketDisconnect:
154        pass

1fastapi
2uvicorn
3websockets