polyfill-rs 0.4.0

//! Order creation and signing functionality
//!
//! This module handles the complex process of creating and signing orders
//! for the Polymarket CLOB, including EIP-712 signature generation.

use crate::auth::{sign_order_message, SignedOrderMessage};
use crate::errors::{PolyfillError, Result};
use crate::types::{
    CreateOrderOptions, MarketOrderArgs, OrderArgs, OrderType, Side, SignedOrderRequest,
};
use alloy_primitives::{keccak256, Address, B256, U256};
use alloy_signer_local::PrivateKeySigner;
use rand::Rng;
use rust_decimal::Decimal;
use rust_decimal::RoundingStrategy::{AwayFromZero, MidpointTowardZero, ToZero};
use std::collections::HashMap;
use std::str::FromStr;
use std::sync::LazyLock;
use std::time::{SystemTime, UNIX_EPOCH};

pub const BYTES32_ZERO: &str = "0x0000000000000000000000000000000000000000000000000000000000000000";

/// Signature types for orders
#[derive(Copy, Clone, Debug, PartialEq, Eq)]
pub enum SigType {
    /// ECDSA EIP712 signatures signed by EOAs
    Eoa = 0,
    /// EIP712 signatures signed by EOAs that own Polymarket Proxy wallets
    PolyProxy = 1,
    /// EIP712 signatures signed by EOAs that own Polymarket Gnosis safes
    PolyGnosisSafe = 2,
    /// EIP-1271 smart contract wallet signatures (V2 orders only)
    Poly1271 = 3,
}

/// Rounding configuration for different tick sizes
pub struct RoundConfig {
    price: u32,
    size: u32,
    amount: u32,
}

/// Contract configuration
pub struct ContractConfig {
    pub exchange: String,
    pub collateral: String,
    pub conditional_tokens: String,
}

/// Order builder for creating and signing orders
pub struct OrderBuilder {
    signer: PrivateKeySigner,
    sig_type: SigType,
    funder: Address,
}

const POLYGON_PROXY_FACTORY: &str = "0xaB45c5A4B0c941a2F231C04C3f49182e1A254052";
const POLYGON_SAFE_FACTORY: &str = "0xaacFeEa03eb1561C4e67d661e40682Bd20E3541b";
const PROXY_INIT_CODE_HASH: &str =
    "0xd21df8dc65880a8606f09fe0ce3df9b8869287ab0b058be05aa9e8af6330a00b";
const SAFE_INIT_CODE_HASH: &str =
    "0x2bce2127ff07fb632d16c8347c4ebf501f4841168bed00d9e6ef715ddb6fcecf";

/// Rounding configurations for different tick sizes
static ROUNDING_CONFIG: LazyLock<HashMap<Decimal, RoundConfig>> = LazyLock::new(|| {
    HashMap::from([
        (
            Decimal::from_str("0.1").unwrap(),
            RoundConfig {
                price: 1,
                size: 2,
                amount: 3,
            },
        ),
        (
            Decimal::from_str("0.01").unwrap(),
            RoundConfig {
                price: 2,
                size: 2,
                amount: 4,
            },
        ),
        (
            Decimal::from_str("0.001").unwrap(),
            RoundConfig {
                price: 3,
                size: 2,
                amount: 5,
            },
        ),
        (
            Decimal::from_str("0.0001").unwrap(),
            RoundConfig {
                price: 4,
                size: 2,
                amount: 6,
            },
        ),
    ])
});

/// Get contract configuration for chain
pub fn get_contract_config(chain_id: u64, neg_risk: bool) -> Option<ContractConfig> {
    match (chain_id, neg_risk) {
        (137, false) => Some(ContractConfig {
            exchange: "0xE111180000d2663C0091e4f400237545B87B996B".to_string(),
            collateral: "0xC011a7E12a19f7B1f670d46F03B03f3342E82DFB".to_string(),
            conditional_tokens: "0x4D97DCd97eC945f40cF65F87097ACe5EA0476045".to_string(),
        }),
        (137, true) => Some(ContractConfig {
            exchange: "0xe2222d279d744050d28e00520010520000310F59".to_string(),
            collateral: "0xC011a7E12a19f7B1f670d46F03B03f3342E82DFB".to_string(),
            conditional_tokens: "0x4D97DCd97eC945f40cF65F87097ACe5EA0476045".to_string(),
        }),
        _ => None,
    }
}

