rustrade-backtest 0.5.2

//! The replay engine.
//!
//! Single-threaded loop: feeds candles to the brain in order, builds an
//! order from each non-`Hold` decision, applies slippage and fees, and
//! updates a synthetic per-symbol position. On position-reducing fills
//! (closes or flips) a [`TradeOutcome`] is emitted into the result.
//!
//! # Perp funding
//!
//! With a [`FundingModel`](crate::FundingModel) configured (off by
//! default), settlements in
//! the window `(previous_candle_time, candle_time]` are booked against
//! each symbol's position *as it stood entering the candle*, before any
//! bracket fill or decision on that candle: `cashflow = −sign(qty) ×
//! rate × |qty| × contract_value × last_mark` (positive rate → longs
//! pay, shorts receive). Cashflows land in cash immediately (so the
//! equity curve carries them) and accrue against the open position; when
//! the position closes the accrual is attributed to the emitted
//! [`TradeOutcome`]s pro-rata by closed quantity. Funding does **not**
//! feed the session-PnL risk gate (parity with live, which gates on
//! trade gross − fee); it does flow into the circuit breaker's win/loss
//! classification via `TradeOutcome::net_pnl`.
//!
//! # Fill models
//!
//! [`BacktestConfig::fill_model`](crate::BacktestConfig::fill_model)
//! selects the fill semantics. The default
//! ([`FillModel::TakerAtClose`]) keeps the historical behaviour:
//! everything fills (or is dropped) on the decision candle, and
//! protective brackets fill at their fixed level. Opting in to
//! [`FillModel::Resting`] gives honest resting-order semantics —
//! non-marketable limits rest across candles (fill on cross at the
//! limit, at the open on a gap through), stops fill at the level or
//! worse, standalone stop/TP protective orders work without the other
//! leg, and same-candle ambiguity resolves pessimistically. The complete
//! convention list lives on [`FillModel`].
//!
//! # Multi-symbol replay
//!
//! Each candle is tagged with a symbol. The engine maintains independent
//! [`Position`] state per symbol but a *single* shared cash balance. When
//! [`Backtest::with_candles`] is used, every candle is assumed to be for
//! the (single) symbol on the config. For multi-symbol runs use
//! [`Backtest::with_symbol_candles`].

use std::collections::BTreeMap;
use std::sync::Arc;

use chrono::{DateTime, TimeZone, Utc};
use rustrade_core::{
    Brain, Candle, Decision, Exchange, Fill, MarketDataEvent, OrderKind, Position, Side,
    SignalType, SizeHint, Symbol,
};
use rustrade_risk::clock::ManualClock;
use rustrade_risk::{CircuitBreaker, PositionSizer, SessionPnl};

use crate::config::{BacktestConfig, FillModel};
use crate::error::{Error, Result};
use crate::fees::FeeModel;
use crate::metrics::TradeOutcome;
use crate::result::BacktestResult;
use crate::slippage::SlippageModel;

/// The replay engine itself. Configure via [`BacktestConfig`], attach a
/// [`Brain`] and one or more candle series, then `.run().await` for the
/// result.
///
/// # Example
///
/// ```no_run
/// # use std::sync::Arc;
/// use rustrade_backtest::{Backtest, BacktestConfig, load_csv};
/// # async fn run(brain: Arc<dyn rustrade_core::Brain>) -> rustrade_backtest::Result<()> {
/// let candles = load_csv("data/btcusdt-1m.csv")?;
/// let result = Backtest::new(
///     BacktestConfig::builder()
///         .symbol("BTCUSDT")
///         .initial_cash(10_000.0)
///         .build()?,
///     brain,
/// )
/// .with_candles(candles)
/// .run()
/// .await?;
///
/// println!("{}", result.summary());
/// # Ok(())
/// # }
/// ```
pub struct Backtest {
    config: BacktestConfig,
    brain: Arc<dyn Brain>,
    /// Per-symbol candle series. Within each series candles are assumed
    /// to be in chronological order; across series the engine merges
    /// chronologically before replay.
    series: Vec<(Symbol, Vec<Candle>)>,
}

impl Backtest {
    /// Construct with a config + brain. The candle series is attached
    /// separately via [`Self::with_candles`] / [`Self::with_symbol_candles`].
    pub fn new(config: BacktestConfig, brain: Arc<dyn Brain>) -> Self {
        Self {
            config,
            brain,
            series: Vec::new(),
        }
    }

    /// Feed a candle series for the (single) symbol on the config.
    /// Convenience wrapper around [`Self::with_symbol_candles`].
    ///
    /// Panics if the config has more than one symbol — use
    /// [`Self::with_symbol_candles`] for multi-symbol backtests.
    pub fn with_candles(mut self, candles: Vec<Candle>) -> Self {
        assert_eq!(
            self.config.symbols.len(),
            1,
            "Backtest::with_candles requires a single-symbol config; \
             this config has {} symbols. Use Backtest::with_symbol_candles instead.",
            self.config.symbols.len()
        );
        let symbol = self.config.symbols[0].clone();
        self.series = vec![(symbol, candles)];
        self
    }

    /// Feed a candle series for a specific symbol. Call multiple times
    /// for multi-symbol backtests; repeated calls for the same symbol
    /// replace the previous series.
    ///
    /// The engine merges all series chronologically before replay, so
    /// the brain sees the global event stream — not per-symbol blocks.
    pub fn with_symbol_candles(mut self, symbol: impl Into<Symbol>, candles: Vec<Candle>) -> Self {
        let symbol = symbol.into();
        self.series.retain(|(s, _)| s != &symbol);
        self.series.push((symbol, candles));
        self
    }

