# /opt/docker/dev/service_finder/backend/app/api/v1/endpoints/expenses.py
from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select, func
from app.api.deps import get_db, get_current_user
from app.models import Asset, AssetCost, OrganizationMember, SystemParameter
from app.schemas.asset_cost import AssetCostCreate
from datetime import datetime

router = APIRouter()

@router.post("/", status_code=201)
async def create_expense(
    expense: AssetCostCreate,
    db: AsyncSession = Depends(get_db),
    current_user = Depends(get_current_user)
):
    """
    Create a new expense (fuel, service, tax, insurance) for an asset.
    Uses AssetCostCreate schema which includes mileage_at_cost, cost_type, etc.
    """
    # Validate asset exists
    stmt = select(Asset).where(Asset.id == expense.asset_id)
    result = await db.execute(stmt)
    asset = result.scalar_one_or_none()
    if not asset:
        raise HTTPException(status_code=404, detail="Asset not found.")
    
    # Dynamic Gatekeeper: Check draft expense limit
    if asset.status == "draft":
        # 1. Get VEHICLE_DRAFT_MAX_EXPENSES parameter
        param_stmt = select(SystemParameter).where(
            SystemParameter.key == "VEHICLE_DRAFT_MAX_EXPENSES",
            SystemParameter.scope_level == "global"
        )
        param_result = await db.execute(param_stmt)
        param = param_result.scalar_one_or_none()
        
        if param:
            limit = param.value.get("limit", 10)  # Default to 10 if not found
        else:
            limit = 10  # Default fallback
        
        # 2. Count existing expenses for this asset
        count_stmt = select(func.count(AssetCost.id)).where(AssetCost.asset_id == asset.id)
        count_result = await db.execute(count_stmt)
        expense_count = count_result.scalar()
        
        # 3. Check if limit reached
        if expense_count >= limit:
            raise HTTPException(
                status_code=status.HTTP_403_FORBIDDEN,
                detail=f"DRAFT_LIMIT_REACHED: Draft vehicles are limited to {limit} expenses. This asset already has {expense_count} expenses."
            )
    
    # Determine organization_id: prefer explicit from payload, fallback to asset fields
    organization_id = expense.organization_id or asset.current_organization_id or asset.owner_org_id
    if not organization_id:
        # B2C fallback: if the asset is owned by a person (not an org),
        # try to find the current user's default organization
        org_stmt = select(OrganizationMember).where(
            OrganizationMember.user_id == current_user.id,
            OrganizationMember.is_verified == True
        ).limit(1)
        org_result = await db.execute(org_stmt)
        org_member = org_result.scalar_one_or_none()
        if org_member:
            organization_id = org_member.organization_id
        else:
            # Last resort: any membership (even unverified)
            org_stmt = select(OrganizationMember).where(
                OrganizationMember.user_id == current_user.id
            ).limit(1)
            org_result = await db.execute(org_stmt)
            org_member = org_result.scalar_one_or_none()
            if org_member:
                organization_id = org_member.organization_id
            else:
                raise HTTPException(status_code=400, detail="Asset has no associated organization.")
    
    # Prepare data JSON for extra fields (mileage_at_cost, description, etc.)
    data = expense.data.copy() if expense.data else {}
    if expense.mileage_at_cost is not None:
        data["mileage_at_cost"] = expense.mileage_at_cost
    if expense.description:
        data["description"] = expense.description
    
    # Create AssetCost instance
    new_cost = AssetCost(
        asset_id=expense.asset_id,
        organization_id=organization_id,
        category_id=expense.category_id,
        amount_net=expense.amount_local,
        currency=expense.currency_local,
        date=expense.date,
        invoice_number=data.get("invoice_number"),
        data=data
    )
    
    db.add(new_cost)
    await db.commit()
    await db.refresh(new_cost)
    
    return {
        "status": "success",
        "id": new_cost.id,
        "asset_id": new_cost.asset_id,
        "category_id": new_cost.category_id,
        "amount_net": new_cost.amount_net,
        "date": new_cost.date
    }