feat(storage): implement the forget object execute message

bdeneux · bdeneux · commit 52f16661c60c · 2023-03-21T14:50:58.000+01:00
diff --git a/contracts/cw-storage/src/contract.rs b/contracts/cw-storage/src/contract.rs
@@ -45,14 +45,19 @@ pub fn execute(
         ExecuteMsg::StoreObject { data, pin } => execute::store_object(deps, info, data, pin),
         ExecuteMsg::PinObject { id } => execute::pin_object(deps, info, id),
         ExecuteMsg::UnpinObject { id } => execute::unpin_object(deps, info, id),
+        ExecuteMsg::ForgetObject { id } => execute::forget_object(deps, info, id),
         _ => Err(NotImplemented {}),
     }
 }
 
 pub mod execute {
     use super::*;
     use crate::state::Limits;
-    use cosmwasm_std::{StdError, Uint128};
+    use crate::ContractError::Pinned;
+    use cosmwasm_std::Order::Ascending;
+    use cosmwasm_std::StdError::NotFound;
+    use cosmwasm_std::{Order, StdError, Uint128};
+    use cw_storage_plus::Bound;
     use std::any::type_name;
 
     pub fn store_object(
@@ -195,6 +200,34 @@ pub mod execute {
 
         Ok(res)
     }
+
+    pub fn forget_object(
+        deps: DepsMut,
+        _info: MessageInfo,
+        object_id: ObjectId,
+    ) -> Result<Response, ContractError> {
+        if pins()
+            .idx
+            .object
+            .prefix(object_id.clone())
+            .keys_raw(deps.storage, None, None, Order::Ascending)
+            .next()
+            .is_some()
+        {
+            return Err(Pinned {});
+        }
+        let object = query::object(deps.as_ref(), object_id.clone())?;
+        BUCKET.update(deps.storage, |mut b| -> Result<_, ContractError> {
+            b.stat.object_count -= Uint128::one();
+            b.stat.size -= object.size;
+            Ok(b)
+        })?;
+
+        objects().remove(deps.storage, object_id.clone())?;
+        Ok(Response::new()
+            .add_attribute("action", "forget_object")
+            .add_attribute("id", object_id))
+    }
 }
 
 #[cfg_attr(not(feature = "library"), entry_point)]
diff --git a/contracts/cw-storage/src/error.rs b/contracts/cw-storage/src/error.rs
@@ -11,6 +11,9 @@ pub enum ContractError {
 
     #[error("{0}")]
     Bucket(#[from] BucketError),
+
+    #[error("Object is pinned")]
+    Pinned {},
 }
 
 #[derive(Error, Debug, Eq, PartialEq)]

Original file line number	Diff line number	Diff line change
`@@ -11,6 +11,9 @@ pub enum ContractError {`
`11`	`11`
`12`	`12`	`#[error("{0}")]`
`13`	`13`	`Bucket(#[from] BucketError),`
	`14`	`+`
	`15`	`+ #[error("Object is pinned")]`
	`16`	`+ Pinned {},`
`14`	`17`	`}`
`15`	`18`
`16`	`19`	`#[derive(Error, Debug, Eq, PartialEq)]`