main.py

from fastapi import FastAPI, WebSocket, WebSocketDisconnect, Request, Response
from typing import List
from fastapi.templating import Jinja2Templates
from pydantic import BaseModel
# locate templates
import uvicorn
templates = Jinja2Templates(directory="templates")
app = FastAPI()
from uuid import UUID
from fastapi_sessions.backends.implementations import InMemoryBackend
# manager
from fastapi_session import SessionMiddleware, get_session

session_manager = SessionManager(secret_key="your-secret-key")

class SocketManager:
    def __init__(self):
        self.active_connections: List[(WebSocket, str)] = []

    async def connect(self, websocket: WebSocket, user: str):
        await websocket.accept()
        self.active_connections.append((websocket, user))

    def disconnect(self, websocket: WebSocket, user: str):
        self.active_connections.remove((websocket, user))

    async def broadcast(self, data):
        for connection in self.active_connections:
            await connection[0].send_json(data)    

manager = SocketManager()


@app.websocket("/api/chat")
async def chat(websocket: WebSocket):
    sender = websocket.cookies.get("X-Authorization")
    if sender:
        await manager.connect(websocket, sender)
        response = {
            "sender": sender,
            "message": "got connected"
        }
        await manager.broadcast(response)
        try:
            while True:
                data = await websocket.receive_json()
                await manager.broadcast(data)
        except WebSocketDisconnect:
            manager.disconnect(websocket, sender)
            response['message'] = "left"
            await manager.broadcast(response)
@app.get("/api/current_user")
def get_user(request: Request):
    return request.cookies.get("X-Authorization")

class RegisterValidator(BaseModel):
    username: str

@app.post("/api/register")
def register_user(user: RegisterValidator, response: Response):
    session = get_session()
    response.set_cookie(key="X-Authorization", value=user.username, httponly=True)
       
@app.get("/")
def get_home(request: Request):
    return templates.TemplateResponse("home.html", {"request": request})
@app.get("/chat")
def get_chat(request: Request):
    return templates.TemplateResponse("chat.html", {"request": request})

if __name__ == "__main__":
    uvicorn.run(app, host="127.0.0.1", port=8000)