|
| 1 | +use ic_cdk::{api::time, management_canister::HttpHeader}; |
| 2 | +use serde::Deserialize; |
| 3 | +use serde_json::from_slice; |
| 4 | +use shared::types::{exchange::ExchangeData, token_id::TokenId}; |
| 5 | + |
| 6 | +use crate::{ |
| 7 | + exchange::supplemental::{SupplementalPriceProvider, SupplementalPricesFuture}, |
| 8 | + types::storable::StoredTokenId, |
| 9 | + utils::http_outcall, |
| 10 | +}; |
| 11 | + |
| 12 | +const DEFAULT_BASE_URL: &str = "https://api.icpswap.com"; |
| 13 | +/// `ICPSwap` token info responses are small JSON objects; keep the cap tight for cycle costs. |
| 14 | +const MAX_RESPONSE_BYTES: u64 = 8_192; |
| 15 | + |
| 16 | +#[derive(Debug, Deserialize)] |
| 17 | +struct IcpSwapEnvelope { |
| 18 | + code: i64, |
| 19 | + #[serde(default)] |
| 20 | + data: Option<IcpSwapTokenBody>, |
| 21 | +} |
| 22 | + |
| 23 | +#[derive(Debug, Deserialize)] |
| 24 | +struct IcpSwapTokenBody { |
| 25 | + #[serde(rename = "tokenLedgerId")] |
| 26 | + _token_ledger_id: String, |
| 27 | + price: String, |
| 28 | + #[serde(rename = "priceChange24H")] |
| 29 | + price_change_24h: String, |
| 30 | +} |
| 31 | + |
| 32 | +fn exchange_data_from_icpswap_envelope( |
| 33 | + parsed: IcpSwapEnvelope, |
| 34 | + timestamp_ns: u64, |
| 35 | +) -> Option<ExchangeData> { |
| 36 | + if parsed.code != 0 { |
| 37 | + return None; |
| 38 | + } |
| 39 | + let token = parsed.data?; |
| 40 | + let price: f64 = token.price.parse().ok()?; |
| 41 | + if !price.is_finite() || price <= 0.0 { |
| 42 | + return None; |
| 43 | + } |
| 44 | + let price_24h_change_pct = token |
| 45 | + .price_change_24h |
| 46 | + .parse() |
| 47 | + .ok() |
| 48 | + .filter(|v: &f64| v.is_finite()); |
| 49 | + Some(ExchangeData { |
| 50 | + timestamp_ns, |
| 51 | + price: Some(price), |
| 52 | + price_24h_change_pct, |
| 53 | + market_cap: None, |
| 54 | + }) |
| 55 | +} |
| 56 | + |
| 57 | +fn parse_icpswap_body(body: &[u8]) -> Option<ExchangeData> { |
| 58 | + let parsed: IcpSwapEnvelope = from_slice(body).ok()?; |
| 59 | + exchange_data_from_icpswap_envelope(parsed, time()) |
| 60 | +} |
| 61 | + |
| 62 | +/// Supplemental USD prices for ICRC ledger tokens via [ICPSwap](https://api.icpswap.com). |
| 63 | +#[derive(Debug, Clone)] |
| 64 | +pub(crate) struct IcpSwapProvider { |
| 65 | + base_url: String, |
| 66 | +} |
| 67 | + |
| 68 | +impl Default for IcpSwapProvider { |
| 69 | + fn default() -> Self { |
| 70 | + Self { |
| 71 | + base_url: DEFAULT_BASE_URL.to_string(), |
| 72 | + } |
| 73 | + } |
| 74 | +} |
| 75 | + |
| 76 | +impl IcpSwapProvider { |
| 77 | + #[expect(dead_code)] |
| 78 | + pub(crate) fn with_base_url(base_url: String) -> Self { |
| 79 | + Self { base_url } |
| 80 | + } |
| 81 | + |
| 82 | + async fn fetch_icrc_token_usd( |
| 83 | + &self, |
| 84 | + ledger_text: &str, |
| 85 | + ) -> Result<Option<ExchangeData>, String> { |
| 86 | + let url = format!( |
| 87 | + "{}/info/token/{ledger_text}", |
| 88 | + self.base_url.trim_end_matches('/') |
| 89 | + ); |
| 90 | + |
| 91 | + let response = http_outcall::get( |
| 92 | + &url, |
| 93 | + vec![HttpHeader { |
| 94 | + name: "Accept".to_string(), |
| 95 | + value: "application/json".to_string(), |
| 96 | + }], |
| 97 | + MAX_RESPONSE_BYTES, |
| 98 | + ) |
| 99 | + .await?; |
| 100 | + |
| 101 | + Ok(parse_icpswap_body(&response.body)) |
| 102 | + } |
| 103 | +} |
| 104 | + |
| 105 | +impl SupplementalPriceProvider for IcpSwapProvider { |
| 106 | + fn id(&self) -> &'static str { |
| 107 | + "icpswap" |
| 108 | + } |
| 109 | + |
| 110 | + fn supplement<'a>(&'a self, missing: &'a [StoredTokenId]) -> SupplementalPricesFuture<'a> { |
| 111 | + Box::pin(async move { |
| 112 | + let mut out = Vec::new(); |
| 113 | + |
| 114 | + for stored in missing { |
| 115 | + let StoredTokenId(TokenId::Icrc(ledger_id)) = stored else { |
| 116 | + continue; |
| 117 | + }; |
| 118 | + |
| 119 | + let ledger_text = ledger_id.to_text(); |
| 120 | + |
| 121 | + match self.fetch_icrc_token_usd(&ledger_text).await { |
| 122 | + Ok(Some(data)) => out.push((stored.clone(), data)), |
| 123 | + Ok(None) => {} |
| 124 | + Err(err) => { |
| 125 | + ic_cdk::println!("ICPSwap price fetch for {ledger_text} failed: {err}"); |
| 126 | + } |
| 127 | + } |
| 128 | + } |
| 129 | + |
| 130 | + Ok(out) |
| 131 | + }) |
| 132 | + } |
| 133 | +} |
| 134 | + |
| 135 | +#[cfg(test)] |
| 136 | +mod tests { |
| 137 | + use super::*; |
| 138 | + |
| 139 | + #[test] |
| 140 | + fn parse_icpswap_body_success() { |
| 141 | + let json = br#"{"code":0,"message":null,"data":{"tokenLedgerId":"ryjl3-tyaaa-aaaaa-aaaba-cai","tokenName":"x","tokenSymbol":"X","price":"1.25","priceChange24H":"-2.5","tvlUSD":"0","tvlUSDChange24H":"0","txCount24H":"0","volumeUSD24H":"0","volumeUSD7D":"0","totalVolumeUSD":"0","priceLow24H":"0","priceHigh24H":"0","priceLow7D":"0","priceHigh7D":"0","priceLow30D":"0","priceHigh30D":"0"}}"#; |
| 142 | + let parsed: IcpSwapEnvelope = serde_json::from_slice(json).unwrap(); |
| 143 | + let d = exchange_data_from_icpswap_envelope(parsed, 99).unwrap(); |
| 144 | + assert_eq!(d.timestamp_ns, 99); |
| 145 | + assert_eq!(d.price, Some(1.25)); |
| 146 | + assert_eq!(d.price_24h_change_pct, Some(-2.5)); |
| 147 | + } |
| 148 | + |
| 149 | + #[test] |
| 150 | + fn parse_icpswap_body_rejects_nonzero_code() { |
| 151 | + let json = br#"{"code":404,"message":"not found","data":null}"#; |
| 152 | + let parsed: IcpSwapEnvelope = serde_json::from_slice(json).unwrap(); |
| 153 | + assert!(exchange_data_from_icpswap_envelope(parsed, 0).is_none()); |
| 154 | + } |
| 155 | + |
| 156 | + #[test] |
| 157 | + fn parse_icpswap_body_filters_non_finite_price_change() { |
| 158 | + let json = |
| 159 | + br#"{"code":0,"data":{"tokenLedgerId":"x","price":"1.0","priceChange24H":"NaN"}}"#; |
| 160 | + let parsed: IcpSwapEnvelope = serde_json::from_slice(json).unwrap(); |
| 161 | + let d = exchange_data_from_icpswap_envelope(parsed, 0).unwrap(); |
| 162 | + assert_eq!(d.price, Some(1.0)); |
| 163 | + assert_eq!(d.price_24h_change_pct, None); |
| 164 | + } |
| 165 | + |
| 166 | + #[test] |
| 167 | + fn parse_icpswap_body_rejects_bad_price() { |
| 168 | + let json = br#"{"code":0,"data":{"tokenLedgerId":"x","price":"0","priceChange24H":"0"}}"#; |
| 169 | + let parsed: IcpSwapEnvelope = serde_json::from_slice(json).unwrap(); |
| 170 | + assert!(exchange_data_from_icpswap_envelope(parsed, 0).is_none()); |
| 171 | + } |
| 172 | +} |
0 commit comments