schwab-sdk 0.4.1

//! Response shapes for the `/orders` and `/accounts/{n}/orders*` GET
//! endpoints. The construction-side types live in
//! [`super::request`](crate::orders::request).

use chrono::{DateTime, Utc};
use rust_decimal::Decimal;
use rust_decimal::serde::float_option as decimal_opt;
use serde::Deserialize;

use crate::accounts::AccountsInstrument;
use crate::orders::OrderId;
use crate::orders::enums::*;
use crate::secrets::AccountNumber;

/// One order, as returned by the read endpoints. Schwab marks almost no
/// field as required, so everything outside the discriminator-bearing
/// enums is `Option`.
///
/// `order_id` is wrapped in [`OrderId`], which serializes transparently as
/// the same `int64`.
#[derive(Debug, Clone, Deserialize, PartialEq, Eq, Hash)]
#[non_exhaustive]
pub struct Order {
    /// Trading session the order is valid in.
    #[serde(default)]
    pub session: Option<Session>,
    /// Time-in-force.
    #[serde(default)]
    pub duration: Option<Duration>,
    /// Order type (market / limit / stop / ...).
    #[serde(default, rename = "orderType")]
    pub order_type: Option<OrderType>,
    /// Scheduled cancel time for time-bound orders.
    #[serde(default, rename = "cancelTime")]
    pub cancel_time: Option<DateTime<Utc>>,
    /// Multi-leg option strategy shape; `NONE` for single-leg orders.
    #[serde(default, rename = "complexOrderStrategyType")]
    pub complex_order_strategy_type: Option<ComplexOrderStrategyType>,
    /// Total order quantity.
    #[serde(default, with = "decimal_opt")]
    pub quantity: Option<Decimal>,
    /// Quantity filled so far.
    #[serde(default, with = "decimal_opt", rename = "filledQuantity")]
    pub filled_quantity: Option<Decimal>,
    /// Quantity still working.
    #[serde(default, with = "decimal_opt", rename = "remainingQuantity")]
    pub remaining_quantity: Option<Decimal>,
    /// Response-only: the venue Schwab routed the order to.
    #[serde(default, rename = "requestedDestination")]
    pub requested_destination: Option<RequestedDestination>,
    /// Schwab-internal name for the routing destination.
    #[serde(default, rename = "destinationLinkName")]
    pub destination_link_name: Option<String>,
    /// Scheduled release time for orders held for later activation.
    #[serde(default, rename = "releaseTime")]
    pub release_time: Option<DateTime<Utc>>,
    /// Stop trigger price, USD.
    #[serde(default, with = "decimal_opt", rename = "stopPrice")]
    pub stop_price: Option<Decimal>,
    /// Reference price the stop is linked to.
    #[serde(default, rename = "stopPriceLinkBasis")]
    pub stop_price_link_basis: Option<StopPriceLinkBasis>,
    /// How the linked stop offset is interpreted.
    #[serde(default, rename = "stopPriceLinkType")]
    pub stop_price_link_type: Option<StopPriceLinkType>,
    /// Offset from the linked reference price.
    #[serde(default, with = "decimal_opt", rename = "stopPriceOffset")]
    pub stop_price_offset: Option<Decimal>,
    /// Which feed triggers the stop (bid / ask / last / mark).
    #[serde(default, rename = "stopType")]
    pub stop_type: Option<StopType>,
    /// Reference price the limit is linked to.
    #[serde(default, rename = "priceLinkBasis")]
    pub price_link_basis: Option<PriceLinkBasis>,
    /// How the linked limit offset is interpreted.
    #[serde(default, rename = "priceLinkType")]
    pub price_link_type: Option<PriceLinkType>,
    /// Limit price, USD.
    #[serde(default, with = "decimal_opt")]
    pub price: Option<Decimal>,
    /// Tax-lot relief method to apply when closing.
    #[serde(default, rename = "taxLotMethod")]
    pub tax_lot_method: Option<TaxLotMethod>,
    /// One entry per order leg.
    #[serde(default, rename = "orderLegCollection")]
    pub order_leg_collection: Vec<OrderLegCollection>,
    /// Activation price for stop / trigger orders.
    #[serde(default, with = "decimal_opt", rename = "activationPrice")]
    pub activation_price: Option<Decimal>,
    /// Schwab special-instruction flag (e.g. all-or-none).
    #[serde(default, rename = "specialInstruction")]
    pub special_instruction: Option<SpecialInstruction>,
    /// Top-level structure of the order envelope.
    #[serde(default, rename = "orderStrategyType")]
    pub order_strategy_type: Option<OrderStrategyType>,
    /// Schwab-assigned order id.
    #[serde(default, rename = "orderId")]
    pub order_id: Option<OrderId>,
    /// `true` if the order can currently be cancelled.
    #[serde(default)]
    pub cancelable: Option<bool>,
    /// `true` if the order can currently be replaced.
    #[serde(default)]
    pub editable: Option<bool>,
    /// Lifecycle status.
    #[serde(default)]
    pub status: Option<ApiOrderStatus>,
    /// Time Schwab recorded the order.
    #[serde(default, rename = "enteredTime")]
    pub entered_time: Option<DateTime<Utc>>,
    /// Time the order reached a terminal state.
    #[serde(default, rename = "closeTime")]
    pub close_time: Option<DateTime<Utc>>,
    /// Response-only: Schwab-assigned classification of the order's origin.
    /// Not settable on the request; consumers cannot use this for
    /// client-side correlation.
    #[serde(default)]
    pub tag: Option<String>,
    /// Plain account number that owns this order.
    #[serde(default, rename = "accountNumber")]
    pub account_number: Option<AccountNumber>,
    /// Per-event activity history (fills, lifecycle actions).
    #[serde(default, rename = "orderActivityCollection")]
    pub order_activity_collection: Vec<OrderActivity>,
    /// Orders that have replaced this one (replace lineage).
    #[serde(default, rename = "replacingOrderCollection")]
    pub replacing_order_collection: Vec<Order>,
    /// Child legs for `OCO` / `TRIGGER` / other compound strategies.
    #[serde(default, rename = "childOrderStrategies")]
    pub child_order_strategies: Vec<Order>,
    /// Schwab's free-form description of the current status (rejection
    /// reason, etc.).
    #[serde(default, rename = "statusDescription")]
    pub status_description: Option<String>,
}

