From 9163912df688f6d59df3fac3deb61c152a3ba484 Mon Sep 17 00:00:00 2001
From: jjy <jjyruby@gmail.com>
Date: Sun, 10 Nov 2019 21:17:20 +0800
Subject: [PATCH] feat: expose gen_proof on MemMMR

---
 src/util.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/util.rs b/src/util.rs
index 1c68af0..94ec617 100644
--- a/src/util.rs
+++ b/src/util.rs
@@ -1,4 +1,4 @@
-use crate::{MMRStore, Merge, Result, MMR};
+use crate::{MMRStore, Merge, MerkleProof, Result, MMR};
 use std::cell::RefCell;
 use std::collections::HashMap;
 use std::fmt::Debug;
@@ -70,4 +70,9 @@ impl<T: Clone + Debug + PartialEq, M: Merge<Item = T>> MemMMR<T, M> {
         mmr.commit()?;
         Ok(pos)
     }
+
+    pub fn gen_proof(&self, pos: u64) -> Result<MerkleProof<T, M>> {
+        let mmr = MMR::<T, M, &MemStore<T>>::new(self.mmr_size, &self.store);
+        mmr.gen_proof(pos)
+    }
 }