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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

use std::str::FromStr;

use colored_json::to_colored_json_auto;
use cosm_tome::{
    clients::{client::CosmTome, tendermint_rpc::TendermintRPC},
    modules::auth::model::Address,
};
use cw20::Cw20QueryMsg;
use inquire::Text;
use interactive_parse::traits::InteractiveParseObj;
use serde::{de::DeserializeOwned, Serialize};
use serde_json::Value;

use crate::{
    contract::Contract,
    error::DeployError,
    file::Config,
    settings::WorkspaceSettings,
    utils::{replace_strings, replace_strings_any},
};

pub async fn query_contract(
    settings: &WorkspaceSettings,
    contract: &impl Contract,
) -> anyhow::Result<Value> {
    println!("Querying");
    let mut config = Config::load(settings)?;
    let msg = contract.query()?;
    let addr = config.get_contract_addr_mut(&contract.to_string())?.clone();
    let value = query(&mut config, addr, msg).await?;
    let color = to_colored_json_auto(&value)?;
    println!("{color}");
    Ok(value)
}

pub async fn query(
    config: &mut Config,
    mut addr: impl AsRef<str> + Serialize + DeserializeOwned + Clone,
    msg: impl Serialize,
) -> anyhow::Result<Value> {
    let mut value = serde_json::to_value(msg)?;
    replace_strings(&mut value, &config.get_active_env()?.contracts)?;
    replace_strings_any(&mut addr, &config.get_active_env()?.contracts)?;
    let chain_info = config.get_active_chain_info()?;
    let client = TendermintRPC::new(
        &chain_info
            .rpc_endpoint
            .clone()
            .ok_or(DeployError::MissingRpc)?,
    )?;
    let cosm_tome = CosmTome::new(chain_info, client);
    let response = cosm_tome
        .wasm_query(Address::from_str(addr.as_ref()).unwrap(), &value)
        .await?;
    let string = String::from_utf8(response.res.data.unwrap()).unwrap();
    Ok(serde_json::from_str::<Value>(string.as_str()).unwrap())
}

pub async fn cw20_query(settings: &WorkspaceSettings) -> anyhow::Result<Value> {
    println!("Querying cw20");
    let mut config = Config::load(settings)?;
    let addr = Text::new("Cw20 Contract Address?")
        .with_help_message("string")
        .prompt()?;
    let msg = Cw20QueryMsg::parse_to_obj()?;
    let value = query(&mut config, addr, msg).await?;
    let color = to_colored_json_auto(&value)?;
    println!("{color}");
    Ok(value)
}

#[cfg(test)]
mod test {
    use cosm_tome::{
        clients::{client::CosmTome, tendermint_rpc::TendermintRPC},
        config::cfg::ChainConfig,
    };

    #[tokio::test]
    async fn test_query() {
        let url = "https://archive-rpc.noria.nextnet.zone:443";
        let client = TendermintRPC::new(url).expect("Failed to create client");
        let chain_cfg = ChainConfig {
            denom: "".to_string(),
            prefix: "".to_string(),
            chain_id: "".to_string(),
            derivation_path: "".to_string(),
            rpc_endpoint: Some("".to_string()),
            grpc_endpoint: None,
            gas_price: 0.025,
            gas_adjustment: 1.2,
        };
        let cosm_tome = CosmTome::new(chain_cfg, client);

        let res = cosm_tome.tendermint_query_latest_block().await.unwrap();

        println!("{:#?}", res);
    }
}