fix: Add new to SwapEncoder trait

Add executor_address in SwapEncoderBuilder::new and not in the builder pattern
This commit is contained in:
Diana Carvalho
2025-01-15 16:51:28 +00:00
parent 6d8cbcd80c
commit 30f2ac9f6b
3 changed files with 13 additions and 27 deletions

View File

@@ -6,33 +6,22 @@ use std::str::FromStr;
pub struct SwapEncoderBuilder {
protocol_system: String,
executor_address: Option<Address>,
executor_address: Address,
}
impl SwapEncoderBuilder {
pub fn new(protocol_system: &str) -> Self {
pub fn new(protocol_system: &str, executor_address: &str) -> Self {
SwapEncoderBuilder {
protocol_system: protocol_system.to_string(),
executor_address: None,
executor_address: Address::from_str(executor_address)
.expect(&format!("Invalid address: {}", executor_address)),
}
}
pub fn executor_address(mut self, address: &str) -> Self {
self.executor_address =
Some(Address::from_str(address).expect(&format!("Invalid address: {}", address)));
self
}
pub fn build(self) -> Result<Box<dyn SwapEncoder>, String> {
let executor_address = self.executor_address.ok_or_else(|| {
format!(
"Executor address must be provided for protocol: {}",
self.protocol_system
)
})?;
match self.protocol_system.as_str() {
"uniswap_v2" => Ok(Box::new(UniswapV2SwapEncoder::new(executor_address))),
"vm:balancer_v2" => Ok(Box::new(BalancerV2SwapEncoder::new(executor_address))),
"uniswap_v2" => Ok(Box::new(UniswapV2SwapEncoder::new(self.executor_address))),
"vm:balancer_v2" => Ok(Box::new(BalancerV2SwapEncoder::new(self.executor_address))),
_ => Err(format!("Unknown protocol system: {}", self.protocol_system)),
}
}

View File

@@ -13,7 +13,7 @@ impl SwapEncoderRegistry {
let mut encoders = HashMap::new();
for (protocol, executor_address) in config.executors {
let builder = SwapEncoderBuilder::new(&protocol).executor_address(&executor_address);
let builder = SwapEncoderBuilder::new(&protocol, &executor_address);
let encoder = builder.build().expect(&format!(
"Failed to build swap encoder for protocol: {}",
protocol

View File

@@ -7,6 +7,7 @@ use anyhow::Error;
use std::str::FromStr;
pub trait SwapEncoder: Sync + Send {
fn new(executor_address: Address) -> Self;
fn encode_swap(&self, swap: Swap, encoding_context: EncodingContext) -> Result<Vec<u8>, Error>;
fn executor_address(&self) -> Address;
}
@@ -15,12 +16,11 @@ pub struct UniswapV2SwapEncoder {
executor_address: Address,
}
impl UniswapV2SwapEncoder {
pub fn new(executor_address: Address) -> Self {
impl UniswapV2SwapEncoder {}
impl SwapEncoder for UniswapV2SwapEncoder {
fn new(executor_address: Address) -> Self {
Self { executor_address }
}
}
impl SwapEncoder for UniswapV2SwapEncoder {
fn encode_swap(&self, swap: Swap, encoding_context: EncodingContext) -> Result<Vec<u8>, Error> {
todo!()
}
@@ -35,17 +35,14 @@ pub struct BalancerV2SwapEncoder {
vault_address: Address,
}
impl BalancerV2SwapEncoder {
pub fn new(executor_address: Address) -> Self {
impl SwapEncoder for BalancerV2SwapEncoder {
fn new(executor_address: Address) -> Self {
Self {
executor_address,
vault_address: Address::from_str("0xba12222222228d8ba445958a75a0704d566bf2c8")
.expect("Invalid string for balancer vault address"),
}
}
}
impl SwapEncoder for BalancerV2SwapEncoder {
fn encode_swap(&self, swap: Swap, encoding_context: EncodingContext) -> Result<Vec<u8>, Error> {
let token_approvals_manager = ProtocolApprovalsManager::new();
let runtime = tokio::runtime::Handle::try_current()