add get tree node helper from @chm-diederichs

holepunchto · Aug 16, 2023 · 8fb53ef · 8fb53ef
1 parent c134b43
commit 8fb53ef
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/lib/merkle-tree.js b/lib/merkle-tree.js
@@ -85,6 +85,17 @@ class MerkleTreeBatch {
  return caps.treeSignableLegacy(hash, this.length, this.fork)
  }
 
+ get (index) {
+ if (index >= this.length * 2) return null
+ if (index < this.treeLength) return this.tree.get(index)
+
+ for (const n of this.nodes) {
+ if (n.index === index) return n
+ }
+
+ return null
+ }
+
  append (buf) {
  const head = this.length * 2
  const ite = flat.iterator(head)