pub fn sig_type_from_u8(signature_type: u8) -> Result<SigType> {
    match signature_type {
        0 => Ok(SigType::Eoa),
        1 => Ok(SigType::PolyProxy),
        2 => Ok(SigType::PolyGnosisSafe),
        3 => Ok(SigType::Poly1271),
        other => Err(PolyfillError::validation(format!(
            "Unsupported signature_type {other}"
        ))),
    }
}

pub fn derive_proxy_wallet(eoa_address: Address, chain_id: u64) -> Result<Address> {
    if chain_id != 137 {
        return Err(PolyfillError::config(
            "Proxy wallet auto-derivation is only configured for Polygon mainnet",
        ));
    }

    let factory = Address::from_str(POLYGON_PROXY_FACTORY)
        .map_err(|e| PolyfillError::config(format!("Invalid proxy factory address: {e}")))?;
    let init_code_hash = B256::from_str(PROXY_INIT_CODE_HASH)
        .map_err(|e| PolyfillError::config(format!("Invalid proxy init code hash: {e}")))?;
    let salt = keccak256(eoa_address);
    Ok(factory.create2(salt, init_code_hash))
}

pub fn derive_safe_wallet(eoa_address: Address, chain_id: u64) -> Result<Address> {
    if chain_id != 137 {
        return Err(PolyfillError::config(
            "Safe wallet auto-derivation is only configured for Polygon mainnet",
        ));
    }

    let factory = Address::from_str(POLYGON_SAFE_FACTORY)
        .map_err(|e| PolyfillError::config(format!("Invalid safe factory address: {e}")))?;
    let init_code_hash = B256::from_str(SAFE_INIT_CODE_HASH)
        .map_err(|e| PolyfillError::config(format!("Invalid safe init code hash: {e}")))?;
    let mut padded = [0_u8; 32];
    padded[12..].copy_from_slice(eoa_address.as_slice());
    let salt = keccak256(padded);
    Ok(factory.create2(salt, init_code_hash))
}

pub fn resolve_funder(
    signer_address: Address,
    chain_id: u64,
    sig_type: SigType,
    funder: Option<Address>,
) -> Result<Option<Address>> {
    match (sig_type, funder) {
        (SigType::Eoa, Some(_)) => Err(PolyfillError::validation(
            "funder cannot be set for EOA signature_type",
        )),
        (SigType::PolyProxy, None) => derive_proxy_wallet(signer_address, chain_id).map(Some),
        (SigType::PolyGnosisSafe, None) => derive_safe_wallet(signer_address, chain_id).map(Some),
        (SigType::Poly1271, None) => Err(PolyfillError::validation(
            "funder is required for Poly1271 signature_type",
        )),
        (_, Some(Address::ZERO)) => Err(PolyfillError::validation("funder cannot be zero address")),
        (_, explicit) => Ok(explicit),
    }
}

/// Generate a random seed for order salt
fn generate_seed() -> u64 {
    let mut rng = rand::thread_rng();
    let y: f64 = rng.gen();
    let timestamp = SystemTime::now()
        .duration_since(UNIX_EPOCH)
        .expect("Time went backwards")
        .as_secs();
    (timestamp as f64 * y) as u64
}

/// Convert decimal to token units (multiply by 1e6)
fn decimal_to_token_u32(amt: Decimal) -> u32 {
    let mut amt = Decimal::from_scientific("1e6").expect("1e6 is not scientific") * amt;
    if amt.scale() > 0 {
        amt = amt.round_dp_with_strategy(0, MidpointTowardZero);
    }
    amt.try_into().expect("Couldn't round decimal to integer")
}

fn parse_round_config(tick_size: Decimal) -> Result<&'static RoundConfig> {
    ROUNDING_CONFIG
        .get(&tick_size)
        .ok_or_else(|| PolyfillError::validation(format!("Unsupported tick size {tick_size}")))
}

pub(crate) fn validate_bytes32_hex(field: &str, value: &str) -> Result<()> {
    if value == BYTES32_ZERO {
        return Ok(());
    }

    if !value.starts_with("0x") {
        return Err(PolyfillError::validation(format!(
            "{field} must be a 0x-prefixed 32-byte hex string"
        )));
    }

    if value.len() != 66 {
        return Err(PolyfillError::validation(format!(
            "{field} must be exactly 32 bytes (64 hex chars)"
        )));
    }

    if !value
        .as_bytes()
        .iter()
        .skip(2)
        .all(|byte| byte.is_ascii_hexdigit())
    {
        return Err(PolyfillError::validation(format!(
            "{field} must contain only hexadecimal characters"
        )));
    }

    Ok(())
}