    /// Run the backtest to completion. Returns the aggregated result.
    pub async fn run(self) -> Result<BacktestResult> {
        let exchange = Exchange::from("backtest");
        let sizer = PositionSizer::new(self.config.sizing.clone());

        let merged = merge_series(&self.series);
        let candles_processed = merged.len();

        // Reject non-finite / non-positive prices up front. A single
        // `NaN` close would otherwise propagate through the equity curve
        // and silently turn every downstream metric (Sharpe, Sortino,
        // drawdown) into `NaN`. Fail loud instead.
        for (symbol, candle) in &merged {
            if let Err(why) = validate_candle(candle) {
                return Err(Error::Data(format!(
                    "{symbol} candle at t={}: {why}",
                    candle.time
                )));
            }
        }

        let mut state = State::new(
            self.config.initial_cash,
            self.config.symbols.iter().cloned(),
        );
        let mut signals_emitted = 0usize;
        let mut orders_filled = 0usize;
        let mut orders_blocked = 0usize;
        let mut trades: Vec<TradeOutcome> = Vec::new();

        // Per-symbol risk gates (session-PnL halt + circuit breaker),
        // mirroring the live execution path's gate sequence. Time is
        // driven by candle timestamps through a shared `ManualClock`, so
        // the UTC-midnight rollover and the breaker's window/cooldown run
        // on replay time and the run stays deterministic.
        let risk_clock = Arc::new(ManualClock::new(0));
        let mut risk: BTreeMap<Symbol, SymbolRisk> = BTreeMap::new();
        if self.config.session_pnl.is_some() || self.config.circuit_breaker.is_some() {
            for s in &self.config.symbols {
                risk.insert(
                    s.clone(),
                    SymbolRisk {
                        session: self
                            .config
                            .session_pnl
                            .clone()
                            .map(|c| SessionPnl::with_clock(s.as_str(), c, risk_clock.clone())),
                        breaker: self
                            .config
                            .circuit_breaker
                            .clone()
                            .map(|c| CircuitBreaker::with_clock(c, risk_clock.clone())),
                    },
                );
            }
        }

        // Re-validate + normalise the funding schedule at run time.
        // `BacktestConfig`'s fields are pub, so a schedule assigned after
        // `build()` (or a config built via struct literal) would bypass
        // the builder's validation — `settlements_between`'s binary
        // search on an unsorted series silently drops or double-books
        // settlements, corrupting funding PnL with no error. Sorting is
        // the same normalisation `build()` applies; NaN rates and
        // duplicate timestamps fail loud instead of poisoning results.
        let funding = self
            .config
            .funding
            .clone()
            .validated()
            .map_err(|why| Error::Config(format!("BacktestConfig.funding: {why}")))?;

        // Per-symbol previous candle timestamp — the funding settlement
        // window for a candle is `(prev_time, candle_time]`, matching the
        // live paper ledger's entry-exclusive / exit-inclusive accrual.
        let funding_on = funding.is_enabled();
        let mut last_seen: BTreeMap<Symbol, i64> = BTreeMap::new();

        // Honest resting-order semantics (see `FillModel::Resting`) —
        // off by default, so legacy runs take none of the paths below.
        let resting_mode = self.config.fill_model == FillModel::Resting;

        for (symbol, candle) in &merged {
            // Advance replay time and roll the risk state over before any
            // decision on this candle — the live counterpart is the
            // periodic `RiskSweepService` tick.
            if !risk.is_empty() {
                risk_clock.set(candle.time.max(0) as u64 / 1_000);
                if let Some(r) = risk.get_mut(symbol) {
                    r.tick();
                }
            }

            // ── Perp funding settlements since this symbol's last candle ────
            // Booked against the position as it stood entering the candle
            // (so a position closed ON a settlement timestamp still pays it,
            // and one opened exactly ON a settlement does not), at the
            // last-known mark. Before a symbol's first candle the position
            // is necessarily flat, so the missing window charges nothing.
            if funding_on {
                let prev = last_seen.insert(symbol.clone(), candle.time);
                if let Some(prev_t) = prev {
                    for (_, rate) in funding.settlements_between(prev_t, candle.time) {
                        state.apply_funding(symbol, rate, self.config.contract_value);
                    }
                }
            }

            // ── Resting protective brackets fill first ──────────────────────
            // Before this candle's decision, fire any SL/TP leg whose level the
            // candle's range crosses — the live counterpart is the exchange
            // filling the reduce-only OCO legs placed at entry. A stop-out loss
            // still feeds the risk gates, same as a manual close.
            {
                let before = trades.len();
                if let Some(fill) = state.check_bracket(
                    symbol,
                    candle,
                    self.config.fees,
                    self.config.slippage,
                    self.config.contract_value,
                    self.config.fill_model,
                    candle_time(candle),
                    &mut trades,
                ) {
                    if let Some(r) = risk.get_mut(symbol) {
                        record_closes_into_risk(r, &trades[before..]);
                    }
                    orders_filled += 1;
                    self.brain
                        .on_fill(&fill)
                        .await
                        .map_err(|e| Error::Brain(e.to_string()))?;
                }
            }

            // ── Resting entry orders fill second (FillModel::Resting) ───────
            // A limit placed on an earlier candle fills when this candle's
            // range reaches its level — at the limit price, or at this
            // candle's open when it gaps through. Protective exits are
            // evaluated first (above): reducing the old position before
            // opening a new one is the pessimistic ordering when both are
            // touchable in the same candle.
            if resting_mode {
                let before = trades.len();
                let fills = state.check_resting_entry(
                    symbol,
                    candle,
                    self.config.fees,
                    self.config.slippage,
                    self.config.contract_value,
                    candle_time(candle),
                    &mut trades,
                );
                if !fills.is_empty() {
                    if let Some(r) = risk.get_mut(symbol) {
                        record_closes_into_risk(r, &trades[before..]);
                    }
                    for fill in &fills {
                        orders_filled += 1;
                        self.brain
                            .on_fill(fill)
                            .await
                            .map_err(|e| Error::Brain(e.to_string()))?;
                    }
                }
            }

            let event = MarketDataEvent::Candle {
                exchange: exchange.clone(),
                symbol: symbol.clone(),
                candle: *candle,
            };

            // Brains see the live position at decision time — same as
            // the live `ExecutionService` does. For symbols not in the
            // config we still route the event through the brain (so
            // multi-symbol brains can filter) but treat the position as
            // FLAT and skip any orders.
            let position = state.position(symbol).copied().unwrap_or(Position::FLAT);
            let decision = self
                .brain
                .on_event(&event, &position)
                .await
                .map_err(|e| Error::Brain(e.to_string()))?;

            let in_config = state.has_symbol(symbol);

            if !in_config || matches!(decision.signal, SignalType::Hold) {
                state.sample_step(
                    symbol,
                    candle.close,
                    self.config.contract_value,
                    candle.time,
                );
                continue;
            }
            signals_emitted += 1;

            // ── Risk gates (parity with the live ExecutionService) ────
            // Gate 1: session-PnL halt; gate 2: circuit breaker. Both
            // block entries (`Buy`/`Sell`) after the signal is counted.
            //
            // Exit exemption (matches `ExecutionService`, rustrade #59): a
            // genuine de-risking exit — a `Close` against a non-flat
            // position — bypasses the halt/breaker checks. `resolve_order`
            // turns exactly this case into a reduce-only fill sized to the
            // current position, so it can only reduce or flatten, never
            // open/increase/flip, and can never take on new risk under a
            // halt. The predicate is intentionally tight: any `Buy`/`Sell`,
            // and a `Close` with nothing to reduce (flat), stays gated. It
            // reads the same `position` snapshot `resolve_order` sizes
            // against, so the exemption decision and the order built can
            // never diverge — the safe default is to gate unless the order
            // is provably reducing.
            let is_reduce_only_exit =
                matches!(decision.signal, SignalType::Close) && position.close_side().is_some();
            if !is_reduce_only_exit
                && let Some(r) = risk.get(symbol)
                && (r
                    .session
                    .as_ref()
                    .is_some_and(SessionPnl::is_session_halted)
                    || r.breaker.as_ref().is_some_and(CircuitBreaker::is_tripped))
            {
                orders_blocked += 1;
                state.sample_step(
                    symbol,
                    candle.close,
                    self.config.contract_value,
                    candle.time,
                );
                continue;
            }

            // Resting mode: any non-`Hold` decision that reaches here (a
            // gate-passing entry, or an exempt reduce-only exit) cancels
            // the symbol's working entry order first (cancel-and-replace;
            // the replacement may be nothing — e.g. a `Close` on a flat
            // position just pulls the resting order). Gated-off entries
            // above leave it resting.
            if resting_mode {
                state.cancel_resting(symbol);
            }

            // Translate the decision into a concrete fill request. For
            // `Close` we use the existing position size. For Buy/Sell we
            // size from the brain's hint just like ExecutionService, and
            // carry the requested order kind + limit price.
            let Some(resolved) = resolve_order(
                &decision,
                &position,
                &sizer,
                candle.close,
                self.config.contract_value,
            ) else {
                state.sample_step(
                    symbol,
                    candle.close,
                    self.config.contract_value,
                    candle.time,
                );
                continue;
            };
            if resolved.qty <= 0.0 {
                state.sample_step(
                    symbol,
                    candle.close,
                    self.config.contract_value,
                    candle.time,
                );
                continue;
            }

            // Decide the fill by order kind and fill model. Legacy
            // (`TakerAtClose`): market / IOC / FOK and closes are immediate
            // takers at the candle close; limit / post-only fill only if the
            // decision candle's range crosses the limit (a post-only that
            // would cross as taker is rejected) and are dropped otherwise —
            // nothing rests across candles. `Resting`: non-marketable limits
            // rest on the synthetic book and fill on later candles (see the
            // resting-entry block above); partial fills still need an order
            // book and remain out of scope.
            let disposition = match self.config.fill_model {
                FillModel::TakerAtClose => match resolve_fill(&resolved, candle) {
                    // Legacy keeps its exact fill semantics (no limit cap):
                    // this path is pinned bit-identical by regression tests.
                    Some((price, is_taker)) => EntryDisposition::Immediate {
                        price,
                        is_taker,
                        limit_cap: None,
                    },
                    None => EntryDisposition::Drop,
                },
                FillModel::Resting => resolve_entry_resting(&resolved, candle),
            };
            let (reference_price, is_taker, limit_cap) = match disposition {
                EntryDisposition::Immediate {
                    price,
                    is_taker,
                    limit_cap,
                } => (price, is_taker, limit_cap),
                EntryDisposition::Rest { limit } => {
                    state.set_resting(
                        symbol,
                        RestingEntry {
                            side: resolved.side,
                            qty: resolved.qty,
                            limit,
                            stop: decision.stop_price.map(|p| p.value()),
                            take_profit: decision.take_profit_price.map(|p| p.value()),
                        },
                    );
                    state.sample_step(
                        symbol,
                        candle.close,
                        self.config.contract_value,
                        candle.time,
                    );
                    continue;
                }
                EntryDisposition::Drop => {
                    state.sample_step(
                        symbol,
                        candle.close,
                        self.config.contract_value,
                        candle.time,
                    );
                    continue;
                }
            };
            // Slippage applies to taker fills (crossing the spread); a
            // resting maker fills at its limit price exactly. A marketable
            // limit / IOC / FOK carries a `limit_cap`: slippage may move the
            // fill toward the limit but never through it (buys cap at the
            // limit as a ceiling, sells as a floor) — a real venue can't
            // print a fill worse than the order's own limit.
            let fill_price = if is_taker {
                let slipped = self.config.slippage.apply(resolved.side, reference_price);
                match limit_cap {
                    Some(limit) => match resolved.side {
                        Side::Buy => slipped.min(limit),
                        Side::Sell => slipped.max(limit),
                    },
                    None => slipped,
                }
            } else {
                reference_price
            };
            let fee = self.config.fees.fee_for(
                fill_price,
                resolved.qty * self.config.contract_value,
                is_taker,
            );

            // Update position state. If this fill reduces or flips the
            // position, emit one or more TradeOutcomes.
            let trades_before = trades.len();
            apply_fill(
                &mut state,
                symbol,
                resolved.side,
                resolved.qty,
                fill_price,
                fee,
                self.config.contract_value,
                candle_time(candle),
                &mut trades,
            );

            // Feed each realised close into the risk gates — the live
            // counterpart is `FillRoutingService`'s auto-PnL recording.
            if let Some(r) = risk.get_mut(symbol) {
                record_closes_into_risk(r, &trades[trades_before..]);
            }

            orders_filled += 1;

            // Inform the brain of the (synthetic) fill — same callback
            // the live `FillRoutingService` would invoke.
            let fill = Fill {
                symbol: symbol.clone(),
                order_id: format!("bt-{orders_filled}"),
                client_id: None,
                side: resolved.side,
                price: rustrade_core::Price(fill_price),
                size: rustrade_core::Volume(resolved.qty),
                fee,
                fee_currency: "QUOTE".into(),
                timestamp: candle_time(candle),
            };
            self.brain
                .on_fill(&fill)
                .await
                .map_err(|e| Error::Brain(e.to_string()))?;

            // Record a simulated protective bracket for a freshly opened
            // entry (mirrors the live ExecutionService::place_brackets).
            // Legacy mode requires BOTH legs (a full OCO); resting mode
            // also honours standalone stop-only / TP-only protections. It
            // is fired intra-candle on later candles; cleared when flat.
            if !resolved.is_close {
                let sl = decision.stop_price.map(|p| p.value());
                let tp = decision.take_profit_price.map(|p| p.value());
                let register = if resting_mode {
                    sl.is_some() || tp.is_some()
                } else {
                    sl.is_some() && tp.is_some()
                };
                if register {
                    let pos = state.position(symbol).copied().unwrap_or(Position::FLAT);
                    if pos.qty != 0.0 {
                        let long = pos.qty > 0.0;
                        // Drop protective legs whose orientation is incoherent
                        // for the surviving position (a reducing fill can carry
                        // levels meant for a different intended position).
                        let (stop, take_profit) = coherent_bracket_legs(long, fill_price, sl, tp);
                        if stop.is_some() || take_profit.is_some() {
                            state.set_bracket(
                                symbol,
                                Bracket {
                                    stop,
                                    take_profit,
                                    long,
                                },
                            );
                        }
                    }
                }
            }

            state.sample_step(
                symbol,
                candle.close,
                self.config.contract_value,
                candle.time,
            );
        }

        let total_fees: f64 = trades.iter().map(|t| t.fee).sum();
        let trades_net: f64 = trades.iter().map(|t| t.net_pnl()).sum();
        let symbol_label = if self.config.symbols.len() == 1 {
            self.config.symbols[0].as_str().to_string()
        } else {
            // Stable, deterministic label for multi-symbol runs.
            let parts: Vec<&str> = self.config.symbols.iter().map(|s| s.as_str()).collect();
            parts.join(",")
        };

        let returns = state.into_returns();
        // Funding on positions still open at the end of the run is
        // realised cash the trade ledger never sees — fold it into the
        // headline PnL so `final_cash` agrees with the equity curve's
        // cash component. Exactly `0.0` with funding off.
        let net_pnl = trades_net + returns.open_funding;
        Ok(BacktestResult {
            symbol: symbol_label,
            initial_cash: self.config.initial_cash,
            final_cash: self.config.initial_cash + net_pnl,
            net_pnl,
            total_fees,
            funding_received: returns.funding_received,
            funding_paid: returns.funding_paid,
            candles_processed,
            signals_emitted,
            orders_filled,
            orders_blocked,
            trades,
            max_drawdown: returns.max_drawdown,
            equity_curve: returns.equity,
            equity_times: returns.equity_times,
            period_returns: returns.period_returns,
            risk_free_rate: self.config.risk_free_rate,
            periods_per_year: self.config.periods_per_year,
        })
    }
}

