From 1005b67b80526949946ba4aec462e0860dc5a20a Mon Sep 17 00:00:00 2001
From: rymnc <43716372+rymnc@users.noreply.github.com>
Date: Thu, 8 Feb 2024 14:11:04 +0530
Subject: [PATCH] feat: diff checker

---
 tools/rln_db_inspector/external_config.nim  |  7 +++
 tools/rln_db_inspector/rln_db_inspector.nim | 53 +++++++++++++++++++++
 vendor/nim-unittest2                        |  2 +-
 3 files changed, 61 insertions(+), 1 deletion(-)

diff --git a/tools/rln_db_inspector/external_config.nim b/tools/rln_db_inspector/external_config.nim
index e6d37484e..d31e755fd 100644
--- a/tools/rln_db_inspector/external_config.nim
+++ b/tools/rln_db_inspector/external_config.nim
@@ -36,12 +36,19 @@ type
       defaultValue: "",
       name: "rln-relay-tree-path" }: string
 
+    rlnRelayTreePathDiff* {.
+      desc: "The path to the rln-relay tree diff",
+      defaultValue: "",
+      name: "rln-relay-tree-path-diff" }: string
+
 
 proc loadConfig*(T: type RlnDbInspectorConf): Result[T, string] =
   try:
     let conf = RlnDbInspectorConf.load()
     if conf.rlnRelayTreePath == "":
       return err("--rln-relay-tree-path must be set")
+    if conf.rlnRelayTreePathDiff == "":
+      return err("--rln-relay-tree-path-diff must be set")
     ok(conf)
   except CatchableError, Exception:
     err(getCurrentExceptionMsg())
diff --git a/tools/rln_db_inspector/rln_db_inspector.nim b/tools/rln_db_inspector/rln_db_inspector.nim
index 864f50612..5415a987a 100644
--- a/tools/rln_db_inspector/rln_db_inspector.nim
+++ b/tools/rln_db_inspector/rln_db_inspector.nim
@@ -31,14 +31,67 @@ when isMainModule:
     error "failure while creating RLN instance", error
     quit(1)
 
+  let rlnInstance2 = createRLNInstance(d=20, tree_path = conf.rlnRelayTreePathDiff).valueOr:
+    error "failure while creating RLN instance", error
+    quit(1)
+
   # 3. get metadata
   let metadata = rlnInstance.getMetadata().valueOr:
     error "failure while getting RLN metadata", error
     quit(1)
 
+  let metadata2 = rlnInstance2.getMetadata().valueOr:
+    error "failure while getting RLN metadata", error
+    quit(1)
+
+
+
   info "RLN metadata", lastProcessedBlock = metadata.lastProcessedBlock, 
                        chainId = metadata.chainId,
                        contractAddress = metadata.contractAddress,
                        validRoots = metadata.validRoots.mapIt(it.inHex())
 
+  info "RLN DIFF metadata", lastProcessedBlock = metadata2.lastProcessedBlock, 
+                       chainId = metadata2.chainId,
+                       contractAddress = metadata2.contractAddress,
+                       validRoots = metadata2.validRoots.mapIt(it.inHex())
+
+  info "RUNNING DIFF CHECK"
+
+  var index: uint = 0
+  while true:
+    let leaf = rlnInstance.getMember(index).valueOr:
+      error "failure while getting RLN member", error
+      quit(1)
+    let leaf2 = rlnInstance2.getMember(index).valueOr:
+      error "failure while getting RLN member", error
+      quit(1)
+
+    debug "leaf", index = index, leaf = leaf.inHex(), leaf2 = leaf2.inHex()
+
+    if leaf != leaf2:
+      error "DIFF FOUND", index = index, leaf = leaf.inHex(), leaf2 = leaf2.inHex()
+
+    # repeat for 0-31 index
+    if leaf[0] == 0 and leaf2[0] == 0 and leaf[1] == 0 and leaf2[1] == 0 and 
+      leaf[2] == 0 and leaf2[2] == 0 and leaf[3] == 0 and leaf2[3] == 0 and 
+      leaf[4] == 0 and leaf2[4] == 0 and leaf[5] == 0 and leaf2[5] == 0 and
+      leaf[6] == 0 and leaf2[6] == 0 and leaf[7] == 0 and leaf2[7] == 0 and
+      leaf[8] == 0 and leaf2[8] == 0 and leaf[9] == 0 and leaf2[9] == 0 and
+      leaf[10] == 0 and leaf2[10] == 0 and leaf[11] == 0 and leaf2[11] == 0 and
+      leaf[12] == 0 and leaf2[12] == 0 and leaf[13] == 0 and leaf2[13] == 0 and
+      leaf[14] == 0 and leaf2[14] == 0 and leaf[15] == 0 and leaf2[15] == 0 and
+      leaf[16] == 0 and leaf2[16] == 0 and leaf[17] == 0 and leaf2[17] == 0 and
+      leaf[18] == 0 and leaf2[18] == 0 and leaf[19] == 0 and leaf2[19] == 0 and
+      leaf[20] == 0 and leaf2[20] == 0 and leaf[21] == 0 and leaf2[21] == 0 and
+      leaf[22] == 0 and leaf2[22] == 0 and leaf[23] == 0 and leaf2[23] == 0 and
+      leaf[24] == 0 and leaf2[24] == 0 and leaf[25] == 0 and leaf2[25] == 0 and
+      leaf[26] == 0 and leaf2[26] == 0 and leaf[27] == 0 and leaf2[27] == 0 and
+      leaf[28] == 0 and leaf2[28] == 0 and leaf[29] == 0 and leaf2[29] == 0 and
+      leaf[30] == 0 and leaf2[30] == 0 and leaf[31] == 0 and leaf2[31] == 0:
+      info "reached end of RLN tree", index = index
+      quit(0)
+    
+    index += 1
+
   quit(0)
diff --git a/vendor/nim-unittest2 b/vendor/nim-unittest2
index db67e2ad7..262b697f3 160000
--- a/vendor/nim-unittest2
+++ b/vendor/nim-unittest2
@@ -1 +1 @@
-Subproject commit db67e2ad76840993ff82015987e3089a7d76f55f
+Subproject commit 262b697f38d6b6f1e7462d3b3ab81d79b894e336