feat: Take Chain object containing native/wrapped addresses
- This way this chain object contains everything we need, we don't need to worry about doing any transformation or calling any supplementary functions inside any of the encoders - Needed to move our new Chain object to a higher level since this is used in the higher-level encoder traits. This required some weird default values in the constants in order to avoid using alloy's hex literal. I could have instead opted to make Bytes parse a string I think, though this would mean possibly returning an error at the constants level, which is not nice either. Question: - Do we want the user to be in charge of passing the native and wrapped token every single time? This may be a bit annoying for the user. For now, I have defaulted to those in constants.rs, this would take 5 mins to remove though if you don't like it, and it would get rid of this complicated bytes initialization.
This commit is contained in:
1
Cargo.lock
generated
1
Cargo.lock
generated
@@ -4242,6 +4242,7 @@ dependencies = [
|
|||||||
"lazy_static",
|
"lazy_static",
|
||||||
"num-bigint",
|
"num-bigint",
|
||||||
"num-traits",
|
"num-traits",
|
||||||
|
"once_cell",
|
||||||
"rstest",
|
"rstest",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ alloy = { version = "0.9.2", features = ["providers", "rpc-types-eth", "eip712",
|
|||||||
alloy-sol-types = { version = "0.8.14", optional = true }
|
alloy-sol-types = { version = "0.8.14", optional = true }
|
||||||
alloy-primitives = { version = "0.8.9", optional = true }
|
alloy-primitives = { version = "0.8.9", optional = true }
|
||||||
tycho-core = { git = "https://github.com/propeller-heads/tycho-indexer.git", package = "tycho-core", tag = "0.46.0" }
|
tycho-core = { git = "https://github.com/propeller-heads/tycho-indexer.git", package = "tycho-core", tag = "0.46.0" }
|
||||||
|
once_cell = "1.20.2"
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
rstest = "0.24.0"
|
rstest = "0.24.0"
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ use std::str::FromStr;
|
|||||||
|
|
||||||
use num_bigint::BigUint;
|
use num_bigint::BigUint;
|
||||||
use tycho_core::{
|
use tycho_core::{
|
||||||
dto::{Chain, ProtocolComponent},
|
dto::{Chain as TychoCoreChain, ProtocolComponent},
|
||||||
Bytes,
|
Bytes,
|
||||||
};
|
};
|
||||||
use tycho_execution::encoding::{
|
use tycho_execution::encoding::{
|
||||||
@@ -10,7 +10,7 @@ use tycho_execution::encoding::{
|
|||||||
strategy_encoder::strategy_encoder_registry::EVMStrategyEncoderRegistry,
|
strategy_encoder::strategy_encoder_registry::EVMStrategyEncoderRegistry,
|
||||||
tycho_encoder::EVMTychoEncoder,
|
tycho_encoder::EVMTychoEncoder,
|
||||||
},
|
},
|
||||||
models::{Solution, Swap},
|
models::{Chain, Solution, Swap},
|
||||||
strategy_encoder::StrategyEncoderRegistry,
|
strategy_encoder::StrategyEncoderRegistry,
|
||||||
tycho_encoder::TychoEncoder,
|
tycho_encoder::TychoEncoder,
|
||||||
};
|
};
|
||||||
@@ -24,11 +24,13 @@ fn main() {
|
|||||||
.expect("Failed to create user address");
|
.expect("Failed to create user address");
|
||||||
let executors_file_path = "src/encoding/config/executor_addresses.json";
|
let executors_file_path = "src/encoding/config/executor_addresses.json";
|
||||||
|
|
||||||
|
let eth_chain = Chain::from_tycho_core_chain(TychoCoreChain::Ethereum, None, None)
|
||||||
|
.expect("Failed to create chain.");
|
||||||
// Initialize the encoder
|
// Initialize the encoder
|
||||||
let strategy_encoder_registry =
|
let strategy_encoder_registry =
|
||||||
EVMStrategyEncoderRegistry::new(Chain::Ethereum, executors_file_path, signer_pk.clone())
|
EVMStrategyEncoderRegistry::new(eth_chain.clone(), executors_file_path, signer_pk.clone())
|
||||||
.expect("Failed to create strategy encoder registry");
|
.expect("Failed to create strategy encoder registry");
|
||||||
let encoder = EVMTychoEncoder::new(strategy_encoder_registry, router_address, Chain::Ethereum)
|
let encoder = EVMTychoEncoder::new(strategy_encoder_registry, router_address, eth_chain)
|
||||||
.expect("Failed to create encoder");
|
.expect("Failed to create encoder");
|
||||||
|
|
||||||
// ------------------- Encode a simple swap -------------------
|
// ------------------- Encode a simple swap -------------------
|
||||||
|
|||||||
19
src/encoding/constants.rs
Normal file
19
src/encoding/constants.rs
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
use std::collections::HashMap;
|
||||||
|
|
||||||
|
use once_cell::sync::Lazy;
|
||||||
|
use tycho_core::{dto::Chain, Bytes};
|
||||||
|
const ETH_NATIVE: &[u8] = &[
|
||||||
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
||||||
|
0x00, 0x00, 0x00, 0x00,
|
||||||
|
];
|
||||||
|
|
||||||
|
const ETH_WRAPPED: &[u8] = &[
|
||||||
|
0xc0, 0x2a, 0xaa, 0x39, 0xb2, 0x23, 0xfe, 0x8d, 0x0a, 0x0e, 0x5c, 0x4f, 0x27, 0xea, 0xd9, 0x08,
|
||||||
|
0x3c, 0x75, 0x6c, 0xc2,
|
||||||
|
];
|
||||||
|
|
||||||
|
pub static NATIVE_ADDRESSES: Lazy<HashMap<Chain, Bytes>> =
|
||||||
|
Lazy::new(|| HashMap::from([(Chain::Ethereum, Bytes::from(ETH_NATIVE))]));
|
||||||
|
|
||||||
|
pub static WRAPPED_ADDRESSES: Lazy<HashMap<Chain, Bytes>> =
|
||||||
|
Lazy::new(|| HashMap::from([(Chain::Ethereum, Bytes::from(ETH_WRAPPED))]));
|
||||||
@@ -12,15 +12,15 @@ use alloy_sol_types::{eip712_domain, sol, SolStruct, SolValue};
|
|||||||
use chrono::Utc;
|
use chrono::Utc;
|
||||||
use num_bigint::BigUint;
|
use num_bigint::BigUint;
|
||||||
use tokio::runtime::Runtime;
|
use tokio::runtime::Runtime;
|
||||||
use tycho_core::{dto::Chain, Bytes};
|
use tycho_core::Bytes;
|
||||||
|
|
||||||
use crate::encoding::{
|
use crate::encoding::{
|
||||||
errors::EncodingError,
|
errors::EncodingError,
|
||||||
evm::{
|
evm::{
|
||||||
approvals::protocol_approvals_manager::get_client,
|
approvals::protocol_approvals_manager::get_client,
|
||||||
models::ChainId,
|
|
||||||
utils::{biguint_to_u256, bytes_to_address, encode_input},
|
utils::{biguint_to_u256, bytes_to_address, encode_input},
|
||||||
},
|
},
|
||||||
|
models::{Chain, ChainId},
|
||||||
};
|
};
|
||||||
|
|
||||||
/// Struct for managing Permit2 operations, including encoding approvals and fetching allowance
|
/// Struct for managing Permit2 operations, including encoding approvals and fetching allowance
|
||||||
@@ -60,7 +60,6 @@ sol! {
|
|||||||
|
|
||||||
impl Permit2 {
|
impl Permit2 {
|
||||||
pub fn new(signer_pk: String, chain: Chain) -> Result<Self, EncodingError> {
|
pub fn new(signer_pk: String, chain: Chain) -> Result<Self, EncodingError> {
|
||||||
let chain_id = ChainId::from(chain);
|
|
||||||
let runtime = Runtime::new()
|
let runtime = Runtime::new()
|
||||||
.map_err(|_| EncodingError::FatalError("Failed to create runtime".to_string()))?;
|
.map_err(|_| EncodingError::FatalError("Failed to create runtime".to_string()))?;
|
||||||
let client = runtime.block_on(get_client())?;
|
let client = runtime.block_on(get_client())?;
|
||||||
@@ -76,7 +75,7 @@ impl Permit2 {
|
|||||||
client,
|
client,
|
||||||
runtime,
|
runtime,
|
||||||
signer,
|
signer,
|
||||||
chain_id,
|
chain_id: chain.id,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -199,11 +198,28 @@ mod tests {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn eth() -> Bytes {
|
||||||
|
Bytes::from_str("0x0000000000000000000000000000000000000000").unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn weth() -> Bytes {
|
||||||
|
Bytes::from_str("0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2").unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn eth_chain() -> Chain {
|
||||||
|
Chain {
|
||||||
|
id: ChainId(1),
|
||||||
|
name: String::from("ethereum"),
|
||||||
|
native_token: eth(),
|
||||||
|
wrapped_token: weth(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_get_existing_allowance() {
|
fn test_get_existing_allowance() {
|
||||||
let signer_pk =
|
let signer_pk =
|
||||||
"4c0883a69102937d6231471b5dbb6204fe512961708279feb1be6ae5538da033".to_string();
|
"4c0883a69102937d6231471b5dbb6204fe512961708279feb1be6ae5538da033".to_string();
|
||||||
let manager = Permit2::new(signer_pk, Chain::Ethereum).unwrap();
|
let manager = Permit2::new(signer_pk, eth_chain()).unwrap();
|
||||||
|
|
||||||
let token = Bytes::from_str("0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48").unwrap();
|
let token = Bytes::from_str("0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48").unwrap();
|
||||||
let owner = Bytes::from_str("0x2c6a3cd97c6283b95ac8c5a4459ebb0d5fd404f4").unwrap();
|
let owner = Bytes::from_str("0x2c6a3cd97c6283b95ac8c5a4459ebb0d5fd404f4").unwrap();
|
||||||
@@ -223,7 +239,7 @@ mod tests {
|
|||||||
// Set up a mock private key for signing
|
// Set up a mock private key for signing
|
||||||
let private_key =
|
let private_key =
|
||||||
"4c0883a69102937d6231471b5dbb6204fe512961708279feb1be6ae5538da033".to_string();
|
"4c0883a69102937d6231471b5dbb6204fe512961708279feb1be6ae5538da033".to_string();
|
||||||
let permit2 = Permit2::new(private_key, Chain::Ethereum).expect("Failed to create Permit2");
|
let permit2 = Permit2::new(private_key, eth_chain()).expect("Failed to create Permit2");
|
||||||
|
|
||||||
let owner = Bytes::from_str("0x2c6a3cd97c6283b95ac8c5a4459ebb0d5fd404f4").unwrap();
|
let owner = Bytes::from_str("0x2c6a3cd97c6283b95ac8c5a4459ebb0d5fd404f4").unwrap();
|
||||||
let spender = Bytes::from_str("0xba12222222228d8ba445958a75a0704d566bf2c8").unwrap();
|
let spender = Bytes::from_str("0xba12222222228d8ba445958a75a0704d566bf2c8").unwrap();
|
||||||
@@ -264,7 +280,7 @@ mod tests {
|
|||||||
"0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80".to_string();
|
"0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80".to_string();
|
||||||
|
|
||||||
let permit2 =
|
let permit2 =
|
||||||
Permit2::new(anvil_private_key, Chain::Ethereum).expect("Failed to create Permit2");
|
Permit2::new(anvil_private_key, eth_chain()).expect("Failed to create Permit2");
|
||||||
|
|
||||||
let token = Bytes::from_str("0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48").unwrap();
|
let token = Bytes::from_str("0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48").unwrap();
|
||||||
let amount = BigUint::from(1000u64);
|
let amount = BigUint::from(1000u64);
|
||||||
|
|||||||
@@ -1,18 +0,0 @@
|
|||||||
use alloy_primitives::hex;
|
|
||||||
use tycho_core::{dto::Chain, Bytes};
|
|
||||||
|
|
||||||
pub fn native_address(chain: Chain) -> Bytes {
|
|
||||||
match chain {
|
|
||||||
Chain::Ethereum => Bytes::from(hex!("0000000000000000000000000000000000000000").to_vec()),
|
|
||||||
// Placeholder values for other chains; update with real addresses
|
|
||||||
_ => Bytes::from(hex!("0000000000000000000000000000000000000000").to_vec()),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn wrapped_address(chain: Chain) -> Bytes {
|
|
||||||
match chain {
|
|
||||||
Chain::Ethereum => Bytes::from(hex!("c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2").to_vec()),
|
|
||||||
// Placeholder values for other chains; update with real addresses
|
|
||||||
_ => Bytes::from(hex!("0000000000000000000000000000000000000000").to_vec()),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,6 +1,4 @@
|
|||||||
pub mod approvals;
|
pub mod approvals;
|
||||||
mod constants;
|
|
||||||
mod models;
|
|
||||||
pub mod strategy_encoder;
|
pub mod strategy_encoder;
|
||||||
mod swap_encoder;
|
mod swap_encoder;
|
||||||
pub mod tycho_encoder;
|
pub mod tycho_encoder;
|
||||||
|
|||||||
@@ -1,20 +0,0 @@
|
|||||||
use tycho_core::dto::Chain;
|
|
||||||
|
|
||||||
pub struct ChainId(u64);
|
|
||||||
|
|
||||||
impl ChainId {
|
|
||||||
pub fn id(&self) -> u64 {
|
|
||||||
self.0
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl From<Chain> for ChainId {
|
|
||||||
fn from(chain: Chain) -> Self {
|
|
||||||
match chain {
|
|
||||||
Chain::Ethereum => ChainId(1),
|
|
||||||
Chain::ZkSync => ChainId(324),
|
|
||||||
Chain::Arbitrum => ChainId(42161),
|
|
||||||
Chain::Starknet => ChainId(0),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,14 +1,12 @@
|
|||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use tycho_core::dto::Chain;
|
|
||||||
|
|
||||||
use crate::encoding::{
|
use crate::encoding::{
|
||||||
errors::EncodingError,
|
errors::EncodingError,
|
||||||
evm::{
|
evm::{
|
||||||
strategy_encoder::strategy_encoders::{ExecutorStrategyEncoder, SplitSwapStrategyEncoder},
|
strategy_encoder::strategy_encoders::{ExecutorStrategyEncoder, SplitSwapStrategyEncoder},
|
||||||
swap_encoder::swap_encoder_registry::SwapEncoderRegistry,
|
swap_encoder::swap_encoder_registry::SwapEncoderRegistry,
|
||||||
},
|
},
|
||||||
models::Solution,
|
models::{Chain, Solution},
|
||||||
strategy_encoder::{StrategyEncoder, StrategyEncoderRegistry},
|
strategy_encoder::{StrategyEncoder, StrategyEncoderRegistry},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -22,7 +20,7 @@ impl StrategyEncoderRegistry for EVMStrategyEncoderRegistry {
|
|||||||
executors_file_path: &str,
|
executors_file_path: &str,
|
||||||
signer_pk: Option<String>,
|
signer_pk: Option<String>,
|
||||||
) -> Result<Self, EncodingError> {
|
) -> Result<Self, EncodingError> {
|
||||||
let swap_encoder_registry = SwapEncoderRegistry::new(executors_file_path, Chain::Ethereum)?;
|
let swap_encoder_registry = SwapEncoderRegistry::new(executors_file_path, chain.clone())?;
|
||||||
|
|
||||||
let mut strategies: HashMap<String, Box<dyn StrategyEncoder>> = HashMap::new();
|
let mut strategies: HashMap<String, Box<dyn StrategyEncoder>> = HashMap::new();
|
||||||
strategies.insert(
|
strategies.insert(
|
||||||
|
|||||||
@@ -7,17 +7,16 @@ use std::{
|
|||||||
use alloy_primitives::{aliases::U24, FixedBytes, U256, U8};
|
use alloy_primitives::{aliases::U24, FixedBytes, U256, U8};
|
||||||
use alloy_sol_types::SolValue;
|
use alloy_sol_types::SolValue;
|
||||||
use num_bigint::BigUint;
|
use num_bigint::BigUint;
|
||||||
use tycho_core::{dto::Chain, keccak256, Bytes};
|
use tycho_core::{keccak256, Bytes};
|
||||||
|
|
||||||
use crate::encoding::{
|
use crate::encoding::{
|
||||||
errors::EncodingError,
|
errors::EncodingError,
|
||||||
evm::{
|
evm::{
|
||||||
approvals::permit2::Permit2,
|
approvals::permit2::Permit2,
|
||||||
constants::{native_address, wrapped_address},
|
|
||||||
swap_encoder::swap_encoder_registry::SwapEncoderRegistry,
|
swap_encoder::swap_encoder_registry::SwapEncoderRegistry,
|
||||||
utils::{biguint_to_u256, bytes_to_address, encode_input, percentage_to_uint24},
|
utils::{biguint_to_u256, bytes_to_address, encode_input, percentage_to_uint24},
|
||||||
},
|
},
|
||||||
models::{EncodingContext, NativeAction, Solution, Swap},
|
models::{Chain, EncodingContext, NativeAction, Solution, Swap},
|
||||||
strategy_encoder::StrategyEncoder,
|
strategy_encoder::StrategyEncoder,
|
||||||
swap_encoder::SwapEncoder,
|
swap_encoder::SwapEncoder,
|
||||||
};
|
};
|
||||||
@@ -62,8 +61,8 @@ pub struct SplitSwapStrategyEncoder {
|
|||||||
swap_encoder_registry: SwapEncoderRegistry,
|
swap_encoder_registry: SwapEncoderRegistry,
|
||||||
permit2: Permit2,
|
permit2: Permit2,
|
||||||
selector: String,
|
selector: String,
|
||||||
wrapped_address: Bytes,
|
|
||||||
native_address: Bytes,
|
native_address: Bytes,
|
||||||
|
wrapped_address: Bytes,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl SplitSwapStrategyEncoder {
|
impl SplitSwapStrategyEncoder {
|
||||||
@@ -73,14 +72,12 @@ impl SplitSwapStrategyEncoder {
|
|||||||
swap_encoder_registry: SwapEncoderRegistry,
|
swap_encoder_registry: SwapEncoderRegistry,
|
||||||
) -> Result<Self, EncodingError> {
|
) -> Result<Self, EncodingError> {
|
||||||
let selector = "swap(uint256,address,address,uint256,bool,bool,uint256,address,((address,uint160,uint48,uint48),address,uint256),bytes,bytes)".to_string();
|
let selector = "swap(uint256,address,address,uint256,bool,bool,uint256,address,((address,uint160,uint48,uint48),address,uint256),bytes,bytes)".to_string();
|
||||||
let wrapped_address = wrapped_address(chain);
|
|
||||||
let native_address = native_address(chain);
|
|
||||||
Ok(Self {
|
Ok(Self {
|
||||||
permit2: Permit2::new(signer_pk, chain)?,
|
permit2: Permit2::new(signer_pk, chain.clone())?,
|
||||||
selector,
|
selector,
|
||||||
swap_encoder_registry,
|
swap_encoder_registry,
|
||||||
wrapped_address,
|
native_address: chain.native_token.clone(),
|
||||||
native_address,
|
wrapped_address: chain.wrapped_token.clone(),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -434,26 +431,37 @@ mod tests {
|
|||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
|
|
||||||
use alloy::hex::encode;
|
use alloy::hex::encode;
|
||||||
|
use alloy_primitives::hex;
|
||||||
use num_bigint::BigUint;
|
use num_bigint::BigUint;
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
use tycho_core::{dto::ProtocolComponent, Bytes};
|
use tycho_core::{
|
||||||
|
dto::{Chain as TychoCoreChain, ProtocolComponent},
|
||||||
|
Bytes,
|
||||||
|
};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::encoding::{
|
use crate::encoding::models::{ChainId, Swap};
|
||||||
evm::constants::{native_address, wrapped_address},
|
|
||||||
models::Swap,
|
fn eth_chain() -> Chain {
|
||||||
};
|
Chain {
|
||||||
|
id: ChainId(1),
|
||||||
|
name: TychoCoreChain::Ethereum.to_string(),
|
||||||
|
native_token: eth(),
|
||||||
|
wrapped_token: weth(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
fn eth() -> Bytes {
|
fn eth() -> Bytes {
|
||||||
native_address(Chain::Ethereum)
|
Bytes::from(hex!("0000000000000000000000000000000000000000").to_vec())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn weth() -> Bytes {
|
fn weth() -> Bytes {
|
||||||
wrapped_address(Chain::Ethereum)
|
Bytes::from(hex!("c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2").to_vec())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_swap_encoder_registry() -> SwapEncoderRegistry {
|
fn get_swap_encoder_registry() -> SwapEncoderRegistry {
|
||||||
SwapEncoderRegistry::new("src/encoding/config/executor_addresses.json", Chain::Ethereum)
|
let eth_chain = eth_chain();
|
||||||
.unwrap()
|
SwapEncoderRegistry::new("src/encoding/config/executor_addresses.json", eth_chain).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@@ -567,8 +575,7 @@ mod tests {
|
|||||||
};
|
};
|
||||||
let swap_encoder_registry = get_swap_encoder_registry();
|
let swap_encoder_registry = get_swap_encoder_registry();
|
||||||
let encoder =
|
let encoder =
|
||||||
SplitSwapStrategyEncoder::new(private_key, Chain::Ethereum, swap_encoder_registry)
|
SplitSwapStrategyEncoder::new(private_key, eth_chain(), swap_encoder_registry).unwrap();
|
||||||
.unwrap();
|
|
||||||
let solution = Solution {
|
let solution = Solution {
|
||||||
exact_out: false,
|
exact_out: false,
|
||||||
given_token: weth,
|
given_token: weth,
|
||||||
@@ -669,8 +676,7 @@ mod tests {
|
|||||||
};
|
};
|
||||||
let swap_encoder_registry = get_swap_encoder_registry();
|
let swap_encoder_registry = get_swap_encoder_registry();
|
||||||
let encoder =
|
let encoder =
|
||||||
SplitSwapStrategyEncoder::new(private_key, Chain::Ethereum, swap_encoder_registry)
|
SplitSwapStrategyEncoder::new(private_key, eth_chain(), swap_encoder_registry).unwrap();
|
||||||
.unwrap();
|
|
||||||
let solution = Solution {
|
let solution = Solution {
|
||||||
exact_out: false,
|
exact_out: false,
|
||||||
given_token: eth(),
|
given_token: eth(),
|
||||||
@@ -718,8 +724,7 @@ mod tests {
|
|||||||
};
|
};
|
||||||
let swap_encoder_registry = get_swap_encoder_registry();
|
let swap_encoder_registry = get_swap_encoder_registry();
|
||||||
let encoder =
|
let encoder =
|
||||||
SplitSwapStrategyEncoder::new(private_key, Chain::Ethereum, swap_encoder_registry)
|
SplitSwapStrategyEncoder::new(private_key, eth_chain(), swap_encoder_registry).unwrap();
|
||||||
.unwrap();
|
|
||||||
let solution = Solution {
|
let solution = Solution {
|
||||||
exact_out: false,
|
exact_out: false,
|
||||||
given_token: dai,
|
given_token: dai,
|
||||||
@@ -808,8 +813,7 @@ mod tests {
|
|||||||
};
|
};
|
||||||
let swap_encoder_registry = get_swap_encoder_registry();
|
let swap_encoder_registry = get_swap_encoder_registry();
|
||||||
let encoder =
|
let encoder =
|
||||||
SplitSwapStrategyEncoder::new(private_key, Chain::Ethereum, swap_encoder_registry)
|
SplitSwapStrategyEncoder::new(private_key, eth_chain(), swap_encoder_registry).unwrap();
|
||||||
.unwrap();
|
|
||||||
let solution = Solution {
|
let solution = Solution {
|
||||||
exact_out: false,
|
exact_out: false,
|
||||||
given_token: weth,
|
given_token: weth,
|
||||||
@@ -836,7 +840,7 @@ mod tests {
|
|||||||
let private_key =
|
let private_key =
|
||||||
"0x123456789abcdef123456789abcdef123456789abcdef123456789abcdef1234".to_string();
|
"0x123456789abcdef123456789abcdef123456789abcdef123456789abcdef1234".to_string();
|
||||||
let swap_encoder_registry = get_swap_encoder_registry();
|
let swap_encoder_registry = get_swap_encoder_registry();
|
||||||
SplitSwapStrategyEncoder::new(private_key, Chain::Ethereum, swap_encoder_registry).unwrap()
|
SplitSwapStrategyEncoder::new(private_key, eth_chain(), swap_encoder_registry).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
|||||||
@@ -1,9 +1,7 @@
|
|||||||
use std::{collections::HashMap, fs};
|
use std::{collections::HashMap, fs};
|
||||||
|
|
||||||
use tycho_core::dto::Chain;
|
|
||||||
|
|
||||||
use crate::encoding::{
|
use crate::encoding::{
|
||||||
errors::EncodingError, evm::swap_encoder::builder::SwapEncoderBuilder,
|
errors::EncodingError, evm::swap_encoder::builder::SwapEncoderBuilder, models::Chain,
|
||||||
swap_encoder::SwapEncoder,
|
swap_encoder::SwapEncoder,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -15,10 +13,10 @@ pub struct SwapEncoderRegistry {
|
|||||||
impl SwapEncoderRegistry {
|
impl SwapEncoderRegistry {
|
||||||
pub fn new(executors_file_path: &str, blockchain: Chain) -> Result<Self, EncodingError> {
|
pub fn new(executors_file_path: &str, blockchain: Chain) -> Result<Self, EncodingError> {
|
||||||
let config_str = fs::read_to_string(executors_file_path)?;
|
let config_str = fs::read_to_string(executors_file_path)?;
|
||||||
let config: HashMap<Chain, HashMap<String, String>> = serde_json::from_str(&config_str)?;
|
let config: HashMap<String, HashMap<String, String>> = serde_json::from_str(&config_str)?;
|
||||||
let mut encoders = HashMap::new();
|
let mut encoders = HashMap::new();
|
||||||
let executors = config
|
let executors = config
|
||||||
.get(&blockchain)
|
.get(&blockchain.name)
|
||||||
.ok_or(EncodingError::FatalError("No executors found for blockchain".to_string()))?;
|
.ok_or(EncodingError::FatalError("No executors found for blockchain".to_string()))?;
|
||||||
for (protocol, executor_address) in executors {
|
for (protocol, executor_address) in executors {
|
||||||
let builder = SwapEncoderBuilder::new(protocol, executor_address);
|
let builder = SwapEncoderBuilder::new(protocol, executor_address);
|
||||||
|
|||||||
@@ -1,12 +1,11 @@
|
|||||||
use std::str::FromStr;
|
use std::str::FromStr;
|
||||||
|
|
||||||
use num_bigint::BigUint;
|
use num_bigint::BigUint;
|
||||||
use tycho_core::{dto::Chain, Bytes};
|
use tycho_core::Bytes;
|
||||||
|
|
||||||
use crate::encoding::{
|
use crate::encoding::{
|
||||||
errors::EncodingError,
|
errors::EncodingError,
|
||||||
evm::constants::{native_address, wrapped_address},
|
models::{Chain, NativeAction, Solution, Transaction},
|
||||||
models::{NativeAction, Solution, Transaction},
|
|
||||||
strategy_encoder::StrategyEncoderRegistry,
|
strategy_encoder::StrategyEncoderRegistry,
|
||||||
tycho_encoder::TychoEncoder,
|
tycho_encoder::TychoEncoder,
|
||||||
};
|
};
|
||||||
@@ -26,14 +25,17 @@ impl<S: StrategyEncoderRegistry> EVMTychoEncoder<S> {
|
|||||||
) -> Result<Self, EncodingError> {
|
) -> Result<Self, EncodingError> {
|
||||||
let router_address = Bytes::from_str(&router_address)
|
let router_address = Bytes::from_str(&router_address)
|
||||||
.map_err(|_| EncodingError::FatalError("Invalid router address".to_string()))?;
|
.map_err(|_| EncodingError::FatalError("Invalid router address".to_string()))?;
|
||||||
let native_address = native_address(chain);
|
if chain.name != *"ethereum" {
|
||||||
let wrapped_address = wrapped_address(chain);
|
|
||||||
if chain != Chain::Ethereum {
|
|
||||||
return Err(EncodingError::InvalidInput(
|
return Err(EncodingError::InvalidInput(
|
||||||
"Currently only Ethereum is supported".to_string(),
|
"Currently only Ethereum is supported".to_string(),
|
||||||
));
|
));
|
||||||
}
|
}
|
||||||
Ok(EVMTychoEncoder { strategy_selector, router_address, native_address, wrapped_address })
|
Ok(EVMTychoEncoder {
|
||||||
|
strategy_selector,
|
||||||
|
router_address,
|
||||||
|
native_address: chain.native_token,
|
||||||
|
wrapped_address: chain.wrapped_token,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -119,27 +121,38 @@ impl<S: StrategyEncoderRegistry> TychoEncoder<S> for EVMTychoEncoder<S> {
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use tycho_core::dto::{Chain, ProtocolComponent};
|
use tycho_core::dto::{Chain as TychoCoreChain, ProtocolComponent};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::encoding::{
|
use crate::encoding::{
|
||||||
models::Swap, strategy_encoder::StrategyEncoder, swap_encoder::SwapEncoder,
|
models::{ChainId, Swap},
|
||||||
|
strategy_encoder::StrategyEncoder,
|
||||||
|
swap_encoder::SwapEncoder,
|
||||||
};
|
};
|
||||||
|
|
||||||
struct MockStrategyRegistry {
|
struct MockStrategyRegistry {
|
||||||
strategy: Box<dyn StrategyEncoder>,
|
strategy: Box<dyn StrategyEncoder>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn eth_chain() -> Chain {
|
||||||
|
Chain {
|
||||||
|
id: ChainId(1),
|
||||||
|
name: TychoCoreChain::Ethereum.to_string(),
|
||||||
|
native_token: eth(),
|
||||||
|
wrapped_token: weth(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
fn dai() -> Bytes {
|
fn dai() -> Bytes {
|
||||||
Bytes::from_str("0x6b175474e89094c44da98b954eedeac495271d0f").unwrap()
|
Bytes::from_str("0x6b175474e89094c44da98b954eedeac495271d0f").unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn eth() -> Bytes {
|
fn eth() -> Bytes {
|
||||||
native_address(Chain::Ethereum)
|
Bytes::from_str("0x0000000000000000000000000000000000000000").unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn weth() -> Bytes {
|
fn weth() -> Bytes {
|
||||||
wrapped_address(Chain::Ethereum)
|
Bytes::from_str("0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2").unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
impl StrategyEncoderRegistry for MockStrategyRegistry {
|
impl StrategyEncoderRegistry for MockStrategyRegistry {
|
||||||
@@ -181,11 +194,11 @@ mod tests {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn get_mocked_tycho_encoder() -> EVMTychoEncoder<MockStrategyRegistry> {
|
fn get_mocked_tycho_encoder() -> EVMTychoEncoder<MockStrategyRegistry> {
|
||||||
let strategy_selector = MockStrategyRegistry::new(Chain::Ethereum, "", None).unwrap();
|
let strategy_selector = MockStrategyRegistry::new(eth_chain(), "", None).unwrap();
|
||||||
EVMTychoEncoder::new(
|
EVMTychoEncoder::new(
|
||||||
strategy_selector,
|
strategy_selector,
|
||||||
"0x1234567890abcdef1234567890abcdef12345678".to_string(),
|
"0x1234567890abcdef1234567890abcdef12345678".to_string(),
|
||||||
Chain::Ethereum,
|
eth_chain(),
|
||||||
)
|
)
|
||||||
.unwrap()
|
.unwrap()
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
pub mod constants;
|
||||||
mod errors;
|
mod errors;
|
||||||
#[cfg(feature = "evm")]
|
#[cfg(feature = "evm")]
|
||||||
pub mod evm;
|
pub mod evm;
|
||||||
|
|||||||
@@ -1,5 +1,13 @@
|
|||||||
use num_bigint::BigUint;
|
use num_bigint::BigUint;
|
||||||
use tycho_core::{dto::ProtocolComponent, Bytes};
|
use tycho_core::{
|
||||||
|
dto::{Chain as TychoCoreChain, ProtocolComponent},
|
||||||
|
Bytes,
|
||||||
|
};
|
||||||
|
|
||||||
|
use crate::encoding::{
|
||||||
|
constants::{NATIVE_ADDRESSES, WRAPPED_ADDRESSES},
|
||||||
|
errors::EncodingError,
|
||||||
|
};
|
||||||
|
|
||||||
#[derive(Clone, Default, Debug)]
|
#[derive(Clone, Default, Debug)]
|
||||||
pub struct Solution {
|
pub struct Solution {
|
||||||
@@ -68,3 +76,65 @@ pub struct EncodingContext {
|
|||||||
pub exact_out: bool,
|
pub exact_out: bool,
|
||||||
pub router_address: Bytes,
|
pub router_address: Bytes,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||||
|
pub struct ChainId(pub u64);
|
||||||
|
|
||||||
|
#[derive(Clone, PartialEq, Eq, Hash)]
|
||||||
|
pub struct Chain {
|
||||||
|
pub id: ChainId,
|
||||||
|
pub name: String,
|
||||||
|
pub native_token: Bytes,
|
||||||
|
pub wrapped_token: Bytes,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ChainId {
|
||||||
|
pub fn id(&self) -> u64 {
|
||||||
|
self.0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<TychoCoreChain> for ChainId {
|
||||||
|
fn from(chain: TychoCoreChain) -> Self {
|
||||||
|
match chain {
|
||||||
|
TychoCoreChain::Ethereum => ChainId(1),
|
||||||
|
TychoCoreChain::ZkSync => ChainId(324),
|
||||||
|
TychoCoreChain::Arbitrum => ChainId(42161),
|
||||||
|
TychoCoreChain::Starknet => ChainId(0),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Chain {
|
||||||
|
pub fn from_tycho_core_chain(
|
||||||
|
chain: TychoCoreChain,
|
||||||
|
native_token: Option<Bytes>,
|
||||||
|
wrapped_token: Option<Bytes>,
|
||||||
|
) -> Result<Self, EncodingError> {
|
||||||
|
let native_token_address = match native_token {
|
||||||
|
Some(token) => token,
|
||||||
|
None => NATIVE_ADDRESSES.get(&chain)
|
||||||
|
.cloned()
|
||||||
|
.ok_or_else(|| EncodingError::InvalidInput(format!(
|
||||||
|
"Native token does not have a default address for chain {:?}. Please pass the native token address",
|
||||||
|
chain
|
||||||
|
)))?,
|
||||||
|
};
|
||||||
|
|
||||||
|
let wrapped_token_address = match wrapped_token {
|
||||||
|
Some(token) => token,
|
||||||
|
None => WRAPPED_ADDRESSES.get(&chain)
|
||||||
|
.cloned()
|
||||||
|
.ok_or_else(|| EncodingError::InvalidInput(format!(
|
||||||
|
"Wrapped token does not have a default address for chain {:?}. Please pass the wrapped token address",
|
||||||
|
chain
|
||||||
|
)))?,
|
||||||
|
};
|
||||||
|
Ok(Chain {
|
||||||
|
id: chain.into(),
|
||||||
|
name: chain.to_string(),
|
||||||
|
native_token: native_token_address,
|
||||||
|
wrapped_token: wrapped_token_address,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,6 +1,10 @@
|
|||||||
use tycho_core::{dto::Chain, Bytes};
|
use tycho_core::Bytes;
|
||||||
|
|
||||||
use crate::encoding::{errors::EncodingError, models::Solution, swap_encoder::SwapEncoder};
|
use crate::encoding::{
|
||||||
|
errors::EncodingError,
|
||||||
|
models::{Chain, Solution},
|
||||||
|
swap_encoder::SwapEncoder,
|
||||||
|
};
|
||||||
|
|
||||||
pub trait StrategyEncoder {
|
pub trait StrategyEncoder {
|
||||||
fn encode_strategy(
|
fn encode_strategy(
|
||||||
|
|||||||
Reference in New Issue
Block a user