beaker 0.1.8

CosmWasm swiss-army knife configured for Osmosis by default, but trivial to make it work for other CosmWasm enabled chain.
Documentation
use crate::attrs_format;
use crate::modules::wasm::config::WasmConfig;
use crate::support::coin::Coins;
use crate::support::cosmos::ResponseValuePicker;
use crate::support::future::block;
use crate::support::gas::Gas;
use crate::support::ops_response::OpResponseDisplay;
use crate::support::state::State;
use crate::{framework::Context, support::cosmos::Client};
use anyhow::anyhow;
use anyhow::Context as _;
use anyhow::Result;
use cosmos_sdk_proto::cosmwasm::wasm::v1::MsgExecuteContractResponse;
use cosmos_sdk_proto::{traits::Message, Any};
use cosmrs::cosmwasm::MsgExecuteContract;
use cosmrs::crypto::secp256k1::SigningKey;
use cosmrs::tx::Msg;
use cosmrs::AccountId;
use serde::Serialize;

use std::{fs, vec};

#[allow(clippy::too_many_arguments)]
pub fn execute<'a, Ctx: Context<'a, WasmConfig>>(
    ctx: &Ctx,
    contract_name: &str,
    label: &str,
    raw: Option<&String>,
    funds: Coins,
    network: &str,
    timeout_height: &u32,
    gas: &Gas,
    signing_key: SigningKey,
    account_sequence: &Option<u64>,
) -> Result<ExecuteResponse> {
    let global_config = ctx.global_config()?;
    let account_prefix = global_config.account_prefix().as_str();

    let network_info = global_config
        .networks()
        .get(network)
        .with_context(|| format!("Unable to find network config: {network}"))?
        .to_owned();

    let client = Client::new(network_info.clone()).to_signing_client(signing_key, account_prefix);
    let state = State::load_by_network(network_info, ctx.root()?)?;

    let contract = state
        .get_ref(network, contract_name)?
        .addresses()
        .get(label)
        .with_context(|| format!("Unable to retrieve contract for {contract_name}:{label}"))?
        .parse::<AccountId>()
        .map_err(|e| anyhow!(e))?;

    let msg_execute_contract = MsgExecuteContract {
        sender: client.signer_account_id(),
        contract,
        msg: raw
            .map(|s| s.as_bytes().to_vec())
            .map(Ok)
            .unwrap_or_else(|| {
                let path = ctx
                    .root()?
                    .join("contracts")
                    .join(contract_name)
                    .join("execute-msgs")
                    .join(format!("{label}.json"));
                fs::read_to_string(&path)
                    .with_context(|| format!("Unable to execute with `{}`", path.to_string_lossy()))
                    .map(|s| s.as_bytes().to_vec())
            })?,
        funds: funds.into(),
    };

    block(async {
        let response = client
            .sign_and_broadcast(
                vec![msg_execute_contract.to_any().unwrap()],
                gas,
                "",
                timeout_height,
                account_sequence,
            )
            .await?;

        let contract_address = response.pick("execute", "_contract_address");

        let deliver_tx_data_bytes = base64::decode(&response.deliver_tx.data[..])?;
        let deliver_tx_data: DeliverTxData =
            DeliverTxData::decode(&mut &deliver_tx_data_bytes[..])?;

        let response: MsgExecuteContractResponse =
            MsgExecuteContractResponse::decode(&mut &deliver_tx_data.data[0].value[..])?;

        let decoded = serde_json::from_slice::<serde_json::Value>(&response.data[..]);

        let execute_response = ExecuteResponse {
            contract_address,
            label: label.to_string(),
            data: decoded.ok(),
        };

        execute_response.log();

        Ok(execute_response)
    })
}

#[derive(Serialize)]
pub struct ExecuteResponse {
    pub label: String,
    pub contract_address: String,
    pub data: Option<serde_json::Value>,
}

impl OpResponseDisplay for ExecuteResponse {
    fn headline() -> &'static str {
        "Contract executed successfully!! 🎉 "
    }
    fn attrs(&self) -> Vec<String> {
        attrs_format! { self | label, contract_address }
    }
}

#[derive(Clone, PartialEq, ::prost::Message)]
pub struct DeliverTxData {
    #[prost(message, repeated, tag = "1")]
    pub data: ::prost::alloc::vec::Vec<Any>,
}