/// One leg of an order (the security being traded plus its side / quantity).
#[derive(Debug, Clone, Default, Deserialize, PartialEq, Eq, Hash)]
#[non_exhaustive]
pub struct OrderLegCollection {
    /// Asset class of the leg.
    #[serde(default, rename = "orderLegType")]
    pub order_leg_type: Option<OrderLegType>,
    /// Schwab-assigned leg id within the order.
    #[serde(default, rename = "legId")]
    pub leg_id: Option<i64>,
    /// Instrument being traded.
    #[serde(default)]
    pub instrument: Option<AccountsInstrument>,
    /// Side / intent (buy / sell / buy-to-cover / ...).
    #[serde(default)]
    pub instruction: Option<Instruction>,
    /// Whether the leg opens or closes a position.
    #[serde(default, rename = "positionEffect")]
    pub position_effect: Option<PositionEffect>,
    /// Leg quantity (shares / contracts / dollars per `quantity_type`).
    #[serde(default, with = "decimal_opt")]
    pub quantity: Option<Decimal>,
    /// How `quantity` is denominated.
    #[serde(default, rename = "quantityType")]
    pub quantity_type: Option<QuantityType>,
    /// Dividend / capital-gains handling for mutual-fund legs.
    #[serde(default, rename = "divCapGains")]
    pub div_cap_gains: Option<DivCapGains>,
    /// Destination symbol for mutual-fund exchanges.
    #[serde(default, rename = "toSymbol")]
    pub to_symbol: Option<String>,
}

