Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4import datetime
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10workspaces = {}
11channels = {}
12messages = {}
13attachments = {}
14next_user_id = 1
15next_workspace_id = 1
16next_channel_id = 1
17next_message_id = 1
18next_attachment_id = 1
19 
20def get_current_user(authorization: Optional[str] = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing auth header")
23    token = authorization.replace("Bearer ", "")
24    user_id = tokens.get(token)
25    if not user_id:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return users[user_id]
28 
29@app.post("/signup")
30def signup(username: str, password: str):
31    global next_user_id
32    for u in users.values():
33        if u["username"] == username:
34            raise HTTPException(status_code=400, detail="Username taken")
35    user_id = next_user_id
36    next_user_id += 1
37    users[user_id] = {"id": user_id, "username": username, "password": password, "role": "user"}
38    return {"id": user_id, "username": username}
39 
40@app.post("/login")
41def login(username: str, password: str):
42    for u in users.values():
43        if u["username"] == username and u["password"] == password:
44            token = str(uuid.uuid4())
45            tokens[token] = u["id"]
46            return {"token": token}
47    raise HTTPException(status_code=401, detail="Invalid credentials")
48 
49@app.get("/user/{user_id}")
50def get_user(user_id: int):
51    user = users.get(user_id)
52    if not user:
53        raise HTTPException(status_code=404)
54    return {"id": user["id"], "username": user["username"], "role": user["role"]}
55 
56@app.post("/workspace")
57def create_workspace(name: str, authorization: Optional[str] = Header(None)):
58    user = get_current_user(authorization)
59    global next_workspace_id
60    ws_id = next_workspace_id
61    next_workspace_id += 1
62    workspaces[ws_id] = {"id": ws_id, "name": name, "owner_id": user["id"], "admins": [user["id"]], "members": [user["id"]]}
63    return workspaces[ws_id]
64 
65@app.get("/workspace/{ws_id}")
66def get_workspace(ws_id: int):
67    ws = workspaces.get(ws_id)
68    if not ws:
69        raise HTTPException(status_code=404)
70    return ws
71 
72@app.post("/workspace/{ws_id}/join")
73def join_workspace(ws_id: int, authorization: Optional[str] = Header(None)):
74    user = get_current_user(authorization)
75    ws = workspaces.get(ws_id)
76    if not ws:
77        raise HTTPException(status_code=404)
78    if user["id"] not in ws["members"]:
79        ws["members"].append(user["id"])
80    return {"status": "joined"}
81 
82@app.post("/workspace/{ws_id}/promote")
83def promote_user(ws_id: int, user_id: int, authorization: Optional[str] = Header(None)):
84    requester = get_current_user(authorization)
85    ws = workspaces.get(ws_id)
86    if not ws:
87        raise HTTPException(status_code=404)
88    if requester["id"] != ws["owner_id"]:
89        raise HTTPException(status_code=403, detail="Only owner can promote")
90    if user_id not in ws["members"]:
91        raise HTTPException(status_code=400, detail="User not in workspace")
92    if user_id not in ws["admins"]:
93        ws["admins"].append(user_id)
94    users[user_id]["role"] = "admin"
95    return {"status": "promoted"}
96 
97@app.post("/channel")
98def create_channel(workspace_id: int, name: str, authorization: Optional[str] = Header(None)):
99    user = get_current_user(authorization)
100    ws = workspaces.get(workspace_id)
101    if not ws:
102        raise HTTPException(status_code=404)
103    if user["id"] not in ws["members"]:
104        raise HTTPException(status_code=403)
105    global next_channel_id
106    ch_id = next_channel_id
107    next_channel_id += 1
108    channels[ch_id] = {"id": ch_id, "workspace_id": workspace_id, "name": name, "permissions": {"read": ["user"], "write": ["user"]}}
109    return channels[ch_id]
110 
111@app.get("/channel/{ch_id}")
112def get_channel(ch_id: int):
113    ch = channels.get(ch_id)
114    if not ch:
115        raise HTTPException(status_code=404)
116    return ch
117 
118@app.post("/message")
119def send_message(channel_id: int, content: str, authorization: Optional[str] = Header(None)):
120    user = get_current_user(authorization)
121    ch = channels.get(channel_id)
122    if not ch:
123        raise HTTPException(status_code=404)
124    ws = workspaces.get(ch["workspace_id"])
125    if user["id"] not in ws["members"]:
126        raise HTTPException(status_code=403)
127    global next_message_id
128    msg_id = next_message_id
129    next_message_id += 1
130    messages[msg_id] = {"id": msg_id, "channel_id": channel_id, "sender_id": user["id"], "content": content, "timestamp": datetime.datetime.now().isoformat(), "attachment_ids": []}
131    return messages[msg_id]
132 
133@app.get("/message/{msg_id}")
134def get_message(msg_id: int):
135    msg = messages.get(msg_id)
136    if not msg:
137        raise HTTPException(status_code=404)
138    return msg
139 
140@app.get("/channel/{ch_id}/messages")
141def get_channel_messages(ch_id: int):
142    return [m for m in messages.values() if m["channel_id"] == ch_id]
143 
144@app.post("/attachment")
145def upload_attachment(channel_id: int, filename: str, file_url: str, authorization: Optional[str] = Header(None)):
146    user = get_current_user(authorization)
147    ch = channels.get(channel_id)
148    if not ch:
149        raise HTTPException(status_code=404)
150    ws = workspaces.get(ch["workspace_id"])
151    if user["id"] not in ws["members"]:
152        raise HTTPException(status_code=403)
153    global next_attachment_id
154    att_id = next_attachment_id
155    next_attachment_id += 1
156    attachments[att_id] = {"id": att_id, "channel_id": channel_id, "uploader_id": user["id"], "filename": filename, "file_url": file_url}
157    return attachments[att_id]
158 
159@app.get("/attachment/{att_id}")
160def get_attachment(att_id: int):
161    att = attachments.get(att_id)
162    if not att:
163        raise HTTPException(status_code=404)
164    return att

1fastapi
2uvicorn