/// Per-symbol risk gates for replay: the same primitives the live bot
/// runs, time-driven by the shared replay clock.
struct SymbolRisk {
    session: Option<SessionPnl>,
    breaker: Option<CircuitBreaker>,
}

impl SymbolRisk {
    /// Roll both primitives forward to the current replay time.
    fn tick(&mut self) {
        if let Some(s) = &mut self.session {
            s.tick();
        }
        if let Some(b) = &mut self.breaker {
            b.tick();
        }
    }
}

// ── Series merging ──────────────────────────────────────────────────────

/// Merge per-symbol candle series into a chronological `(symbol, candle)`
/// stream. Stable for equal timestamps: ties preserve the *order the
/// series were attached*, then the order candles appear within their
/// series. This keeps multi-symbol runs deterministic even if two
/// exchanges produce identical timestamps.
fn merge_series(series: &[(Symbol, Vec<Candle>)]) -> Vec<(Symbol, Candle)> {
    let total: usize = series.iter().map(|(_, c)| c.len()).sum();
    let mut out: Vec<(Symbol, Candle, usize)> = Vec::with_capacity(total);
    for (series_idx, (sym, candles)) in series.iter().enumerate() {
        for c in candles {
            out.push((sym.clone(), *c, series_idx));
        }
    }
    // Sort by (time, series order) — stable for matching timestamps
    // within the same series.
    out.sort_by(|a, b| a.1.time.cmp(&b.1.time).then(a.2.cmp(&b.2)));
    out.into_iter().map(|(s, c, _)| (s, c)).collect()
}

// ── State + helpers ─────────────────────────────────────────────────────

/// Mutable in-loop state: per-symbol position, shared realised cash,
/// equity HWM, drawdown, and the equity / per-period returns sample
/// stream used by Sharpe / Sortino.
struct State {
    // `BTreeMap`, not `HashMap`: `equity_now` sums unrealised PnL by
    // iterating this map, and float addition is not associative. A
    // `HashMap`'s per-process-randomized iteration order would make a
    // multi-symbol equity curve (and thus Sharpe/Sortino) differ in the
    // last ULP between otherwise-identical runs, breaking the engine's
    // determinism guarantee. `BTreeMap` iterates in sorted symbol order,
    // so the summation order is fixed across runs.
    positions: BTreeMap<Symbol, Position>,
    cash: f64,
    equity_hwm: f64,
    max_drawdown: f64,
    // Sampled once per candle in `sample_step`, so Sharpe/Sortino see
    // the full price path even on Hold ticks.
    last_equity: f64,
    equity_curve: Vec<f64>,
    period_returns: Vec<f64>,
    // Cached marks per symbol (last close seen) so we can compute the
    // total portfolio equity at any sample boundary even when only one
    // symbol's price has just changed. `BTreeMap` for the same
    // determinism reason as `positions`.
    last_marks: BTreeMap<Symbol, f64>,
    // Simulated reduce-only OCO brackets (SL stop-market + TP) per open
    // position — the backtest counterpart of the live
    // `ExecutionService::place_brackets`: recorded when a bracketed market
    // entry fills, fired intra-candle, cleared when the position goes flat.
    brackets: BTreeMap<Symbol, Bracket>,
    // Resting entry orders (`FillModel::Resting` only) — at most one per
    // symbol, GTC. Placed when a non-marketable limit / post-only entry
    // reaches the synthetic book at a decision candle's close; filled by a
    // later candle crossing the level; cancelled (replaced) by the next
    // gate-passing non-`Hold` decision. `BTreeMap` for the same
    // determinism reason as `positions`.
    resting: BTreeMap<Symbol, RestingEntry>,
    // Funding cashflow accrued against each *open* position (already in
    // `cash`; tracked here so closes can attribute it to their
    // `TradeOutcome`s). Cleared when the position goes flat or flips.
    // `BTreeMap` for the same determinism reason as `positions`.
    accrued_funding: BTreeMap<Symbol, f64>,
    // Run-level funding totals: sums of the positive (received) and
    // negative (paid, stored positive) settlement cashflows.
    funding_received: f64,
    funding_paid: f64,
    // Candle timestamp of each equity sample; seeded alongside
    // `equity_curve`'s initial-cash sample with the first candle's time.
    equity_times: Vec<i64>,
}

struct ReturnsSummary {
    max_drawdown: f64,
    equity: Vec<f64>,
    equity_times: Vec<i64>,
    period_returns: Vec<f64>,
    funding_received: f64,
    funding_paid: f64,
    /// Funding accrued on positions still open at the end of the run —
    /// realised cash not attributed to any `TradeOutcome`.
    open_funding: f64,
}

