feat: Rename shortcut methods of encoder builder
Also moved executors_file_path to be an independent attribute --- don't change below this line --- ENG-4286 Took 19 minutes
This commit is contained in:
@@ -23,7 +23,7 @@ fn main() {
|
|||||||
// Initialize the encoder
|
// Initialize the encoder
|
||||||
let encoder = EVMEncoderBuilder::new()
|
let encoder = EVMEncoderBuilder::new()
|
||||||
.chain(Chain::Ethereum)
|
.chain(Chain::Ethereum)
|
||||||
.tycho_router_with_permit2(None, swapper_pk)
|
.initialize_tycho_router_with_permit2(swapper_pk)
|
||||||
.expect("Failed to create encoder builder")
|
.expect("Failed to create encoder builder")
|
||||||
.build()
|
.build()
|
||||||
.expect("Failed to build encoder");
|
.expect("Failed to build encoder");
|
||||||
|
|||||||
@@ -42,27 +42,21 @@ use tycho_execution::encoding::{
|
|||||||
pub struct Cli {
|
pub struct Cli {
|
||||||
#[command(subcommand)]
|
#[command(subcommand)]
|
||||||
pub command: Commands,
|
pub command: Commands,
|
||||||
|
#[arg(short, long)]
|
||||||
|
executors_file_path: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Subcommand)]
|
#[derive(Subcommand)]
|
||||||
pub enum Commands {
|
pub enum Commands {
|
||||||
/// Use the Tycho router encoding strategy
|
/// Use the Tycho router encoding strategy
|
||||||
TychoRouter {
|
TychoRouter,
|
||||||
#[arg(short, long)]
|
|
||||||
config_path: Option<String>,
|
|
||||||
},
|
|
||||||
/// Use the Tycho router encoding strategy with Permit2 approval and token in transfer
|
/// Use the Tycho router encoding strategy with Permit2 approval and token in transfer
|
||||||
TychoRouterPermit2 {
|
TychoRouterPermit2 {
|
||||||
#[arg(short, long)]
|
|
||||||
config_path: Option<String>,
|
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
swapper_pk: String,
|
swapper_pk: String,
|
||||||
},
|
},
|
||||||
/// Use the direct execution encoding strategy
|
/// Use the direct execution encoding strategy
|
||||||
DirectExecution {
|
DirectExecution,
|
||||||
#[arg(short, long)]
|
|
||||||
config_path: Option<String>,
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||||
@@ -82,12 +76,16 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|||||||
|
|
||||||
let mut builder = EVMEncoderBuilder::new().chain(chain);
|
let mut builder = EVMEncoderBuilder::new().chain(chain);
|
||||||
|
|
||||||
builder = match cli.command {
|
if let Some(config_path) = cli.executors_file_path {
|
||||||
Commands::TychoRouter { config_path } => builder.tycho_router(config_path)?,
|
builder = builder.executors_file_path(config_path);
|
||||||
Commands::TychoRouterPermit2 { config_path, swapper_pk } => {
|
|
||||||
builder.tycho_router_with_permit2(config_path, swapper_pk)?
|
|
||||||
}
|
}
|
||||||
Commands::DirectExecution { config_path } => builder.direct_execution(config_path)?,
|
|
||||||
|
builder = match cli.command {
|
||||||
|
Commands::TychoRouter => builder.initialize_tycho_router()?,
|
||||||
|
Commands::TychoRouterPermit2 { swapper_pk } => {
|
||||||
|
builder.initialize_tycho_router_with_permit2(swapper_pk)?
|
||||||
|
}
|
||||||
|
Commands::DirectExecution => builder.initialize_direct_execution()?,
|
||||||
};
|
};
|
||||||
let encoder = builder.build()?;
|
let encoder = builder.build()?;
|
||||||
let transactions = encoder.encode_router_calldata(vec![solution])?;
|
let transactions = encoder.encode_router_calldata(vec![solution])?;
|
||||||
|
|||||||
@@ -16,6 +16,7 @@ use crate::encoding::{
|
|||||||
pub struct EVMEncoderBuilder {
|
pub struct EVMEncoderBuilder {
|
||||||
strategy: Option<Box<dyn StrategyEncoder>>,
|
strategy: Option<Box<dyn StrategyEncoder>>,
|
||||||
chain: Option<Chain>,
|
chain: Option<Chain>,
|
||||||
|
executors_file_path: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for EVMEncoderBuilder {
|
impl Default for EVMEncoderBuilder {
|
||||||
@@ -26,13 +27,20 @@ impl Default for EVMEncoderBuilder {
|
|||||||
|
|
||||||
impl EVMEncoderBuilder {
|
impl EVMEncoderBuilder {
|
||||||
pub fn new() -> Self {
|
pub fn new() -> Self {
|
||||||
EVMEncoderBuilder { chain: None, strategy: None }
|
EVMEncoderBuilder { chain: None, strategy: None, executors_file_path: None }
|
||||||
}
|
}
|
||||||
pub fn chain(mut self, chain: Chain) -> Self {
|
pub fn chain(mut self, chain: Chain) -> Self {
|
||||||
self.chain = Some(chain);
|
self.chain = Some(chain);
|
||||||
self
|
self
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Sets the `executors_file_path` manually.
|
||||||
|
/// If it's not set, the default path will be used (config/executor_addresses.json)
|
||||||
|
pub fn executors_file_path(mut self, executors_file_path: String) -> Self {
|
||||||
|
self.executors_file_path = Some(executors_file_path);
|
||||||
|
self
|
||||||
|
}
|
||||||
|
|
||||||
/// Sets the `strategy_encoder` manually.
|
/// Sets the `strategy_encoder` manually.
|
||||||
///
|
///
|
||||||
/// **Note**: This method should not be used in combination with `tycho_router` or
|
/// **Note**: This method should not be used in combination with `tycho_router` or
|
||||||
@@ -44,12 +52,17 @@ impl EVMEncoderBuilder {
|
|||||||
|
|
||||||
/// Shortcut method to initialize a `SplitSwapStrategyEncoder` without any approval nor token in
|
/// Shortcut method to initialize a `SplitSwapStrategyEncoder` without any approval nor token in
|
||||||
/// transfer. **Note**: Should not be used at the same time as `strategy_encoder`.
|
/// transfer. **Note**: Should not be used at the same time as `strategy_encoder`.
|
||||||
pub fn tycho_router(self, executors_file_path: Option<String>) -> Result<Self, EncodingError> {
|
pub fn initialize_tycho_router(self) -> Result<Self, EncodingError> {
|
||||||
if let Some(chain) = self.chain {
|
if let Some(chain) = self.chain {
|
||||||
let swap_encoder_registry = SwapEncoderRegistry::new(executors_file_path, chain)?;
|
let swap_encoder_registry =
|
||||||
|
SwapEncoderRegistry::new(self.executors_file_path.clone(), chain)?;
|
||||||
let strategy =
|
let strategy =
|
||||||
Box::new(SplitSwapStrategyEncoder::new(chain, swap_encoder_registry, None)?);
|
Box::new(SplitSwapStrategyEncoder::new(chain, swap_encoder_registry, None)?);
|
||||||
Ok(EVMEncoderBuilder { chain: Some(chain), strategy: Some(strategy) })
|
Ok(EVMEncoderBuilder {
|
||||||
|
chain: Some(chain),
|
||||||
|
strategy: Some(strategy),
|
||||||
|
executors_file_path: self.executors_file_path,
|
||||||
|
})
|
||||||
} else {
|
} else {
|
||||||
Err(EncodingError::FatalError(
|
Err(EncodingError::FatalError(
|
||||||
"Please set the chain before setting the tycho router".to_string(),
|
"Please set the chain before setting the tycho router".to_string(),
|
||||||
@@ -59,19 +72,23 @@ impl EVMEncoderBuilder {
|
|||||||
|
|
||||||
/// Shortcut method to initialize a `SplitSwapStrategyEncoder` with Permit2 approval and token
|
/// Shortcut method to initialize a `SplitSwapStrategyEncoder` with Permit2 approval and token
|
||||||
/// in transfer. **Note**: Should not be used at the same time as `strategy_encoder`.
|
/// in transfer. **Note**: Should not be used at the same time as `strategy_encoder`.
|
||||||
pub fn tycho_router_with_permit2(
|
pub fn initialize_tycho_router_with_permit2(
|
||||||
self,
|
self,
|
||||||
executors_file_path: Option<String>,
|
|
||||||
swapper_pk: String,
|
swapper_pk: String,
|
||||||
) -> Result<Self, EncodingError> {
|
) -> Result<Self, EncodingError> {
|
||||||
if let Some(chain) = self.chain {
|
if let Some(chain) = self.chain {
|
||||||
let swap_encoder_registry = SwapEncoderRegistry::new(executors_file_path, chain)?;
|
let swap_encoder_registry =
|
||||||
|
SwapEncoderRegistry::new(self.executors_file_path.clone(), chain)?;
|
||||||
let strategy = Box::new(SplitSwapStrategyEncoder::new(
|
let strategy = Box::new(SplitSwapStrategyEncoder::new(
|
||||||
chain,
|
chain,
|
||||||
swap_encoder_registry,
|
swap_encoder_registry,
|
||||||
Some(swapper_pk),
|
Some(swapper_pk),
|
||||||
)?);
|
)?);
|
||||||
Ok(EVMEncoderBuilder { chain: Some(chain), strategy: Some(strategy) })
|
Ok(EVMEncoderBuilder {
|
||||||
|
chain: Some(chain),
|
||||||
|
strategy: Some(strategy),
|
||||||
|
executors_file_path: self.executors_file_path,
|
||||||
|
})
|
||||||
} else {
|
} else {
|
||||||
Err(EncodingError::FatalError(
|
Err(EncodingError::FatalError(
|
||||||
"Please set the chain before setting the tycho router".to_string(),
|
"Please set the chain before setting the tycho router".to_string(),
|
||||||
@@ -81,14 +98,16 @@ impl EVMEncoderBuilder {
|
|||||||
|
|
||||||
/// Shortcut method to initialize an `ExecutorStrategyEncoder`.
|
/// Shortcut method to initialize an `ExecutorStrategyEncoder`.
|
||||||
/// **Note**: Should not be used at the same time as `strategy_encoder`.
|
/// **Note**: Should not be used at the same time as `strategy_encoder`.
|
||||||
pub fn direct_execution(
|
pub fn initialize_direct_execution(self) -> Result<Self, EncodingError> {
|
||||||
self,
|
|
||||||
executors_file_path: Option<String>,
|
|
||||||
) -> Result<Self, EncodingError> {
|
|
||||||
if let Some(chain) = self.chain {
|
if let Some(chain) = self.chain {
|
||||||
let swap_encoder_registry = SwapEncoderRegistry::new(executors_file_path, chain)?;
|
let swap_encoder_registry =
|
||||||
|
SwapEncoderRegistry::new(self.executors_file_path.clone(), chain)?;
|
||||||
let strategy = Box::new(ExecutorStrategyEncoder::new(swap_encoder_registry));
|
let strategy = Box::new(ExecutorStrategyEncoder::new(swap_encoder_registry));
|
||||||
Ok(EVMEncoderBuilder { chain: Some(chain), strategy: Some(strategy) })
|
Ok(EVMEncoderBuilder {
|
||||||
|
chain: Some(chain),
|
||||||
|
strategy: Some(strategy),
|
||||||
|
executors_file_path: self.executors_file_path,
|
||||||
|
})
|
||||||
} else {
|
} else {
|
||||||
Err(EncodingError::FatalError(
|
Err(EncodingError::FatalError(
|
||||||
"Please set the chain before setting the strategy".to_string(),
|
"Please set the chain before setting the strategy".to_string(),
|
||||||
|
|||||||
Reference in New Issue
Block a user