From 861a776d05a301de075e8bd29e2a9d0633a41dce Mon Sep 17 00:00:00 2001
From: Danish Arora <danisharora099@gmail.com>
Date: Wed, 2 Apr 2025 17:33:10 +0530
Subject: [PATCH] chore: treeIndex as number

---
 packages/rln/src/contract/rln_contract.ts       | 4 ++--
 packages/rln/src/contract/rln_light_contract.ts | 4 ++--
 packages/rln/src/keystore/types.ts              | 2 +-
 packages/rln/src/rln.ts                         | 2 +-
 4 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/packages/rln/src/contract/rln_contract.ts b/packages/rln/src/contract/rln_contract.ts
index 3943208c5d..f08b188501 100644
--- a/packages/rln/src/contract/rln_contract.ts
+++ b/packages/rln/src/contract/rln_contract.ts
@@ -476,7 +476,7 @@ export class RLNContract {
         identity,
         membership: {
           address,
-          treeIndex: `0x${membershipId}`,
+          treeIndex: parseInt(membershipId),
           chainId: network.chainId.toString(),
           rateLimit: decodedData.membershipRateLimit.toNumber()
         }
@@ -602,7 +602,7 @@ export class RLNContract {
         identity,
         membership: {
           address,
-          treeIndex: `0x${membershipId}`,
+          treeIndex: parseInt(membershipId),
           chainId: network.chainId.toString(),
           rateLimit: decodedData.membershipRateLimit.toNumber()
         }
diff --git a/packages/rln/src/contract/rln_light_contract.ts b/packages/rln/src/contract/rln_light_contract.ts
index 0798633979..533c85e203 100644
--- a/packages/rln/src/contract/rln_light_contract.ts
+++ b/packages/rln/src/contract/rln_light_contract.ts
@@ -395,7 +395,7 @@ export class RLNLightContract {
         identity,
         membership: {
           address,
-          treeIndex: `0x${membershipId}`,
+          treeIndex: parseInt(membershipId),
           chainId: network.chainId.toString(),
           rateLimit: decodedData.membershipRateLimit.toNumber()
         }
@@ -526,7 +526,7 @@ export class RLNLightContract {
         identity,
         membership: {
           address,
-          treeIndex: `0x${membershipId}`,
+          treeIndex: parseInt(membershipId),
           chainId: network.chainId.toString(),
           rateLimit: decodedData.membershipRateLimit.toNumber()
         }
diff --git a/packages/rln/src/keystore/types.ts b/packages/rln/src/keystore/types.ts
index f9b75f56fe..89d240edfe 100644
--- a/packages/rln/src/keystore/types.ts
+++ b/packages/rln/src/keystore/types.ts
@@ -10,7 +10,7 @@ export type Password = string | Uint8Array;
 export type MembershipInfo = {
   chainId: string;
   address: string;
-  treeIndex: `0x${string}`; // hexadecimal string
+  treeIndex: number;
   rateLimit: number;
 };
 
diff --git a/packages/rln/src/rln.ts b/packages/rln/src/rln.ts
index 28d41a7d17..1610f1c918 100644
--- a/packages/rln/src/rln.ts
+++ b/packages/rln/src/rln.ts
@@ -288,7 +288,7 @@ export class RLNInstance {
     return createRLNEncoder({
       encoder: createEncoder(options),
       rlnInstance: this,
-      index: parseInt(credentials.membership.treeIndex),
+      index: credentials.membership.treeIndex,
       credential: credentials.identity
     });
   }