init

Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,54 @@
+from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect
+from pydantic import BaseModel
+from typing import Dict
+import json
+from cryptography.hazmat.primitives.asymmetric import rsa, padding
+from cryptography.hazmat.primitives import serialization, hashes
+
+app = FastAPI()
+
+# In-memory storage for public keys
+public_keys: Dict[str, str] = {}
+messages: Dict[str, list] = {}
+
+class Message(BaseModel):
+    recipient: str
+    message: str
+
+@app.post("/register")
+async def register(username: str, public_key: str):
+    public_keys[username] = public_key
+    return {"status": "Public key registered successfully"}
+
+@app.get("/public_key/{username}")
+async def get_public_key(username: str):
+    if username in public_keys:
+        return {"public_key": public_keys[username]}
+    raise HTTPException(status_code=404, detail="User not found")
+
+@app.post("/send_message")
+async def send_message(message: Message):
+    if message.recipient not in messages:
+        messages[message.recipient] = []
+    messages[message.recipient].append(message.message)
+    return {"status": "Message sent"}
+
+@app.websocket("/ws/{username}")
+async def websocket_endpoint(websocket: WebSocket, username: str):
+    await websocket.accept()
+    while True:
+        try:
+            data = await websocket.receive_text()
+            message_data = json.loads(data)
+            recipient = message_data.get("recipient")
+            message = message_data.get("message")
+            
+            if recipient in messages:
+                messages[recipient].append(message)
+            else:
+                messages[recipient] = [message]
+
+            # Send the new message to the recipient
+            await websocket.send_text(json.dumps({"recipient": recipient, "message": message}))
+        except WebSocketDisconnect:
+            break

requirements.txt

@@ -0,0 +1,2 @@
+fastapi
+cryptography