fn normalize_optional_bytes32(field: &str, value: Option<&str>) -> Result<String> {
    let value = value.unwrap_or(BYTES32_ZERO);
    validate_bytes32_hex(field, value)?;
    Ok(value.to_string())
}

pub fn adjust_buy_amount_for_fees(
    amount: Decimal,
    price: Decimal,
    user_usdc_balance: Decimal,
    fee_rate: Decimal,
    fee_exponent: u32,
    builder_taker_fee_rate: Decimal,
) -> Result<Decimal> {
    if price <= Decimal::ZERO {
        return Err(PolyfillError::validation(
            "Market buy fee adjustment requires a positive price",
        ));
    }

    let base = price * (Decimal::ONE - price);
    let base_f64: f64 = base
        .try_into()
        .map_err(|_| PolyfillError::validation(format!("Invalid fee base {base}")))?;
    let exp_f64: f64 = Decimal::from(fee_exponent)
        .try_into()
        .map_err(|_| PolyfillError::validation(format!("Invalid fee exponent {fee_exponent}")))?;
    let platform_fee_rate = fee_rate
        * Decimal::try_from(base_f64.powf(exp_f64)).map_err(|_| {
            PolyfillError::validation(format!(
                "Invalid platform fee rate for price {price} and exponent {fee_exponent}"
            ))
        })?;

    let platform_fee = amount / price * platform_fee_rate;
    let total_cost = amount + platform_fee + amount * builder_taker_fee_rate;

    let raw = if user_usdc_balance <= total_cost {
        let divisor = Decimal::ONE + platform_fee_rate / price + builder_taker_fee_rate;
        user_usdc_balance / divisor
    } else {
        amount
    };

    let adjusted = raw.trunc_with_scale(6);
    if adjusted.is_zero() {
        return Err(PolyfillError::validation(format!(
            "user_usdc_balance {user_usdc_balance} too small to cover fees at price {price}; \
             fee-adjusted amount truncated to zero"
        )));
    }

    Ok(adjusted)
}

impl OrderBuilder {
    /// Create a new order builder
    pub fn new(
        signer: PrivateKeySigner,
        sig_type: Option<SigType>,
        funder: Option<Address>,
    ) -> Self {
        let sig_type = sig_type.unwrap_or(SigType::Eoa);
        let funder = funder.unwrap_or(signer.address());

        OrderBuilder {
            signer,
            sig_type,
            funder,
        }
    }

    /// Get signature type as u8
    pub fn get_sig_type(&self) -> u8 {
        self.sig_type as u8
    }

    /// Fix amount rounding according to configuration
    fn fix_amount_rounding(&self, mut amt: Decimal, round_config: &RoundConfig) -> Decimal {
        if amt.scale() > round_config.amount {
            amt = amt.round_dp_with_strategy(round_config.amount + 4, AwayFromZero);
            if amt.scale() > round_config.amount {
                amt = amt.round_dp_with_strategy(round_config.amount, ToZero);
            }
        }
        amt
    }

    /// Get order amounts (maker and taker) for a regular order
    fn get_order_amounts(
        &self,
        side: Side,
        size: Decimal,
        price: Decimal,
        round_config: &RoundConfig,
    ) -> (u32, u32) {
        let raw_price = price.round_dp_with_strategy(round_config.price, MidpointTowardZero);

        match side {
            Side::BUY => {
                let raw_taker_amt = size.round_dp_with_strategy(round_config.size, ToZero);
                let raw_maker_amt = raw_taker_amt * raw_price;
                let raw_maker_amt = self.fix_amount_rounding(raw_maker_amt, round_config);
                (
                    decimal_to_token_u32(raw_maker_amt),
                    decimal_to_token_u32(raw_taker_amt),
                )
            },
            Side::SELL => {
                let raw_maker_amt = size.round_dp_with_strategy(round_config.size, ToZero);
                let raw_taker_amt = raw_maker_amt * raw_price;
                let raw_taker_amt = self.fix_amount_rounding(raw_taker_amt, round_config);

                (
                    decimal_to_token_u32(raw_maker_amt),
                    decimal_to_token_u32(raw_taker_amt),
                )
            },
        }
    }

