zint/
contract.rs

1//! Contract Instance
2
3use crate::{lookup, Bytes32, Info, EVM};
4use anyhow::{anyhow, Result};
5use std::fs;
6use zinkc::{Artifact, Compiler, Config, Constructor, InitStorage};
7
8/// Contract instance for testing.
9#[derive(Default)]
10pub struct Contract {
11    /// If enable dispatcher.
12    pub dispatcher: bool,
13    /// The artifact of the contract.
14    pub artifact: Artifact,
15    /// The source WASM of the contract.
16    pub wasm: Vec<u8>,
17    /// Bytecode constructor
18    pub constructor: Constructor,
19    /// Address in evm
20    pub address: [u8; 20],
21}
22
23impl<T> From<T> for Contract
24where
25    T: AsRef<[u8]>,
26{
27    fn from(wasm: T) -> Self {
28        crate::setup_logger();
29
30        Self {
31            wasm: wasm.as_ref().into(),
32            dispatcher: true,
33            ..Default::default()
34        }
35    }
36}
37
38impl Contract {
39    /// Get the bytecode of the contract.
40    pub fn bytecode(&self) -> Result<Vec<u8>> {
41        let bytecode = self
42            .constructor
43            .finish(self.artifact.runtime_bytecode.clone().into())
44            .map(|v| v.to_vec())?;
45
46        tracing::debug!("runtime bytecode: {}", hex::encode(&bytecode));
47        Ok(bytecode)
48    }
49
50    /// Preset the storage of the contract, similar with the concept `constructor`
51    /// in solidity, but just in time.
52    pub fn construct(&mut self, storage: InitStorage) -> Result<&mut Self> {
53        self.constructor.storage(storage)?;
54        Ok(self)
55    }
56
57    /// Compile WASM to EVM bytecode.
58    pub fn compile(mut self) -> Result<Self> {
59        let config = Config::default().dispatcher(self.dispatcher);
60        let compiler = Compiler::new(config);
61        self.artifact = compiler.compile(&self.wasm)?;
62
63        // tracing::debug!("abi: {:#}", self.json_abi()?);
64        tracing::debug!("bytecode: {}", hex::encode(&self.artifact.runtime_bytecode));
65        Ok(self)
66    }
67
68    /// Deploy self to evm
69    pub fn deploy<'e>(&mut self) -> Result<EVM<'e>> {
70        let mut evm = EVM::default();
71        let info = evm.deploy(&self.bytecode()?)?;
72
73        self.address.copy_from_slice(&info.address);
74        Ok(evm)
75    }
76
77    /// Load zink contract defined in the current
78    /// package.
79    ///
80    /// NOTE: This only works if the current contract
81    /// is not an example.
82    pub fn current() -> Result<Self> {
83        Self::search(&lookup::pkg_name()?)
84    }
85
86    /// Encode call data
87    pub fn encode<Param>(&self, inputs: impl AsRef<[Param]>) -> Result<Vec<u8>>
88    where
89        Param: Bytes32,
90    {
91        let mut calldata = Vec::new();
92        let mut inputs = inputs.as_ref();
93        if self.dispatcher {
94            if inputs.is_empty() {
95                return Err(anyhow!("no selector provided"));
96            }
97
98            calldata.extend_from_slice(&zabi::selector::parse(&inputs[0].to_vec()));
99            inputs = &inputs[1..];
100        }
101
102        for input in inputs {
103            calldata.extend_from_slice(&input.to_bytes32());
104        }
105
106        tracing::debug!("calldata: {}", hex::encode(&calldata));
107        Ok(calldata)
108    }
109
110    /// Execute the contract.
111    pub fn execute<Param>(&mut self, inputs: impl AsRef<[Param]>) -> Result<Info>
112    where
113        Param: Bytes32,
114    {
115        EVM::interp(&self.artifact.runtime_bytecode, &self.encode(inputs)?)
116    }
117
118    /// Get the JSON ABI of the contract.
119    pub fn json_abi(&self) -> Result<String> {
120        serde_json::to_string_pretty(&self.artifact.abi).map_err(Into::into)
121    }
122
123    /// Disable dispatcher.
124    pub fn pure(mut self) -> Self {
125        self.dispatcher = false;
126        self
127    }
128
129    /// Search for zink contract in the target directory.
130    pub fn search(name: &str) -> Result<Self> {
131        // TODO(g4titanx): `Contract::search` to fail properly
132        // when the contract file isn’t found
133        crate::setup_logger();
134        let wasm = lookup::wasm(name)?;
135        zinkc::utils::wasm_opt(&wasm, &wasm)?;
136
137        tracing::debug!("loading contract from {}", wasm.display());
138        Ok(Self::from(fs::read(wasm)?))
139    }
140}
zint/contract.rs

zint/
contract.rs
