odos-sdk 8.0.0

// SPDX-FileCopyrightText: 2025 Semiotic AI, Inc.
//
// SPDX-License-Identifier: Apache-2.0

use std::fmt::Display;

use alloy_primitives::{Address, U256};
use bon::Builder;
use serde::{Deserialize, Serialize};
use url::Url;

use crate::{error_code::TraceId, OdosError, Result};

#[cfg(feature = "v2")]
use {
    crate::OdosRouterV2::{inputTokenInfo, outputTokenInfo, swapTokenInfo},
    crate::OdosV2Router::{swapCall, OdosV2RouterCalls},
    alloy_primitives::Bytes,
    tracing::debug,
};

#[cfg(feature = "v3")]
use {
    crate::IOdosRouterV3::swapTokenInfo as v3SwapTokenInfo, crate::OdosV3Router::OdosV3RouterCalls,
};

/// API host tier for the Odos API
///
/// Odos provides two API host tiers:
/// - **Public**: Standard API available to all users at <https://api.odos.xyz>
/// - **Enterprise**: Premium API with enhanced features at <https://enterprise-api.odos.xyz>
///
/// Use in combination with [`ApiVersion`] via the [`Endpoint`] type for complete
/// endpoint configuration.
///
/// # Examples
///
/// ```rust
/// use odos_sdk::{ApiHost, ApiVersion, Endpoint};
///
/// // Use directly with Endpoint
/// let endpoint = Endpoint::new(ApiHost::Public, ApiVersion::V2);
///
/// // Or use convenience methods
/// let endpoint = Endpoint::public_v2();
/// ```
#[derive(Copy, Clone, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Deserialize, Serialize)]
#[serde(rename_all = "lowercase")]
pub enum ApiHost {
    /// Public API endpoint <https://docs.odos.xyz/build/api-docs>
    ///
    /// Standard API available to all users. Suitable for most use cases.
    Public,
    /// Enterprise API endpoint <https://docs.odos.xyz/build/enterprise-api>
    ///
    /// Premium API with enhanced features, higher rate limits, and dedicated support.
    /// Requires an API key obtained through the Odos Enterprise program.
    Enterprise,
}

impl ApiHost {
    /// Get the base URL for the API host
    ///
    /// Returns the root URL for the selected host tier without any path segments.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::ApiHost;
    ///
    /// let public = ApiHost::Public;
    /// assert_eq!(public.base_url().as_str(), "https://api.odos.xyz/");
    ///
    /// let enterprise = ApiHost::Enterprise;
    /// assert_eq!(enterprise.base_url().as_str(), "https://enterprise-api.odos.xyz/");
    /// ```
    pub fn base_url(&self) -> Url {
        match self {
            ApiHost::Public => Url::parse("https://api.odos.xyz/").unwrap(),
            ApiHost::Enterprise => Url::parse("https://enterprise-api.odos.xyz/").unwrap(),
        }
    }
}

/// Version of the Odos API
///
/// Odos provides multiple API versions with different features and response formats:
/// - **V2**: Stable production version with comprehensive swap routing
/// - **V3**: Latest version with enhanced features and optimizations
///
/// Use in combination with [`ApiHost`] via the [`Endpoint`] type for complete
/// endpoint configuration.
///
/// # Examples
///
/// ```rust
/// use odos_sdk::{ApiHost, ApiVersion, Endpoint};
///
/// // Recommended: Use V2 for production
/// let endpoint = Endpoint::new(ApiHost::Public, ApiVersion::V2);
///
/// // Or use V3 for latest features
/// let endpoint = Endpoint::new(ApiHost::Public, ApiVersion::V3);
/// ```
#[derive(Copy, Clone, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Deserialize, Serialize)]
#[serde(rename_all = "lowercase")]
pub enum ApiVersion {
    /// API version 2 - Stable production version
    ///
    /// Recommended for most production use cases. Provides comprehensive
    /// swap routing with extensive DEX coverage.
    V2,
    /// API version 3 - Latest version with enhanced features
    ///
    /// Includes optimizations and new features. Check the Odos documentation
    /// for specific enhancements over V2.
    V3,
}

impl ApiVersion {
    /// Get the path segment for this version
    ///
    /// Returns the path component to use in API URLs (e.g., "v2", "v3").
    fn path(&self) -> &'static str {
        match self {
            ApiVersion::V2 => "v2",
            ApiVersion::V3 => "v3",
        }
    }
}