    /// Get order amounts for a market order
    fn get_market_order_amounts(
        &self,
        side: Side,
        amount: Decimal,
        price: Decimal,
        round_config: &RoundConfig,
    ) -> (u32, u32) {
        let raw_price = price.round_dp_with_strategy(round_config.price, MidpointTowardZero);

        match side {
            Side::BUY => {
                let raw_maker_amt = amount.round_dp_with_strategy(round_config.size, ToZero);
                let raw_taker_amt =
                    self.fix_amount_rounding(raw_maker_amt / raw_price, round_config);

                (
                    decimal_to_token_u32(raw_maker_amt),
                    decimal_to_token_u32(raw_taker_amt),
                )
            },
            Side::SELL => {
                let raw_maker_amt = amount.round_dp_with_strategy(round_config.size, ToZero);
                let raw_taker_amt =
                    self.fix_amount_rounding(raw_maker_amt * raw_price, round_config);

                (
                    decimal_to_token_u32(raw_maker_amt),
                    decimal_to_token_u32(raw_taker_amt),
                )
            },
        }
    }

    /// Calculate market price from order book levels
    pub fn calculate_market_price(
        &self,
        positions: &[crate::types::BookLevel],
        amount_to_match: Decimal,
        side: Side,
        order_type: OrderType,
    ) -> Result<Decimal> {
        let mut sum = Decimal::ZERO;
        let mut last_price = None;

        for level in positions {
            sum += match side {
                Side::BUY => level.size * level.price,
                Side::SELL => level.size,
            };
            last_price = Some(level.price);
            if sum >= amount_to_match {
                return Ok(level.price);
            }
        }

        match (order_type, last_price) {
            (OrderType::FAK, Some(price)) => Ok(price),
            _ => Err(PolyfillError::order(
                format!(
                    "Not enough liquidity to create market order with amount {}",
                    amount_to_match
                ),
                crate::errors::OrderErrorKind::InsufficientBalance,
            )),
        }
    }

    /// Create a market order
    pub fn create_market_order(
        &self,
        chain_id: u64,
        order_args: &MarketOrderArgs,
        price: Decimal,
        options: &CreateOrderOptions,
    ) -> Result<SignedOrderRequest> {
        if !matches!(order_args.order_type, OrderType::FOK | OrderType::FAK) {
            return Err(PolyfillError::validation(
                "Market orders only support FOK and FAK order types",
            ));
        }

        let tick_size = options
            .tick_size
            .ok_or_else(|| PolyfillError::validation("Cannot create order without tick size"))?;
        let round_config = parse_round_config(tick_size)?;

        let (maker_amount, taker_amount) =
            self.get_market_order_amounts(order_args.side, order_args.amount, price, round_config);

        let neg_risk = options
            .neg_risk
            .ok_or_else(|| PolyfillError::validation("Cannot create order without neg_risk"))?;

        let contract_config = get_contract_config(chain_id, neg_risk).ok_or_else(|| {
            PolyfillError::config("No contract found with given chain_id and neg_risk")
        })?;

        let exchange_address = Address::from_str(&contract_config.exchange)
            .map_err(|e| PolyfillError::config(format!("Invalid exchange address: {}", e)))?;

        self.build_signed_order(
            order_args.token_id.clone(),
            order_args.side,
            chain_id,
            exchange_address,
            maker_amount,
            taker_amount,
            0,
            order_args.builder_code.as_deref(),
            order_args.metadata.as_deref(),
        )
    }

