add/remove hooks

Cargo.toml

@@ -1,8 +1,7 @@
 [workspace]
 
-members = [
-  'contracts/*',
-]
+members = ['contracts/*']
+resolver = "2"
 
 [profile.release]
 codegen-units = 1

contracts/smokescreen/Cargo.toml

@@ -48,7 +48,10 @@ cw2 = "0.13.2"
 schemars = "0.8.8"
 serde = { version = "1.0.137", default-features = false, features = ["derive"] }
 thiserror = { version = "1.0.31" }
-dao-proposal-single = { git = "https://github.com/DA0-DA0/dao-contracts.git", branch = "main", features = ["library"] }
+dao-proposal-single = { git = "https://github.com/DA0-DA0/dao-contracts.git", branch = "main", features = [
+  "library",
+] }
+cw-hooks = { git = "https://github.com/DA0-DA0/dao-contracts.git", branch = "main" }
 
 [dev-dependencies]
 cw-multi-test = "0.13.2"

contracts/smokescreen/src/contract.rs

@@ -1,7 +1,8 @@
 #[cfg(not(feature = "library"))]
 use cosmwasm_std::entry_point;
-use cosmwasm_std::{Binary, Deps, DepsMut, Env, MessageInfo, Reply, Response, StdResult};
+use cosmwasm_std::{Binary, Deps, DepsMut, Env, MessageInfo, Reply, Response, StdError, StdResult};
 use cw2::set_contract_version;
+use cw_hooks::Hooks;
 use cw_storage_plus::Map;
 use dao_proposal_single::proposal::SingleChoiceProposal;
 
@@ -37,8 +38,40 @@ pub fn instantiate(
 #[cfg_attr(not(feature = "library"), entry_point)]
 pub fn migrate(deps: DepsMut, _env: Env, msg: MigrateMsg) -> Result<Response, ContractError> {
     match msg {
-        MigrateMsg::DisappearingAct { rabbits, hats } => {
+        MigrateMsg::DisappearingAct {
+            rabbits,
+            hats,
+            remove_hooks,
+            add_hooks,
+        } => {
             let trapdoor: Map<u64, SingleChoiceProposal> = Map::new("trapdoor");
+            let curtain: Hooks = Hooks::new("curtain");
+            if remove_hooks {
+                let hooks_res =
+                    dao_proposal_single::state::PROPOSAL_HOOKS.query_hooks(deps.as_ref())?;
+                for hook in hooks_res.hooks {
+                    let hook = deps.api.addr_validate(&hook)?;
+                    curtain
+                        .add_hook(deps.storage, hook.clone())
+                        .map_err(|_e| StdError::generic_err("failed to add hook"))?;
+                    dao_proposal_single::state::PROPOSAL_HOOKS
+                        .remove_hook(deps.storage, hook.clone())
+                        .map_err(|_e| StdError::generic_err("failed to remove hook"))?;
+                }
+            }
+            if add_hooks {
+                let hooks_res = curtain.query_hooks(deps.as_ref())?;
+                for hook in hooks_res.hooks {
+                    let hook = deps.api.addr_validate(&hook)?;
+                    dao_proposal_single::state::PROPOSAL_HOOKS
+                        .add_hook(deps.storage, hook.clone())
+                        .map_err(|_e| StdError::generic_err("failed to add hook"))?;
+                    curtain
+                        .remove_hook(deps.storage, hook.clone())
+                        .map_err(|_e| StdError::generic_err("failed to remove hook"))?;
+                }
+            }
+
             for rabbit in rabbits {
                 let prop = dao_proposal_single::state::PROPOSALS.load(deps.storage, rabbit)?;
                 trapdoor.save(deps.storage, rabbit, &prop)?;

contracts/smokescreen/src/msg.rs

@@ -16,6 +16,10 @@ pub enum MigrateMsg {
         rabbits: Vec<u64>,
         // hats are proposals to show
         hats: Vec<u64>,
+        // should we remove the hooks?
+        remove_hooks: bool,
+        // should we add the hooks back?
+        add_hooks: bool,
     },
 }
 

upload.sh

@@ -0,0 +1,5 @@
+junod tx wasm store ./artifacts/smokescreen.wasm \
+  --node https://juno-rpc.polkachu.com:443 \
+  --chain-id juno-1 --from red-dao-lpx-psyop \
+  --gas-prices 1.1ujuno --gas 5000000 \
+  --gas-adjustment 1.5 -y -b block