impl State {
    fn new(initial_cash: f64, symbols: impl IntoIterator<Item = Symbol>) -> Self {
        let mut positions = BTreeMap::new();
        for s in symbols {
            positions.insert(s, Position::FLAT);
        }
        Self {
            positions,
            cash: initial_cash,
            equity_hwm: initial_cash,
            max_drawdown: 0.0,
            last_equity: initial_cash,
            equity_curve: vec![initial_cash],
            period_returns: Vec::new(),
            last_marks: BTreeMap::new(),
            brackets: BTreeMap::new(),
            resting: BTreeMap::new(),
            accrued_funding: BTreeMap::new(),
            funding_received: 0.0,
            funding_paid: 0.0,
            equity_times: Vec::new(),
        }
    }

    fn has_symbol(&self, sym: &Symbol) -> bool {
        self.positions.contains_key(sym)
    }

    fn position(&self, sym: &Symbol) -> Option<&Position> {
        self.positions.get(sym)
    }

    fn position_mut(&mut self, sym: &Symbol) -> &mut Position {
        self.positions.entry(sym.clone()).or_insert(Position::FLAT)
    }

    /// Record a protective OCO bracket for a freshly opened position.
    fn set_bracket(&mut self, sym: &Symbol, bracket: Bracket) {
        self.brackets.insert(sym.clone(), bracket);
    }

    /// Place a resting entry order (`FillModel::Resting` only). At most
    /// one per symbol — a new placement replaces the previous one.
    fn set_resting(&mut self, sym: &Symbol, entry: RestingEntry) {
        self.resting.insert(sym.clone(), entry);
    }

    /// Cancel the symbol's resting entry order, if any.
    fn cancel_resting(&mut self, sym: &Symbol) {
        self.resting.remove(sym);
    }

    /// Fire a resting protective bracket leg if this candle's range crosses
    /// it, closing the whole position and emitting the `TradeOutcome`.
    /// Pessimistic: when a single bar spans BOTH legs the **stop** fills
    /// first. Fill-price conventions by mode:
    ///
    /// - [`FillModel::TakerAtClose`] (legacy): both legs fill at their
    ///   fixed trigger level as takers with slippage — gaps beyond the
    ///   level are not modelled.
    /// - [`FillModel::Resting`]: the stop-market leg crosses the spread —
    ///   taker with slippage, at the level *or worse* (a candle gapping
    ///   through fills at its open). The take-profit leg is a resting
    ///   limit — maker, no slippage, at the level or at the open when the
    ///   candle gaps through it (price improvement).
    ///
    /// Returns the synthetic [`Fill`] for the brain callback, or `None` if
    /// nothing fired. Self-heals a stale bracket (flat or flipped position).
    #[allow(clippy::too_many_arguments)]
    fn check_bracket(
        &mut self,
        sym: &Symbol,
        candle: &Candle,
        fees: FeeModel,
        slippage: SlippageModel,
        contract_value: f64,
        fill_model: FillModel,
        when: DateTime<Utc>,
        trades: &mut Vec<TradeOutcome>,
    ) -> Option<Fill> {
        let bracket = self.brackets.get(sym).copied()?;
        let pos = self.positions.get(sym).copied().unwrap_or(Position::FLAT);
        // Stale guard: the position is flat, or has flipped to the other side.
        if pos.qty == 0.0 || bracket.long != (pos.qty > 0.0) {
            self.brackets.remove(sym);
            return None;
        }
        let long = bracket.long;
        let close_side = if long { Side::Sell } else { Side::Buy };
        // Leg selection, stop first (pessimistic in both modes). For a
        // long the stop sits below the market and the TP above; mirrored
        // for a short. `crossed` is the honest gap-aware fill price;
        // legacy mode overrides it with the fixed level below.
        let stop_hit = bracket.stop.and_then(|level| {
            let crossed = if long {
                cross_down(level, candle)
            } else {
                cross_up(level, candle)
            };
            crossed.map(|px| (level, px, true))
        });
        let (level, crossed, is_stop) = stop_hit.or_else(|| {
            bracket.take_profit.and_then(|level| {
                let crossed = if long {
                    cross_up(level, candle)
                } else {
                    cross_down(level, candle)
                };
                crossed.map(|px| (level, px, false))
            })
        })?;
        let (fill_price, is_taker) = match fill_model {
            // Legacy: fixed-level fill, taker + slippage on both legs.
            FillModel::TakerAtClose => (slippage.apply(close_side, level), true),
            FillModel::Resting => {
                if is_stop {
                    // Stop-market: crosses the spread at the trigger (or
                    // at the open on a gap through) — taker + slippage.
                    (slippage.apply(close_side, crossed), true)
                } else {
                    // Take-profit: a resting limit — maker, no slippage,
                    // gap through → fill at the open (price improvement).
                    (crossed, false)
                }
            }
        };
        let qty = pos.qty.abs();
        let fee = fees.fee_for(fill_price, qty * contract_value, is_taker);
        apply_fill(
            self,
            sym,
            close_side,
            qty,
            fill_price,
            fee,
            contract_value,
            when,
            trades,
        );
        self.brackets.remove(sym);
        Some(Fill {
            symbol: sym.clone(),
            order_id: "bt-bracket".to_string(),
            client_id: None,
            side: close_side,
            price: rustrade_core::Price(fill_price),
            size: rustrade_core::Volume(qty),
            fee,
            fee_currency: "QUOTE".into(),
            timestamp: when,
        })
    }

    /// Fill the symbol's resting entry order if this candle's range
    /// reaches its level (`FillModel::Resting` only). The fill is a maker
    /// — the order rested on the book — so it takes the maker fee rate
    /// and no slippage, at the limit price on a cross or at this candle's
    /// open when it gaps through the level.
    ///
    /// If the fill leaves a position open and the original decision
    /// carried protective levels, the bracket is attached — and resolved
    /// **conservatively** for this candle: the entry and its stop can
    /// both lie inside the candle's range with the OHLC path unknown, so
    /// the stop is assumed to have been hit *after* the entry filled
    /// (immediate stop-out, the worse outcome). The take-profit gets no
    /// such benefit of the doubt — its level may have printed *before*
    /// the entry existed — so it only becomes eligible from the next
    /// candle.
    ///
    /// Returns the emitted fills in order (entry, then a possible
    /// same-candle stop-out); empty when nothing fired.
    #[allow(clippy::too_many_arguments)]
    fn check_resting_entry(
        &mut self,
        sym: &Symbol,
        candle: &Candle,
        fees: FeeModel,
        slippage: SlippageModel,
        contract_value: f64,
        when: DateTime<Utc>,
        trades: &mut Vec<TradeOutcome>,
    ) -> Vec<Fill> {
        let Some(entry) = self.resting.get(sym).copied() else {
            return Vec::new();
        };
        let crossed = match entry.side {
            Side::Buy => cross_down(entry.limit, candle),
            Side::Sell => cross_up(entry.limit, candle),
        };
        let Some(entry_price) = crossed else {
            return Vec::new();
        };
        self.resting.remove(sym);

        let entry_fee = fees.fee_for(entry_price, entry.qty * contract_value, false);
        apply_fill(
            self,
            sym,
            entry.side,
            entry.qty,
            entry_price,
            entry_fee,
            contract_value,
            when,
            trades,
        );
        let mut fills = vec![Fill {
            symbol: sym.clone(),
            order_id: "bt-resting".to_string(),
            client_id: None,
            side: entry.side,
            price: rustrade_core::Price(entry_price),
            size: rustrade_core::Volume(entry.qty),
            fee: entry_fee,
            fee_currency: "QUOTE".into(),
            timestamp: when,
        }];

        // Post-fill position. This fill may have reduced, flattened, or
        // flipped a position that still carries a protective bracket from
        // its own (now-dead) entry. A bracket left over from a position
        // this fill flattened or flipped is stale — a real venue would
        // have cancelled its reduce-only legs the moment the book went
        // flat or crossed to the other side — so drop it before attaching
        // (or not) this entry's own protection. Without this, a dead
        // position's legs fire against the brand-new position on a later
        // candle (the exact phantom order state the honest model removes).
        let pos = self.positions.get(sym).copied().unwrap_or(Position::FLAT);
        if let Some(existing) = self.brackets.get(sym).copied() {
            let flat = pos.qty == 0.0;
            let flipped = existing.long != (pos.qty > 0.0);
            if flat || flipped {
                self.brackets.remove(sym);
            }
        }

        // Attach the decision's protective bracket to whatever position
        // the fill leaves open (mirrors the immediate-fill path; partial
        // stop-only / TP-only brackets are honoured in resting mode).
        if entry.stop.is_none() && entry.take_profit.is_none() {
            return fills;
        }
        if pos.qty == 0.0 {
            return fills;
        }
        let long = pos.qty > 0.0;
        // Drop protective legs whose orientation is incoherent for the
        // surviving position (a reducing fill can carry levels meant for a
        // different intended position); an incoherent leg would fire almost
        // instantly against the survivor.
        let (stop, take_profit) =
            coherent_bracket_legs(long, entry_price, entry.stop, entry.take_profit);
        if stop.is_none() && take_profit.is_none() {
            return fills;
        }
        self.set_bracket(
            sym,
            Bracket {
                stop,
                take_profit,
                long,
            },
        );

        // Conservative same-candle stop check (see the method docs): the
        // freshly attached stop may fire on the very candle that filled
        // the entry; the TP may not.
        let Some(stop) = stop else {
            return fills;
        };
        let stop_crossed = if long {
            cross_down(stop, candle)
        } else {
            cross_up(stop, candle)
        };
        let Some(stop_px) = stop_crossed else {
            return fills;
        };
        let close_side = if long { Side::Sell } else { Side::Buy };
        let fill_price = slippage.apply(close_side, stop_px);
        let qty = pos.qty.abs();
        let stop_fee = fees.fee_for(fill_price, qty * contract_value, true);
        apply_fill(
            self,
            sym,
            close_side,
            qty,
            fill_price,
            stop_fee,
            contract_value,
            when,
            trades,
        );
        self.brackets.remove(sym);
        fills.push(Fill {
            symbol: sym.clone(),
            order_id: "bt-bracket".to_string(),
            client_id: None,
            side: close_side,
            price: rustrade_core::Price(fill_price),
            size: rustrade_core::Volume(qty),
            fee: stop_fee,
            fee_currency: "QUOTE".into(),
            timestamp: when,
        });
        fills
    }

