"""落账逻辑：按订单类型复用现有扣款+加仓 / 减仓+入账仓储方法。

供两处调用：未命中分支（申购/赎回接口）与放行处置（release）。不 commit，
由外层事务统一提交，保证「落账 + 写流水 / 改订单状态」原子性。
"""
from __future__ import annotations

from sqlalchemy.ext.asyncio import AsyncSession

from model.trade_order import TradeOrder
from repositories.fin_account import FinAccountRepo
from repositories.fin_holdings import FinHoldingsRepo
from utils.exceptions import ParamError


async def settle(db: AsyncSession, order: TradeOrder) -> None:
    """执行落账（申购扣款+加仓 / 赎回减仓+入账）。不 commit。

    - 申购：余额不足返回 False 时抛 ParamError；
    - 赎回：份额不足返回 False 时抛 ParamError。
    """
    account_repo = FinAccountRepo(db)
    holdings_repo = FinHoldingsRepo(db)

    if order.order_type == "申购":
        if not await account_repo.deduct_balance(order.customer_id, order.amount):
            raise ParamError("可用余额不足")
        await holdings_repo.upsert(
            order.customer_id, order.product_id, order.shares, order.amount
        )
    elif order.order_type == "赎回":
        if not await holdings_repo.redeem(
            order.customer_id, order.product_id, order.shares
        ):
            raise ParamError("可赎回份额不足")
        await account_repo.credit_balance(order.customer_id, order.amount)
    else:
        raise ParamError(f"未知订单类型：{order.order_type}")