/// One lifecycle event in an order's activity history (a fill or an order
/// action).
#[derive(Debug, Clone, Default, Deserialize, PartialEq, Eq, Hash)]
#[non_exhaustive]
pub struct OrderActivity {
    /// Whether this row is an execution or an order action.
    #[serde(default, rename = "activityType")]
    pub activity_type: Option<OrderActivityType>,
    /// For executions, the kind of execution.
    #[serde(default, rename = "executionType")]
    pub execution_type: Option<ExecutionType>,
    /// Quantity affected by this activity.
    #[serde(default, with = "decimal_opt")]
    pub quantity: Option<Decimal>,
    /// Order quantity still working after this activity.
    #[serde(default, with = "decimal_opt", rename = "orderRemainingQuantity")]
    pub order_remaining_quantity: Option<Decimal>,
    /// Per-leg detail for executions.
    #[serde(default, rename = "executionLegs")]
    pub execution_legs: Vec<ExecutionLeg>,
}

/// One executed leg within an [`OrderActivity`] fill row.
#[derive(Debug, Clone, Default, Deserialize, PartialEq, Eq, Hash)]
#[non_exhaustive]
pub struct ExecutionLeg {
    /// Schwab-assigned leg id this fill is against.
    #[serde(default, rename = "legId")]
    pub leg_id: Option<i64>,
    /// Fill price, USD.
    #[serde(default, with = "decimal_opt")]
    pub price: Option<Decimal>,
    /// Quantity filled in this leg.
    #[serde(default, with = "decimal_opt")]
    pub quantity: Option<Decimal>,
    /// Quantity that was mis-marked at fill time.
    #[serde(default, with = "decimal_opt", rename = "mismarkedQuantity")]
    pub mismarked_quantity: Option<Decimal>,
    /// Schwab-internal instrument id of the security filled.
    #[serde(default, rename = "instrumentId")]
    pub instrument_id: Option<i64>,
    /// Execution time.
    #[serde(default)]
    pub time: Option<DateTime<Utc>>,
}

#[cfg(test)]
mod tests {
    use super::*;
    use rust_decimal_macros::dec;

    #[test]
    fn filled_equity_order_parses_with_execution() {
        let json = r#"{
            "orderId": 100000001,
            "accountNumber": 12345678,
            "status": "FILLED",
            "orderType": "LIMIT",
            "session": "NORMAL",
            "duration": "DAY",
            "orderStrategyType": "SINGLE",
            "complexOrderStrategyType": "NONE",
            "quantity": 10.0,
            "filledQuantity": 10.0,
            "remainingQuantity": 0.0,
            "price": 145.32,
            "enteredTime": "2024-03-15T15:30:00.000Z",
            "closeTime": "2024-03-15T15:30:02.500Z",
            "cancelable": false,
            "editable": false,
            "orderLegCollection": [{
                "orderLegType": "EQUITY",
                "legId": 1,
                "instruction": "BUY",
                "positionEffect": "OPENING",
                "quantity": 10.0,
                "quantityType": "SHARES",
                "instrument": {
                    "assetType": "EQUITY",
                    "symbol": "AAPL",
                    "cusip": "037833100",
                    "instrumentId": 12345
                }
            }],
            "orderActivityCollection": [{
                "activityType": "EXECUTION",
                "executionType": "FILL",
                "quantity": 10.0,
                "orderRemainingQuantity": 0.0,
                "executionLegs": [{
                    "legId": 1,
                    "price": 145.32,
                    "quantity": 10.0,
                    "mismarkedQuantity": 0.0,
                    "instrumentId": 12345,
                    "time": "2024-03-15T15:30:02.500Z"
                }]
            }]
        }"#;
        let order: Order = serde_json::from_str(json).unwrap();
        assert_eq!(order.order_id, Some(OrderId::new(100000001)));
        assert_eq!(
            order
                .account_number
                .as_ref()
                .map(AccountNumber::expose_secret),
            Some("12345678"),
        );
        assert_eq!(order.status, Some(ApiOrderStatus::Filled));
        assert_eq!(order.order_type, Some(OrderType::Limit));
        assert_eq!(order.order_strategy_type, Some(OrderStrategyType::Single));
        assert_eq!(order.quantity, Some(dec!(10.0)));
        assert_eq!(order.filled_quantity, Some(dec!(10.0)));
        assert_eq!(order.price, Some(dec!(145.32)));
        assert_eq!(order.cancelable, Some(false));

