use std::{collections::HashSet, str::FromStr};

use num_bigint::BigUint;
use tycho_common::Bytes;

use crate::encoding::{
    errors::EncodingError,
    evm::{
        group_swaps::group_swaps,
        strategy_encoder::strategy_encoders::{
            SequentialSwapStrategyEncoder, SingleSwapStrategyEncoder, SplitSwapStrategyEncoder,
        },
        swap_encoder::swap_encoder_registry::SwapEncoderRegistry,
    },
    models::{Chain, EncodingContext, NativeAction, Solution, Transaction, TransferType},
    strategy_encoder::StrategyEncoder,
    tycho_encoder::TychoEncoder,
};

/// Encodes solutions to be used by the TychoRouter.
///
/// # Fields
/// * `single_swap_strategy`: Encoder for single swaps
/// * `sequential_swap_strategy`: Encoder for sequential swaps
/// * `split_swap_strategy`: Encoder for split swaps
/// * `native_address`: Address of the chain's native token
/// * `wrapped_address`: Address of the chain's wrapped native token
#[derive(Clone)]
pub struct TychoRouterEncoder {
    single_swap_strategy: SingleSwapStrategyEncoder,
    sequential_swap_strategy: SequentialSwapStrategyEncoder,
    split_swap_strategy: SplitSwapStrategyEncoder,
    native_address: Bytes,
    wrapped_address: Bytes,
}

impl TychoRouterEncoder {
    pub fn new(
        chain: Chain,
        swap_encoder_registry: SwapEncoderRegistry,
        swapper_pk: Option<String>,
        router_address: Bytes,
        token_in_already_in_router: bool,
    ) -> Result<Self, EncodingError> {
        let native_address = chain.native_token()?;
        let wrapped_address = chain.wrapped_token()?;
        Ok(TychoRouterEncoder {
            single_swap_strategy: SingleSwapStrategyEncoder::new(
                chain.clone(),
                swap_encoder_registry.clone(),
                swapper_pk.clone(),
                router_address.clone(),
                token_in_already_in_router,
            )?,
            sequential_swap_strategy: SequentialSwapStrategyEncoder::new(
                chain.clone(),
                swap_encoder_registry.clone(),
                swapper_pk.clone(),
                router_address.clone(),
                token_in_already_in_router,
            )?,
            split_swap_strategy: SplitSwapStrategyEncoder::new(
                chain,
                swap_encoder_registry,
                None,
                router_address.clone(),
                token_in_already_in_router,
            )?,
            native_address,
            wrapped_address,
        })
    }
}

impl TychoEncoder for TychoRouterEncoder {
    fn encode_calldata(&self, solutions: Vec<Solution>) -> Result<Vec<Transaction>, EncodingError> {
        let mut transactions: Vec<Transaction> = Vec::new();
        for solution in solutions.iter() {
            self.validate_solution(solution)?;
            let (contract_interaction, target_address) = if solution.swaps.len() == 1 {
                self.single_swap_strategy
                    .encode_strategy(solution.clone())?
            } else if solution
                .swaps
                .iter()
                .all(|swap| swap.split == 0.0)
            {
                self.sequential_swap_strategy
                    .encode_strategy(solution.clone())?
            } else {
                self.split_swap_strategy
                    .encode_strategy(solution.clone())?
            };

            let value = if solution.given_token == self.native_address {
                solution.given_amount.clone()
            } else {
                BigUint::ZERO
            };

            transactions.push(Transaction {
                value,
                data: contract_interaction,
                to: target_address,
            });
        }
        Ok(transactions)
    }