    /// Book one funding settlement against a symbol's open position: the
    /// position pays `sign(qty) × rate × notional` (so a positive rate
    /// debits a long and credits a short), with notional valued at the
    /// last-known mark — the price as of the candle *before* the
    /// settlement, the backtest counterpart of the live ledger's
    /// entry-notional approximation. No-op on flat positions and on
    /// symbols with no mark yet (a position can't predate its first
    /// candle). The cashflow lands in `cash` immediately and accrues
    /// against the position for later trade attribution.
    fn apply_funding(&mut self, sym: &Symbol, rate: f64, contract_value: f64) {
        let Some(pos) = self.positions.get(sym) else {
            return;
        };
        if pos.qty == 0.0 {
            return;
        }
        let Some(&mark) = self.last_marks.get(sym) else {
            return;
        };
        let notional = pos.qty.abs() * contract_value * mark;
        let cashflow = -pos.qty.signum() * rate * notional;
        self.cash += cashflow;
        *self.accrued_funding.entry(sym.clone()).or_insert(0.0) += cashflow;
        if cashflow >= 0.0 {
            self.funding_received += cashflow;
        } else {
            self.funding_paid -= cashflow;
        }
    }

    /// Record the latest close for a symbol, then sample portfolio
    /// equity. The equity curve always grows by one sample per candle —
    /// even on `Hold` ticks — so Sharpe/Sortino see the full price path.
    fn sample_step(&mut self, sym: &Symbol, close: f64, contract_value: f64, time: i64) {
        self.last_marks.insert(sym.clone(), close);
        let equity = self.equity_now(contract_value);

        // Timestamp bookkeeping: the seed sample (initial cash, pushed in
        // `State::new` before any candle is known) borrows the first
        // candle's timestamp.
        if self.equity_times.is_empty() {
            self.equity_times.push(time);
        }
        self.equity_times.push(time);

        // Drawdown bookkeeping.
        if equity > self.equity_hwm {
            self.equity_hwm = equity;
        }
        let dd = equity - self.equity_hwm;
        if dd < self.max_drawdown {
            self.max_drawdown = dd;
        }

        self.equity_curve.push(equity);
        // Per-period simple return on prior equity (skip the first
        // sample — no prior period). Use prior equity to avoid divide-
        // by-zero on a fully-drained account.
        let prev = self.last_equity;
        if prev > 0.0 {
            self.period_returns.push((equity - prev) / prev);
        } else {
            self.period_returns.push(0.0);
        }
        self.last_equity = equity;
    }

    /// Total portfolio equity = realised cash + sum of unrealised PnL
    /// across all symbols using their last-known marks.
    fn equity_now(&self, contract_value: f64) -> f64 {
        let mut equity = self.cash;
        for (sym, pos) in &self.positions {
            if let Some(entry) = pos.entry_price
                && let Some(mark) = self.last_marks.get(sym)
            {
                let pnl_per_unit = (mark - entry) * pos.qty.signum();
                equity += pnl_per_unit * pos.qty.abs() * contract_value;
            }
        }
        equity
    }

    fn into_returns(self) -> ReturnsSummary {
        // `BTreeMap` iteration order keeps this sum deterministic.
        let open_funding: f64 = self.accrued_funding.values().sum();
        ReturnsSummary {
            max_drawdown: self.max_drawdown,
            equity: self.equity_curve,
            equity_times: self.equity_times,
            period_returns: self.period_returns,
            funding_received: self.funding_received,
            funding_paid: self.funding_paid,
            open_funding,
        }
    }
}

/// Validate a candle's OHLCV fields are usable: prices finite and
/// strictly positive, volume finite and non-negative. Returns a
/// human-readable reason on the first offending field.
///
/// OHLC *ordering* (`high >= low`, etc.) is intentionally not enforced —
/// the goal is to keep `NaN`/`inf`/negative values out of the
/// mark-to-market math, not to police exchange data quality.
pub(crate) fn validate_candle(c: &Candle) -> std::result::Result<(), String> {
    for (name, v) in [
        ("open", c.open),
        ("high", c.high),
        ("low", c.low),
        ("close", c.close),
    ] {
        if !v.is_finite() || v <= 0.0 {
            return Err(format!("{name}={v} (prices must be finite and > 0)"));
        }
    }
    if !c.volume.is_finite() || c.volume < 0.0 {
        return Err(format!("volume={} (must be finite and >= 0)", c.volume));
    }
    Ok(())
}

/// A decision resolved into a concrete fill request for the engine.
struct ResolvedOrder {
    side: Side,
    qty: f64,
    is_close: bool,
    kind: OrderKind,
    /// Limit price (quote currency) for resting kinds; `None` for market
    /// and close orders.
    limit_price: Option<f64>,
}

/// A simulated reduce-only bracket attached to an open position — the
/// backtest counterpart of the live `ExecutionService::place_brackets` SL
/// (stop-market) + TP legs. Whichever leg the candle range crosses first
/// closes the whole position (one-cancels-the-other). Legacy mode always
/// records both legs; [`FillModel::Resting`] also honours standalone
/// stop-only / TP-only protections.
#[derive(Clone, Copy)]
struct Bracket {
    /// Stop-loss trigger price, if a stop leg exists.
    stop: Option<f64>,
    /// Take-profit trigger price, if a TP leg exists.
    take_profit: Option<f64>,
    /// `true` when protecting a long (stop below entry, TP above); `false`
    /// for a short (stop above, TP below).
    long: bool,
}

/// A resting entry order on the synthetic book (`FillModel::Resting`
/// only): a non-marketable limit / post-only entry waiting for a later
/// candle to cross its level. Quantity is fixed at placement (sized at
/// the decision candle's close, like the immediate path). Carries the
/// decision's protective levels so the bracket attaches when the entry
/// eventually fills.
#[derive(Clone, Copy)]
struct RestingEntry {
    /// Direction of the entry.
    side: Side,
    /// Contracts to fill (whole-order fills only — no book depth).
    qty: f64,
    /// Resting limit level in quote currency.
    limit: f64,
    /// Stop-loss level to attach on fill, if the decision carried one.
    stop: Option<f64>,
    /// Take-profit level to attach on fill, if the decision carried one.
    take_profit: Option<f64>,
}

// ── Honest level-cross fills ────────────────────────────────────────────
//
// The two primitives behind every resting fill. A level *below* the
// market path is a buy limit or a sell stop; a level *above* is a sell
// limit or a buy stop. Both fire on the same trigger their live
// counterparts use; the honest fill price is the level itself when the
// candle trades through it, or the candle's **open** when the candle
// already opens at/through the level (a gap — the level never traded, so
// the open was the first price the market offered). This one formula
// yields price *improvement* for gapped limit fills and *worse-than-level*
// fills for gapped stops, exactly as a real book would.

/// Fill price for a resting order at `level` when the market crosses
/// **down** through it (buy limit below the market, or a long's sell
/// stop). `None` when the candle's range never reaches the level.
fn cross_down(level: f64, candle: &Candle) -> Option<f64> {
    (candle.low <= level).then(|| candle.open.min(level))
}

/// Fill price for a resting order at `level` when the market crosses
/// **up** through it (sell limit above the market, or a short's buy
/// stop). `None` when the candle's range never reaches the level.
fn cross_up(level: f64, candle: &Candle) -> Option<f64> {
    (candle.high >= level).then(|| candle.open.max(level))
}