/// Complete API endpoint configuration combining host tier and API version
///
/// The `Endpoint` type provides an ergonomic way to configure both the API host
/// tier (Public/Enterprise) and version (V2/V3) together.
///
/// # Examples
///
/// ## Using convenience constructors (recommended)
///
/// ```rust
/// use odos_sdk::{ClientConfig, Endpoint};
///
/// // Public API V2 (default, recommended for production)
/// let config = ClientConfig {
///     endpoint: Endpoint::public_v2(),
///     ..Default::default()
/// };
///
/// // Enterprise API V3 (latest features)
/// let config = ClientConfig {
///     endpoint: Endpoint::enterprise_v3(),
///     ..Default::default()
/// };
/// ```
///
/// ## Using explicit construction
///
/// ```rust
/// use odos_sdk::{Endpoint, ApiHost, ApiVersion};
///
/// let endpoint = Endpoint::new(ApiHost::Enterprise, ApiVersion::V2);
/// assert_eq!(endpoint.quote_url().as_str(), "https://enterprise-api.odos.xyz/sor/quote/v2");
/// ```
///
/// ## Migration from old API
///
/// ```rust
/// use odos_sdk::{ClientConfig, Endpoint};
///
/// // Old way (still works but deprecated)
/// // let config = ClientConfig {
/// //     endpoint: EndpointBase::Public,
/// //     endpoint_version: EndpointVersion::V2,
/// //     ..Default::default()
/// // };
///
/// // New way
/// let config = ClientConfig {
///     endpoint: Endpoint::public_v2(),
///     ..Default::default()
/// };
/// ```
#[derive(Copy, Clone, Debug, Eq, PartialEq, Hash, Deserialize, Serialize)]
pub struct Endpoint {
    host: ApiHost,
    version: ApiVersion,
}

impl Endpoint {
    /// Create a new endpoint with specific host and version
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::{Endpoint, ApiHost, ApiVersion};
    ///
    /// let endpoint = Endpoint::new(ApiHost::Public, ApiVersion::V2);
    /// ```
    pub const fn new(host: ApiHost, version: ApiVersion) -> Self {
        Self { host, version }
    }

    /// Public API V2 endpoint (default, recommended for production)
    ///
    /// This is the recommended configuration for most production use cases.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::Endpoint;
    ///
    /// let endpoint = Endpoint::public_v2();
    /// assert_eq!(endpoint.quote_url().as_str(), "https://api.odos.xyz/sor/quote/v2");
    /// ```
    pub const fn public_v2() -> Self {
        Self::new(ApiHost::Public, ApiVersion::V2)
    }

    /// Public API V3 endpoint
    ///
    /// Use for latest features and optimizations on the public API.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::Endpoint;
    ///
    /// let endpoint = Endpoint::public_v3();
    /// assert_eq!(endpoint.quote_url().as_str(), "https://api.odos.xyz/sor/quote/v3");
    /// ```
    pub const fn public_v3() -> Self {
        Self::new(ApiHost::Public, ApiVersion::V3)
    }

    /// Enterprise API V2 endpoint
    ///
    /// Use for enterprise tier with V2 API. Requires an API key.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::Endpoint;
    ///
    /// let endpoint = Endpoint::enterprise_v2();
    /// assert_eq!(endpoint.quote_url().as_str(), "https://enterprise-api.odos.xyz/sor/quote/v2");
    /// ```
    pub const fn enterprise_v2() -> Self {
        Self::new(ApiHost::Enterprise, ApiVersion::V2)
    }

    /// Enterprise API V3 endpoint
    ///
    /// Use for enterprise tier with latest V3 features. Requires an API key.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::Endpoint;
    ///
    /// let endpoint = Endpoint::enterprise_v3();
    /// assert_eq!(endpoint.quote_url().as_str(), "https://enterprise-api.odos.xyz/sor/quote/v3");
    /// ```
    pub const fn enterprise_v3() -> Self {
        Self::new(ApiHost::Enterprise, ApiVersion::V3)
    }

    /// Get the quote URL for this endpoint
    ///
    /// Constructs the full URL for the quote endpoint by combining the base URL
    /// with the appropriate version path.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::Endpoint;
    ///
    /// let endpoint = Endpoint::public_v2();
    /// assert_eq!(endpoint.quote_url().as_str(), "https://api.odos.xyz/sor/quote/v2");
    ///
    /// let endpoint = Endpoint::enterprise_v3();
    /// assert_eq!(endpoint.quote_url().as_str(), "https://enterprise-api.odos.xyz/sor/quote/v3");
    /// ```
    pub fn quote_url(&self) -> Url {
        self.host
            .base_url()
            .join(&format!("sor/quote/{}", self.version.path()))
            .unwrap()
    }

