feat: add remove entry from map

packages/cw-storey/tests/smoke_test.rs

@@ -28,6 +28,12 @@ fn map() {
     let map = Map::<String, Item<u32>>::new(0);
 
     map.access(&mut storage).entry_mut("foo").set(&42).unwrap();
+
+    assert_eq!(map.access(&storage).entry("foo").get().unwrap(), Some(42));
+
+    map.access(&mut storage).entry_remove("foo");
+
+    assert_eq!(map.access(&storage).entry("foo").get().unwrap(), None);
 }
 
 #[test]

packages/storey/src/containers/map.rs

@@ -1,7 +1,9 @@
 use std::{borrow::Borrow, marker::PhantomData};
 
 use crate::storage::IterableStorage;
+use crate::storage::Storage;
 use crate::storage::StorageBranch;
+use crate::storage::StorageMut;
 
 use super::IterableAccessor;
 use super::Storable;
@@ -155,6 +157,7 @@ impl<K, V, S> MapAccess<K, V, S>
 where
     K: Key,
     V: Storable,
+    S: Storage,
 {
     /// Returns an immutable accessor for the inner container of this map.
     ///
@@ -192,7 +195,14 @@ where
 
         V::access_impl(StorageBranch::new(&self.storage, key))
     }
+}
 
+impl<K, V, S> MapAccess<K, V, S>
+where
+    K: Key,
+    V: Storable,
+    S: StorageMut,
+{
     /// Returns a mutable accessor for the inner container of this map.
     ///
     /// # Examples
@@ -227,15 +237,54 @@ where
         K: Borrow<Q>,
         Q: Key + ?Sized,
     {
-        let len = key.bytes().len();
-        let bytes = key.bytes();
-        let mut key = Vec::with_capacity(len + 1);
-
-        key.push(len as u8);
-        key.extend_from_slice(bytes);
+        let key = length_prefixed_key(key);
 
         V::access_impl(StorageBranch::new(&mut self.storage, key))
     }
+
+    /// Removes the inner container for a given key.
+    ///
+    /// # Examples
+    ///
+    /// ```
+    /// # use mocks::encoding::TestEncoding;
+    /// # use mocks::backend::TestStorage;
+    /// use storey::containers::{Item, Map};
+    /// pub fn main()
+    /// let mut storage = TestStorage::new();
+    /// let map = Map::<String, Item<u64, TestEncoding>>::new(0);
+    /// let mut access = map.access(&mut storage);
+    ///
+    /// access.entry_mut("foo").set(&1337).unwrap();
+    /// assert_eq!(access.entry("foo").get().unwrap(), Some(1337));
+    ///
+    /// access.entry_remove("foo");
+    /// assert_eq!(access.entry("foo").get().unwrap(), None);
+    /// ```
+    ///
+    /// ```rust
+    /// # use mocks::encoding::TestEncoding;
+    /// # use mocks::backend::TestStorage;
+    /// use storey::containers::{Item, Map};
+    ///
+    /// let mut storage = TestStorage::new();
+    /// let map = Map::<String, Map<String, Item<u64, TestEncoding>>>::new(0);
+    /// let mut access = map.access(&mut storage);
+    ///
+    /// access.entry_mut("foo").entry_mut("bar").set(&1337).unwrap();
+    /// assert_eq!(access.entry("foo").entry("bar").get().unwrap(), Some(1337));
+    ///
+    /// access.entry_remove("foo");
+    /// assert_eq!(access.entry("foo").get().unwrap(), None);
+    /// ```
+    pub fn entry_remove<Q>(&mut self, key: &Q)
+    where
+        Q: Key + ?Sized,
+    {
+        let key = length_prefixed_key(key);
+
+        self.storage.remove(&key);
+    }
 }
 
 fn length_prefixed_key<K: Key + ?Sized>(key: &K) -> Vec<u8> {
@@ -313,7 +362,6 @@ mod tests {
 
     use mocks::backend::TestStorage;
     use mocks::encoding::TestEncoding;
-    use storey_storage::Storage as _;
 
     #[test]
     fn map() {
@@ -331,6 +379,9 @@ mod tests {
             storage.get(&[0, 3, 102, 111, 111]),
             Some(1337u64.to_le_bytes().to_vec())
         );
+        map.access(&mut storage).entry_remove("foo");
+
+        assert_eq!(map.access(&storage).entry("foo").get().unwrap(), None);
         assert_eq!(map.access(&storage).entry("bar").get().unwrap(), None);
     }