    /// Create a regular order
    pub fn create_order(
        &self,
        chain_id: u64,
        order_args: &OrderArgs,
        options: &CreateOrderOptions,
    ) -> Result<SignedOrderRequest> {
        let tick_size = options
            .tick_size
            .ok_or_else(|| PolyfillError::validation("Cannot create order without tick size"))?;
        let round_config = parse_round_config(tick_size)?;

        let (maker_amount, taker_amount) = self.get_order_amounts(
            order_args.side,
            order_args.size,
            order_args.price,
            round_config,
        );

        let neg_risk = options
            .neg_risk
            .ok_or_else(|| PolyfillError::validation("Cannot create order without neg_risk"))?;

        let contract_config = get_contract_config(chain_id, neg_risk).ok_or_else(|| {
            PolyfillError::config("No contract found with given chain_id and neg_risk")
        })?;

        let exchange_address = Address::from_str(&contract_config.exchange)
            .map_err(|e| PolyfillError::config(format!("Invalid exchange address: {}", e)))?;

        self.build_signed_order(
            order_args.token_id.clone(),
            order_args.side,
            chain_id,
            exchange_address,
            maker_amount,
            taker_amount,
            order_args.expiration.unwrap_or(0),
            order_args.builder_code.as_deref(),
            order_args.metadata.as_deref(),
        )
    }

