Add update_tiers() to program

public-awesome · Aug 30, 2023 · 20b34c3 · 20b34c3
1 parent 589afbf
commit 20b34c3
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 10 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/contracts/vip/collection/src/contract.rs b/contracts/vip/collection/src/contract.rs
@@ -73,7 +73,7 @@ pub fn query(deps: Deps, _env: Env, msg: QueryMsg) -> StdResult<Binary> {
 /// sum the staked amounts.
 pub fn query_total_staked(deps: Deps, owner: String) -> StdResult<Binary> {
     // TODO: get all tokens by owner of `address` (token_id)
-    // TODO: iterate through metdata to get total stake weight
+    // TODO: iterate through metadata to get total stake weight
 
     todo!()
 }

diff --git a/contracts/vip/program/Cargo.toml b/contracts/vip/program/Cargo.toml
@@ -35,6 +35,7 @@ library = []
 cosmwasm-schema         = { workspace = true }
 cosmwasm-std            = { workspace = true, features = ["staking"] }
 cosmwasm-storage        = { workspace = true }
+cw-ownable              = { workspace = true }
 cw-storage-plus         = { workspace = true }
 cw2                     = { workspace = true }
 cw721                   = { workspace = true }

diff --git a/contracts/vip/program/src/contract.rs b/contracts/vip/program/src/contract.rs
@@ -1,6 +1,8 @@
 #[cfg(not(feature = "library"))]
 use cosmwasm_std::entry_point;
-use cosmwasm_std::{to_binary, Binary, Deps, DepsMut, Env, MessageInfo, Response, StdResult};
+use cosmwasm_std::{
+    to_binary, Binary, Deps, DepsMut, Env, MessageInfo, Response, StdResult, Uint128,
+};
 use cw2::set_contract_version;
 use stargaze_vip_collection::state::Metadata;
 
@@ -16,12 +18,11 @@ const CONTRACT_VERSION: &str = env!("CARGO_PKG_VERSION");
 pub fn instantiate(
     deps: DepsMut,
     _env: Env,
-    _info: MessageInfo,
+    info: MessageInfo,
     msg: InstantiateMsg,
 ) -> Result<Response, ContractError> {
     set_contract_version(deps.storage, CONTRACT_NAME, CONTRACT_VERSION)?;
-
-    // TODO: add cw_ownable so an admin can update tier limits
+    cw_ownable::initialize_owner(deps.storage, deps.api, Some(&info.sender.as_str()))?;
 
     COLLECTION.save(deps.storage, &deps.api.addr_validate(&msg.collection)?)?;
 
@@ -32,12 +33,26 @@ pub fn instantiate(
 
 #[cfg_attr(not(feature = "library"), entry_point)]
 pub fn execute(
-    _deps: DepsMut,
+    deps: DepsMut,
     _env: Env,
-    _info: MessageInfo,
-    _msg: ExecuteMsg,
+    info: MessageInfo,
+    msg: ExecuteMsg,
+) -> Result<Response, ContractError> {
+    match msg {
+        ExecuteMsg::UpdateTiers { tiers } => execute_update_tiers(deps, info, tiers),
+    }
+}
+
+pub fn execute_update_tiers(
+    deps: DepsMut,
+    info: MessageInfo,
+    tiers: Vec<Uint128>,
 ) -> Result<Response, ContractError> {
-    unimplemented!()
+    cw_ownable::assert_owner(deps.storage, &info.sender)
+        .map_err(|_| ContractError::Unauthorized {})?;
+    TIERS.save(deps.storage, &tiers)?;
+
+    Ok(Response::new())
 }
 
 #[cfg_attr(not(feature = "library"), entry_point)]

diff --git a/contracts/vip/program/src/msg.rs b/contracts/vip/program/src/msg.rs
@@ -8,7 +8,9 @@ pub struct InstantiateMsg {
 }
 
 #[cw_serde]
-pub enum ExecuteMsg {}
+pub enum ExecuteMsg {
+    UpdateTiers { tiers: Vec<Uint128> },
+}
 
 #[cw_serde]
 #[derive(QueryResponses)]