        assert_eq!(order.order_leg_collection.len(), 1);
        let leg = &order.order_leg_collection[0];
        assert_eq!(leg.instruction, Some(Instruction::Buy));
        assert_eq!(leg.position_effect, Some(PositionEffect::Opening));
        assert_eq!(leg.quantity, Some(dec!(10.0)));
        assert_eq!(leg.quantity_type, Some(QuantityType::Shares));

        assert_eq!(order.order_activity_collection.len(), 1);
        let activity = &order.order_activity_collection[0];
        assert_eq!(activity.activity_type, Some(OrderActivityType::Execution));
        assert_eq!(activity.execution_type, Some(ExecutionType::Fill));
        assert_eq!(activity.execution_legs.len(), 1);
        let exec = &activity.execution_legs[0];
        assert_eq!(exec.price, Some(dec!(145.32)));
        assert_eq!(exec.quantity, Some(dec!(10.0)));
    }

    #[test]
    fn working_order_with_no_fills_parses() {
        let json = r#"{
            "orderId": 100000002,
            "status": "WORKING",
            "orderType": "LIMIT",
            "orderStrategyType": "SINGLE",
            "quantity": 5.0,
            "filledQuantity": 0.0,
            "remainingQuantity": 5.0,
            "price": 140.00,
            "cancelable": true,
            "editable": true,
            "orderLegCollection": [{
                "orderLegType": "EQUITY",
                "instruction": "BUY",
                "quantity": 5.0,
                "instrument": {
                    "assetType": "EQUITY",
                    "symbol": "AAPL"
                }
            }]
        }"#;
        let order: Order = serde_json::from_str(json).unwrap();
        assert_eq!(order.status, Some(ApiOrderStatus::Working));
        assert_eq!(order.filled_quantity, Some(dec!(0.0)));
        assert_eq!(order.remaining_quantity, Some(dec!(5.0)));
        assert!(order.order_activity_collection.is_empty());
        assert_eq!(order.cancelable, Some(true));
    }

    #[test]
    fn trigger_strategy_parses_with_child_orders() {
        let json = r#"{
            "orderId": 100000003,
            "orderStrategyType": "TRIGGER",
            "orderType": "LIMIT",
            "price": 34.97,
            "quantity": 10.0,
            "orderLegCollection": [{
                "instruction": "BUY",
                "quantity": 10.0,
                "instrument": { "assetType": "EQUITY", "symbol": "XYZ" }
            }],
            "childOrderStrategies": [{
                "orderId": 100000004,
                "orderStrategyType": "SINGLE",
                "orderType": "LIMIT",
                "price": 42.03,
                "quantity": 10.0,
                "orderLegCollection": [{
                    "instruction": "SELL",
                    "quantity": 10.0,
                    "instrument": { "assetType": "EQUITY", "symbol": "XYZ" }
                }]
            }]
        }"#;
        let order: Order = serde_json::from_str(json).unwrap();
        assert_eq!(order.order_strategy_type, Some(OrderStrategyType::Trigger));
        assert_eq!(order.child_order_strategies.len(), 1);
        let child = &order.child_order_strategies[0];
        assert_eq!(child.order_id, Some(OrderId::new(100000004)));
        assert_eq!(child.order_strategy_type, Some(OrderStrategyType::Single));
        assert_eq!(child.price, Some(dec!(42.03)));
    }

    #[test]
    fn account_number_accepts_both_string_and_int_forms() {
        // The OpenAPI spec types `accountNumber` here as `int64`. For
        // robustness, check that both string and int forms decode to the
        // same `AccountNumber`, and the resulting value is redacted in Debug.
        let as_int: Order =
            serde_json::from_str(r#"{"orderId": 1, "accountNumber": 12345678}"#).unwrap();
        let as_str: Order =
            serde_json::from_str(r#"{"orderId": 1, "accountNumber": "12345678"}"#).unwrap();

        assert_eq!(
            as_int
                .account_number
                .as_ref()
                .map(AccountNumber::expose_secret),
            Some("12345678"),
        );
        assert_eq!(as_int.account_number, as_str.account_number);

        let debug = format!("{:?}", as_str.account_number.as_ref().unwrap());
        assert!(!debug.contains("12345678"), "Debug leaked: {debug}");
        assert!(debug.contains("REDACTED"), "expected REDACTED in {debug}");
    }

    #[test]
    fn missing_account_number_decodes_as_none() {
        let order: Order = serde_json::from_str(r#"{"orderId": 1}"#).unwrap();
        assert!(order.account_number.is_none());
    }

    #[test]
    fn empty_collections_default_to_empty_vecs() {
        let json = r#"{"orderId": 1}"#;
        let order: Order = serde_json::from_str(json).unwrap();
        assert!(order.order_leg_collection.is_empty());
        assert!(order.order_activity_collection.is_empty());
        assert!(order.child_order_strategies.is_empty());
        assert!(order.replacing_order_collection.is_empty());
    }

    #[test]
    fn oco_strategy_parses_with_two_child_orders_and_no_top_level_legs() {
        // OCO ("one cancels other"): the parent carries no orderLegCollection;
        // each child is an independent SINGLE strategy with its own leg.
        let json = r#"{
            "orderId": 100000005,
            "orderStrategyType": "OCO",
            "childOrderStrategies": [
                {
                    "orderId": 100000006,
                    "orderStrategyType": "SINGLE",
                    "orderType": "LIMIT",
                    "price": 155.00,
                    "quantity": 10.0,
                    "orderLegCollection": [{
                        "instruction": "SELL",
                        "quantity": 10.0,
                        "instrument": { "assetType": "EQUITY", "symbol": "AAPL" }
                    }]
                },
                {
                    "orderId": 100000007,
                    "orderStrategyType": "SINGLE",
                    "orderType": "STOP",
                    "stopPrice": 135.00,
                    "quantity": 10.0,
                    "orderLegCollection": [{
                        "instruction": "SELL",
                        "quantity": 10.0,
                        "instrument": { "assetType": "EQUITY", "symbol": "AAPL" }
                    }]
                }
            ]
        }"#;
        let order: Order = serde_json::from_str(json).unwrap();
        assert_eq!(order.order_id, Some(OrderId::new(100000005)));
        assert_eq!(order.order_strategy_type, Some(OrderStrategyType::Oco));
        assert!(order.order_leg_collection.is_empty());

        assert_eq!(order.child_order_strategies.len(), 2);
        let limit_leg = &order.child_order_strategies[0];
        assert_eq!(limit_leg.order_id, Some(OrderId::new(100000006)));
        assert_eq!(
            limit_leg.order_strategy_type,
            Some(OrderStrategyType::Single)
        );
        assert_eq!(limit_leg.order_type, Some(OrderType::Limit));
        assert_eq!(limit_leg.price, Some(dec!(155.00)));
        assert_eq!(limit_leg.order_leg_collection.len(), 1);
        assert_eq!(
            limit_leg.order_leg_collection[0].instruction,
            Some(Instruction::Sell)
        );

        let stop_leg = &order.child_order_strategies[1];
        assert_eq!(stop_leg.order_id, Some(OrderId::new(100000007)));
        assert_eq!(stop_leg.order_type, Some(OrderType::Stop));
        assert_eq!(stop_leg.stop_price, Some(dec!(135.00)));
        assert_eq!(stop_leg.order_leg_collection.len(), 1);
    }
}