use core::marker::PhantomData;
use ethereum_standards::IERC20;
use frame_support::traits::{fungible::Inspect, OriginTrait};
use frame_system::pallet_prelude::OriginFor;
use pallet_revive::{
precompiles::alloy::{
primitives::{Address, U256 as EU256},
sol_types::SolCall,
},
AddressMapper, ContractResult, ExecConfig, MomentOf,
};
use sp_core::{Get, H160, H256, U256};
use sp_runtime::Weight;
use xcm::latest::prelude::*;
use xcm_executor::{
traits::{ConvertLocation, Error as MatchError, MatchesFungibles, TransactAsset},
AssetsInHolding,
};
type BalanceOf<T> = <<T as pallet_revive::Config>::Currency as Inspect<
<T as frame_system::Config>::AccountId,
>>::Balance;
pub struct ERC20Transactor<
T,
Matcher,
AccountIdConverter,
GasLimit,
StorageDepositLimit,
AccountId,
TransfersCheckingAccount,
>(
PhantomData<(
T,
Matcher,
AccountIdConverter,
GasLimit,
StorageDepositLimit,
AccountId,
TransfersCheckingAccount,
)>,
);
impl<
AccountId: Eq + Clone,
T: pallet_revive::Config<AccountId = AccountId>,
AccountIdConverter: ConvertLocation<AccountId>,
Matcher: MatchesFungibles<H160, u128>,
GasLimit: Get<Weight>,
StorageDepositLimit: Get<BalanceOf<T>>,
TransfersCheckingAccount: Get<AccountId>,
> TransactAsset
for ERC20Transactor<
T,
Matcher,
AccountIdConverter,
GasLimit,
StorageDepositLimit,
AccountId,
TransfersCheckingAccount,
>
where
BalanceOf<T>: Into<U256> + TryFrom<U256>,
MomentOf<T>: Into<U256>,
T::Hash: frame_support::traits::IsType<H256>,
{
fn can_check_in(_origin: &Location, _what: &Asset, _context: &XcmContext) -> XcmResult {
Err(XcmError::Unimplemented)
}
fn check_in(_origin: &Location, _what: &Asset, _context: &XcmContext) {
}
fn can_check_out(_destination: &Location, _what: &Asset, _context: &XcmContext) -> XcmResult {
Err(XcmError::Unimplemented)
}
fn check_out(_destination: &Location, _what: &Asset, _context: &XcmContext) {
}
fn withdraw_asset_with_surplus(
what: &Asset,
who: &Location,
_context: Option<&XcmContext>,
) -> Result<(AssetsInHolding, Weight), XcmError> {
tracing::trace!(
target: "xcm::transactor::erc20::withdraw",
?what, ?who,
);
let (asset_id, amount) = Matcher::matches_fungibles(what)?;
let who = AccountIdConverter::convert_location(who)
.ok_or(MatchError::AccountIdConversionFailed)?;
let checking_account_eth = T::AddressMapper::to_address(&TransfersCheckingAccount::get());
let checking_address = Address::from(Into::<[u8; 20]>::into(checking_account_eth));
let gas_limit = GasLimit::get();
let data =
IERC20::transferCall { to: checking_address, value: EU256::from(amount) }.abi_encode();
let ContractResult { result, gas_consumed, storage_deposit, .. } =
pallet_revive::Pallet::<T>::bare_call(
OriginFor::<T>::signed(who.clone()),
asset_id,
U256::zero(),
gas_limit,
StorageDepositLimit::get(),
data,
ExecConfig::new_substrate_tx(),
);
let surplus = gas_limit.saturating_sub(gas_consumed);
tracing::trace!(target: "xcm::transactor::erc20::withdraw", ?gas_consumed, ?surplus, ?storage_deposit);
if let Ok(return_value) = result {
tracing::trace!(target: "xcm::transactor::erc20::withdraw", ?return_value, "Return value by withdraw_asset");
if return_value.did_revert() {
tracing::debug!(target: "xcm::transactor::erc20::withdraw", "ERC20 contract reverted");
Err(XcmError::FailedToTransactAsset("ERC20 contract reverted"))
} else {
let is_success = IERC20::transferCall::abi_decode_returns_validate(&return_value.data).map_err(|error| {
tracing::debug!(target: "xcm::transactor::erc20::withdraw", ?error, "ERC20 contract result couldn't decode");
XcmError::FailedToTransactAsset("ERC20 contract result couldn't decode")
})?;
if is_success {
tracing::trace!(target: "xcm::transactor::erc20::withdraw", "ERC20 contract was successful");
Ok((what.clone().into(), surplus))
} else {
tracing::debug!(target: "xcm::transactor::erc20::withdraw", "contract transfer failed");
Err(XcmError::FailedToTransactAsset("ERC20 contract transfer failed"))
}
}
} else {
tracing::debug!(target: "xcm::transactor::erc20::withdraw", ?result, "Error");
Err(XcmError::FailedToTransactAsset("ERC20 contract execution errored"))
}
}
fn deposit_asset_with_surplus(
what: &Asset,
who: &Location,
_context: Option<&XcmContext>,
) -> Result<Weight, XcmError> {
tracing::trace!(
target: "xcm::transactor::erc20::deposit",
?what, ?who,
);
let (asset_id, amount) = Matcher::matches_fungibles(what)?;
let who = AccountIdConverter::convert_location(who)
.ok_or(MatchError::AccountIdConversionFailed)?;
let eth_address = T::AddressMapper::to_address(&who);
let address = Address::from(Into::<[u8; 20]>::into(eth_address));
let data = IERC20::transferCall { to: address, value: EU256::from(amount) }.abi_encode();
let gas_limit = GasLimit::get();
let ContractResult { result, gas_consumed, storage_deposit, .. } =
pallet_revive::Pallet::<T>::bare_call(
OriginFor::<T>::signed(TransfersCheckingAccount::get()),
asset_id,
U256::zero(),
gas_limit,
StorageDepositLimit::get(),
data,
ExecConfig::new_substrate_tx(),
);
let surplus = gas_limit.saturating_sub(gas_consumed);
tracing::trace!(target: "xcm::transactor::erc20::deposit", ?gas_consumed, ?surplus, ?storage_deposit);
if let Ok(return_value) = result {
tracing::trace!(target: "xcm::transactor::erc20::deposit", ?return_value, "Return value");
if return_value.did_revert() {
tracing::debug!(target: "xcm::transactor::erc20::deposit", "Contract reverted");
Err(XcmError::FailedToTransactAsset("ERC20 contract reverted"))
} else {
let is_success = IERC20::transferCall::abi_decode_returns_validate(&return_value.data).map_err(|error| {
tracing::debug!(target: "xcm::transactor::erc20::deposit", ?error, "ERC20 contract result couldn't decode");
XcmError::FailedToTransactAsset("ERC20 contract result couldn't decode")
})?;
if is_success {
tracing::trace!(target: "xcm::transactor::erc20::deposit", "ERC20 contract was successful");
Ok(surplus)
} else {
tracing::debug!(target: "xcm::transactor::erc20::deposit", "contract transfer failed");
Err(XcmError::FailedToTransactAsset("ERC20 contract transfer failed"))
}
}
} else {
tracing::debug!(target: "xcm::transactor::erc20::deposit", ?result, "Error");
Err(XcmError::FailedToTransactAsset("ERC20 contract execution errored"))
}
}
}