"""Rebalancing API endpoints."""
from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy.orm import Session

from app.database import get_db
from app.schemas.portfolio import (
    RebalancingRequest,
    RebalancingResponse
)
from app.services.rebalancing_service import RebalancingService, PortfolioService

router = APIRouter()


@router.post("/calculate", response_model=RebalancingResponse)
async def calculate_rebalancing(
    request: RebalancingRequest,
    db: Session = Depends(get_db)
):
    """
    리밸런싱 계산.

    Args:
        request: 리밸런싱 요청
        db: 데이터베이스 세션

    Returns:
        리밸런싱 추천
    """
    try:
        # 포트폴리오 조회
        portfolio = PortfolioService.get_portfolio(request.portfolio_id, db)

        if not portfolio:
            raise HTTPException(
                status_code=status.HTTP_404_NOT_FOUND,
                detail="포트폴리오를 찾을 수 없습니다"
            )

        # 현재 보유량을 딕셔너리로 변환
        current_holdings = {
            holding.ticker: holding.quantity
            for holding in request.current_holdings
        }

        # 리밸런싱 계산
        result = RebalancingService.calculate_rebalancing(
            portfolio_id=request.portfolio_id,
            current_holdings=current_holdings,
            cash=request.cash,
            db_session=db
        )

        # 응답 구성
        return {
            'portfolio': portfolio,
            **result
        }

    except ValueError as e:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail=str(e)
        )
    except Exception as e:
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"리밸런싱 계산 오류: {str(e)}"
        )