    /// Raises an `EncodingError` if the solution is not considered valid.
    ///
    /// A solution is considered valid if all the following conditions are met:
    /// * The solution is not exact out.
    /// * The solution has at least one swap.
    /// * If the solution is wrapping, the given token is the chain's native token and the first
    ///   swap's input is the chain's wrapped token.
    /// * If the solution is unwrapping, the checked token is the chain's native token and the last
    ///   swap's output is the chain's wrapped token.
    /// * The token cannot appear more than once in the solution unless it is the first and last
    ///   token (i.e. a true cyclical swap).
    fn validate_solution(&self, solution: &Solution) -> Result<(), EncodingError> {
        if solution.exact_out {
            return Err(EncodingError::FatalError(
                "Currently only exact input solutions are supported".to_string(),
            ));
        }
        if solution.swaps.is_empty() {
            return Err(EncodingError::FatalError("No swaps found in solution".to_string()));
        }
        if let Some(native_action) = solution.clone().native_action {
            if native_action == NativeAction::Wrap {
                if solution.given_token != self.native_address {
                    return Err(EncodingError::FatalError(
                        "Native token must be the input token in order to wrap".to_string(),
                    ));
                }
                if let Some(first_swap) = solution.swaps.first() {
                    if first_swap.token_in != self.wrapped_address {
                        return Err(EncodingError::FatalError(
                            "Wrapped token must be the first swap's input in order to wrap"
                                .to_string(),
                        ));
                    }
                }
            } else if native_action == NativeAction::Unwrap {
                if solution.checked_token != self.native_address {
                    return Err(EncodingError::FatalError(
                        "Native token must be the output token in order to unwrap".to_string(),
                    ));
                }
                if let Some(last_swap) = solution.swaps.last() {
                    if last_swap.token_out != self.wrapped_address {
                        return Err(EncodingError::FatalError(
                            "Wrapped token must be the last swap's output in order to unwrap"
                                .to_string(),
                        ));
                    }
                }
            }
        }

        let mut solution_tokens = vec![];
        let mut split_tokens_already_considered = HashSet::new();
        for (i, swap) in solution.swaps.iter().enumerate() {
            // so we don't count the split tokens more than once
            if swap.split != 0.0 {
                if !split_tokens_already_considered.contains(&swap.token_in) {
                    solution_tokens.push(swap.token_in.clone());
                    split_tokens_already_considered.insert(swap.token_in.clone());
                }
            } else {
                // it might be the last swap of the split or a regular swap
                if !split_tokens_already_considered.contains(&swap.token_in) {
                    solution_tokens.push(swap.token_in.clone());
                }
            }
            if i == solution.swaps.len() - 1 {
                solution_tokens.push(swap.token_out.clone());
            }
        }

        if solution_tokens.len() !=
            solution_tokens
                .iter()
                .cloned()
                .collect::<HashSet<Bytes>>()
                .len()
        {
            if let Some(last_swap) = solution.swaps.last() {
                if solution.swaps[0].token_in != last_swap.token_out {
                    return Err(EncodingError::FatalError(
                        "Cyclical swaps are only allowed if they are the first and last token of a solution".to_string(),
                    ));
                } else {
                    // it is a valid cyclical swap
                    // we don't support any wrapping or unwrapping in this case
                    if let Some(_native_action) = solution.clone().native_action {
                        return Err(EncodingError::FatalError(
                            "Wrapping/Unwrapping is not available in cyclical swaps".to_string(),
                        ));
                    }
                }
            }
        }
        Ok(())
    }
}

/// Represents an encoder for one swap to be executed directly against an Executor.
///
/// This is useful when you want to bypass the Tycho Router, use your own Router contract and
/// just need the calldata for a particular swap.
///
/// # Fields
/// * `swap_encoder_registry`: Registry of swap encoders
/// * `native_address`: Address of the chain's native token
#[derive(Clone)]
pub struct TychoExecutorEncoder {
    swap_encoder_registry: SwapEncoderRegistry,
    native_address: Bytes,
}

impl TychoExecutorEncoder {
    pub fn new(
        chain: Chain,
        swap_encoder_registry: SwapEncoderRegistry,
    ) -> Result<Self, EncodingError> {
        let native_address = chain.native_token()?;
        Ok(TychoExecutorEncoder { swap_encoder_registry, native_address })
    }

    fn encode_executor_calldata(
        &self,
        solution: Solution,
    ) -> Result<(Vec<u8>, Bytes), EncodingError> {
        let grouped_swaps = group_swaps(solution.clone().swaps);
        let number_of_groups = grouped_swaps.len();
        if number_of_groups > 1 {
            return Err(EncodingError::InvalidInput(format!(
                "Tycho executor encoder only supports one swap. Found {number_of_groups}"
            )))
        }

        let grouped_swap = grouped_swaps
            .first()
            .ok_or_else(|| EncodingError::FatalError("Swap grouping failed".to_string()))?;

        let receiver = solution.receiver;

        let swap_encoder = self
            .swap_encoder_registry
            .get_encoder(&grouped_swap.protocol_system)
            .ok_or_else(|| {
                EncodingError::InvalidInput(format!(
                    "Swap encoder not found for protocol: {}",
                    grouped_swap.protocol_system
                ))
            })?;

        let mut grouped_protocol_data: Vec<u8> = vec![];
        for swap in grouped_swap.swaps.iter() {
            let encoding_context = EncodingContext {
                receiver: receiver.clone(),
                exact_out: solution.exact_out,
                router_address: None,
                group_token_in: grouped_swap.token_in.clone(),
                group_token_out: grouped_swap.token_out.clone(),
                transfer_type: TransferType::TransferToProtocol,
            };
            let protocol_data = swap_encoder.encode_swap(swap.clone(), encoding_context.clone())?;
            grouped_protocol_data.extend(protocol_data);
        }

        let executor_address = Bytes::from_str(swap_encoder.executor_address())
            .map_err(|_| EncodingError::FatalError("Invalid executor address".to_string()))?;

        Ok((grouped_protocol_data, executor_address))
    }
}