    /// Build and sign an order
    #[allow(clippy::too_many_arguments)]
    fn build_signed_order(
        &self,
        token_id: String,
        side: Side,
        chain_id: u64,
        exchange: Address,
        maker_amount: u32,
        taker_amount: u32,
        expiration: u64,
        builder_code: Option<&str>,
        metadata: Option<&str>,
    ) -> Result<SignedOrderRequest> {
        let seed = generate_seed();
        let timestamp = SystemTime::now()
            .duration_since(UNIX_EPOCH)
            .expect("Time went backwards")
            .as_millis();

        let u256_token_id = U256::from_str_radix(&token_id, 10)
            .map_err(|e| PolyfillError::validation(format!("Incorrect tokenId format: {}", e)))?;
        let builder = normalize_optional_bytes32("builder_code", builder_code)?;
        let metadata = normalize_optional_bytes32("metadata", metadata)?;

        let order = SignedOrderMessage {
            salt: U256::from(seed),
            maker: self.funder,
            signer: self.signer.address(),
            token_id: u256_token_id,
            maker_amount: U256::from(maker_amount),
            taker_amount: U256::from(taker_amount),
            side: side as u8,
            signature_type: self.sig_type as u8,
            timestamp: U256::from(timestamp),
            metadata: B256::from_str(&metadata).map_err(|e| {
                PolyfillError::validation(format!("Invalid metadata bytes32 value: {e}"))
            })?,
            builder: B256::from_str(&builder).map_err(|e| {
                PolyfillError::validation(format!("Invalid builder_code bytes32 value: {e}"))
            })?,
        };

        let signature = sign_order_message(&self.signer, order, chain_id, exchange)?;

        Ok(SignedOrderRequest {
            salt: seed,
            maker: self.funder.to_checksum(None),
            signer: self.signer.address().to_checksum(None),
            token_id,
            maker_amount: maker_amount.to_string(),
            taker_amount: taker_amount.to_string(),
            expiration: expiration.to_string(),
            side: side.as_str().to_string(),
            signature_type: self.sig_type as u8,
            timestamp: timestamp.to_string(),
            metadata,
            builder,
            signature,
        })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use alloy_signer_local::PrivateKeySigner;
    use serde_json::Value;

    fn test_builder() -> OrderBuilder {
        let signer: PrivateKeySigner =
            "0x1234567890123456789012345678901234567890123456789012345678901234"
                .parse()
                .expect("valid private key");
        OrderBuilder::new(signer, None, None)
    }

    #[test]
    fn test_decimal_to_token_u32() {
        let result = decimal_to_token_u32(Decimal::from_str("1.5").unwrap());
        assert_eq!(result, 1_500_000);
    }

    #[test]
    fn test_generate_seed() {
        let seed1 = generate_seed();
        let seed2 = generate_seed();
        assert_ne!(seed1, seed2);
    }

    #[test]
    fn test_decimal_to_token_u32_edge_cases() {
        // Test zero
        let result = decimal_to_token_u32(Decimal::ZERO);
        assert_eq!(result, 0);

        // Test small decimal
        let result = decimal_to_token_u32(Decimal::from_str("0.000001").unwrap());
        assert_eq!(result, 1);

        // Test large number
        let result = decimal_to_token_u32(Decimal::from_str("1000.0").unwrap());
        assert_eq!(result, 1_000_000_000);
    }

    #[test]
    fn test_get_contract_config() {
        // Test Polygon mainnet
        let config = get_contract_config(137, false).expect("polygon config");
        assert_eq!(
            config.exchange,
            "0xE111180000d2663C0091e4f400237545B87B996B"
        );
        assert_eq!(
            config.collateral,
            "0xC011a7E12a19f7B1f670d46F03B03f3342E82DFB"
        );
        assert_eq!(
            config.conditional_tokens,
            "0x4D97DCd97eC945f40cF65F87097ACe5EA0476045"
        );

        // Test with neg risk
        let config_neg = get_contract_config(137, true).expect("neg risk polygon config");
        assert_eq!(
            config_neg.exchange,
            "0xe2222d279d744050d28e00520010520000310F59"
        );
        assert_eq!(
            config_neg.collateral,
            "0xC011a7E12a19f7B1f670d46F03B03f3342E82DFB"
        );

        // Test unsupported chain
        let config_unsupported = get_contract_config(999, false);
        assert!(config_unsupported.is_none());
    }

    #[test]
    fn test_signature_type_from_u8() {
        assert_eq!(sig_type_from_u8(0).unwrap(), SigType::Eoa);
        assert_eq!(sig_type_from_u8(1).unwrap(), SigType::PolyProxy);
        assert_eq!(sig_type_from_u8(2).unwrap(), SigType::PolyGnosisSafe);
        assert_eq!(sig_type_from_u8(3).unwrap(), SigType::Poly1271);
        assert!(sig_type_from_u8(4).is_err());
    }

    #[test]
    fn test_derive_polygon_funder_addresses() {
        let eoa = Address::from_str("0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266").unwrap();
        assert_eq!(
            derive_safe_wallet(eoa, 137).unwrap(),
            Address::from_str("0xd93b25Cb943D14d0d34FBAf01fc93a0F8b5f6e47").unwrap()
        );
        assert_eq!(
            derive_proxy_wallet(eoa, 137).unwrap(),
            Address::from_str("0x365f0cA36ae1F641E02Fe3b7743673DA42A13a70").unwrap()
        );
    }

    #[test]
    fn test_normalize_optional_bytes32_defaults_to_zero() {
        assert_eq!(
            normalize_optional_bytes32("builder_code", None).unwrap(),
            BYTES32_ZERO
        );
    }

    #[test]
    fn test_normalize_optional_bytes32_rejects_invalid_hex() {
        let err = normalize_optional_bytes32("metadata", Some("deadbeef")).unwrap_err();
        assert!(matches!(err, PolyfillError::Validation { .. }));
    }

    #[test]
    fn test_create_order_serializes_v2_fields_without_legacy_fields() {
        let builder = test_builder();
        let order = builder
            .create_order(
                137,
                &OrderArgs {
                    token_id: "123456".to_string(),
                    price: Decimal::from_str("0.45").unwrap(),
                    size: Decimal::from_str("12.34").unwrap(),
                    side: Side::BUY,
                    expiration: Some(1_900_000_000),
                    builder_code: Some(BYTES32_ZERO.to_string()),
                    metadata: None,
                },
                &CreateOrderOptions {
                    tick_size: Some(Decimal::from_str("0.01").unwrap()),
                    neg_risk: Some(false),
                },
            )
            .unwrap();

        let serialized = serde_json::to_value(&order).unwrap();
        let object = serialized.as_object().unwrap();
        assert!(object.contains_key("timestamp"));
        assert!(object.contains_key("metadata"));
        assert!(object.contains_key("builder"));
        assert!(object.contains_key("expiration"));
        assert!(!object.contains_key("taker"));
        assert!(!object.contains_key("nonce"));
        assert!(!object.contains_key("feeRateBps"));
        assert_eq!(order.builder, BYTES32_ZERO);
        assert_eq!(order.metadata, BYTES32_ZERO);
    }

    #[test]
    fn test_create_market_order_supports_fak() {
        let builder = test_builder();
        let order = builder
            .create_market_order(
                137,
                &MarketOrderArgs {
                    token_id: "123456".to_string(),
                    amount: Decimal::from_str("10.0").unwrap(),
                    side: Side::BUY,
                    order_type: OrderType::FAK,
                    price_limit: None,
                    user_usdc_balance: None,
                    builder_code: None,
                    metadata: None,
                },
                Decimal::from_str("0.25").unwrap(),
                &CreateOrderOptions {
                    tick_size: Some(Decimal::from_str("0.01").unwrap()),
                    neg_risk: Some(false),
                },
            )
            .unwrap();

        assert_eq!(order.side, "BUY");
        assert!(!order.timestamp.is_empty());
    }

    #[test]
    fn test_adjust_buy_amount_for_fees_uses_builder_rate_decimal() {
        let adjusted = adjust_buy_amount_for_fees(
            Decimal::from_str("100").unwrap(),
            Decimal::from_str("0.5").unwrap(),
            Decimal::from_str("100").unwrap(),
            Decimal::ZERO,
            0,
            Decimal::from_str("0.01").unwrap(),
        )
        .unwrap();

        assert_eq!(adjusted, Decimal::from_str("99.009900").unwrap());
    }

    #[test]
    fn test_adjust_buy_amount_for_fees_rejects_zero_after_truncation() {
        let err = adjust_buy_amount_for_fees(
            Decimal::from_str("1").unwrap(),
            Decimal::from_str("0.5").unwrap(),
            Decimal::from_str("0.0000009").unwrap(),
            Decimal::ZERO,
            0,
            Decimal::ZERO,
        )
        .unwrap_err();

        assert!(matches!(err, PolyfillError::Validation { .. }));
    }

    #[test]
    fn test_market_order_amounts_differ_for_buy_and_sell() {
        let builder = test_builder();
        let round_config = parse_round_config(Decimal::from_str("0.01").unwrap()).unwrap();

        let (buy_maker, buy_taker) = builder.get_market_order_amounts(
            Side::BUY,
            Decimal::from_str("10").unwrap(),
            Decimal::from_str("0.25").unwrap(),
            round_config,
        );
        let (sell_maker, sell_taker) = builder.get_market_order_amounts(
            Side::SELL,
            Decimal::from_str("10").unwrap(),
            Decimal::from_str("0.25").unwrap(),
            round_config,
        );

        assert_eq!(buy_maker, 10_000_000);
        assert_eq!(buy_taker, 40_000_000);
        assert_eq!(sell_maker, 10_000_000);
        assert_eq!(sell_taker, 2_500_000);
    }

    #[test]
    fn test_calculate_market_price_returns_last_level_for_fak() {
        let builder = test_builder();
        let levels = vec![
            crate::types::BookLevel {
                price: Decimal::from_str("0.40").unwrap(),
                size: Decimal::from_str("2.0").unwrap(),
            },
            crate::types::BookLevel {
                price: Decimal::from_str("0.45").unwrap(),
                size: Decimal::from_str("1.0").unwrap(),
            },
        ];

        let price = builder
            .calculate_market_price(
                &levels,
                Decimal::from_str("10.0").unwrap(),
                Side::SELL,
                OrderType::FAK,
            )
            .unwrap();
        assert_eq!(price, Decimal::from_str("0.45").unwrap());
    }

    #[test]
    fn test_signed_order_json_uses_camel_case_wire_shape() {
        let builder = test_builder();
        let order = builder
            .create_order(
                137,
                &OrderArgs {
                    token_id: "123456".to_string(),
                    price: Decimal::from_str("0.55").unwrap(),
                    size: Decimal::from_str("5.0").unwrap(),
                    side: Side::SELL,
                    expiration: Some(1_900_000_000),
                    builder_code: None,
                    metadata: Some(BYTES32_ZERO.to_string()),
                },
                &CreateOrderOptions {
                    tick_size: Some(Decimal::from_str("0.01").unwrap()),
                    neg_risk: Some(true),
                },
            )
            .unwrap();

        let json = serde_json::to_value(order).unwrap();
        assert!(matches!(json.get("tokenId"), Some(Value::String(_))));
        assert!(matches!(json.get("makerAmount"), Some(Value::String(_))));
        assert!(matches!(json.get("takerAmount"), Some(Value::String(_))));
        assert!(matches!(json.get("signatureType"), Some(Value::Number(_))));
    }

    #[test]
    fn test_seed_generation_uniqueness() {
        let mut seeds = std::collections::HashSet::new();

        // Generate 1000 seeds and ensure they're all unique
        for _ in 0..1000 {
            let seed = generate_seed();
            assert!(seeds.insert(seed), "Duplicate seed generated");
        }
    }

    #[test]
    fn test_seed_generation_range() {
        for _ in 0..100 {
            let seed = generate_seed();
            // Seeds should be positive and within reasonable range
            assert!(seed > 0);
            assert!(seed < u64::MAX);
        }
    }
}