From fa147c89c6c98a03bba745538f4e68d4eaed5146 Mon Sep 17 00:00:00 2001
From: Kris Nuttycombe <kris@nutty.land>
Date: Mon, 11 Mar 2024 13:03:03 -0600
Subject: [PATCH] shardtree: Add `ShardTree::store_mut`

---
 shardtree/CHANGELOG.md | 2 +-
 shardtree/src/lib.rs   | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/shardtree/CHANGELOG.md b/shardtree/CHANGELOG.md
index 01938a5..1845943 100644
--- a/shardtree/CHANGELOG.md
+++ b/shardtree/CHANGELOG.md
@@ -8,7 +8,7 @@ and this project adheres to Rust's notion of
 ## Unreleased
 
 ## Added
-* `Shardtree::store`
+* `Shardtree::{store, store_mut}`
 - `ShardTree::insert_frontier`
 
 ## Changed
diff --git a/shardtree/src/lib.rs b/shardtree/src/lib.rs
index 918247a..1e7073b 100644
--- a/shardtree/src/lib.rs
+++ b/shardtree/src/lib.rs
@@ -97,6 +97,11 @@ impl<
         &self.store
     }
 
+    /// Returns a mutable reference to the underlying [`ShardStore`].
+    pub fn store_mut(&mut self) -> &mut S {
+        &mut self.store
+    }
+
     /// Returns the root address of the tree.
     pub fn root_addr() -> Address {
         Address::from_parts(Level::from(DEPTH), 0)