/// Keep only the protective legs whose price sits on the correct side of
/// the fill for the surviving position's direction: a long's stop must be
/// below the entry and its take-profit above; a short is mirrored. A
/// reducing or flipping fill can leave a bracket built from levels meant
/// for a *different* intended position (e.g. a long that emits a partial
/// `Sell` carrying a stop above the market for the short it wanted); an
/// incoherent leg would fire almost immediately against the survivor. Such
/// legs are dropped. Returns the coherent `(stop, take_profit)`; either or
/// both may be `None`, in which case the caller registers no bracket.
fn coherent_bracket_legs(
    long: bool,
    entry_price: f64,
    stop: Option<f64>,
    take_profit: Option<f64>,
) -> (Option<f64>, Option<f64>) {
    let stop = stop.filter(|&s| {
        if long {
            s < entry_price
        } else {
            s > entry_price
        }
    });
    let take_profit = take_profit.filter(|&t| {
        if long {
            t > entry_price
        } else {
            t < entry_price
        }
    });
    (stop, take_profit)
}

/// Feed each realised close into a symbol's risk gates — the session-PnL
/// halt and the consecutive-loss circuit breaker — exactly as the live
/// `FillRoutingService` does. Shared by the manual-fill and bracket-fill
/// paths so the two can never diverge.
fn record_closes_into_risk(r: &mut SymbolRisk, closes: &[TradeOutcome]) {
    for t in closes {
        if let Some(session) = &mut r.session {
            session.record_close(t.gross_pnl, t.fee);
        }
        if let Some(breaker) = &mut r.breaker {
            let net = t.net_pnl();
            if net > 0.0 {
                breaker.record_win();
            } else if net < 0.0 {
                breaker.record_loss();
            }
        }
    }
}

/// Resolve a `Decision` into a [`ResolvedOrder`].
fn resolve_order(
    decision: &Decision,
    position: &Position,
    sizer: &PositionSizer,
    price: f64,
    contract_value: f64,
) -> Option<ResolvedOrder> {
    match decision.signal {
        SignalType::Hold => None,
        SignalType::Close => {
            let close_side = position.close_side()?;
            Some(ResolvedOrder {
                side: close_side,
                qty: position.qty.abs(),
                is_close: true,
                kind: OrderKind::Market,
                limit_price: None,
            })
        }
        SignalType::Buy | SignalType::Sell => {
            let side = if matches!(decision.signal, SignalType::Buy) {
                Side::Buy
            } else {
                Side::Sell
            };
            let contracts = size_from_hint(sizer, decision.size_hint, price, contract_value);
            if contracts == 0 {
                None
            } else {
                Some(ResolvedOrder {
                    side,
                    qty: contracts as f64,
                    is_close: false,
                    kind: decision.order_kind,
                    limit_price: decision.limit_price.map(|p| p.value()),
                })
            }
        }
    }
}

/// What happens to a resolved entry order on its decision candle.
#[derive(Clone, Copy)]
enum EntryDisposition {
    /// Fill now at `price` (`is_taker` selects the fee rate and whether
    /// slippage applies). `limit_cap` bounds the post-slippage taker fill
    /// at the order's own limit — `Some(limit)` for a marketable limit /
    /// IOC / FOK (slippage can never print a fill through the limit a real
    /// venue would honour), `None` for a market order (no protective
    /// price, slippage applies uncapped).
    Immediate {
        price: f64,
        is_taker: bool,
        limit_cap: Option<f64>,
    },
    /// Rest on the synthetic book at `limit` (`FillModel::Resting` only).
    Rest { limit: f64 },
    /// Dropped without a fill: a marketable post-only, a non-marketable
    /// IOC/FOK (resting mode), or a limit the decision candle never
    /// crossed (legacy mode).
    Drop,
}

/// Resolve an entry order under [`FillModel::Resting`]. The order reaches
/// the book at the decision candle's **close** — the market price at
/// decision time — so marketability is judged against the close and the
/// decision candle itself can never fill a resting order (its range
/// printed before the order existed; no lookahead).
///
/// - Market orders and closes: taker at the close, same as legacy.
/// - Limit: marketable (buy limit ≥ close / sell limit ≤ close) crosses
///   immediately at the close as taker; otherwise rests at the limit.
/// - PostOnly: rejected when marketable; otherwise rests (fills are
///   always maker).
/// - IOC / FOK: taker at the close when marketable, else cancelled. The
///   engine has no book depth, so both fill in full or not at all and
///   behave identically.
/// - A missing limit price falls back to the close (mirroring the live
///   execution layer's event-price fallback), which makes the order
///   immediately marketable.
fn resolve_entry_resting(resolved: &ResolvedOrder, candle: &Candle) -> EntryDisposition {
    // A market order (or a close) has no protective limit: it crosses at
    // the close as a taker and slippage applies uncapped.
    if resolved.is_close || matches!(resolved.kind, OrderKind::Market) {
        return EntryDisposition::Immediate {
            price: candle.close,
            is_taker: true,
            limit_cap: None,
        };
    }
    let limit = resolved.limit_price.unwrap_or(candle.close);
    let marketable = match resolved.side {
        Side::Buy => limit >= candle.close,
        Side::Sell => limit <= candle.close,
    };
    // A marketable limit / IOC / FOK crosses now at the close, but its own
    // limit caps the taker fill — slippage can push the reference toward
    // the limit yet never through it (a real venue fills at the limit or
    // rests the remainder; it can't print worse than the limit).
    let taker_capped = EntryDisposition::Immediate {
        price: candle.close,
        is_taker: true,
        limit_cap: Some(limit),
    };
    match resolved.kind {
        OrderKind::Ioc | OrderKind::Fok => {
            if marketable {
                taker_capped
            } else {
                EntryDisposition::Drop
            }
        }
        OrderKind::PostOnly => {
            if marketable {
                EntryDisposition::Drop
            } else {
                EntryDisposition::Rest { limit }
            }
        }
        OrderKind::Limit | OrderKind::Market => {
            if marketable {
                taker_capped
            } else {
                EntryDisposition::Rest { limit }
            }
        }
    }
}

/// Decide whether and at what reference price a resolved order fills on
/// `candle` — the legacy [`FillModel::TakerAtClose`] path. Returns
/// `(reference_price, is_taker)`; the caller applies slippage to taker
/// fills. Returns `None` when a resting limit doesn't cross this candle
/// (or a post-only would cross as taker).
///
/// - **Market / IOC / FOK and closes** fill immediately at the candle
///   close as takers.
/// - **Limit / PostOnly** rest: a buy fills when `low` trades down to the
///   limit, a sell when `high` trades up to it. A limit already marketable
///   at the candle open fills at the open as a taker (a post-only in that
///   case is rejected); a non-marketable limit fills at its limit price as
///   a maker. A missing limit falls back to the close, mirroring the live
///   execution layer's event-price fallback.
fn resolve_fill(resolved: &ResolvedOrder, candle: &Candle) -> Option<(f64, bool)> {
    if resolved.is_close
        || matches!(
            resolved.kind,
            OrderKind::Market | OrderKind::Ioc | OrderKind::Fok
        )
    {
        return Some((candle.close, true));
    }

    let limit = resolved.limit_price.unwrap_or(candle.close);
    let (fills, price, marketable) = match resolved.side {
        Side::Buy => (
            candle.low <= limit,
            limit.min(candle.open),
            limit >= candle.open,
        ),
        Side::Sell => (
            candle.high >= limit,
            limit.max(candle.open),
            limit <= candle.open,
        ),
    };
    if !fills {
        return None;
    }
    if matches!(resolved.kind, OrderKind::PostOnly) && marketable {
        return None;
    }
    Some((price, marketable))
}

fn size_from_hint(sizer: &PositionSizer, hint: SizeHint, price: f64, contract_value: f64) -> u32 {
    match hint {
        SizeHint::Default => sizer.contracts(price, contract_value),
        SizeHint::MarginFraction(f) => {
            let f = f.clamp(0.0, 1.0);
            let margin = sizer.config().margin_per_trade * f;
            sizer.contracts_with_margin(margin, price, contract_value)
        }
        SizeHint::NotionalUsd(n) => {
            let leverage = sizer.config().leverage.max(1);
            let margin = n / f64::from(leverage);
            sizer.contracts_with_margin(margin, price, contract_value)
        }
        SizeHint::Quantity(q) => {
            let raw = q.value().max(0.0).floor() as u32;
            raw.min(sizer.config().max_contracts)
        }
    }
}

