wallet/backend/app/routes/wallet.py

"""Wallet entry routes."""

from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session
from app.database import get_db
from app.models import User, WalletEntry, EntryType
from app.schemas import (
    WalletEntryCreate, 
    WalletEntryUpdate, 
    WalletEntryResponse,
    DashboardMetrics
)
from app.auth import get_current_user

router = APIRouter(prefix="/wallet", tags=["wallet"])


@router.post("/entries", response_model=WalletEntryResponse)
def create_entry(
    entry_data: WalletEntryCreate,
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """Create a new wallet entry."""
    new_entry = WalletEntry(
        user_id=current_user.id,
        amount=entry_data.amount,
        entry_type=entry_data.entry_type,
        description=entry_data.description
    )
    db.add(new_entry)
    db.commit()
    db.refresh(new_entry)
    return new_entry


@router.get("/entries", response_model=list[WalletEntryResponse])
def get_entries(
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db),
    skip: int = 0,
    limit: int = 100
):
    """Get all wallet entries for the current user."""
    entries = db.query(WalletEntry).filter(
        WalletEntry.user_id == current_user.id
    ).order_by(WalletEntry.created_at.desc()).offset(skip).limit(limit).all()
    return entries


@router.get("/entries/{entry_id}", response_model=WalletEntryResponse)
def get_entry(
    entry_id: int,
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """Get a specific wallet entry."""
    entry = db.query(WalletEntry).filter(
        (WalletEntry.id == entry_id) & (WalletEntry.user_id == current_user.id)
    ).first()
    
    if not entry:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="Entry not found"
        )
    return entry


@router.put("/entries/{entry_id}", response_model=WalletEntryResponse)
def update_entry(
    entry_id: int,
    entry_data: WalletEntryUpdate,
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """Update a wallet entry."""
    entry = db.query(WalletEntry).filter(
        (WalletEntry.id == entry_id) & (WalletEntry.user_id == current_user.id)
    ).first()
    
    if not entry:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="Entry not found"
        )
    
    if entry_data.amount is not None:
        entry.amount = entry_data.amount
    if entry_data.entry_type is not None:
        entry.entry_type = entry_data.entry_type
    if entry_data.description is not None:
        entry.description = entry_data.description
    
    db.commit()
    db.refresh(entry)
    return entry


@router.delete("/entries/{entry_id}", status_code=status.HTTP_204_NO_CONTENT)
def delete_entry(
    entry_id: int,
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """Delete a wallet entry."""
    entry = db.query(WalletEntry).filter(
        (WalletEntry.id == entry_id) & (WalletEntry.user_id == current_user.id)
    ).first()
    
    if not entry:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail="Entry not found"
        )
    
    db.delete(entry)
    db.commit()


@router.get("/metrics", response_model=DashboardMetrics)
def get_metrics(
    current_user: User = Depends(get_current_user),
    db: Session = Depends(get_db)
):
    """Get dashboard metrics for the current user."""
    entries = db.query(WalletEntry).filter(
        WalletEntry.user_id == current_user.id
    ).all()
    
    total_income = sum(e.amount for e in entries if e.entry_type == EntryType.INCOME)
    total_expenses = sum(e.amount for e in entries if e.entry_type == EntryType.EXPENSE)
    total_balance = total_income - total_expenses
    
    return {
        "total_balance": total_balance,
        "total_income": total_income,
        "total_expenses": total_expenses,
        "entry_count": len(entries)
    }