    /// Get the assemble URL for this endpoint
    ///
    /// The assemble endpoint is version-independent and constructs transaction data
    /// from a previously obtained quote path ID.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::Endpoint;
    ///
    /// let endpoint = Endpoint::public_v2();
    /// assert_eq!(endpoint.assemble_url().as_str(), "https://api.odos.xyz/sor/assemble");
    /// ```
    pub fn assemble_url(&self) -> Url {
        self.host.base_url().join("sor/assemble").unwrap()
    }

    /// Get the API host tier
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::{Endpoint, ApiHost};
    ///
    /// let endpoint = Endpoint::public_v2();
    /// assert_eq!(endpoint.host(), ApiHost::Public);
    /// ```
    pub const fn host(&self) -> ApiHost {
        self.host
    }

    /// Get the API version
    ///
    /// # Examples
    ///
    /// ```rust
    /// use odos_sdk::{Endpoint, ApiVersion};
    ///
    /// let endpoint = Endpoint::public_v2();
    /// assert_eq!(endpoint.version(), ApiVersion::V2);
    /// ```
    pub const fn version(&self) -> ApiVersion {
        self.version
    }
}

impl Default for Endpoint {
    /// Returns the default endpoint: Public API V2
    ///
    /// This is the recommended configuration for most production use cases.
    fn default() -> Self {
        Self::public_v2()
    }
}

/// Input token for the Odos quote API
#[derive(Clone, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct InputToken {
    token_address: Address,
    // Odos API error message: "Input Amount should be positive integer in string form with < 64 digits[0x6]"
    amount: String,
}

impl InputToken {
    pub fn new(token_address: Address, amount: U256) -> Self {
        Self {
            token_address,
            amount: amount.to_string(),
        }
    }
}

impl From<(Address, U256)> for InputToken {
    fn from((token_address, amount): (Address, U256)) -> Self {
        Self::new(token_address, amount)
    }
}

impl Display for InputToken {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "InputToken {{ token_address: {}, amount: {} }}",
            self.token_address, self.amount
        )
    }
}

/// Output token for the Odos quote API
#[derive(Clone, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct OutputToken {
    token_address: Address,
    proportion: u32,
}

impl OutputToken {
    pub fn new(token_address: Address, proportion: u32) -> Self {
        Self {
            token_address,
            proportion,
        }
    }
}

impl From<(Address, u32)> for OutputToken {
    fn from((token_address, proportion): (Address, u32)) -> Self {
        Self::new(token_address, proportion)
    }
}

impl Display for OutputToken {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "OutputToken {{ token_address: {}, proportion: {} }}",
            self.token_address, self.proportion
        )
    }
}

/// Request to the Odos quote API: <https://docs.odos.xyz/build/api-docs>
///
/// # Using Type-Safe Newtypes
///
/// You can use the type-safe [`Slippage`](crate::Slippage), [`Chain`](crate::Chain),
/// and [`ReferralCode`](crate::ReferralCode) types with their conversion methods:
///
/// ```rust
/// use odos_sdk::{QuoteRequest, Slippage, Chain, ReferralCode};
/// use alloy_primitives::Address;
///
/// let request = QuoteRequest::builder()
///     .chain_id(Chain::ethereum().id())
///     .slippage_limit_percent(Slippage::percent(0.5).unwrap().as_percent())
///     .referral_code(ReferralCode::NONE.code())
///     // ... other fields
///     # .input_tokens(vec![])
///     # .output_tokens(vec![])
///     # .user_addr(Address::ZERO)
///     # .compact(false)
///     # .simple(false)
///     # .disable_rfqs(false)
///     .build();
/// ```
#[derive(Builder, Clone, Debug, Default, PartialEq, PartialOrd, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct QuoteRequest {
    chain_id: u64,
    input_tokens: Vec<InputToken>,
    output_tokens: Vec<OutputToken>,
    slippage_limit_percent: f64,
    user_addr: Address,
    compact: bool,
    simple: bool,
    referral_code: u32,
    disable_rfqs: bool,
    #[builder(default)]
    source_blacklist: Vec<String>,
}

