pedidos-express/routes/orders.py

from logging import getLogger
from typing import List
from uuid import uuid4

from pydantic import BaseModel

from fastapi import APIRouter, Depends

from toteat import toteat as fudo
from models.sales import ItemWeb, OrderWeb
from models.items import Product
from models.user import User
from services.fudo_service import add_product_to_fudo
from services.data_service import DataServiceFactory
from config.messages import ErrorResponse, SuccessResponse
from auth.security import get_current_user
from utils.responses import error_response, success_response

logger = getLogger(__name__)

# Data services initialization
user_data_service = DataServiceFactory.get_user_service()
product_data_service = DataServiceFactory.get_product_service()
sale_data_service = DataServiceFactory.get_sales_service()

order_router = APIRouter()
name_promo = "Cervezas"


class ComparePricesResponse(BaseModel):
    product: Product
    oldPrice: int
    newPrice: int
    isAvailable: bool


def compare_prices(products: List[Product], items: List[ItemWeb]) -> list:
    """Compare prices of products and items and return the cheapest product"""
    # Initialize a dictionary to store the prices
    prices: List[ComparePricesResponse] = []
    for product, item in zip(products, items):
        if product.status == 0:
            prices.append(
                ComparePricesResponse(
                    product=product,
                    oldPrice=item.price,
                    newPrice=product.price,
                    isAvailable=False
                )
            )
            continue
        if product.price != item.price:
            prices.append(
                ComparePricesResponse(
                    product=product,
                    oldPrice=item.price,
                    newPrice=product.price,
                    isAvailable=True
                )
            )

    return list(map(lambda x: x.model_dump(), prices))


@order_router.post("/send")
async def printer_order(order: OrderWeb, current_user: User = Depends(get_current_user)):
    """Process printer order"""
    logger.info(f"Printer order received from user {current_user.email} for table {order.table}")
    
    # Extract order data
    items = order.items
    table = order.table

    # Input validation
    if not items or not table:
        logger.warning(f"Invalid order data from user {current_user.email}: missing items or table")
        return error_response(message=ErrorResponse.MISSING_FIELDS, status_code=400)

    if not isinstance(table, int):
        logger.warning(f"Invalid table type from user {current_user.email}: {type(table)}")
       
        return error_response(message=ErrorResponse.INVALID_TABLE_TYPE, status_code=400)

    logger.info(f"Processing order for table {table} with {len(items)} items")



    # Get products data
    try:
        products = await product_data_service.get_products([item.id for item in items])
        # Me aseguro de que los items y los productos esten en el mismo orden
        products = list(sorted(products, key=lambda x: x.id))
        items = list(sorted(items, key=lambda x: x.id))
        logger.info(f"Retrieved {len(products)} products from database")
        
        
    except Exception as e:
        error_msg = f"Error getting products: {e}"
        logger.error(error_msg)
        return error_response(message=error_msg, status_code=500)
    # Comparo los precios de los items con los productos
    prices = compare_prices(products, items)
    if prices:
        return success_response(data=prices, message="El estado de los productos y items coincide con el de la venta", status_code=409)

    # Add products to Fudo
    product_errors = []

    beers_for_promo = 0
    
    try:
        fudo.get_token()
        logger.info("Fudo token obtained successfully")
        
        for item, product in zip(items, products):
            try:
               
                if product.type == name_promo and current_user.name != "Guest":
                    beers_for_promo += item.quantity
                    logger.debug(f"Added {item.quantity} beers for promotion calculation")
                
                fudo_product = add_product_to_fudo(item.id, item.quantity, table)
                
                logger.info(f"Added product {item.id} to table {table} with quantity {item.quantity} in fudo")
            
                if not fudo_product: 
                    error_msg = f"Error adding product {item.id} to table {table}."
                    product_errors.append(error_msg)
                    logger.error(error_msg)
                   
                    
            except Exception as e:
                error_msg = f"Error processing product {item.id}: {e}"
                logger.error(error_msg)
                product_errors.append(error_msg)
                
                
    except Exception as e:
        error_msg = f"Error with Fudo integration: {e}"
        logger.error(error_msg)
        
        return error_response(message=error_msg, status_code=500)

    if product_errors:
        logger.error(f"Product errors occurred: {product_errors}")
        
        return error_response(error={"errors": product_errors}, message=ErrorResponse.PRODUCT_ADD_ERROR, status_code=424)


    
    # Get active sale
    try:
        active_sale_id = fudo.get_active_sale(fudo.get_table(table))
        if not active_sale_id:
            error_msg = f"No active sale found for table {table}"
            logger.error(error_msg)
            
            return error_response(message=error_msg, status_code=404)
            
        active_sale_id = active_sale_id['id']
        logger.info(f"Active sale found for table {table}: {active_sale_id}")
    except Exception as e:
        error_msg = f"Error retrieving active sale for table {table}: {e}"
        logger.error(error_msg)
        
        return error_response(message=error_msg, status_code=500)

    # Update user reward progress
    try:
        new_progress = current_user.reward_progress + beers_for_promo * 10
        user_data_service.set_reward_progress(current_user.id, new_progress)
        
        logger.info(f"Updated reward progress for user {current_user.email}: {current_user.reward_progress} -> {new_progress}")
        
        
    except Exception as e:
        error_msg = f"Error updating reward progress for user {current_user.id}: {e}"
        logger.error(error_msg)
        # Don't fail the order for this, just log it
        new_progress = current_user.reward_progress

    # Create sale record
    try:
        sale = sale_data_service.create(
            order.customerId,
            active_sale_id or uuid4().hex,
            order.totalAmount,
            order.table,
            [item.id for item in items],
            [item.quantity for item in items]
        )   
        
        if sale > 0:
            logger.info(f"Sale created successfully: ID {sale}")
        else:
            error_msg = "Failed to create sale record"
            logger.error(error_msg)
            return error_response(message=error_msg, status_code=500)
            
    except Exception as e:
        error_msg = f"Error creating sale record: {e}"
        logger.error(error_msg)
        
        return error_response(message=error_msg, status_code=500)

    logger.info(f"Order processing completed successfully for table {table}, sale ID: {sale}")
    return success_response(data={"new_progress": new_progress}, message=SuccessResponse.ORDER_SUCCESS)