impl TychoEncoder for TychoExecutorEncoder {
    fn encode_calldata(&self, solutions: Vec<Solution>) -> Result<Vec<Transaction>, EncodingError> {
        let mut transactions: Vec<Transaction> = Vec::new();
        let solution = solutions
            .first()
            .ok_or(EncodingError::FatalError("No solutions found".to_string()))?;
        self.validate_solution(solution)?;

        let (contract_interaction, target_address) =
            self.encode_executor_calldata(solution.clone())?;

        let value = if solution.given_token == self.native_address {
            solution.given_amount.clone()
        } else {
            BigUint::ZERO
        };

        transactions.push(Transaction { value, data: contract_interaction, to: target_address });
        Ok(transactions)
    }

    /// Raises an `EncodingError` if the solution is not considered valid.
    ///
    /// A solution is considered valid if all the following conditions are met:
    /// * The solution is not exact out.
    fn validate_solution(&self, solution: &Solution) -> Result<(), EncodingError> {
        if solution.exact_out {
            return Err(EncodingError::FatalError(
                "Currently only exact input solutions are supported".to_string(),
            ));
        }
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use std::str::FromStr;

    use tycho_common::models::{protocol::ProtocolComponent, Chain as TychoCommonChain};

    use super::*;
    use crate::encoding::models::Swap;

    fn dai() -> Bytes {
        Bytes::from_str("0x6b175474e89094c44da98b954eedeac495271d0f").unwrap()
    }

    fn eth() -> Bytes {
        Bytes::from_str("0x0000000000000000000000000000000000000000").unwrap()
    }

    fn weth() -> Bytes {
        Bytes::from_str("0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2").unwrap()
    }

    fn usdc() -> Bytes {
        Bytes::from_str("0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48").unwrap()
    }

    fn wbtc() -> Bytes {
        Bytes::from_str("0x2260FAC5E5542a773Aa44fBCfeDf7C193bc2C599").unwrap()
    }

    fn get_swap_encoder_registry() -> SwapEncoderRegistry {
        SwapEncoderRegistry::new(
            Some("config/test_executor_addresses.json".to_string()),
            TychoCommonChain::Ethereum.into(),
        )
        .unwrap()
    }

    mod router_encoder {
        use super::*;

        fn get_mocked_tycho_router_encoder() -> TychoRouterEncoder {
            TychoRouterEncoder::new(
                TychoCommonChain::Ethereum.into(),
                get_swap_encoder_registry(),
                None,
                Bytes::from_str("0x3Ede3eCa2a72B3aeCC820E955B36f38437D01395").unwrap(),
                false,
            )
            .unwrap()
        }
        #[test]
        fn test_encode_router_calldata() {
            let encoder = get_mocked_tycho_router_encoder();
            let eth_amount_in = BigUint::from(1000u32);
            let swap = Swap {
                component: ProtocolComponent {
                    id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                    protocol_system: "uniswap_v2".to_string(),
                    ..Default::default()
                },
                token_in: weth(),
                token_out: dai(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                given_amount: eth_amount_in.clone(),
                given_token: eth(),
                checked_token: dai(),
                swaps: vec![swap],
                receiver: Bytes::from_str("0xcd09f75E2BF2A4d11F3AB23f1389FcC1621c0cc2").unwrap(),
                native_action: Some(NativeAction::Wrap),
                ..Default::default()
            };

            let transactions = encoder.encode_calldata(vec![solution]);

            assert!(transactions.is_ok());
            let transactions = transactions.unwrap();
            assert_eq!(transactions.len(), 1);
            assert_eq!(transactions[0].value, eth_amount_in);
            assert_eq!(
                transactions[0].to,
                Bytes::from_str("0x3ede3eca2a72b3aecc820e955b36f38437d01395").unwrap()
            );
        }

        #[test]
        fn test_validate_fails_for_exact_out() {
            let encoder = get_mocked_tycho_router_encoder();
            let solution = Solution {
                exact_out: true, // This should cause an error
                ..Default::default()
            };
            let result = encoder.validate_solution(&solution);

            assert!(result.is_err());
            assert_eq!(
                result.err().unwrap(),
                EncodingError::FatalError(
                    "Currently only exact input solutions are supported".to_string()
                )
            );
        }

        #[test]
        fn test_validate_passes_for_wrap() {
            let encoder = get_mocked_tycho_router_encoder();
            let swap = Swap {
                component: ProtocolComponent {
                    id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                    protocol_system: "uniswap_v2".to_string(),
                    ..Default::default()
                },
                token_in: weth(),
                token_out: dai(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                given_token: eth(),
                checked_token: dai(),
                checked_amount: None,
                swaps: vec![swap],
                native_action: Some(NativeAction::Wrap),
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_ok());
        }

        #[test]
        fn test_validate_fails_for_wrap_wrong_input() {
            let encoder = get_mocked_tycho_router_encoder();
            let swap = Swap {
                component: ProtocolComponent {
                    id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                    protocol_system: "uniswap_v2".to_string(),
                    ..Default::default()
                },
                token_in: weth(),
                token_out: dai(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                given_token: weth(),
                swaps: vec![swap],
                native_action: Some(NativeAction::Wrap),
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_err());
            assert_eq!(
                result.err().unwrap(),
                EncodingError::FatalError(
                    "Native token must be the input token in order to wrap".to_string()
                )
            );
        }

        #[test]
        fn test_validate_fails_for_wrap_wrong_first_swap() {
            let encoder = get_mocked_tycho_router_encoder();
            let swap = Swap {
                component: ProtocolComponent {
                    id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                    protocol_system: "uniswap_v2".to_string(),
                    ..Default::default()
                },
                token_in: eth(),
                token_out: dai(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                given_token: eth(),
                swaps: vec![swap],
                native_action: Some(NativeAction::Wrap),
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_err());
            assert_eq!(
                result.err().unwrap(),
                EncodingError::FatalError(
                    "Wrapped token must be the first swap's input in order to wrap".to_string()
                )
            );
        }

        #[test]
        fn test_validate_fails_no_swaps() {
            let encoder = get_mocked_tycho_router_encoder();
            let solution = Solution {
                exact_out: false,
                given_token: eth(),
                swaps: vec![],
                native_action: Some(NativeAction::Wrap),
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_err());
            assert_eq!(
                result.err().unwrap(),
                EncodingError::FatalError("No swaps found in solution".to_string())
            );
        }

        #[test]
        fn test_validate_passes_for_unwrap() {
            let encoder = get_mocked_tycho_router_encoder();
            let swap = Swap {
                component: ProtocolComponent {
                    id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                    protocol_system: "uniswap_v2".to_string(),
                    ..Default::default()
                },
                token_in: dai(),
                token_out: weth(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                checked_token: eth(),
                checked_amount: None,
                swaps: vec![swap],
                native_action: Some(NativeAction::Unwrap),
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_ok());
        }

        #[test]
        fn test_validate_fails_for_unwrap_wrong_output() {
            let encoder = get_mocked_tycho_router_encoder();
            let swap = Swap {
                component: ProtocolComponent {
                    id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                    protocol_system: "uniswap_v2".to_string(),
                    ..Default::default()
                },
                token_in: dai(),
                token_out: weth(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                given_token: dai(),
                checked_token: weth(),
                swaps: vec![swap],
                native_action: Some(NativeAction::Unwrap),
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_err());
            assert_eq!(
                result.err().unwrap(),
                EncodingError::FatalError(
                    "Native token must be the output token in order to unwrap".to_string()
                )
            );
        }

        #[test]
        fn test_validate_fails_for_unwrap_wrong_last_swap() {
            let encoder = get_mocked_tycho_router_encoder();
            let swap = Swap {
                component: ProtocolComponent {
                    id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                    protocol_system: "uniswap_v2".to_string(),
                    ..Default::default()
                },
                token_in: dai(),
                token_out: eth(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                checked_token: eth(),
                swaps: vec![swap],
                native_action: Some(NativeAction::Unwrap),
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_err());
            assert_eq!(
                result.err().unwrap(),
                EncodingError::FatalError(
                    "Wrapped token must be the last swap's output in order to unwrap".to_string()
                )
            );
        }

        #[test]
        fn test_validate_cyclical_swap() {
            // This validation passes because the cyclical swap is the first and last token
            //      50% ->  WETH
            // DAI -              -> DAI
            //      50% -> WETH
            // (some of the pool addresses in this test are fake)
            let encoder = get_mocked_tycho_router_encoder();
            let swaps = vec![
                Swap {
                    component: ProtocolComponent {
                        id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: dai(),
                    token_out: weth(),
                    split: 0.5f64,
                },
                Swap {
                    component: ProtocolComponent {
                        id: "0x0000000000000000000000000000000000000000".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: dai(),
                    token_out: weth(),
                    split: 0f64,
                },
                Swap {
                    component: ProtocolComponent {
                        id: "0x0000000000000000000000000000000000000000".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: weth(),
                    token_out: dai(),
                    split: 0f64,
                },
            ];

            let solution = Solution {
                exact_out: false,
                given_token: dai(),
                checked_token: dai(),
                swaps,
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_ok());
        }

        #[test]
        fn test_validate_cyclical_swap_fail() {
            // This test should fail because the cyclical swap is not the first and last token
            // DAI -> WETH -> USDC -> DAI -> WBTC
            // (some of the pool addresses in this test are fake)
            let encoder = get_mocked_tycho_router_encoder();
            let swaps = vec![
                Swap {
                    component: ProtocolComponent {
                        id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: dai(),
                    token_out: weth(),
                    split: 0f64,
                },
                Swap {
                    component: ProtocolComponent {
                        id: "0xB4e16d0168e52d35CaCD2c6185b44281Ec28C9Dc".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: weth(),
                    token_out: usdc(),
                    split: 0f64,
                },
                Swap {
                    component: ProtocolComponent {
                        id: "0x0000000000000000000000000000000000000000".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: usdc(),
                    token_out: dai(),
                    split: 0f64,
                },
                Swap {
                    component: ProtocolComponent {
                        id: "0x0000000000000000000000000000000000000000".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: dai(),
                    token_out: wbtc(),
                    split: 0f64,
                },
            ];

            let solution = Solution {
                exact_out: false,
                given_token: dai(),
                checked_token: wbtc(),
                swaps,
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_err());
            assert_eq!(
            result.err().unwrap(),
            EncodingError::FatalError(
                "Cyclical swaps are only allowed if they are the first and last token of a solution".to_string()
            )
        );
        }
        #[test]
        fn test_validate_cyclical_swap_split_output() {
            // This validation passes because it is a valid cyclical swap
            //             -> WETH
            // WETH -> DAI
            //             -> WETH
            // (some of the pool addresses in this test are fake)
            let encoder = get_mocked_tycho_router_encoder();
            let swaps = vec![
                Swap {
                    component: ProtocolComponent {
                        id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: weth(),
                    token_out: dai(),
                    split: 0f64,
                },
                Swap {
                    component: ProtocolComponent {
                        id: "0x0000000000000000000000000000000000000000".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: dai(),
                    token_out: weth(),
                    split: 0.5f64,
                },
                Swap {
                    component: ProtocolComponent {
                        id: "0x0000000000000000000000000000000000000000".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: dai(),
                    token_out: weth(),
                    split: 0f64,
                },
            ];

            let solution = Solution {
                exact_out: false,
                given_token: weth(),
                checked_token: weth(),
                swaps,
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_ok());
        }

        #[test]
        fn test_validate_cyclical_swap_native_action_fail() {
            // This validation fails because there is a native action with a valid cyclical swap
            // ETH -> WETH -> DAI -> WETH
            // (some of the pool addresses in this test are fake)
            let encoder = get_mocked_tycho_router_encoder();
            let swaps = vec![
                Swap {
                    component: ProtocolComponent {
                        id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: weth(),
                    token_out: dai(),
                    split: 0f64,
                },
                Swap {
                    component: ProtocolComponent {
                        id: "0x0000000000000000000000000000000000000000".to_string(),
                        protocol_system: "uniswap_v2".to_string(),
                        ..Default::default()
                    },
                    token_in: dai(),
                    token_out: weth(),
                    split: 0f64,
                },
            ];

            let solution = Solution {
                exact_out: false,
                given_token: eth(),
                checked_token: weth(),
                swaps,
                native_action: Some(NativeAction::Wrap),
                ..Default::default()
            };

            let result = encoder.validate_solution(&solution);

            assert!(result.is_err());
            assert_eq!(
                result.err().unwrap(),
                EncodingError::FatalError(
                    "Wrapping/Unwrapping is not available in cyclical swaps"
                        .to_string()
                        .to_string()
                )
            );
        }
    }

    mod executor_encoder {
        use std::{collections::HashMap, str::FromStr};

        use alloy::hex::encode;
        use num_bigint::{BigInt, BigUint};
        use tycho_common::{models::protocol::ProtocolComponent, Bytes};

        use super::*;
        use crate::encoding::models::{Solution, Swap};

        #[test]
        fn test_executor_encoder_encode() {
            let swap_encoder_registry = get_swap_encoder_registry();
            let encoder =
                TychoExecutorEncoder::new(TychoCommonChain::Ethereum.into(), swap_encoder_registry)
                    .unwrap();

            let token_in = weth();
            let token_out = dai();

            let swap = Swap {
                component: ProtocolComponent {
                    id: "0xA478c2975Ab1Ea89e8196811F51A7B7Ade33eB11".to_string(),
                    protocol_system: "uniswap_v2".to_string(),
                    ..Default::default()
                },
                token_in: token_in.clone(),
                token_out: token_out.clone(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                given_token: token_in,
                given_amount: BigUint::from(1000000000000000000u64),
                expected_amount: Some(BigUint::from(1000000000000000000u64)),
                checked_token: token_out,
                checked_amount: None,
                sender: Bytes::from_str("0x0000000000000000000000000000000000000000").unwrap(),
                // The receiver was generated with `makeAddr("bob") using forge`
                receiver: Bytes::from_str("0x1d96f2f6bef1202e4ce1ff6dad0c2cb002861d3e").unwrap(),
                swaps: vec![swap],
                slippage: None,
                native_action: None,
            };

            let transactions = encoder
                .encode_calldata(vec![solution])
                .unwrap();
            let transaction = transactions
                .first()
                .expect("Expected at least one transaction");
            let hex_protocol_data = encode(&transaction.data);
            assert_eq!(
                transaction.to,
                Bytes::from_str("0x5615deb798bb3e4dfa0139dfa1b3d433cc23b72f").unwrap()
            );
            assert_eq!(
                hex_protocol_data,
                String::from(concat!(
                    // in token
                    "c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2",
                    // component id
                    "a478c2975ab1ea89e8196811f51a7b7ade33eb11",
                    // receiver
                    "1d96f2f6bef1202e4ce1ff6dad0c2cb002861d3e",
                    // zero for one
                    "00",
                    // transfer type
                    "00",
                ))
            );
        }

        #[test]
        fn test_executor_encoder_too_many_swaps() {
            let swap_encoder_registry = get_swap_encoder_registry();
            let encoder =
                TychoExecutorEncoder::new(TychoCommonChain::Ethereum.into(), swap_encoder_registry)
                    .unwrap();

            let token_in = weth();
            let token_out = dai();

            let swap = Swap {
                component: ProtocolComponent {
                    protocol_system: "uniswap_v2".to_string(),
                    ..Default::default()
                },
                token_in: token_in.clone(),
                token_out: token_out.clone(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                given_token: token_in,
                given_amount: BigUint::from(1000000000000000000u64),
                expected_amount: Some(BigUint::from(1000000000000000000u64)),
                checked_token: token_out,
                checked_amount: None,
                sender: Bytes::from_str("0x0000000000000000000000000000000000000000").unwrap(),
                receiver: Bytes::from_str("0x1d96f2f6bef1202e4ce1ff6dad0c2cb002861d3e").unwrap(),
                swaps: vec![swap.clone(), swap],
                slippage: None,
                native_action: None,
            };

            let result = encoder.encode_calldata(vec![solution]);
            assert!(result.is_err());
        }

        #[test]
        fn test_executor_encoder_grouped_swaps() {
            let swap_encoder_registry = get_swap_encoder_registry();
            let encoder =
                TychoExecutorEncoder::new(TychoCommonChain::Ethereum.into(), swap_encoder_registry)
                    .unwrap();

            let eth = eth();
            let usdc = usdc();
            let pepe = Bytes::from_str("0x6982508145454Ce325dDbE47a25d4ec3d2311933").unwrap();

            // Fee and tick spacing information for this test is obtained by querying the
            // USV4 Position Manager contract: 0xbd216513d74c8cf14cf4747e6aaa6420ff64ee9e
            // Using the poolKeys function with the first 25 bytes of the pool id
            let pool_fee_usdc_eth = Bytes::from(BigInt::from(3000).to_signed_bytes_be());
            let tick_spacing_usdc_eth = Bytes::from(BigInt::from(60).to_signed_bytes_be());
            let mut static_attributes_usdc_eth: HashMap<String, Bytes> = HashMap::new();
            static_attributes_usdc_eth.insert("key_lp_fee".into(), pool_fee_usdc_eth);
            static_attributes_usdc_eth.insert("tick_spacing".into(), tick_spacing_usdc_eth);

            let pool_fee_eth_pepe = Bytes::from(BigInt::from(25000).to_signed_bytes_be());
            let tick_spacing_eth_pepe = Bytes::from(BigInt::from(500).to_signed_bytes_be());
            let mut static_attributes_eth_pepe: HashMap<String, Bytes> = HashMap::new();
            static_attributes_eth_pepe.insert("key_lp_fee".into(), pool_fee_eth_pepe);
            static_attributes_eth_pepe.insert("tick_spacing".into(), tick_spacing_eth_pepe);

            let swap_usdc_eth = Swap {
                component: ProtocolComponent {
                    id: "0xdce6394339af00981949f5f3baf27e3610c76326a700af57e4b3e3ae4977f78d"
                        .to_string(),
                    protocol_system: "uniswap_v4".to_string(),
                    static_attributes: static_attributes_usdc_eth,
                    ..Default::default()
                },
                token_in: usdc.clone(),
                token_out: eth.clone(),
                split: 0f64,
            };

            let swap_eth_pepe = Swap {
                component: ProtocolComponent {
                    id: "0xecd73ecbf77219f21f129c8836d5d686bbc27d264742ddad620500e3e548e2c9"
                        .to_string(),
                    protocol_system: "uniswap_v4".to_string(),
                    static_attributes: static_attributes_eth_pepe,
                    ..Default::default()
                },
                token_in: eth.clone(),
                token_out: pepe.clone(),
                split: 0f64,
            };

            let solution = Solution {
                exact_out: false,
                given_token: usdc,
                given_amount: BigUint::from_str("1000_000000").unwrap(),
                checked_token: pepe,
                expected_amount: Some(BigUint::from_str("105_152_000000000000000000").unwrap()),
                checked_amount: None,
                slippage: None,
                sender: Bytes::from_str("0xcd09f75E2BF2A4d11F3AB23f1389FcC1621c0cc2").unwrap(),
                receiver: Bytes::from_str("0xcd09f75E2BF2A4d11F3AB23f1389FcC1621c0cc2").unwrap(),
                swaps: vec![swap_usdc_eth, swap_eth_pepe],
                ..Default::default()
            };

            let transactions = encoder
                .encode_calldata(vec![solution])
                .unwrap();
            let transaction = transactions
                .first()
                .expect("Expected at least one transaction");
            let hex_protocol_data = encode(&transaction.data);
            assert_eq!(
                transaction.to,
                Bytes::from_str("0xf62849f9a0b5bf2913b396098f7c7019b51a820a").unwrap()
            );
            assert_eq!(
                hex_protocol_data,
                String::from(concat!(
                    // group in token
                    "a0b86991c6218b36c1d19d4a2e9eb0ce3606eb48",
                    // group out token
                    "6982508145454ce325ddbe47a25d4ec3d2311933",
                    // zero for one
                    "00",
                    // transfer type
                    "00",
                    // receiver
                    "cd09f75e2bf2a4d11f3ab23f1389fcc1621c0cc2",
                    // first pool intermediary token (ETH)
                    "0000000000000000000000000000000000000000",
                    // fee
                    "000bb8",
                    // tick spacing
                    "00003c",
                    // second pool intermediary token (PEPE)
                    "6982508145454ce325ddbe47a25d4ec3d2311933",
                    // fee
                    "0061a8",
                    // tick spacing
                    "0001f4"
                ))
            );
        }
    }
}
