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

use cosmwasm_std::{
    Addr, BankMsg, Binary, Coin, CosmosMsg, QuerierWrapper, StdResult, to_binary, Uint128, WasmMsg,
};
use cw20::Cw20ExecuteMsg;
use serde::Serialize;

use crate::terra::extract_tax;

pub fn native_send(
    querier: &QuerierWrapper,
    denom: String,
    recipient: &Addr,
    amount_with_tax: Uint128,
) -> StdResult<CosmosMsg> {
    let tax = extract_tax(querier, denom.to_string(), amount_with_tax)?;

    Ok(CosmosMsg::Bank(BankMsg::Send {
        to_address: recipient.to_string(),
        amount: vec![Coin {
            amount: amount_with_tax.checked_sub(tax)?,
            denom,
        }],
    }))
}

pub fn cw20_transfer(token: &Addr, recipient: &Addr, amount: Uint128) -> CosmosMsg {
    CosmosMsg::Wasm(WasmMsg::Execute {
        contract_addr: token.to_string(),
        funds: vec![],
        msg: to_binary(&Cw20ExecuteMsg::Transfer {
            recipient: recipient.to_string(),
            amount,
        })
            .unwrap(),
    })
}

pub fn wasm_instantiate(code_id: u64, admin: Option<Addr>, msg: Binary) -> CosmosMsg {
    CosmosMsg::Wasm(WasmMsg::Instantiate {
        admin: admin.map(|v| v.to_string()),
        code_id,
        msg,
        funds: vec![],
        label: String::new(),
    })
}

pub fn wasm_execute<T>(contract: &Addr, msg: &T) -> CosmosMsg
where
    T: Serialize + ?Sized {
    wasm_execute_bin(contract, to_binary(&msg).unwrap())
}

pub fn wasm_execute_with_funds<T>(contract: &Addr, funds: Vec<Coin>, msg: &T) -> CosmosMsg
where
    T: Serialize + ?Sized {
    wasm_execute_bin_with_funds(contract, funds, to_binary(msg).unwrap())
}

pub fn wasm_execute_bin(contract: &Addr, msg: Binary) -> CosmosMsg {
    wasm_execute_bin_with_funds(contract, vec![], msg)
}

pub fn wasm_execute_bin_with_funds(contract: &Addr, funds: Vec<Coin>, msg: Binary) -> CosmosMsg {
    CosmosMsg::Wasm(WasmMsg::Execute {
        contract_addr: contract.to_string(),
        funds,
        msg,
    })
}