main.py

from fastapi import Depends, FastAPI, WebSocket
from config.database import create_start_app_handler, create_stop_app_handler
from sqlalchemy.orm import Session
from typing import Dict
from userService.v1.routers import userrouter
from companyService.v1.routers import companyrouter
from busService.v1.routers import busrouter
from busService.v1.socket_manager import relay_data
from config.middlewares import is_bus_or_authenticated


tags_metadata = [
    {
        "name": "users",
        "description": "Operations with users and authentication",
    },
    {
        "name": "company",
        "description": "Operations with companies",
    },
    {
        "name": "bus",
        "description": "Operations with bus",
    },
]


app = FastAPI()
app.add_event_handler("startup", create_start_app_handler(app))
app.add_event_handler("shutdown", create_stop_app_handler(app))
# userService.v1.models.Base.metadata.create_all(Engine)


app.include_router(userrouter, prefix="/api/v1")
app.include_router(companyrouter, prefix="/api/v1")
app.include_router(busrouter, prefix="/api/v1")


"""@app.get("/")
async def root(db: Session = Depends()) -> Dict:
    return {"message": "Hello Tom"}
"""


@app.websocket("/wb/{bus_id}")
async def track_bus(
    websocket: WebSocket,
    bus_id: int
    # token: str,
    # is_bus: bool = Depends(is_bus_or_authenticated),
):
    """if is_bus == "unverified":
        await websocket.close()
    await relay_data(websocket, bus_id, is_bus == "owner")
    """
    await relay_data(websocket, bus_id, True)