mirror of
https://github.com/pezkuwichain/revive.git
synced 2026-06-11 23:01:02 +00:00
revive-runner: add a utility binary for local contract execution (#284)
I had this in mind for a while but never implemented a standalone binary so far because I always end up writing an integration test anyways. However, using a standalone version of the pallet based on the revive-runner crate is something people filing in bug reports do anyways, for example: https://github.com/paritytech/revive/issues/266 https://github.com/paritytech/contract-issues/issues/54 --------- Signed-off-by: Cyrill Leutwiler <bigcyrill@hotmail.com>
This commit is contained in:
@@ -0,0 +1,93 @@
|
||||
use std::path::PathBuf;
|
||||
|
||||
use clap::Parser;
|
||||
|
||||
use revive_runner::{Code, OptionalHex, Specs, SpecsAction::*, TestAddress};
|
||||
|
||||
/// Execute revive PolkaVM contracts locally.
|
||||
#[derive(Parser)]
|
||||
#[command(version, about, long_about = None)]
|
||||
struct Arguments {
|
||||
/// The hex encoded calldata for the contract call.
|
||||
#[arg(short, long)]
|
||||
calldata: Option<String>,
|
||||
|
||||
/// The hex encoded calldata for the contract deployment.
|
||||
#[arg(short, long)]
|
||||
deploy_calldata: Option<String>,
|
||||
|
||||
/// The hex encoded contract code blob to instantiate and execute.
|
||||
#[arg(short, long)]
|
||||
blob: Option<String>,
|
||||
|
||||
/// The contract code to instantiate and execute.
|
||||
#[arg(short, long)]
|
||||
file: Option<PathBuf>,
|
||||
|
||||
/// The origin account used to initiate the deploy and call transactions.
|
||||
#[arg(short, long)]
|
||||
origin: Option<TestAddress>,
|
||||
|
||||
/// The value the call transaction is endowed with.
|
||||
#[arg(short, long)]
|
||||
value: Option<u128>,
|
||||
|
||||
/// The value the deploy transaction is endowed with.
|
||||
#[arg(long)]
|
||||
deploy_value: Option<u128>,
|
||||
}
|
||||
|
||||
fn main() -> anyhow::Result<()> {
|
||||
env_logger::init();
|
||||
|
||||
let arguments = Arguments::parse();
|
||||
|
||||
let code = match (arguments.blob, arguments.file) {
|
||||
(Some(blob), None) => hex::decode(blob)
|
||||
.map_err(|error| anyhow::anyhow!("expected hex encoded PVM blob: {error}"))?,
|
||||
(None, Some(file)) => std::fs::read(&file).map_err(|error| {
|
||||
anyhow::anyhow!("unable to read PVM file {}: {error}", file.display())
|
||||
})?,
|
||||
_ => anyhow::bail!("should either provide a PVM blob or a PVM file"),
|
||||
};
|
||||
let calldata = match arguments.calldata {
|
||||
Some(calldata) => hex::decode(calldata)
|
||||
.map_err(|error| anyhow::anyhow!("expected hex encoded calldata: {error}"))?,
|
||||
None => vec![],
|
||||
};
|
||||
let deploy_calldata = match arguments.deploy_calldata {
|
||||
Some(calldata) => hex::decode(calldata)
|
||||
.map_err(|error| anyhow::anyhow!("expected hex encoded calldata: {error}"))?,
|
||||
None => vec![],
|
||||
};
|
||||
let origin = arguments.origin.unwrap_or(TestAddress::Alice);
|
||||
|
||||
let actions = vec![
|
||||
Instantiate {
|
||||
origin: origin.clone(),
|
||||
value: arguments.deploy_value.unwrap_or(0),
|
||||
gas_limit: None,
|
||||
storage_deposit_limit: None,
|
||||
code: Code::Bytes(code),
|
||||
data: deploy_calldata,
|
||||
salt: OptionalHex::default(),
|
||||
},
|
||||
Call {
|
||||
origin,
|
||||
dest: TestAddress::Instantiated(0),
|
||||
value: arguments.value.unwrap_or(0),
|
||||
gas_limit: None,
|
||||
storage_deposit_limit: None,
|
||||
data: calldata,
|
||||
},
|
||||
];
|
||||
|
||||
Specs {
|
||||
actions,
|
||||
differential: false,
|
||||
..Default::default()
|
||||
}
|
||||
.run();
|
||||
|
||||
Ok(())
|
||||
}
|
||||
@@ -1,9 +1,11 @@
|
||||
use std::time::Instant;
|
||||
use std::{str::FromStr, time::Instant};
|
||||
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
use crate::*;
|
||||
use alloy_primitives::{keccak256, Address};
|
||||
use alloy_primitives::keccak256;
|
||||
#[cfg(feature = "revive-solidity")]
|
||||
use alloy_primitives::Address;
|
||||
#[cfg(feature = "revive-solidity")]
|
||||
use revive_differential::{Evm, EvmLog};
|
||||
#[cfg(feature = "revive-solidity")]
|
||||
@@ -156,6 +158,39 @@ impl TestAddress {
|
||||
}
|
||||
}
|
||||
|
||||
impl FromStr for TestAddress {
|
||||
type Err = &'static str;
|
||||
|
||||
fn from_str(value: &str) -> Result<Self, Self::Err> {
|
||||
value.try_into()
|
||||
}
|
||||
}
|
||||
|
||||
impl TryFrom<&str> for TestAddress {
|
||||
type Error = &'static str;
|
||||
|
||||
fn try_from(value: &str) -> Result<Self, Self::Error> {
|
||||
match value {
|
||||
"alice" => Ok(Self::Alice),
|
||||
"bob" => Ok(Self::Bob),
|
||||
"charlie" => Ok(Self::Charlie),
|
||||
value => {
|
||||
if let Ok(value) = value.parse() {
|
||||
return Ok(Self::Instantiated(value));
|
||||
}
|
||||
|
||||
if let Ok(value) = hex::decode(value) {
|
||||
if value.len() == 20 {
|
||||
return Ok(Self::AccountId(H160(value.try_into().unwrap())));
|
||||
}
|
||||
}
|
||||
|
||||
Err("can not parse into test address")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Specs for a contract test
|
||||
#[derive(Clone, Debug, Serialize, Deserialize)]
|
||||
#[serde(default)]
|
||||
@@ -212,6 +247,7 @@ impl Specs {
|
||||
/// Helper to allow not specifying the code bytes or path directly in the runner.json
|
||||
/// - Replace `Code::Bytes(bytes)` if `bytes` are empty: read `contract_file`
|
||||
/// - Replace `Code::Solidity{ path, ..}` if `path` is not provided: replace `path` with `contract_file`
|
||||
#[allow(unused_variables)]
|
||||
pub fn replace_empty_code(&mut self, contract_name: &str, contract_path: &str) {
|
||||
for action in self.actions.iter_mut() {
|
||||
let code = match action {
|
||||
|
||||
Reference in New Issue
Block a user