1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

//! TestEnv bindings for Casper's InMemoryWasmTestBuilder.

use crate::env::ENV;
use casper_types::{bytesrepr::Bytes, RuntimeArgs};
use odra::types::{event::EventError, Address as OdraAddress, EventData, OdraError};
use odra_casper_shared::casper_address::CasperAddress;

pub mod env;

/// Returns backend name.
#[no_mangle]
pub fn backend_name() -> String {
    "CasperVM".to_string()
}

/// Deploy WASM file with arguments.
#[no_mangle]
pub fn register_contract(name: &str, args: &RuntimeArgs) -> OdraAddress {
    ENV.with(|env| {
        let wasm_name = format!("{}.wasm", name);
        env.borrow_mut().deploy_contract(&wasm_name, args.clone());

        let contract_package_hash = format!("{}_package_hash", name);
        let contract_package_hash = env
            .borrow()
            .get_contract_package_hash(&contract_package_hash);
        let casper_address: CasperAddress = contract_package_hash.into();
        OdraAddress::try_from(casper_address).unwrap()
    })
}

/// Call contract under a given address.
#[no_mangle]
pub fn call_contract(
    addr: &OdraAddress,
    entrypoint: &str,
    args: &RuntimeArgs,
    has_return: bool,
) -> Option<Bytes> {
    ENV.with(|env| {
        let casper_address = CasperAddress::try_from(*addr).unwrap();
        let contract_package_hash = casper_address.as_contract_package_hash().unwrap();
        env.borrow_mut().call_contract(
            *contract_package_hash,
            entrypoint,
            args.to_owned(),
            has_return,
        )
    })
}

/// Set the caller address for the next [call_contract].
#[no_mangle]
pub fn set_caller(address: &OdraAddress) {
    ENV.with(|env| {
        let casper_address = CasperAddress::try_from(*address).unwrap();
        env.borrow_mut().set_caller(casper_address)
    })
}

/// Returns predefined account.
#[no_mangle]
pub fn get_account(n: usize) -> OdraAddress {
    ENV.with(|env| OdraAddress::try_from(env.borrow().get_account(n)).unwrap())
}

/// Return possible error, from the previous execution.
#[no_mangle]
pub fn get_error() -> Option<OdraError> {
    ENV.with(|env| env.borrow().get_error())
}

/// Returns an event from the given contract.
#[no_mangle]
pub fn get_event(address: &OdraAddress, index: i32) -> Result<EventData, EventError> {
    ENV.with(|env| {
        let casper_address = CasperAddress::try_from(*address).unwrap();
        env.borrow().get_event(casper_address, index)
    })
}