bakery-ia/services/distribution/app/api/shipments.py

"""
Shipment API endpoints for distribution service
"""

from fastapi import APIRouter, Depends, HTTPException, Query
from typing import List, Optional
from datetime import date, timedelta

from app.api.dependencies import get_distribution_service
from shared.auth.tenant_access import verify_tenant_permission_dep

router = APIRouter()


@router.get("/tenants/{tenant_id}/distribution/shipments")
async def get_shipments(
    tenant_id: str,
    date_from: Optional[date] = Query(None, description="Start date for shipment filtering"),
    date_to: Optional[date] = Query(None, description="End date for shipment filtering"),
    status: Optional[str] = Query(None, description="Filter by shipment status"),
    distribution_service: object = Depends(get_distribution_service),
    verified_tenant: str = Depends(verify_tenant_permission_dep)
):
    """
    List shipments with optional filtering
    """
    try:
        # If no date range specified, default to today
        if not date_from and not date_to:
            date_from = date.today()
            date_to = date.today()
        elif not date_to:
            date_to = date_from

        shipments = []
        current_date = date_from
        while current_date <= date_to:
            daily_shipments = await distribution_service.get_shipments_for_date(tenant_id, current_date)
            shipments.extend(daily_shipments)
            current_date = current_date + timedelta(days=1)
            
        if status:
            shipments = [s for s in shipments if s.get('status') == status]
            
        return {"shipments": shipments}
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Failed to get shipments: {str(e)}")


@router.put("/tenants/{tenant_id}/distribution/shipments/{shipment_id}/status")
async def update_shipment_status(
    tenant_id: str,
    shipment_id: str,
    status_update: dict,  # Should be a proper Pydantic model
    distribution_service: object = Depends(get_distribution_service),
    verified_tenant: str = Depends(verify_tenant_permission_dep)
):
    """
    Update shipment status
    """
    try:
        new_status = status_update.get('status')
        if not new_status:
            raise HTTPException(status_code=400, detail="Status is required")

        user_id = "temp_user_id"  # Would come from auth context
        result = await distribution_service.update_shipment_status(
            shipment_id=shipment_id,
            new_status=new_status,
            user_id=user_id,
            metadata=status_update.get('metadata')
        )
        return result
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Failed to update shipment status: {str(e)}")


@router.post("/tenants/{tenant_id}/distribution/shipments/{shipment_id}/delivery-proof")
async def upload_delivery_proof(
    tenant_id: str,
    shipment_id: str,
    delivery_proof: dict,  # Should be a proper Pydantic model
    distribution_service: object = Depends(get_distribution_service),
    verified_tenant: str = Depends(verify_tenant_permission_dep)
):
    """
    Upload delivery proof (signature, photo, etc.)
    """
    try:
        # Implementation would handle signature/photo upload
        # This is a placeholder until proper models are created
        raise HTTPException(status_code=501, detail="Delivery proof upload endpoint not yet implemented")
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Failed to upload delivery proof: {str(e)}")


@router.get("/tenants/{tenant_id}/distribution/shipments/{shipment_id}")
async def get_shipment_detail(
    tenant_id: str,
    shipment_id: str,
    distribution_service: object = Depends(get_distribution_service),
    verified_tenant: str = Depends(verify_tenant_permission_dep)
):
    """
    Get detailed information about a specific shipment
    """
    try:
        # Implementation would fetch detailed shipment information
        # This is a placeholder until repositories are created
        raise HTTPException(status_code=501, detail="Shipment detail endpoint not yet implemented")
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Failed to get shipment details: {str(e)}")