0e70e827a0
This was confusing me when I had all of them open at the same time. Better to be extra explicit --- don't change below this line --- ENG-4081 Took 2 minutes
33 lines
977 B
Rust
33 lines
977 B
Rust
use tycho_core::models::Chain;
|
|
|
|
use crate::encoding::{
|
|
errors::EncodingError,
|
|
evm::strategy_encoder::strategy_encoders::{
|
|
SplitSwapStrategyEncoder, StraightToPoolStrategyEncoder,
|
|
},
|
|
models::Solution,
|
|
strategy_encoder::{StrategyEncoder, StrategySelector},
|
|
};
|
|
|
|
pub struct EVMStrategySelector;
|
|
|
|
impl StrategySelector for EVMStrategySelector {
|
|
fn select_strategy(
|
|
&self,
|
|
solution: &Solution,
|
|
signer: Option<String>,
|
|
chain: Chain,
|
|
) -> Result<Box<dyn StrategyEncoder>, EncodingError> {
|
|
if solution.straight_to_pool {
|
|
Ok(Box::new(StraightToPoolStrategyEncoder {}))
|
|
} else {
|
|
let signer_pk = signer.ok_or_else(|| {
|
|
EncodingError::FatalError(
|
|
"Signer is required for SplitSwapStrategyEncoder".to_string(),
|
|
)
|
|
})?;
|
|
Ok(Box::new(SplitSwapStrategyEncoder::new(signer_pk, chain).unwrap()))
|
|
}
|
|
}
|
|
}
|