minor changes to simulation

Author: chaitanyaprem

.gitignore

@@ -43,6 +43,7 @@ node_modules/
 
 # RLN / keystore
 rlnKeystore.json
+rln_keystore*.json
 *.tar.gz
 
 # Nimbus Build System

simulations/mixnet/setup_credentials.nim

@@ -10,11 +10,7 @@
 ##
 ## Usage: nim c -r setup_credentials.nim
 
-import
-  std/[os, strformat, options],
-  chronicles,
-  chronos,
-  results
+import std/[os, strformat, options], chronicles, chronos, results
 
 import
   mix_rln_spam_protection/credentials,
@@ -23,9 +19,9 @@ import
   mix_rln_spam_protection/types
 
 const
-  KeystorePassword = "mix-rln-password"  # Must match protocol.nim
-  DefaultUserMessageLimit = 100'u64      # Network-wide default rate limit
-  SpammerUserMessageLimit = 3'u64        # Lower limit for spammer testing
+  KeystorePassword = "mix-rln-password" # Must match protocol.nim
+  DefaultUserMessageLimit = 100'u64 # Network-wide default rate limit
+  SpammerUserMessageLimit = 3'u64 # Lower limit for spammer testing
 
   # Peer IDs derived from nodekeys in config files
   # config.toml:   nodekey = "f98e3fba96c32e8d1967d460f1b79457380e1a895f7971cecc8528abe733781a"
@@ -35,18 +31,23 @@ const
   # config4.toml:  nodekey = "3ce887b3c34b7a92dd2868af33941ed1dbec4893b054572cd5078da09dd923d4"
   # chat2mix.sh:   nodekey = "cb6fe589db0e5d5b48f7e82d33093e4d9d35456f4aaffc2322c473a173b2ac49"
   # chat2mix1.sh:  nodekey = "35eace7ccb246f20c487e05015ca77273d8ecaed0ed683de3d39bf4f69336feb"
-  # spammer.sh:    nodekey = "390460d51bdedbe8414ba104f3a1b648331e888545cb6580117ead47d6171368"
 
   # Node info: (peerId, userMessageLimit)
   NodeConfigs = [
-    ("16Uiu2HAmPiEs2ozjjJF2iN2Pe2FYeMC9w4caRHKYdLdAfjgbWM6o", DefaultUserMessageLimit),  # config.toml (service node)
-    ("16Uiu2HAmLtKaFaSWDohToWhWUZFLtqzYZGPFuXwKrojFVF6az5UF", DefaultUserMessageLimit),  # config1.toml (mix node 1)
-    ("16Uiu2HAmTEDHwAziWUSz6ZE23h5vxG2o4Nn7GazhMor4bVuMXTrA", DefaultUserMessageLimit),  # config2.toml (mix node 2)
-    ("16Uiu2HAmPwRKZajXtfb1Qsv45VVfRZgK3ENdfmnqzSrVm3BczF6f", DefaultUserMessageLimit),  # config3.toml (mix node 3)
-    ("16Uiu2HAmRhxmCHBYdXt1RibXrjAUNJbduAhzaTHwFCZT4qWnqZAu", DefaultUserMessageLimit),  # config4.toml (mix node 4)
-    ("16Uiu2HAm1QxSjNvNbsT2xtLjRGAsBLVztsJiTHr9a3EK96717hpj", DefaultUserMessageLimit),  # chat2mix client 1
-    ("16Uiu2HAmC9h26U1C83FJ5xpE32ghqya8CaZHX1Y7qpfHNnRABscN", DefaultUserMessageLimit),  # chat2mix client 2
-    ("16Uiu2HAmQwJtaL3r54D7WzCoBaUPdb1XRakHpWeEVqSPnYYXHQDL", SpammerUserMessageLimit),  # spammer (lower limit for testing)
+    ("16Uiu2HAmPiEs2ozjjJF2iN2Pe2FYeMC9w4caRHKYdLdAfjgbWM6o", DefaultUserMessageLimit),
+      # config.toml (service node)
+    ("16Uiu2HAmLtKaFaSWDohToWhWUZFLtqzYZGPFuXwKrojFVF6az5UF", DefaultUserMessageLimit),
+      # config1.toml (mix node 1)
+    ("16Uiu2HAmTEDHwAziWUSz6ZE23h5vxG2o4Nn7GazhMor4bVuMXTrA", DefaultUserMessageLimit),
+      # config2.toml (mix node 2)
+    ("16Uiu2HAmPwRKZajXtfb1Qsv45VVfRZgK3ENdfmnqzSrVm3BczF6f", DefaultUserMessageLimit),
+      # config3.toml (mix node 3)
+    ("16Uiu2HAmRhxmCHBYdXt1RibXrjAUNJbduAhzaTHwFCZT4qWnqZAu", DefaultUserMessageLimit),
+      # config4.toml (mix node 4)
+    ("16Uiu2HAm1QxSjNvNbsT2xtLjRGAsBLVztsJiTHr9a3EK96717hpj", DefaultUserMessageLimit),
+      # chat2mix client 1
+    ("16Uiu2HAmC9h26U1C83FJ5xpE32ghqya8CaZHX1Y7qpfHNnRABscN", DefaultUserMessageLimit),
+      # chat2mix client 2
   ]
 
 proc setupCredentialsAndTree() {.async.} =
@@ -56,15 +57,16 @@ proc setupCredentialsAndTree() {.async.} =
   echo "Generating credentials for ", NodeConfigs.len, " nodes...\n"
 
   # Generate credentials for all nodes
-  var allCredentials: seq[tuple[peerId: string, cred: IdentityCredential, rateLimit: uint64]]
+  var allCredentials:
+    seq[tuple[peerId: string, cred: IdentityCredential, rateLimit: uint64]]
   for (peerId, rateLimit) in NodeConfigs:
     let cred = generateCredentials().valueOr:
       echo "Failed to generate credentials for ", peerId, ": ", error
       quit(1)
 
     allCredentials.add((peerId: peerId, cred: cred, rateLimit: rateLimit))
     echo "Generated credentials for ", peerId
-    echo "  idCommitment: ", cred.idCommitment.toHex()[0..15], "..."
+    echo "  idCommitment: ", cred.idCommitment.toHex()[0 .. 15], "..."
     echo "  userMessageLimit: ", rateLimit
 
   echo ""
@@ -85,10 +87,13 @@ proc setupCredentialsAndTree() {.async.} =
   # Register all credentials in the tree with their specific rate limits
   echo "Registering all credentials in the Merkle tree..."
   for i, entry in allCredentials:
-    let index = (await groupManager.registerWithLimit(entry.cred.idCommitment, entry.rateLimit)).valueOr:
+    let index = (
+      await groupManager.registerWithLimit(entry.cred.idCommitment, entry.rateLimit)
+    ).valueOr:
       echo "Failed to register credential for ", entry.peerId, ": ", error
       quit(1)
-    echo "  Registered ", entry.peerId, " at index ", index, " (limit: ", entry.rateLimit, ")"
+    echo "  Registered ",
+      entry.peerId, " at index ", index, " (limit: ", entry.rateLimit, ")"
 
   echo ""
 
@@ -113,7 +118,7 @@ proc setupCredentialsAndTree() {.async.} =
       KeystorePassword,
       keystorePath,
       some(MembershipIndex(i)),
-      some(entry.rateLimit)
+      some(entry.rateLimit),
     )
     if saveResult.isErr:
       echo "Failed to save keystore for ", entry.peerId, ": ", saveResult.error