from fastapi import APIRouter, Request, HTTPException
from datetime import date
from database import session
from decimal import Decimal

# Import your models
from app.models.auto import Auto_Delivery, Tickets_Auto_Delivery
from app.models.pricing import Pricing_Oil_Oil

# Import the new estimator class
from app.script.fuel_estimator import FuelEstimator

router = APIRouter(
    prefix="/confirm",
    tags=["confirm"],
    responses={404: {"description": "Not found"}},
)


@router.put("/auto/update/{autoid}")
async def update_auto(autoid: int, request: Request):
    """
    Confirms a delivery, updates the ticket, and triggers the self-correcting
    K-Factor (house_factor) refinement.
    """
    try:
        request_body = await request.json()
        gallons_delivered = Decimal(request_body['gallons_delivered'])

        auto_customer = session.query(Auto_Delivery).filter(Auto_Delivery.id == autoid).first()
        if not auto_customer:
            raise HTTPException(status_code=404, detail="Auto Delivery profile not found")

        if auto_customer.open_ticket_id is None:
            raise HTTPException(status_code=400, detail="No open ticket found for this auto-delivery. Cannot confirm delivery.")
        
        ticket = session.query(Tickets_Auto_Delivery).filter(Tickets_Auto_Delivery.id == auto_customer.open_ticket_id).first()
        if not ticket:
             raise HTTPException(status_code=404, detail=f"Open ticket with ID {auto_customer.open_ticket_id} not found.")

        ticket.gallons_delivered = gallons_delivered
        ticket.total_amount_customer = ticket.price_per_gallon * gallons_delivered

        estimator = FuelEstimator(session=session)
        estimator.refine_factor_after_delivery(ticket=ticket)

        session.add(ticket)
        session.add(auto_customer)

        session.commit()

        return {"ok": True, "message": "Delivery confirmed and customer factor refined successfully."}

    except Exception as e:
        session.rollback()
        raise HTTPException(status_code=500, detail=f"An internal error occurred: {str(e)}")


@router.post("/auto/create/{autoid}")
async def create_auto_ticket(autoid: int, request: Request):
    """
    Creates a new delivery ticket and links it to the auto-delivery customer profile.
    """
    request_body = await request.json()
    gallons_delivered = request_body['gallons_delivered'] # Estimated gallons for pre-auth
    gallons_delivered = Decimal(gallons_delivered)
    payment_type = request_body.get('payment_type')
    payment_card_id = request_body.get('payment_card_id')
    payment_status = request_body.get('payment_status')

    get_auto_delivery = session.query(Auto_Delivery).filter(Auto_Delivery.id == autoid).first()
    get_todays_price = session.query(Pricing_Oil_Oil).order_by(Pricing_Oil_Oil.id.desc()).first()
    
    todays_price = Decimal(get_todays_price.price_for_customer)
    total_amount = gallons_delivered * todays_price

    create_new_ticket = Tickets_Auto_Delivery(
        customer_id=get_auto_delivery.customer_id,
        account_number=get_auto_delivery.account_number,
        customer_town=get_auto_delivery.customer_town,
        customer_state=get_auto_delivery.customer_state,
        customer_address=get_auto_delivery.customer_address,
        customer_zip=get_auto_delivery.customer_zip,
        customer_full_name=get_auto_delivery.customer_full_name,
        oil_prices_id=get_todays_price.id,
        gallons_delivered=gallons_delivered,
        price_per_gallon=get_todays_price.price_for_customer,
        total_amount_customer=total_amount,
        fill_date=date.today(),
        payment_type=payment_type,
        payment_card_id=payment_card_id,
        payment_status=payment_status,
    )

    session.add(create_new_ticket)
    session.flush()

    get_auto_delivery.open_ticket_id = create_new_ticket.id
    session.add(get_auto_delivery)

    session.commit()

    return ({"ok": True, "auto_ticket_id":create_new_ticket.id}), 200


@router.put("/auto/close_ticket/{ticket_id}")
async def close_ticket(ticket_id: int):
    """
    Closes an auto ticket by updating payment_status and unlinking from the auto delivery profile.
    """
    ticket = session.query(Tickets_Auto_Delivery).filter(Tickets_Auto_Delivery.id == ticket_id).first()
    if ticket:
        ticket.payment_status = 3 # Assuming 3 means closed/paid
        session.add(ticket)

    delivery = session.query(Auto_Delivery).filter(Auto_Delivery.open_ticket_id == ticket_id).first()
    if delivery:
        delivery.open_ticket_id = None
        session.add(delivery)

    session.commit()
    return {"ok": True}, 200


@router.delete("/auto/delete_ticket/{ticket_id}")
async def delete_ticket(ticket_id: int):
    """
    Deletes an auto ticket and unlinks its reference, used for cleanup (e.g., failed payment).
    """
    ticket = session.query(Tickets_Auto_Delivery).filter(Tickets_Auto_Delivery.id == ticket_id).first()
    if ticket:
        session.delete(ticket)

    delivery = session.query(Auto_Delivery).filter(Auto_Delivery.open_ticket_id == ticket_id).first()
    if delivery:
        delivery.open_ticket_id = None
        session.add(delivery)

    session.commit()
    return {"ok": True}, 200


@router.post("/auto/create/nopreauth/{autoid}")
async def create_auto_ticket_no_preauth(autoid: int, request: Request):
    """
    Creates a new delivery ticket without pre-authorization details.
    """
    request_body = await request.json()
    gallons_delivered = request_body['gallons_delivered']
    gallons_delivered = Decimal(gallons_delivered)
    payment_type = 1
    payment_card_id = 0
    payment_status = 3

    get_auto_delivery = session.query(Auto_Delivery).filter(Auto_Delivery.id == autoid).first()
    get_todays_price = session.query(Pricing_Oil_Oil).order_by(Pricing_Oil_Oil.id.desc()).first()
    
    todays_price = Decimal(get_todays_price.price_for_customer)
    total_amount = gallons_delivered * todays_price

    create_new_ticket = Tickets_Auto_Delivery(
        customer_id=get_auto_delivery.customer_id,
        account_number=get_auto_delivery.account_number,
        customer_town=get_auto_delivery.customer_town,
        customer_state=get_auto_delivery.customer_state,
        customer_address=get_auto_delivery.customer_address,
        customer_zip=get_auto_delivery.customer_zip,
        customer_full_name=get_auto_delivery.customer_full_name,
        oil_prices_id=get_todays_price.id,
        gallons_delivered=gallons_delivered,
        price_per_gallon=get_todays_price.price_for_customer,
        total_amount_customer=total_amount,
        fill_date=date.today(),
        payment_type=payment_type,
        payment_card_id=payment_card_id,
        payment_status=payment_status,
    )

    session.add(create_new_ticket)
    session.commit()
    
    return ({"ok": True, "auto_ticket_id":create_new_ticket.id}), 200