feat: PathQuery depth

grovedb/build.rs

@@ -13,7 +13,10 @@ fn main() {
     let grovedbg_zip_path = out_dir.join("grovedbg.zip");
 
     if !grovedbg_zip_path.exists() {
-        let response = reqwest::blocking::get(format!("https://github.com/dashpay/grovedbg/releases/download/{GROVEDBG_VERSION}/grovedbg-{GROVEDBG_VERSION}.zip"))
+        let response = reqwest::blocking::get(format!(
+                "https://github.com/dashpay/grovedbg/releases/download/{GROVEDBG_VERSION}/\
+grovedbg-{GROVEDBG_VERSION}.zip"
+            ))
         .expect("can't download GroveDBG artifact");
 
         let mut grovedbg_zip = File::create(&grovedbg_zip_path).unwrap();

grovedb/src/query/mod.rs

@@ -134,6 +134,12 @@ impl PathQuery {
         Self { path, query }
     }
 
+    /// The max depth od the query, this is the maximum layers we could get back
+    /// from grovedb
+    pub fn max_depth(&self) -> u32 {
+        self.query.query.max_depth()
+    }
+
     /// Gets the path of all terminal keys
     pub fn terminal_keys(
         &self,
@@ -1634,6 +1640,8 @@ mod tests {
             },
         };
 
+        assert_eq!(path_query.max_depth(), 4);
+
         {
             let path = vec![];
             let first = path_query

merk/src/proofs/query/mod.rs

@@ -70,6 +70,20 @@ pub struct SubqueryBranch {
     pub subquery: Option<Box<Query>>,
 }
 
+impl SubqueryBranch {
+    /// Returns the depth of the subquery branch
+    /// This depth is how many GroveDB layers down we could query at maximum
+    #[inline]
+    pub fn max_depth(&self) -> u32 {
+        let subquery_path_depth = self
+            .subquery_path
+            .as_ref()
+            .map_or(0, |path| path.len() as u32);
+        let subquery_depth = self.subquery.as_ref().map_or(0, |query| query.max_depth());
+        subquery_path_depth + subquery_depth
+    }
+}
+
 #[cfg(any(feature = "full", feature = "verify"))]
 /// `Query` represents one or more keys or ranges of keys, which can be used to
 /// resolve a proof which will include all the requested values.
@@ -476,6 +490,23 @@ impl Query {
         // checks if all searched for items are keys
         self.items.iter().all(|a| a.is_key())
     }
+
+    /// Returns the depth of the subquery branch
+    /// This depth is how many GroveDB layers down we could query at maximum
+    pub fn max_depth(&self) -> u32 {
+        let default_subquery_branch_depth = self.default_subquery_branch.max_depth();
+        let conditional_subquery_branches_max_depth = self
+            .conditional_subquery_branches
+            .as_ref()
+            .map_or(0, |condition_subqueries| {
+                condition_subqueries
+                    .values()
+                    .map(|conditional_subquery_branch| conditional_subquery_branch.max_depth())
+                    .max()
+                    .unwrap_or_default()
+            });
+        1 + default_subquery_branch_depth.max(conditional_subquery_branches_max_depth)
+    }
 }
 
 #[cfg(feature = "full")]