komple-framework-utils 1.1.1-beta

Package used for utility methods in Komple Framework.
Documentation 
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

use cosmwasm_std::{Coin, DepsMut, MessageInfo, StdError, Uint128};
use cw20::{Cw20QueryMsg, TokenInfoResponse};
use komple_framework_types::modules::fee::FundInfo;
use thiserror::Error;

pub fn check_single_amount(info: &MessageInfo, amount: Uint128) -> Result<(), FundsError> {
    if info.funds.len() != 1 {
        return Err(FundsError::MissingFunds {});
    };
    let sent_fund = info.funds.get(0).unwrap();
    if sent_fund.amount != amount {
        return Err(FundsError::InvalidFunds {
            got: sent_fund.amount.to_string(),
            expected: amount.to_string(),
        });
    }
    Ok(())
}

pub fn check_single_coin(info: &MessageInfo, expected: Coin) -> Result<(), FundsError> {
    if info.funds.len() != 1 {
        return Err(FundsError::MissingFunds {});
    };
    let sent_fund = info.funds.get(0).unwrap();
    if sent_fund.denom != expected.denom {
        return Err(FundsError::InvalidDenom {
            got: sent_fund.denom.to_string(),
            expected: expected.denom,
        });
    }
    if sent_fund.amount != expected.amount {
        return Err(FundsError::InvalidFunds {
            got: sent_fund.amount.to_string(),
            expected: expected.amount.to_string(),
        });
    }
    Ok(())
}

/// Check c20 fund information in `FundInfo`.
pub fn check_cw20_fund_info(deps: &DepsMut, fund_info: &FundInfo) -> Result<(), FundsError> {
    if fund_info.cw20_address.is_none() {
        return Err(FundsError::InvalidCw20Token {});
    };
    let res: TokenInfoResponse = deps.querier.query_wasm_smart(
        fund_info.cw20_address.as_ref().unwrap(),
        &Cw20QueryMsg::TokenInfo {},
    )?;
    if fund_info.denom != res.symbol {
        return Err(FundsError::InvalidCw20Token {});
    };
    Ok(())
}

#[derive(Error, Debug, PartialEq)]
pub enum FundsError {
    #[error("{0}")]
    Std(#[from] StdError),

    #[error("Invalid denom! Got: {got} - Expected: {expected}")]
    InvalidDenom { got: String, expected: String },

    #[error("Invalid funds! Got: {got} - Expected: {expected}")]
    InvalidFunds { got: String, expected: String },

    #[error("No funds found!")]
    MissingFunds {},

    #[error("Invalid cw20 token")]
    InvalidCw20Token {},
}