riptide_amm/math/
quote.rs

1use super::SingleSideLiquidity;
2
3use super::error::{CoreError, INVALID_ORACLE_DATA};
4
5use super::oracle::{build_liquidity, build_price, consume_liquidity, OraclePayload};
6
7use borsh::BorshDeserialize;
8
9use riptide_amm_macros::alias;
10
11#[cfg(feature = "wasm")]
12use riptide_amm_macros::wasm_expose;
13
14#[derive(Debug, Clone, Copy, Eq, PartialEq)]
15#[cfg_attr(feature = "wasm", wasm_expose)]
16pub enum QuoteType {
17    TokenAExactIn,
18    TokenAExactOut,
19    TokenBExactIn,
20    TokenBExactOut,
21}
22
23impl QuoteType {
24    pub(crate) fn new(amount_is_token_a: bool, amount_is_input: bool) -> Self {
25        match (amount_is_token_a, amount_is_input) {
26            (true, true) => QuoteType::TokenAExactIn,
27            (true, false) => QuoteType::TokenAExactOut,
28            (false, true) => QuoteType::TokenBExactIn,
29            (false, false) => QuoteType::TokenBExactOut,
30        }
31    }
32
33    pub fn exact_in(&self) -> bool {
34        matches!(self, QuoteType::TokenAExactIn | QuoteType::TokenBExactIn)
35    }
36
37    pub fn exact_out(&self) -> bool {
38        matches!(self, QuoteType::TokenAExactOut | QuoteType::TokenBExactOut)
39    }
40
41    #[alias(output_is_token_b, a_to_b)]
42    pub fn input_is_token_a(&self) -> bool {
43        matches!(self, QuoteType::TokenAExactIn | QuoteType::TokenBExactOut)
44    }
45
46    #[alias(output_is_token_a, b_to_a)]
47    pub fn input_is_token_b(&self) -> bool {
48        matches!(self, QuoteType::TokenBExactIn | QuoteType::TokenAExactOut)
49    }
50}
51
52#[derive(Debug, Clone, Copy, Eq, PartialEq)]
53#[cfg_attr(feature = "wasm", wasm_expose)]
54pub struct Quote {
55    pub amount_in: u64,
56    pub amount_out: u64,
57    pub quote_type: QuoteType,
58}
59
60#[derive(Default, Debug, Clone, Eq, PartialEq)]
61#[cfg_attr(feature = "wasm", wasm_expose)]
62pub struct Prices {
63    pub oracle_price_q64_64: u128,
64    pub best_bid_price_q64_64: u128,
65    pub best_ask_price_q64_64: u128,
66    pub ask_spread_per_m: i32,
67    pub bid_spread_per_m: i32,
68}
69
70#[derive(Default, Debug, Clone, Eq, PartialEq)]
71#[cfg_attr(feature = "wasm", wasm_expose)]
72pub struct Price {
73    pub oracle_price_q64_64: u128,
74    pub best_price_q64_64: u128,
75    pub spread_per_m: i32,
76}
77
78#[cfg_attr(feature = "wasm", wasm_expose)]
79pub fn quote_exact_in(
80    amount: u64,
81    amount_is_token_a: bool,
82    oracle_data: &[u8],
83    reserves_a: u64,
84    reserves_b: u64,
85) -> Result<Quote, CoreError> {
86    let mut oracle_data = oracle_data;
87    let payload = OraclePayload::deserialize(&mut oracle_data).map_err(|_| INVALID_ORACLE_DATA)?;
88    let quote_type = QuoteType::new(amount_is_token_a, true);
89    let liquidity = build_liquidity(&payload, quote_type, reserves_a, reserves_b)?;
90    consume_liquidity(amount, quote_type, &liquidity)
91}
92
93#[cfg_attr(feature = "wasm", wasm_expose)]
94pub fn quote_exact_out(
95    amount: u64,
96    amount_is_token_a: bool,
97    oracle_data: &[u8],
98    reserves_a: u64,
99    reserves_b: u64,
100) -> Result<Quote, CoreError> {
101    let mut oracle_data = oracle_data;
102    let payload = OraclePayload::deserialize(&mut oracle_data).map_err(|_| INVALID_ORACLE_DATA)?;
103    let quote_type = QuoteType::new(amount_is_token_a, false);
104    let liquidity = build_liquidity(&payload, quote_type, reserves_a, reserves_b)?;
105    consume_liquidity(amount, quote_type, &liquidity)
106}
107
108#[cfg_attr(feature = "wasm", wasm_expose)]
109pub fn bid_price(
110    oracle_data: &[u8],
111    liquidity: SingleSideLiquidity,
112    reserves_a: u64,
113    reserves_b: u64,
114) -> Result<Price, CoreError> {
115    let mut oracle_data = oracle_data;
116    let payload = OraclePayload::deserialize(&mut oracle_data).map_err(|_| INVALID_ORACLE_DATA)?;
117    build_price(
118        &liquidity,
119        &payload.data,
120        QuoteType::TokenAExactIn,
121        reserves_a,
122        reserves_b,
123    )
124}
125
126#[cfg_attr(feature = "wasm", wasm_expose)]
127pub fn ask_price(
128    oracle_data: &[u8],
129    liquidity: SingleSideLiquidity,
130    reserves_a: u64,
131    reserves_b: u64,
132) -> Result<Price, CoreError> {
133    let mut oracle_data = oracle_data;
134    let payload = OraclePayload::deserialize(&mut oracle_data).map_err(|_| INVALID_ORACLE_DATA)?;
135    build_price(
136        &liquidity,
137        &payload.data,
138        QuoteType::TokenBExactIn,
139        reserves_a,
140        reserves_b,
141    )
142}
143
144#[cfg_attr(feature = "wasm", wasm_expose)]
145pub fn bid_liquidity(
146    oracle_data: &[u8],
147    reserves_a: u64,
148    reserves_b: u64,
149) -> Result<SingleSideLiquidity, CoreError> {
150    let mut oracle_data = oracle_data;
151    let payload = OraclePayload::deserialize(&mut oracle_data).map_err(|_| INVALID_ORACLE_DATA)?;
152    build_liquidity(&payload, QuoteType::TokenAExactIn, reserves_a, reserves_b)
153}
154
155#[cfg_attr(feature = "wasm", wasm_expose)]
156pub fn ask_liquidity(
157    oracle_data: &[u8],
158    reserves_a: u64,
159    reserves_b: u64,
160) -> Result<SingleSideLiquidity, CoreError> {
161    let mut oracle_data = oracle_data;
162    let payload = OraclePayload::deserialize(&mut oracle_data).map_err(|_| INVALID_ORACLE_DATA)?;
163    build_liquidity(&payload, QuoteType::TokenBExactIn, reserves_a, reserves_b)
164}
riptide_amm/math/quote.rs

riptide_amm/math/
quote.rs
