from fastapi import APIRouter , Request ,HTTPException , WebSocket
from controllers import websocket_controller as wc
from controllers import ws_controller as w
import logging
import aiohttp

router = APIRouter(prefix="/websockets")

@router.websocket("/ws")
async def get_data(websocket:WebSocket):
    await websocket.accept()
    
    json = await websocket.receive_json()
    access_token = json['access_token']
    brand_name = json['brand_name'] if json.get('brand_name') is not None else None
    access_token or aiohttp.web.HTTPUnauthorized(text="Access token is missing or invalid")
    
    # access_token = await websocket.receive_text()
    # brand_name = "test"
    logging.info(f"brand_name: f{brand_name}")
    logging.info(f"access_token : {access_token}")
    await w.websocket_main(access_token ,brand_name, websocket)