"""Schemas for v1 (Consumer-Driven Contract) API endpoints."""

from __future__ import annotations

from datetime import datetime
from decimal import Decimal
from typing import Any, Literal

from pydantic import BaseModel, Field


# ---------------------------------------------------------------------------
# GET /v1/assets
# ---------------------------------------------------------------------------

class AssetItem(BaseModel):
    asset_id: str
    symbol: str
    status: str
    cash: float
    quantity: float
    broker_type: str
    strategy_id: str | None = None
    strategy_type: str | None = None
    strategy_params: dict[str, Any] = {}
    updated_at: datetime | None = None
    # Additional informational fields
    name: str | None = None
    price: float | None = None
    profit_buffer: float | None = None
    tags: list[str] | None = None
    owner_user_id: str | None = None
    created_at: datetime | None = None
    last_trade_at: datetime | None = None


class AssetsResponse(BaseModel):
    assets: list[AssetItem]
    total: int


# ---------------------------------------------------------------------------
# GET /v1/user_info
# ---------------------------------------------------------------------------

class BrokerItem(BaseModel):
    broker: str
    account_ref: str
    credential_payload: dict[str, Any] | None = None
    credential_version: int = 1
    last_rotated_at: datetime | None = None


class UserInfoResponse(BaseModel):
    runtime_id: str | None = None
    daemon_id: str | None = None
    trading_enabled: bool
    disabled_reason: str | None = None
    brokers: list[BrokerItem] = []
    # Additional informational fields
    user_id: str
    username: str
    email: str | None = None
    permission: str


# ---------------------------------------------------------------------------
# POST /v1/assets/{asset_id}/ledger/trades
# ---------------------------------------------------------------------------

class LedgerTradeRequest(BaseModel):
    order_id: str = Field(..., min_length=1)
    execution_id: str = Field(..., min_length=1)
    side: Literal["buy", "sell"]
    quantity: Decimal = Field(..., gt=0)
    price: Decimal = Field(..., gt=0)
    fee_amount: Decimal = Field(default=Decimal("0"), ge=0)
    tax_amount: Decimal = Field(default=Decimal("0"), ge=0)
    executed_at: datetime


class LedgerTradeResponse(BaseModel):
    ledger_entry_id: int
    asset_id: str
    order_id: str
    execution_id: str
    side: str
    quantity: float
    price: float
    fee_amount: float
    tax_amount: float
    executed_at: datetime
    recorded_at: datetime | None = None
    status: Literal["new", "duplicate"] = "new"