/// Single quote response from the Odos quote API: <https://docs.odos.xyz/build/api-docs>
#[derive(Clone, Debug, PartialEq, PartialOrd, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct SingleQuoteResponse {
    block_number: u64,
    data_gas_estimate: u64,
    gas_estimate: f64,
    gas_estimate_value: f64,
    gwei_per_gas: f64,
    in_amounts: Vec<String>,
    in_tokens: Vec<Address>,
    in_values: Vec<f64>,
    net_out_value: f64,
    out_amounts: Vec<String>,
    out_tokens: Vec<Address>,
    out_values: Vec<f64>,
    /// Partner fee percentage. Defaults to 0.0 if not present (V3 API compatibility).
    #[serde(default)]
    partner_fee_percent: f64,
    path_id: String,
    path_viz: Option<String>,
    percent_diff: f64,
    price_impact: f64,
}

impl SingleQuoteResponse {
    /// Get the first input amount of the quote.
    pub fn in_amount(&self) -> Option<&String> {
        self.in_amounts.first()
    }

    /// Get the data gas estimate of the quote
    pub fn data_gas_estimate(&self) -> u64 {
        self.data_gas_estimate
    }

    /// Get the block number of the quote
    pub fn get_block_number(&self) -> u64 {
        self.block_number
    }

    /// Get the gas estimate of the quote
    pub fn gas_estimate(&self) -> f64 {
        self.gas_estimate
    }

    /// Get the estimated gas cost value of the quote.
    pub fn gas_estimate_value(&self) -> f64 {
        self.gas_estimate_value
    }

    /// Get the gas price used by the quote in gwei.
    pub fn gwei_per_gas(&self) -> f64 {
        self.gwei_per_gas
    }

    /// Get the in amounts of the quote
    pub fn in_amounts_iter(&self) -> impl Iterator<Item = &String> {
        self.in_amounts.iter()
    }

    /// Get the in amount of the quote
    pub fn in_amount_u256(&self) -> Result<U256> {
        let amount_str = self
            .in_amounts_iter()
            .next()
            .ok_or_else(|| OdosError::missing_data("Missing input amount"))?;
        let amount: u128 = amount_str
            .parse()
            .map_err(|_| OdosError::invalid_input("Invalid input amount format"))?;
        Ok(U256::from(amount))
    }

    /// Get the out amount of the quote
    pub fn out_amount(&self) -> Option<&String> {
        self.out_amounts.first()
    }

    /// Get the out amounts of the quote
    pub fn out_amounts_iter(&self) -> impl Iterator<Item = &String> {
        self.out_amounts.iter()
    }

    /// Get the in tokens of the quote
    pub fn in_tokens_iter(&self) -> impl Iterator<Item = &Address> {
        self.in_tokens.iter()
    }

    /// Get the in token of the quote
    pub fn first_in_token(&self) -> Option<&Address> {
        self.in_tokens.first()
    }

    pub fn out_tokens_iter(&self) -> impl Iterator<Item = &Address> {
        self.out_tokens.iter()
    }

    /// Get the out token of the quote
    pub fn first_out_token(&self) -> Option<&Address> {
        self.out_tokens.first()
    }

    /// Get the out values of the quote
    pub fn out_values_iter(&self) -> impl Iterator<Item = &f64> {
        self.out_values.iter()
    }

    /// Get the path id of the quote
    pub fn path_id(&self) -> &str {
        &self.path_id
    }

    /// Get the path id as a vector of bytes
    pub fn path_definition_as_vec_u8(&self) -> Vec<u8> {
        self.path_id().as_bytes().to_vec()
    }

    /// Get the swap input token and amount
    pub fn swap_input_token_and_amount(&self) -> Result<(Address, U256)> {
        let input_token = *self
            .in_tokens_iter()
            .next()
            .ok_or_else(|| OdosError::missing_data("Missing input token"))?;
        let input_amount_in_u256 = self.in_amount_u256()?;

        Ok((input_token, input_amount_in_u256))
    }

    /// Get the price impact of the quote
    pub fn price_impact(&self) -> f64 {
        self.price_impact
    }

    /// Get the net output value of the quote.
    pub fn net_out_value(&self) -> f64 {
        self.net_out_value
    }

    /// Get the partner fee percent applied to the quote.
    pub fn partner_fee_percent(&self) -> f64 {
        self.partner_fee_percent
    }
}

