31 lines
942 B
Rust
31 lines
942 B
Rust
use tycho_core::models::Chain;
|
|
|
|
use crate::encoding::{
|
|
errors::EncodingError,
|
|
evm::strategy_encoder::strategy_encoders::{ExecutorStrategyEncoder, SplitSwapStrategyEncoder},
|
|
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.direct_execution {
|
|
Ok(Box::new(ExecutorStrategyEncoder {}))
|
|
} 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)?))
|
|
}
|
|
}
|
|
}
|