implement getProofElements part 2

Author: michael-spengler

src/merkle-tree.spec.ts

@@ -26,7 +26,7 @@ Deno.test("should return the correct proof for an element at a specific index",
     const exampleArray = ["dog", "horse", "cow", "chicken"]
     const merkleTree = new MerkleTree(exampleArray)
 
-    const proof = merkleTree.getProof(exampleArray.indexOf("dog"))
+    const proof = merkleTree.getProofElements(exampleArray.indexOf("dog"))
 
     const expectedProof = ["fd62862b6dc213bee77c2badd6311528253c6cb3107e03c16051aa15584eca1c", "a12b2f7a5ddb20963c22654f6b22a6955c9956a20c76a0e8f169a437aafb4c98"]
 

src/merkle-tree.ts

@@ -83,14 +83,20 @@ export class MerkleTree {
         return this.hashes.filter((e: IMerkleHashes) => e.level === 2)[0].hashes[0]
     }
 
-    public getProof(investigatedEntryIndex: number): string[] {
+    public getProofElements(investigatedEntryIndex: number): string[] {
         let level = 0
         let levels = this.hashes[this.hashes.length - 1].level
+        let proofElements: string[] = []
 
         console.log(this.hashes)
         console.log(`levels: ${levels}`)
 
-        return [this.hashes.filter((e: IMerkleHashes) => e.level === 0)[0].hashes[1], this.hashes.filter((e: IMerkleHashes) => e.level === 1)[0].hashes[1]]
+        while (level < levels) {
+            proofElements.push(this.hashes.filter((e: IMerkleHashes) => e.level === level)[investigatedEntryIndex].hashes[1])
+            level++
+        }
+
+        return proofElements
     }
 
     private checkInput(array: any[]) {

usage-example.ts

@@ -1,11 +1,12 @@
-import { MerkleTree, Helper } from "https://deno.land/x/merkletrees/mod.ts"
+// import { MerkleTree, Helper } from "https://deno.land/x/merkletrees/mod.ts"
+import { MerkleTree, Helper } from "./mod.ts"
 
 const exampleArray = ["dog", "horse", "cow", "chicken"]
 
 const merkleTree = new MerkleTree(exampleArray)
 
 const investigatedEntry = "dog"
-const proof = merkleTree.getProof(exampleArray.indexOf(investigatedEntry))
+const proof = merkleTree.getProofElements(exampleArray.indexOf(investigatedEntry))
 const investigatedEntryHashed = Helper.sha256(investigatedEntry)
 const rootHash = merkleTree.getRootHash()
 const isValid = merkleTree.verify(proof, investigatedEntryHashed, rootHash, exampleArray.indexOf(investigatedEntry))