/// Error response from the Odos API
///
/// When the Odos API returns an error, it includes:
/// - `detail`: Human-readable error message
/// - `traceId`: UUID for tracking the error in Odos logs
/// - `errorCode`: Numeric error code indicating the specific error type
///
/// Example error response:
/// ```json
/// {
///   "detail": "Error getting quote, please try again",
///   "traceId": "10becdc8-a021-4491-8201-a17b657204e0",
///   "errorCode": 2999
/// }
/// ```
#[derive(Clone, Debug, PartialEq, Eq, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct OdosApiErrorResponse {
    /// Human-readable error message
    pub detail: String,
    /// Trace ID for debugging (UUID)
    pub trace_id: TraceId,
    /// Numeric error code
    pub error_code: u16,
}

/// Swap inputs for the Odos assemble API
///
/// Available only when the `v2` feature is enabled.
#[cfg(feature = "v2")]
#[derive(Clone, Debug)]
pub struct SwapInputs {
    executor: Address,
    path_definition: Bytes,
    input_token_info: inputTokenInfo,
    output_token_info: outputTokenInfo,
    value_out_min: U256,
}

#[cfg(feature = "v2")]
impl TryFrom<OdosV2RouterCalls> for SwapInputs {
    type Error = OdosError;

    fn try_from(swap: OdosV2RouterCalls) -> std::result::Result<Self, Self::Error> {
        match swap {
            OdosV2RouterCalls::swap(call) => {
                debug!(
                    swap_type = "V2Router",
                    input.token = %call.tokenInfo.inputToken,
                    input.amount_wei = %call.tokenInfo.inputAmount,
                    output.token = %call.tokenInfo.outputToken,
                    output.min_wei = %call.tokenInfo.outputMin,
                    executor = %call.executor,
                    "Extracting swap inputs from V2 router call"
                );

                let swapCall {
                    executor,
                    pathDefinition,
                    referralCode,
                    tokenInfo,
                } = call;

                let _referral_code = referralCode;

                let swapTokenInfo {
                    inputToken,
                    inputAmount,
                    inputReceiver,
                    outputMin,
                    outputQuote,
                    outputReceiver,
                    outputToken,
                } = tokenInfo;

                let _output_quote = outputQuote;

                Ok(Self {
                    executor,
                    path_definition: pathDefinition,
                    input_token_info: inputTokenInfo {
                        tokenAddress: inputToken,
                        amountIn: inputAmount,
                        receiver: inputReceiver,
                    },
                    output_token_info: outputTokenInfo {
                        tokenAddress: outputToken,
                        relativeValue: U256::from(1),
                        receiver: outputReceiver,
                    },
                    value_out_min: outputMin,
                })
            }
            _ => Err(OdosError::invalid_input("Unexpected OdosV2RouterCalls")),
        }
    }
}

#[cfg(feature = "v3")]
impl TryFrom<OdosV3RouterCalls> for SwapInputs {
    type Error = OdosError;

    fn try_from(swap: OdosV3RouterCalls) -> std::result::Result<Self, Self::Error> {
        match swap {
            OdosV3RouterCalls::swap(call) => {
                debug!(
                    swap_type = "V3Router",
                    input.token = %call.tokenInfo.inputToken,
                    input.amount_wei = %call.tokenInfo.inputAmount,
                    output.token = %call.tokenInfo.outputToken,
                    output.min_wei = %call.tokenInfo.outputMin,
                    executor = %call.executor,
                    "Extracting swap inputs from V3 router call"
                );

                let v3SwapTokenInfo {
                    inputToken,
                    inputAmount,
                    inputReceiver,
                    outputMin,
                    outputQuote,
                    outputReceiver,
                    outputToken,
                } = call.tokenInfo;

                let _output_quote = outputQuote;
                let _referral_info = call.referralInfo;

                Ok(Self {
                    executor: call.executor,
                    path_definition: call.pathDefinition,
                    input_token_info: inputTokenInfo {
                        tokenAddress: inputToken,
                        amountIn: inputAmount,
                        receiver: inputReceiver,
                    },
                    output_token_info: outputTokenInfo {
                        tokenAddress: outputToken,
                        relativeValue: U256::from(1),
                        receiver: outputReceiver,
                    },
                    value_out_min: outputMin,
                })
            }
            _ => Err(OdosError::invalid_input("Unexpected OdosV3RouterCalls")),
        }
    }
}

#[cfg(feature = "v2")]
impl SwapInputs {
    /// Get the executor of the swap
    pub fn executor(&self) -> Address {
        self.executor
    }

    /// Get the path definition of the swap
    pub fn path_definition(&self) -> &Bytes {
        &self.path_definition
    }

    /// Get the token address of the swap
    pub fn token_address(&self) -> Address {
        self.input_token_info.tokenAddress
    }

