"""Portfolio repository contract."""

from __future__ import annotations

from typing import Protocol

from app.domain.models import LedgerEntry, Portfolio, StrategySet, User


class PortfolioRepository(Protocol):
    def ping(self) -> None: ...

    def get_user(self, user_id: str) -> User: ...

    def list_users(self) -> list[User]: ...

    def create_user(self, username: str, *, email: str | None = None) -> User: ...

    def update_user(self, user_id: str, *, username: str | None = None, email: str | None = None) -> User: ...

    def delete_user(self, user_id: str) -> None: ...

    def get(self, portfolio_id: str) -> Portfolio: ...

    def get_or_create(
        self,
        portfolio_id: str,
        symbol: str,
        *,
        broker: str = "kis",
        status: str = "enabled",
        active_strategy_id: str | None = None,
        avabile_strategies: list[str] | None = None,
        initial_cash: float = 0.0,
    ) -> Portfolio: ...

    def create_portfolio(
        self,
        portfolio_id: str,
        *,
        broker: str,
        status: str,
        symbol: str,
        cash_balance: float,
        asset_quantity: float,
        asset_price: float,
        profit_buffer: float,
        active_strategy_id: str | None,
        avabile_strategies: list[str] | None,
    ) -> Portfolio: ...

    def update_portfolio(self, portfolio_id: str, **fields: object) -> Portfolio: ...

    def list_portfolios(self, *, broker: str | None = None, status: str | None = None) -> list[Portfolio]: ...

    def get_or_create_strategy_set(self, name: str, config: str) -> StrategySet: ...

    def get_strategy_set(self, id: str) -> StrategySet | None: ...

    def list_strategy_sets(self) -> list[StrategySet]: ...

    def delete_strategy_set(self, id: str) -> None: ...

    def save(
        self,
        portfolio: Portfolio,
        *,
        current_price: float,
        action: str,
        trade_price: float,
        quantity: float,
        fee_amount: float,
        tax_amount: float,
        total_value_after: float,
        strategy_set_id: str = "manual",
        note: str = "",
    ) -> None: ...

    def list_recent_trades(self, portfolio_id: str, limit: int = 5) -> list[LedgerEntry]: ...

    def reset(self, portfolio_id: str) -> Portfolio: ...

    def adjust_buffer(self, portfolio_id: str, amount: float, action: str, note: str = "") -> Portfolio: ...

    def transfer_buffer(self, portfolio_id: str, amount: float, direction: str, note: str = "") -> Portfolio: ...

    def delete(self, portfolio_id: str, *, clear_ledger: bool = True) -> None: ...