From 00782b80c07993d877174b352d1d0c7b07651179 Mon Sep 17 00:00:00 2001
From: Rene Vergara <rene@vergara.network>
Date: Wed, 6 Nov 2024 07:29:48 -0600
Subject: [PATCH] feat: add sapling tree root test

---
 librustzcash-wrapper/src/lib.rs | 12 ++++++++++++
 src/C/Zcash.chs                 |  7 +++++++
 src/ZcashHaskell/Sapling.hs     |  5 +++++
 3 files changed, 24 insertions(+)

diff --git a/librustzcash-wrapper/src/lib.rs b/librustzcash-wrapper/src/lib.rs
index 8170650..7a03a49 100644
--- a/librustzcash-wrapper/src/lib.rs
+++ b/librustzcash-wrapper/src/lib.rs
@@ -1602,6 +1602,18 @@ pub extern "C" fn rust_wrapper_combine_sapling_nodes(
     }
 }
 
+#[no_mangle]
+pub extern "C" fn rust_wrapper_get_sapling_root(
+    level: u8,
+    out: *mut u8,
+    out_len: &mut usize
+    ){
+    let tree: CommitmentTree<Node, 32> = CommitmentTree::empty();
+    let root = tree.root();
+    let h = Hhex { bytes: root.to_bytes().to_vec() };
+    marshall_to_haskell_var(&h, out, out_len, RW);
+}
+
 #[no_mangle]
 pub extern "C" fn rust_wrapper_read_orchard_frontier(
     tree: *const u8,
diff --git a/src/C/Zcash.chs b/src/C/Zcash.chs
index 06b486a..8e8947b 100644
--- a/src/C/Zcash.chs
+++ b/src/C/Zcash.chs
@@ -262,6 +262,13 @@ import ZcashHaskell.Types
   -> `()'
 #}
 
+{# fun unsafe rust_wrapper_get_sapling_root as rustWrapperGetSaplingRootTest
+     { `Int8'
+     , getVarBuffer `Buffer HexString'&
+     }
+  -> `()'
+#}
+
 {# fun unsafe rust_wrapper_read_orchard_node as rustWrapperReadOrchardNode
      { toBorshVar* `BS.ByteString'&
      , getVarBuffer `Buffer HexString'&
diff --git a/src/ZcashHaskell/Sapling.hs b/src/ZcashHaskell/Sapling.hs
index c7fe8e2..362a8e9 100644
--- a/src/ZcashHaskell/Sapling.hs
+++ b/src/ZcashHaskell/Sapling.hs
@@ -20,6 +20,7 @@ module ZcashHaskell.Sapling where
 import C.Zcash
   ( rustWrapperCombineSaplingNodes
   , rustWrapperDecodeSaplingAddress
+  , rustWrapperGetSaplingRootTest
   , rustWrapperIsShielded
   , rustWrapperReadSaplingCommitmentTree
   , rustWrapperReadSaplingFrontier
@@ -206,6 +207,10 @@ combineSaplingNodes level n1 n2 =
       withPureBorshVarBuffer $
       rustWrapperCombineSaplingNodes level (toBytes n1) (toBytes n2)
 
+getSaplingRootTest :: Int8 -> HexString
+getSaplingRootTest level =
+  withPureBorshVarBuffer $ rustWrapperGetSaplingRootTest level
+
 getSaplingFrontier :: SaplingCommitmentTree -> Maybe SaplingFrontier
 getSaplingFrontier tree =
   if sf_pos updatedTree > 1