    /// Get the amount in of the swap
    pub fn amount_in(&self) -> U256 {
        self.input_token_info.amountIn
    }

    /// Get the receiver of the swap
    pub fn receiver(&self) -> Address {
        self.input_token_info.receiver
    }

    /// Get the relative value of the swap
    pub fn relative_value(&self) -> U256 {
        self.output_token_info.relativeValue
    }

    /// Get the output token address of the swap
    pub fn output_token_address(&self) -> Address {
        self.output_token_info.tokenAddress
    }

    /// Get the value out min of the swap
    pub fn value_out_min(&self) -> U256 {
        self.value_out_min
    }
}

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

    #[test]
    fn test_api_host_base_url() {
        assert_eq!(ApiHost::Public.base_url().as_str(), "https://api.odos.xyz/");
        assert_eq!(
            ApiHost::Enterprise.base_url().as_str(),
            "https://enterprise-api.odos.xyz/"
        );
    }

    #[test]
    fn test_api_version_path() {
        assert_eq!(ApiVersion::V2.path(), "v2");
        assert_eq!(ApiVersion::V3.path(), "v3");
    }

    #[test]
    fn test_endpoint_constructors() {
        let endpoint = Endpoint::public_v2();
        assert_eq!(endpoint.host(), ApiHost::Public);
        assert_eq!(endpoint.version(), ApiVersion::V2);

        let endpoint = Endpoint::public_v3();
        assert_eq!(endpoint.host(), ApiHost::Public);
        assert_eq!(endpoint.version(), ApiVersion::V3);

        let endpoint = Endpoint::enterprise_v2();
        assert_eq!(endpoint.host(), ApiHost::Enterprise);
        assert_eq!(endpoint.version(), ApiVersion::V2);

        let endpoint = Endpoint::enterprise_v3();
        assert_eq!(endpoint.host(), ApiHost::Enterprise);
        assert_eq!(endpoint.version(), ApiVersion::V3);

        let endpoint = Endpoint::new(ApiHost::Public, ApiVersion::V2);
        assert_eq!(endpoint.host(), ApiHost::Public);
        assert_eq!(endpoint.version(), ApiVersion::V2);
    }

    #[test]
    fn test_endpoint_quote_urls() {
        assert_eq!(
            Endpoint::public_v2().quote_url().as_str(),
            "https://api.odos.xyz/sor/quote/v2"
        );
        assert_eq!(
            Endpoint::public_v3().quote_url().as_str(),
            "https://api.odos.xyz/sor/quote/v3"
        );
        assert_eq!(
            Endpoint::enterprise_v2().quote_url().as_str(),
            "https://enterprise-api.odos.xyz/sor/quote/v2"
        );
        assert_eq!(
            Endpoint::enterprise_v3().quote_url().as_str(),
            "https://enterprise-api.odos.xyz/sor/quote/v3"
        );
    }

    #[test]
    fn test_endpoint_assemble_urls() {
        assert_eq!(
            Endpoint::public_v2().assemble_url().as_str(),
            "https://api.odos.xyz/sor/assemble"
        );
        assert_eq!(
            Endpoint::public_v3().assemble_url().as_str(),
            "https://api.odos.xyz/sor/assemble"
        );
        assert_eq!(
            Endpoint::enterprise_v2().assemble_url().as_str(),
            "https://enterprise-api.odos.xyz/sor/assemble"
        );
        assert_eq!(
            Endpoint::enterprise_v3().assemble_url().as_str(),
            "https://enterprise-api.odos.xyz/sor/assemble"
        );
    }

    #[test]
    fn test_endpoint_default() {
        let endpoint = Endpoint::default();
        assert_eq!(endpoint.host(), ApiHost::Public);
        assert_eq!(endpoint.version(), ApiVersion::V2);
        assert_eq!(
            endpoint.quote_url().as_str(),
            "https://api.odos.xyz/sor/quote/v2"
        );
    }

    #[test]
    fn test_endpoint_equality() {
        assert_eq!(
            Endpoint::public_v2(),
            Endpoint::new(ApiHost::Public, ApiVersion::V2)
        );
        assert_eq!(
            Endpoint::enterprise_v3(),
            Endpoint::new(ApiHost::Enterprise, ApiVersion::V3)
        );
        assert_ne!(Endpoint::public_v2(), Endpoint::public_v3());
        assert_ne!(Endpoint::public_v2(), Endpoint::enterprise_v2());
    }
}