from fastapi import APIRouter, Depends
from db.handlers.orders import OrdersHandler
from utils import requestDict, logger


router = APIRouter()


@router.get("/")
async def get_requests():
    return await OrdersHandler.countNew()


@router.post("/")
async def post_requests(
    reqData: dict = Depends(requestDict),
):
    logger.info("Обновление заказа")
    response = {"status": "error"}
    orderId = reqData.get("body").get("orderId")
    userId = reqData.get("body").get("userId")
    status = reqData.get("body").get("status")
    comment = reqData.get("body").get("comment")
    if (orderId is not None and userId is not None) and (
        status is not None or comment is not None
    ):
        result = await OrdersHandler.update(int(orderId), userId, status, comment)
        if "errorMessage" in result.keys():
            response["message"] = result["errorMessage"]
        else:
            response["status"] = "ok"
            if "orders" in result.keys():
                response["data"] = result["orders"]
    return response


@router.post("/new", summary="Добавление нового заказа")
async def add_order(reqData: dict = Depends(requestDict)):
    logger.info(f"Добавление нового заказа")
    userId = reqData.get("body").get("userId")
    customer_comment = reqData.get("body").get("customer_comment")
    return await OrdersHandler.new(userId, customer_comment)