/// Apply a fill to the synthetic position. Emits one [`TradeOutcome`]
/// per closed quantity (so a flip from +5 to -3 emits one close-5 trade).
#[allow(clippy::too_many_arguments)]
fn apply_fill(
    state: &mut State,
    symbol: &Symbol,
    side: Side,
    qty: f64,
    fill_price: f64,
    fee: f64,
    contract_value: f64,
    when: DateTime<Utc>,
    trades: &mut Vec<TradeOutcome>,
) {
    // Signed delta to the position quantity from this fill.
    let signed_qty = match side {
        Side::Buy => qty,
        Side::Sell => -qty,
    };

    let (old_qty, old_entry) = {
        let p = state.position_mut(symbol);
        (p.qty, p.entry_price)
    };
    let new_qty = old_qty + signed_qty;

    // The realised-PnL portion is whatever quantity *reduces* the
    // existing position. Anything beyond that opens a new position in
    // the opposite direction.
    let closing_qty = if old_qty.signum() != signed_qty.signum() && old_qty != 0.0 {
        old_qty.abs().min(qty)
    } else {
        0.0
    };
    let opening_qty = qty - closing_qty;

    if closing_qty > 0.0 {
        let entry = old_entry.unwrap_or(fill_price);
        let direction = old_qty.signum();
        let gross = (fill_price - entry) * direction * closing_qty * contract_value;
        // Fee is apportioned by closing fraction so a single fill that
        // both closes and reopens charges fees pro-rata to each side.
        let fee_share = if qty > 0.0 {
            fee * (closing_qty / qty)
        } else {
            0.0
        };
        // Attribute the position's accrued funding to this close,
        // pro-rata by closed quantity. The cash itself was booked at
        // each settlement — this only moves the attribution onto the
        // trade ledger. A full close (or flip) takes the whole accrual,
        // so no float-division dust is left behind.
        let accrued = state.accrued_funding.get(symbol).copied().unwrap_or(0.0);
        let closes_all = closing_qty >= old_qty.abs();
        let funding_share = if closes_all {
            accrued
        } else {
            accrued * (closing_qty / old_qty.abs())
        };
        if closes_all {
            state.accrued_funding.remove(symbol);
        } else if let Some(a) = state.accrued_funding.get_mut(symbol) {
            *a -= funding_share;
        }
        trades.push(TradeOutcome {
            symbol: symbol.as_str().to_string(),
            close_side: side,
            qty: closing_qty,
            entry_price: entry,
            exit_price: fill_price,
            gross_pnl: gross,
            fee: fee_share,
            funding: funding_share,
            closed_at: when,
        });
        state.cash += gross - fee_share;
    }

    let new_position = if opening_qty > 0.0 {
        // The fee component charged to opening.
        let fee_open = if qty > 0.0 {
            fee * (opening_qty / qty)
        } else {
            0.0
        };
        state.cash -= fee_open;
        // New entry price: if we were FLAT or fully closed first, this
        // is the fresh entry; if we'd somehow added to an existing
        // position (same-side fill), it's a weighted average. Brains
        // don't pyramid in normal use since they emit one direction at
        // a time, but handle it correctly anyway.
        let new_position_qty_after_close = old_qty + side_sign(side) * closing_qty;
        let post_open_qty = new_position_qty_after_close + side_sign(side) * opening_qty;
        let entry = if new_position_qty_after_close == 0.0 {
            fill_price
        } else {
            let prev_entry = old_entry.unwrap_or(fill_price);
            let prev_notional = prev_entry * new_position_qty_after_close.abs();
            let new_notional = fill_price * opening_qty;
            (prev_notional + new_notional) / post_open_qty.abs()
        };
        Position {
            qty: post_open_qty,
            entry_price: Some(entry),
            unrealised_pnl: 0.0,
        }
    } else if new_qty == 0.0 {
        Position::FLAT
    } else {
        Position {
            qty: new_qty,
            entry_price: old_entry,
            unrealised_pnl: 0.0,
        }
    };
    *state.position_mut(symbol) = new_position;
}

fn side_sign(side: Side) -> f64 {
    match side {
        Side::Buy => 1.0,
        Side::Sell => -1.0,
    }
}

fn candle_time(c: &Candle) -> DateTime<Utc> {
    Utc.timestamp_millis_opt(c.time)
        .single()
        .unwrap_or_else(Utc::now)
}

#[cfg(test)]
mod tests {
    use super::*;
    use async_trait::async_trait;
    use rustrade_core::{BrainHealth, Decision, MarketDataEvent, Position, Result as CoreResult};
    use rustrade_risk::SizingConfig;

    /// Brain that always emits the configured signal.
    struct FixedBrain {
        signal: SignalType,
    }
    #[async_trait]
    impl Brain for FixedBrain {
        fn name(&self) -> &str {
            "fixed"
        }
        async fn on_event(&self, _e: &MarketDataEvent, _p: &Position) -> CoreResult<Decision> {
            Ok(match self.signal {
                SignalType::Hold => Decision::hold(),
                SignalType::Buy => Decision::buy(1.0),
                SignalType::Sell => Decision::sell(1.0),
                SignalType::Close => Decision::close(),
            })
        }
        async fn health(&self) -> BrainHealth {
            BrainHealth::ok()
        }
    }

    fn flat_series(n: usize, price: f64) -> Vec<Candle> {
        (0..n)
            .map(|i| Candle {
                time: i as i64 * 60_000,
                open: price,
                high: price,
                low: price,
                close: price,
                volume: 1.0,
            })
            .collect()
    }

    fn ramp_series(n: usize, start: f64, step: f64) -> Vec<Candle> {
        (0..n)
            .map(|i| {
                let p = start + step * i as f64;
                Candle {
                    time: i as i64 * 60_000,
                    open: p,
                    high: p,
                    low: p,
                    close: p,
                    volume: 1.0,
                }
            })
            .collect()
    }

    fn cfg() -> BacktestConfig {
        BacktestConfig::builder()
            .symbol("BTCUSDT")
            .initial_cash(10_000.0)
            .sizing(SizingConfig {
                margin_per_trade: 1_000.0,
                leverage: 1,
                max_contracts: 100,
            })
            .build()
            .unwrap()
    }

    #[tokio::test]
    async fn hold_brain_produces_no_trades() {
        let result = Backtest::new(
            cfg(),
            Arc::new(FixedBrain {
                signal: SignalType::Hold,
            }),
        )
        .with_candles(flat_series(50, 100.0))
        .run()
        .await
        .unwrap();
        assert_eq!(result.signals_emitted, 0);
        assert_eq!(result.orders_filled, 0);
        assert_eq!(result.trades.len(), 0);
        assert_eq!(result.net_pnl, 0.0);
        assert_eq!(result.candles_processed, 50);
        // Equity curve always seeds the initial cash, then one sample
        // per candle.
        assert_eq!(result.equity_curve.len(), 51);
        assert_eq!(result.period_returns.len(), 50);
    }

    #[tokio::test]
    async fn buy_then_close_realises_pnl_on_uptrend() {
        // Buy on every candle. Position opens once; subsequent buys add
        // to it (pyramiding). Test just runs to completion and asserts
        // we accumulated *some* position and saw no trade close yet.
        let result = Backtest::new(
            cfg(),
            Arc::new(FixedBrain {
                signal: SignalType::Buy,
            }),
        )
        .with_candles(ramp_series(20, 100.0, 1.0))
        .run()
        .await
        .unwrap();
        // Every candle emits Buy → orders_filled equals candle count
        // (sizer always returns ≥ 1 contract here).
        assert_eq!(result.orders_filled, 20);
        // No reducing fills yet → no completed trades.
        assert_eq!(result.trades.len(), 0);
        assert_eq!(result.net_pnl, 0.0);
    }

    #[tokio::test]
    async fn determinism_two_runs_same_inputs() {
        let series = ramp_series(30, 100.0, 0.5);
        let r1 = Backtest::new(
            cfg(),
            Arc::new(FixedBrain {
                signal: SignalType::Buy,
            }),
        )
        .with_candles(series.clone())
        .run()
        .await
        .unwrap();
        let r2 = Backtest::new(
            cfg(),
            Arc::new(FixedBrain {
                signal: SignalType::Buy,
            }),
        )
        .with_candles(series)
        .run()
        .await
        .unwrap();
        assert_eq!(r1.candles_processed, r2.candles_processed);
        assert_eq!(r1.signals_emitted, r2.signals_emitted);
        assert_eq!(r1.orders_filled, r2.orders_filled);
        assert_eq!(r1.trades.len(), r2.trades.len());
        assert!((r1.net_pnl - r2.net_pnl).abs() < 1e-12);
        assert_eq!(r1.equity_curve, r2.equity_curve);
    }

    #[tokio::test]
    async fn close_against_flat_is_noop() {
        let result = Backtest::new(
            cfg(),
            Arc::new(FixedBrain {
                signal: SignalType::Close,
            }),
        )
        .with_candles(flat_series(10, 100.0))
        .run()
        .await
        .unwrap();
        assert_eq!(result.orders_filled, 0);
        assert_eq!(result.trades.len(), 0);
    }

