2021-10-21 16:51:01 +00:00
|
|
|
use std::fs;
|
2021-11-13 17:37:31 +00:00
|
|
|
use std::path::Path;
|
2021-10-21 16:51:01 +00:00
|
|
|
|
2022-01-30 16:18:05 +00:00
|
|
|
use web3::{
|
|
|
|
api::Web3,
|
|
|
|
contract::{Contract, Options},
|
|
|
|
transports::Http,
|
|
|
|
};
|
|
|
|
|
|
|
|
use crate::config::BlockchainConfig;
|
|
|
|
use super::api::connect;
|
|
|
|
use super::errors::EthereumError;
|
2022-06-14 18:55:04 +00:00
|
|
|
use super::sync::{
|
|
|
|
get_current_block_number,
|
|
|
|
CHAIN_REORG_MAX_DEPTH,
|
|
|
|
};
|
2022-01-30 16:18:05 +00:00
|
|
|
use super::utils::parse_address;
|
|
|
|
|
2022-01-26 16:59:27 +00:00
|
|
|
pub const ADAPTER: &str = "IAdapter";
|
2022-01-30 20:58:05 +00:00
|
|
|
pub const SUBSCRIPTION: &str = "ISubscription";
|
|
|
|
pub const ERC20: &str = "IERC20";
|
2022-01-26 16:59:27 +00:00
|
|
|
pub const ERC721: &str = "IERC721Metadata";
|
2021-10-21 16:51:01 +00:00
|
|
|
|
|
|
|
#[derive(thiserror::Error, Debug)]
|
|
|
|
pub enum ArtifactError {
|
|
|
|
#[error("io error")]
|
|
|
|
IoError(#[from] std::io::Error),
|
|
|
|
|
|
|
|
#[error("json error")]
|
|
|
|
JsonError(#[from] serde_json::Error),
|
|
|
|
|
|
|
|
#[error("key error")]
|
|
|
|
KeyError,
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn load_abi(
|
2021-11-13 17:37:31 +00:00
|
|
|
contract_dir: &Path,
|
2021-10-21 16:51:01 +00:00
|
|
|
contract_name: &str,
|
|
|
|
) -> Result<Vec<u8>, ArtifactError> {
|
|
|
|
let contract_artifact_path = contract_dir.join(format!("{}.json", contract_name));
|
|
|
|
let contract_artifact = fs::read_to_string(contract_artifact_path)?;
|
2022-01-26 16:59:27 +00:00
|
|
|
let contract_artifact_value: serde_json::Value =
|
|
|
|
serde_json::from_str(&contract_artifact)?;
|
2021-10-21 16:51:01 +00:00
|
|
|
let contract_abi = contract_artifact_value.get("abi")
|
|
|
|
.ok_or(ArtifactError::KeyError)?
|
|
|
|
.to_string().as_bytes().to_vec();
|
|
|
|
Ok(contract_abi)
|
|
|
|
}
|
2022-01-30 16:18:05 +00:00
|
|
|
|
|
|
|
pub struct ContractSet {
|
|
|
|
pub web3: Web3<Http>,
|
2022-06-14 18:55:04 +00:00
|
|
|
// Last synced block
|
|
|
|
pub current_block: u64,
|
2022-01-30 16:18:05 +00:00
|
|
|
|
|
|
|
#[allow(dead_code)]
|
|
|
|
pub adapter: Contract<Http>,
|
|
|
|
|
|
|
|
pub collectible: Contract<Http>,
|
2022-01-30 20:58:05 +00:00
|
|
|
pub subscription: Contract<Http>,
|
2022-01-30 16:18:05 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
pub async fn get_contracts(
|
|
|
|
config: &BlockchainConfig,
|
2022-06-14 18:55:04 +00:00
|
|
|
storage_dir: &Path,
|
2022-01-30 16:18:05 +00:00
|
|
|
) -> Result<ContractSet, EthereumError> {
|
|
|
|
let web3 = connect(&config.api_url)?;
|
2022-05-28 00:04:21 +00:00
|
|
|
let chain_id = web3.eth().chain_id().await?;
|
|
|
|
if chain_id != config.ethereum_chain_id().into() {
|
|
|
|
return Err(EthereumError::ImproperlyConfigured("incorrect chain ID"));
|
|
|
|
};
|
2022-01-30 16:18:05 +00:00
|
|
|
let adapter_abi = load_abi(&config.contract_dir, ADAPTER)?;
|
|
|
|
let adapter_address = parse_address(&config.contract_address)?;
|
|
|
|
let adapter = Contract::from_json(
|
|
|
|
web3.eth(),
|
|
|
|
adapter_address,
|
|
|
|
&adapter_abi,
|
|
|
|
)?;
|
|
|
|
|
|
|
|
let collectible_address = adapter.query(
|
|
|
|
"collectible",
|
|
|
|
(), None, Options::default(), None,
|
|
|
|
).await?;
|
2022-06-14 10:29:14 +00:00
|
|
|
let collectible_abi = load_abi(&config.contract_dir, ERC721)?;
|
2022-01-30 16:18:05 +00:00
|
|
|
let collectible = Contract::from_json(
|
|
|
|
web3.eth(),
|
|
|
|
collectible_address,
|
2022-06-14 10:29:14 +00:00
|
|
|
&collectible_abi,
|
2022-01-30 16:18:05 +00:00
|
|
|
)?;
|
|
|
|
log::info!("collectible item contract address is {:?}", collectible.address());
|
2022-01-30 20:58:05 +00:00
|
|
|
|
|
|
|
let subscription_address = adapter.query(
|
|
|
|
"subscription",
|
|
|
|
(), None, Options::default(), None,
|
|
|
|
).await?;
|
|
|
|
let subscription_abi = load_abi(&config.contract_dir, SUBSCRIPTION)?;
|
|
|
|
let subscription = Contract::from_json(
|
|
|
|
web3.eth(),
|
|
|
|
subscription_address,
|
|
|
|
&subscription_abi,
|
|
|
|
)?;
|
|
|
|
log::info!("subscription contract address is {:?}", subscription.address());
|
|
|
|
|
2022-06-14 18:55:04 +00:00
|
|
|
let current_block = get_current_block_number(&web3, storage_dir).await?;
|
|
|
|
log::info!("current block is {}", current_block);
|
|
|
|
// Take reorgs into account
|
|
|
|
let current_block = current_block.saturating_sub(CHAIN_REORG_MAX_DEPTH);
|
|
|
|
|
2022-01-30 16:18:05 +00:00
|
|
|
let contract_set = ContractSet {
|
|
|
|
web3,
|
2022-06-14 18:55:04 +00:00
|
|
|
current_block,
|
2022-01-30 16:18:05 +00:00
|
|
|
adapter,
|
|
|
|
collectible,
|
2022-01-30 20:58:05 +00:00
|
|
|
subscription,
|
2022-01-30 16:18:05 +00:00
|
|
|
};
|
|
|
|
Ok(contract_set)
|
|
|
|
}
|