forked from jl777/SuperNET
-
Notifications
You must be signed in to change notification settings - Fork 98
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(tokens): custom token activation for evm (#2141)
Adds support for enabling custom EVM (ERC20, PLG20, etc..) tokens without requiring them to be in the coins config. This allows users to interact with any ERC20 token by providing the contract address.
- Loading branch information
Showing
35 changed files
with
1,064 additions
and
308 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,107 @@ | ||
| use crate::eth::web3_transport::Web3Transport; | ||
| use crate::eth::{EthCoin, ERC20_CONTRACT}; | ||
| use crate::{CoinsContext, MmCoinEnum}; | ||
| use ethabi::Token; | ||
| use ethereum_types::Address; | ||
| use futures_util::TryFutureExt; | ||
| use mm2_core::mm_ctx::MmArc; | ||
| use mm2_err_handle::mm_error::MmResult; | ||
| use web3::types::{BlockId, BlockNumber, CallRequest}; | ||
| use web3::{Transport, Web3}; | ||
|
|
||
| async fn call_erc20_function<T: Transport>( | ||
| web3: &Web3<T>, | ||
| token_addr: Address, | ||
| function_name: &str, | ||
| ) -> Result<Vec<Token>, String> { | ||
| let function = try_s!(ERC20_CONTRACT.function(function_name)); | ||
| let data = try_s!(function.encode_input(&[])); | ||
| let request = CallRequest { | ||
| from: Some(Address::default()), | ||
| to: Some(token_addr), | ||
| gas: None, | ||
| gas_price: None, | ||
| value: Some(0.into()), | ||
| data: Some(data.into()), | ||
| ..CallRequest::default() | ||
| }; | ||
|
|
||
| let res = web3 | ||
| .eth() | ||
| .call(request, Some(BlockId::Number(BlockNumber::Latest))) | ||
| .map_err(|e| ERRL!("{}", e)) | ||
| .await?; | ||
| function.decode_output(&res.0).map_err(|e| ERRL!("{}", e)) | ||
| } | ||
|
|
||
| pub(crate) async fn get_token_decimals(web3: &Web3<Web3Transport>, token_addr: Address) -> Result<u8, String> { | ||
| let tokens = call_erc20_function(web3, token_addr, "decimals").await?; | ||
| let Some(token) = tokens.into_iter().next() else { | ||
| return ERR!("No value returned from decimals() call"); | ||
| }; | ||
| let Token::Uint(dec) = token else { | ||
| return ERR!("Expected Uint token for decimals, got {:?}", token); | ||
| }; | ||
| Ok(dec.as_u64() as u8) | ||
| } | ||
|
|
||
| async fn get_token_symbol(coin: &EthCoin, token_addr: Address) -> Result<String, String> { | ||
| let web3 = try_s!(coin.web3().await); | ||
| let tokens = call_erc20_function(&web3, token_addr, "symbol").await?; | ||
| let Some(token) = tokens.into_iter().next() else { | ||
| return ERR!("No value returned from symbol() call"); | ||
| }; | ||
| let Token::String(symbol) = token else { | ||
| return ERR!("Expected String token for symbol, got {:?}", token); | ||
| }; | ||
| Ok(symbol) | ||
| } | ||
|
|
||
| #[derive(Serialize)] | ||
| pub struct Erc20TokenInfo { | ||
| pub symbol: String, | ||
| pub decimals: u8, | ||
| } | ||
|
|
||
| pub async fn get_erc20_token_info(coin: &EthCoin, token_addr: Address) -> Result<Erc20TokenInfo, String> { | ||
| let symbol = get_token_symbol(coin, token_addr).await?; | ||
| let web3 = try_s!(coin.web3().await); | ||
| let decimals = get_token_decimals(&web3, token_addr).await?; | ||
| Ok(Erc20TokenInfo { symbol, decimals }) | ||
| } | ||
|
|
||
| /// Finds if an ERC20 token is in coins config by its contract address and returns its ticker. | ||
| pub fn get_erc20_ticker_by_contract_address(ctx: &MmArc, platform: &str, contract_address: &str) -> Option<String> { | ||
| ctx.conf["coins"].as_array()?.iter().find_map(|coin| { | ||
| let protocol = coin.get("protocol")?; | ||
| let protocol_type = protocol.get("type")?.as_str()?; | ||
| if protocol_type != "ERC20" { | ||
| return None; | ||
| } | ||
| let protocol_data = protocol.get("protocol_data")?; | ||
| let coin_platform = protocol_data.get("platform")?.as_str()?; | ||
| let coin_contract_address = protocol_data.get("contract_address")?.as_str()?; | ||
|
|
||
| if coin_platform == platform && coin_contract_address == contract_address { | ||
| coin.get("coin")?.as_str().map(|s| s.to_string()) | ||
| } else { | ||
| None | ||
| } | ||
| }) | ||
| } | ||
|
|
||
| /// Finds an enabled ERC20 token by its contract address and returns it as `MmCoinEnum`. | ||
| pub async fn get_enabled_erc20_by_contract( | ||
| ctx: &MmArc, | ||
| contract_address: Address, | ||
| ) -> MmResult<Option<MmCoinEnum>, String> { | ||
| let cctx = CoinsContext::from_ctx(ctx)?; | ||
| let coins = cctx.coins.lock().await; | ||
|
|
||
| Ok(coins.values().find_map(|coin| match &coin.inner { | ||
| MmCoinEnum::EthCoin(eth_coin) if eth_coin.erc20_token_address() == Some(contract_address) => { | ||
| Some(coin.inner.clone()) | ||
| }, | ||
| _ => None, | ||
| })) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.