    #[test]
    fn merge_series_interleaves_by_timestamp() {
        let s1 = Symbol::from("AAA");
        let s2 = Symbol::from("BBB");
        let series = vec![
            (
                s1.clone(),
                vec![
                    Candle {
                        time: 1000,
                        open: 1.0,
                        high: 1.0,
                        low: 1.0,
                        close: 1.0,
                        volume: 0.0,
                    },
                    Candle {
                        time: 3000,
                        open: 1.0,
                        high: 1.0,
                        low: 1.0,
                        close: 1.0,
                        volume: 0.0,
                    },
                ],
            ),
            (
                s2.clone(),
                vec![
                    Candle {
                        time: 2000,
                        open: 2.0,
                        high: 2.0,
                        low: 2.0,
                        close: 2.0,
                        volume: 0.0,
                    },
                    Candle {
                        time: 3000,
                        open: 2.0,
                        high: 2.0,
                        low: 2.0,
                        close: 2.0,
                        volume: 0.0,
                    },
                ],
            ),
        ];
        let merged = merge_series(&series);
        let times: Vec<i64> = merged.iter().map(|(_, c)| c.time).collect();
        assert_eq!(times, vec![1000, 2000, 3000, 3000]);
        // Tie at t=3000 is broken by series-insertion order — AAA first.
        assert_eq!(merged[2].0, s1);
        assert_eq!(merged[3].0, s2);
    }

    #[tokio::test]
    async fn multi_symbol_routes_to_each_symbol_state() {
        // Brain that goes long on AAA and short on BBB; FlipBrain takes
        // both sides simultaneously to verify per-symbol position state.
        struct SymBrain;
        #[async_trait]
        impl Brain for SymBrain {
            fn name(&self) -> &str {
                "sym"
            }
            async fn on_event(&self, e: &MarketDataEvent, _p: &Position) -> CoreResult<Decision> {
                match e.symbol().as_str() {
                    "AAA" => Ok(Decision::buy(1.0)),
                    "BBB" => Ok(Decision::sell(1.0)),
                    _ => Ok(Decision::hold()),
                }
            }
            async fn health(&self) -> BrainHealth {
                BrainHealth::ok()
            }
        }

        let cfg = BacktestConfig::builder()
            .symbols(["AAA", "BBB"])
            .initial_cash(100_000.0)
            .sizing(SizingConfig {
                margin_per_trade: 1_000.0,
                leverage: 1,
                max_contracts: 100,
            })
            .build()
            .unwrap();
        let result = Backtest::new(cfg, Arc::new(SymBrain))
            .with_symbol_candles("AAA", flat_series(5, 100.0))
            .with_symbol_candles("BBB", flat_series(5, 200.0))
            .run()
            .await
            .unwrap();
        // 5 AAA + 5 BBB = 10 orders (every candle, both symbols).
        assert_eq!(result.candles_processed, 10);
        assert_eq!(result.orders_filled, 10);
        // No closes — no completed trades yet.
        assert_eq!(result.trades.len(), 0);
        // Symbol label is the concatenated list.
        assert_eq!(result.symbol, "AAA,BBB");
    }

    // ── Honest level-cross helpers ──────────────────────────────────────

    fn range_candle(open: f64, high: f64, low: f64, close: f64) -> Candle {
        Candle {
            time: 0,
            open,
            high,
            low,
            close,
            volume: 1.0,
        }
    }

    #[test]
    fn cross_down_fills_at_level_on_intrabar_cross() {
        // Opens above the level, trades down through it → the level traded.
        let c = range_candle(100.0, 101.0, 94.0, 96.0);
        assert_eq!(cross_down(95.0, &c), Some(95.0));
    }

    #[test]
    fn cross_down_fills_at_open_on_gap_through() {
        // Opens already below the level → the level never traded; the open
        // was the first price the market offered.
        let c = range_candle(90.0, 92.0, 89.0, 91.0);
        assert_eq!(cross_down(95.0, &c), Some(90.0));
    }

    #[test]
    fn cross_down_none_when_low_never_reaches_level() {
        let c = range_candle(100.0, 101.0, 96.0, 97.0);
        assert_eq!(cross_down(95.0, &c), None);
    }

    #[test]
    fn cross_up_fills_at_level_on_intrabar_cross() {
        let c = range_candle(100.0, 106.0, 99.0, 104.0);
        assert_eq!(cross_up(105.0, &c), Some(105.0));
    }

    #[test]
    fn cross_up_fills_at_open_on_gap_through() {
        let c = range_candle(110.0, 112.0, 108.0, 111.0);
        assert_eq!(cross_up(105.0, &c), Some(110.0));
    }

    #[test]
    fn cross_up_none_when_high_never_reaches_level() {
        let c = range_candle(100.0, 104.0, 99.0, 103.0);
        assert_eq!(cross_up(105.0, &c), None);
    }

    #[test]
    fn cross_fires_on_exact_touch() {
        // Boundary: a low/high exactly ON the level counts as a cross and
        // fills at the level (open is on the far side, so no gap).
        let c = range_candle(100.0, 105.0, 95.0, 100.0);
        assert_eq!(cross_down(95.0, &c), Some(95.0));
        assert_eq!(cross_up(105.0, &c), Some(105.0));
    }

    // ── Candle validation ───────────────────────────────────────────────

    fn good_candle() -> Candle {
        Candle {
            time: 0,
            open: 1.0,
            high: 1.0,
            low: 1.0,
            close: 1.0,
            volume: 1.0,
        }
    }

    #[test]
    fn validate_candle_accepts_finite_positive() {
        assert!(validate_candle(&good_candle()).is_ok());
        // Zero volume is legitimate (illiquid candle).
        let c = Candle {
            volume: 0.0,
            ..good_candle()
        };
        assert!(validate_candle(&c).is_ok());
    }

    #[test]
    fn validate_candle_rejects_non_finite_and_non_positive_prices() {
        for bad in [f64::NAN, f64::INFINITY, f64::NEG_INFINITY, 0.0, -1.0] {
            let c = Candle {
                close: bad,
                ..good_candle()
            };
            assert!(
                validate_candle(&c).is_err(),
                "close={bad} should be rejected"
            );
        }
    }

    #[test]
    fn validate_candle_rejects_negative_or_nan_volume() {
        for bad in [-1.0, f64::NAN, f64::INFINITY] {
            let c = Candle {
                volume: bad,
                ..good_candle()
            };
            assert!(
                validate_candle(&c).is_err(),
                "volume={bad} should be rejected"
            );
        }
    }

    #[tokio::test]
    async fn run_rejects_non_finite_candle() {
        // A NaN close passed straight through `with_candles` (bypassing
        // the CSV loader) must still be caught — otherwise it poisons
        // the equity curve and every metric silently.
        let mut series = flat_series(5, 100.0);
        series[2].close = f64::NAN;
        let err = Backtest::new(
            cfg(),
            Arc::new(FixedBrain {
                signal: SignalType::Hold,
            }),
        )
        .with_candles(series)
        .run()
        .await
        .unwrap_err();
        assert!(matches!(err, Error::Data(_)), "got {err:?}");
    }

    #[tokio::test]
    async fn multi_symbol_equity_curve_deterministic_across_runs() {
        // Regression test for the HashMap-iteration-order determinism
        // bug: with two simultaneously-open positions, `equity_now` sums
        // unrealised PnL across the per-symbol map. Float addition isn't
        // associative, so a randomized map order would make the equity
        // curve differ run-to-run. `BTreeMap` fixes the order. Prices
        // are deliberately awkward so reordering *would* change the ULPs.
        struct DualLong;
        #[async_trait]
        impl Brain for DualLong {
            fn name(&self) -> &str {
                "dual-long"
            }
            async fn on_event(&self, e: &MarketDataEvent, p: &Position) -> CoreResult<Decision> {
                if p.qty == 0.0 && matches!(e, MarketDataEvent::Candle { .. }) {
                    Ok(Decision::buy(1.0))
                } else {
                    Ok(Decision::hold())
                }
            }
            async fn health(&self) -> BrainHealth {
                BrainHealth::ok()
            }
        }

        let run = || async {
            let cfg = BacktestConfig::builder()
                .symbols(["AAA", "BBB", "CCC"])
                .initial_cash(1_000_000.0)
                .sizing(SizingConfig {
                    margin_per_trade: 1_000.0,
                    leverage: 1,
                    max_contracts: 100,
                })
                .build()
                .unwrap();
            Backtest::new(cfg, Arc::new(DualLong))
                .with_symbol_candles("AAA", ramp_series(40, 100.13, 0.37))
                .with_symbol_candles("BBB", ramp_series(40, 250.07, -0.19))
                .with_symbol_candles("CCC", ramp_series(40, 33.31, 0.53))
                .run()
                .await
                .unwrap()
        };

        let r1 = run().await;
        let r2 = run().await;
        // Bit-exact across runs — not just approximately equal.
        assert_eq!(r1.equity_curve, r2.equity_curve);
        assert_eq!(r1.period_returns, r2.period_returns);
        assert_eq!(r1.net_pnl.to_bits(), r2.net_pnl.to_bits());
        assert_eq!(r1.max_drawdown.to_bits(), r2.max_drawdown.to_bits());
    }
}