variable-manager-pkg 1.0.4

pub mod msgs {
    use std::collections::BTreeMap;

    use cosmwasm_schema::{cw_serde, QueryResponses};
    use cosmwasm_std::Addr;

    use crate::definitions::Config;

    use super::definitions::Variable;

    #[cw_serde]
    pub struct InstantiateMsg {
        pub owners: Vec<String>,
    }

    #[cw_serde]
    pub enum ExecuteMsg {
        RegisterVariable(RegisterVariableMsg),
        RegisterVariables(Vec<RegisterVariableMsg>),
        UpdateVariable(RegisterVariableMsg),
        UpdateVariables(Vec<RegisterVariableMsg>),
        RemoveVariable(RemoveVariableMsg),
        UpdateOwners(UpdateOwnerMsg),
    }

    #[cw_serde]
    pub struct RegisterVariableMsg {
        pub key: String,
        pub value: Variable,
    }

    impl RegisterVariableMsg {
        pub fn new(key: String, value: Variable) -> Self {
            Self { key, value }
        }
    }

    #[cw_serde]
    pub struct RemoveVariableMsg {
        pub key: String,
    }

    #[cw_serde]
    pub struct UpdateOwnerMsg {
        pub add: Option<Vec<String>>,
        pub remove: Option<Vec<String>>,
    }

    #[cw_serde]
    #[derive(QueryResponses)]
    pub enum QueryMsg {
        #[returns(Addr)]
        GetVariable { key: String },
        #[returns(BTreeMap<String, Variable>)]
        GetVariables { keys: Vec<String> },
        #[returns(Vec<(String, Variable)>)]
        AllVariables {
            start_after: Option<String>,
            limit: Option<u32>,
        },
        #[returns(Config)]
        Config {},
    }

    #[cw_serde]
    pub struct MigrateMsg {}
}

pub mod definitions {

    use std::fmt::Display;

    use cosmwasm_schema::cw_serde;
    use cosmwasm_std::{from_json, Addr, Binary, Decimal, Deps, StdError, StdResult, Uint128};
    use serde::de::DeserializeOwned;

    #[cw_serde]
    pub struct Config {
        pub owners: Vec<Addr>,
    }

    impl Config {
        pub fn validate_owner(&self, address: &Addr) -> StdResult<()> {
            if !self.owners.contains(address) {
                return Err(StdError::generic_err(format!(
                    "{} is not an owner",
                    address
                )));
            }

            Ok(())
        }
        pub fn validate(&self) -> StdResult<()> {
            if self.owners.is_empty() {
                return Err(StdError::generic_err("Invalid 0 owners. Needed at least 1"));
            }

            Ok(())
        }
    }

    #[cw_serde]
    pub enum Variable {
        String(String),
        Addr(Addr),
        Uint128(Uint128),
        U64(u64),
        Decimal(Decimal),
        Binary(Binary),
    }

    impl Variable {
        pub fn unwrap_string(&self) -> StdResult<String> {
            if let Variable::String(val) = self {
                Ok(val.clone())
            } else {
                Err(StdError::generic_err(format!(
                    "Variable is not String, {:?}",
                    self
                )))
            }
        }

        pub fn unwrap_addr(&self) -> StdResult<Addr> {
            if let Variable::Addr(val) = self {
                Ok(val.clone())
            } else {
                Err(StdError::generic_err(format!(
                    "Variable is not Addr, {:?}",
                    self
                )))
            }
        }

        pub fn unwrap_uint128(&self) -> StdResult<Uint128> {
            if let Variable::Uint128(val) = self {
                Ok(*val)
            } else {
                Err(StdError::generic_err(format!(
                    "Variable is not Uint128, {:?}",
                    self
                )))
            }
        }

        pub fn unwrap_u64(&self) -> StdResult<u64> {
            if let Variable::U64(val) = self {
                Ok(*val)
            } else {
                Err(StdError::generic_err(format!(
                    "Variable is not u64, {:?}",
                    self
                )))
            }
        }

        pub fn unwrap_decimal(&self) -> StdResult<Decimal> {
            if let Variable::Decimal(val) = self {
                Ok(*val)
            } else {
                Err(StdError::generic_err(format!(
                    "Variable is not Decimal, {:?}",
                    self
                )))
            }
        }

        pub fn unwrap_binary<T: DeserializeOwned>(&self) -> StdResult<T> {
            if let Variable::Binary(val) = self {
                from_json::<T>(val)
            } else {
                Err(StdError::generic_err(format!(
                    "Variable is not Decimal, {:?}",
                    self
                )))
            }
        }

        pub fn validate(self, deps: Deps) -> StdResult<Variable> {
            if let Variable::Addr(val) = &self {
                deps.api.addr_validate(val.as_ref())?;
            }

            Ok(self)
        }
    }

    impl Display for Variable {
        fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
            match self {
                Variable::String(val) => write!(f, "String: {val}"),
                Variable::Addr(val) => write!(f, "Addr: {val}"),
                Variable::Uint128(val) => write!(f, "Uint128: {val}"),
                Variable::U64(val) => write!(f, "u64: {val}"),
                Variable::Decimal(val) => write!(f, "Decimal: {val}"),
                Variable::Binary(val) => write!(f, "Binary: {val}"),
            }
        }
    }
}

pub mod helper {
    use std::collections::BTreeMap;

    use cosmwasm_std::{QuerierWrapper, StdResult};

    use crate::definitions::Variable;

    use super::msgs::QueryMsg;

    pub fn variable_manager_get_variable(
        querier: &QuerierWrapper,
        key: impl Into<String>,
        address_manager_addr: impl Into<String>,
    ) -> StdResult<Variable> {
        querier.query_wasm_smart(
            address_manager_addr,
            &QueryMsg::GetVariable { key: key.into() },
        )
    }

    pub fn variable_manager_get_variables(
        querier: &QuerierWrapper,
        keys: Vec<impl Into<String>>,
        address_manager_addr: impl Into<String>,
    ) -> StdResult<BTreeMap<String, Variable>> {
        querier.query_wasm_smart(
            address_manager_addr,
            &QueryMsg::GetVariables {
                keys: keys.into_iter().map(|val| val.into()).collect(),
            },
        )
    }
}