Update app.py

app.py

@@ -1,66 +1,69 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from pydantic import BaseModel
-from datetime import datetime
-from fastapi.openapi.docs import get_swagger_ui_html
+from collections import deque
+from datetime import datetime, timedelta
+from typing import Optional
+import uuid
+
+app = FastAPI(title="Chat Handshake API")
+
+# ---------- Configuration ----------
+EXPIRATION_MINUTES = 5
+MAX_MESSAGES = 1000
+LOG_FILE = "activity.log"
+
+# ---------- In-memory message storage ----------
+messages = deque(maxlen=MAX_MESSAGES)
+
+# ---------- Pydantic model ----------
+class Message(BaseModel):
+    type: str                 # "offer", "answer", "chat", "connect"
+    from_user: str
+    to_user: Optional[str] = None
+    content: Optional[str] = None
+
+# ---------- Helper functions ----------
+def expire_messages():
+    cutoff = datetime.utcnow() - timedelta(minutes=EXPIRATION_MINUTES)
+    while messages and messages[0]["timestamp"] < cutoff:
+        messages.popleft()
+
+def log_message(msg: dict, ip: str):
+    with open(LOG_FILE, "a") as f:
+        f.write(f"{datetime.utcnow()} {ip} {msg}\n")
+
+# ---------- API Endpoints ----------
+
+@app.post("/write")
+async def write_message(msg: Message, request: Request):
+    expire_messages()
+    message_id = str(uuid.uuid4())
+    entry = {
+        "id": message_id,
+        "type": msg.type,
+        "from": msg.from_user,
+        "to": msg.to_user,
+        "content": msg.content,
+        "timestamp": datetime.utcnow()
+    }
+    messages.append(entry)
+    log_message(entry, request.client.host)
+    return {"status": "ok", "id": message_id}
+
+@app.get("/read")
+async def read_messages(username: str):
+    expire_messages()
+    not_mine = [m for m in messages if m["from"] != username]
+    return {"messages": not_mine}
+
+@app.get("/readall")
+async def read_all_messages():
+    expire_messages()
+    return {"messages": list(messages)}
+
+@app.post("/retract")
+async def retract_message(message_id: str):
+    global messages
+    messages = deque([m for m in messages if m["id"] != message_id], maxlen=MAX_MESSAGES)
+    return {"status": "ok"}
 
-
-app = FastAPI(title="1HIT RTC Signaling API")
-
-# ---------- Data Models ----------
-class Offer(BaseModel):
-    offer: str
-
-class Answer(BaseModel):
-    answer: str
-
-class IceCandidate(BaseModel):
-    candidate: str
-
-
-# ---------- Basic Endpoints ----------
-
-# ---------- 4️⃣ Explicit Swagger docs endpoint ----------
-@app.get("/docs", include_in_schema=False)
-async def custom_swagger_ui_html():
-    return get_swagger_ui_html(
-        openapi_url=app.openapi_url,
-        title=app.title + " - Swagger UI",
-        oauth2_redirect_url=app.swagger_ui_oauth2_redirect_url,
-        swagger_js_url="https://unpkg.com/swagger-ui-dist@5.9.0/swagger-ui-bundle.js",
-        swagger_css_url="https://unpkg.com/swagger-ui-dist@5.9.0/swagger-ui.css",
-    )
-
-@app.get("/")
-def root():
-    return {"message": "Hello HF!"}
-
-@app.get("/status")
-def status():
-    return {"status": "running", "time": datetime.utcnow()}
-
-
-@app.get("/time")
-def get_time():
-    return {"time": datetime.utcnow().strftime("%H:%M:%S UTC")}
-
-
-@app.post("/echo")
-def echo_message(payload: dict):
-    return {"you_sent": payload}
-
-
-# ---------- RTC Signaling Endpoints ----------
-@app.post("/rtc/offer")
-def rtc_offer(data: Offer):
-    # In production: store, forward, or route the offer to another client
-    return {"received_offer": data.offer}
-
-
-@app.post("/rtc/answer")
-def rtc_answer(data: Answer):
-    return {"received_answer": data.answer}
-
-
-@app.post("/rtc/ice")
-def rtc_ice(data: IceCandidate):
-    return {"received_candidate": data.candidate}