@waku/rln 0.1.5-053bb95.0 → 0.1.5-35b50c3.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (61) hide show
  1. package/bundle/_virtual/utils.js +2 -2
  2. package/bundle/_virtual/utils2.js +2 -2
  3. package/bundle/index.js +2 -1
  4. package/bundle/packages/rln/dist/contract/constants.js +6 -2
  5. package/bundle/packages/rln/dist/contract/rln_base_contract.js +279 -144
  6. package/bundle/packages/rln/dist/contract/rln_contract.js +74 -89
  7. package/bundle/packages/rln/dist/credentials_manager.js +1 -1
  8. package/bundle/packages/rln/dist/identity.js +0 -8
  9. package/bundle/packages/rln/dist/keystore/keystore.js +28 -15
  10. package/bundle/packages/rln/node_modules/@chainsafe/bls-keystore/node_modules/ethereum-cryptography/random.js +1 -1
  11. package/bundle/packages/rln/node_modules/@chainsafe/bls-keystore/node_modules/ethereum-cryptography/utils.js +2 -2
  12. package/bundle/packages/rln/node_modules/@noble/hashes/_sha2.js +1 -1
  13. package/bundle/packages/rln/node_modules/@noble/hashes/hmac.js +1 -1
  14. package/bundle/packages/rln/node_modules/@noble/hashes/pbkdf2.js +1 -1
  15. package/bundle/packages/rln/node_modules/@noble/hashes/scrypt.js +1 -1
  16. package/bundle/packages/rln/node_modules/@noble/hashes/sha256.js +1 -1
  17. package/bundle/packages/rln/node_modules/@noble/hashes/sha512.js +1 -1
  18. package/bundle/packages/rln/node_modules/@noble/hashes/utils.js +1 -1
  19. package/dist/.tsbuildinfo +1 -1
  20. package/dist/contract/constants.d.ts +1 -1
  21. package/dist/contract/constants.js +1 -1
  22. package/dist/contract/constants.js.map +1 -1
  23. package/dist/contract/index.d.ts +1 -0
  24. package/dist/contract/index.js +1 -0
  25. package/dist/contract/index.js.map +1 -1
  26. package/dist/contract/rln_base_contract.d.ts +27 -25
  27. package/dist/contract/rln_base_contract.js +279 -144
  28. package/dist/contract/rln_base_contract.js.map +1 -1
  29. package/dist/contract/rln_contract.d.ts +4 -24
  30. package/dist/contract/rln_contract.js +74 -89
  31. package/dist/contract/rln_contract.js.map +1 -1
  32. package/dist/contract/types.d.ts +5 -0
  33. package/dist/contract/types.js.map +1 -1
  34. package/dist/credentials_manager.js +1 -1
  35. package/dist/credentials_manager.js.map +1 -1
  36. package/dist/identity.d.ts +0 -1
  37. package/dist/identity.js +0 -8
  38. package/dist/identity.js.map +1 -1
  39. package/dist/index.d.ts +2 -1
  40. package/dist/index.js +1 -0
  41. package/dist/index.js.map +1 -1
  42. package/dist/keystore/keystore.d.ts +1 -0
  43. package/dist/keystore/keystore.js +28 -15
  44. package/dist/keystore/keystore.js.map +1 -1
  45. package/dist/keystore/types.d.ts +2 -2
  46. package/package.json +1 -1
  47. package/src/contract/constants.ts +1 -1
  48. package/src/contract/index.ts +1 -0
  49. package/src/contract/rln_base_contract.ts +427 -216
  50. package/src/contract/rln_contract.ts +95 -120
  51. package/src/contract/types.ts +5 -0
  52. package/src/credentials_manager.ts +1 -1
  53. package/src/identity.ts +0 -9
  54. package/src/index.ts +3 -1
  55. package/src/keystore/keystore.ts +54 -29
  56. package/src/keystore/types.ts +2 -2
  57. package/bundle/packages/rln/dist/contract/errors.js +0 -62
  58. package/dist/contract/errors.d.ts +0 -30
  59. package/dist/contract/errors.js +0 -61
  60. package/dist/contract/errors.js.map +0 -1
  61. package/src/contract/errors.ts +0 -75
@@ -1,10 +1,19 @@
1
1
  import { ethers } from "ethers";
2
2
  import { IdentityCredential } from "../identity.js";
3
3
  import { DecryptedCredentials } from "../keystore/types.js";
4
- import { Member, MembershipInfo, RLNContractInitOptions } from "./types.js";
4
+ import { CustomQueryOptions, FetchMembersOptions, Member, MembershipInfo, RLNContractInitOptions } from "./types.js";
5
5
  export declare class RLNBaseContract {
6
6
  contract: ethers.Contract;
7
+ private deployBlock;
7
8
  private rateLimit;
9
+ protected _members: Map<number, Member>;
10
+ private _membersFilter;
11
+ private _membershipErasedFilter;
12
+ private _membersExpiredFilter;
13
+ /**
14
+ * Constructor for RLNBaseContract.
15
+ * Allows injecting a mocked contract for testing purposes.
16
+ */
8
17
  constructor(options: RLNContractInitOptions);
9
18
  /**
10
19
  * Gets the current rate limit for this contract instance
@@ -48,26 +57,19 @@ export declare class RLNBaseContract {
48
57
  * @param newRateLimit The new rate limit to use
49
58
  */
50
59
  setRateLimit(newRateLimit: number): Promise<void>;
51
- /**
52
- * Gets all members in the given range
53
- * @param startIndex Start index (inclusive)
54
- * @param endIndex End index (exclusive)
55
- */
56
- getMembersInRange(startIndex: number, endIndex: number): Promise<Member[]>;
57
- /**
58
- * Gets all current members
59
- */
60
- getAllMembers(): Promise<Member[]>;
61
- /**
62
- * Gets the member index if it exists, or null if it doesn't
63
- * Throws only on actual errors (invalid input, network issues, etc)
64
- */
65
- private getMemberIndex;
66
- getMembershipInfo(idCommitment: string): Promise<MembershipInfo>;
67
- extendMembership(idCommitment: string): Promise<void>;
68
- eraseMembership(idCommitment: string, eraseFromMembershipSet?: boolean): Promise<void>;
69
- registerMembership(idCommitment: string, rateLimit?: number): Promise<void>;
70
- withdraw(token: string, holder: string): Promise<void>;
60
+ get members(): Member[];
61
+ fetchMembers(options?: FetchMembersOptions): Promise<void>;
62
+ static queryFilter(contract: ethers.Contract, options: CustomQueryOptions): Promise<ethers.Event[]>;
63
+ processEvents(events: ethers.Event[]): void;
64
+ static splitToChunks(from: number, to: number, step: number): Array<[number, number]>;
65
+ static takeN<T>(array: T[], size: number): Iterable<T[]>;
66
+ static ignoreErrors<T>(promise: Promise<T>, defaultValue: T): Promise<T>;
67
+ subscribeToMembers(): void;
68
+ getMembershipInfo(idCommitmentBigInt: bigint): Promise<MembershipInfo | undefined>;
69
+ extendMembership(idCommitmentBigInt: bigint): Promise<ethers.ContractTransaction>;
70
+ eraseMembership(idCommitmentBigInt: bigint, eraseFromMembershipSet?: boolean): Promise<ethers.ContractTransaction>;
71
+ withdraw(token: string, from: string): Promise<void>;
72
+ registerMembership(idCommitmentBigInt: bigint, rateLimit?: number): Promise<ethers.ContractTransaction>;
71
73
  registerWithIdentity(identity: IdentityCredential): Promise<DecryptedCredentials | undefined>;
72
74
  registerWithPermitAndErase(identity: IdentityCredential, permit: {
73
75
  owner: string;
@@ -81,8 +83,8 @@ export declare class RLNBaseContract {
81
83
  * @throws Error if the rate limit is outside the allowed range
82
84
  */
83
85
  private validateRateLimit;
84
- /**
85
- * Helper to confirm a transaction and extract event data
86
- */
87
- private confirmTransaction;
86
+ private get membersFilter();
87
+ private get membershipErasedFilter();
88
+ private get membersExpiredFilter();
89
+ private getMemberIndex;
88
90
  }
@@ -1,18 +1,51 @@
1
+ /* eslint-disable no-console */
1
2
  import { Logger } from "@waku/utils";
2
3
  import { ethers } from "ethers";
3
4
  import { RLN_ABI } from "./abi.js";
4
5
  import { DEFAULT_RATE_LIMIT, RATE_LIMIT_PARAMS } from "./constants.js";
5
- import { InvalidMembershipError, InvalidRateLimitError, MembershipExistsError, MembershipNotFoundError, RateLimitExceededError, RLNContractError, TransactionError } from "./errors.js";
6
6
  import { MembershipState } from "./types.js";
7
7
  const log = new Logger("waku:rln:contract:base");
8
8
  export class RLNBaseContract {
9
9
  contract;
10
+ deployBlock;
10
11
  rateLimit;
12
+ _members = new Map();
13
+ _membersFilter;
14
+ _membershipErasedFilter;
15
+ _membersExpiredFilter;
16
+ /**
17
+ * Constructor for RLNBaseContract.
18
+ * Allows injecting a mocked contract for testing purposes.
19
+ */
11
20
  constructor(options) {
12
21
  const { address, signer, rateLimit = DEFAULT_RATE_LIMIT, contract } = options;
13
- this.validateRateLimit(rateLimit);
22
+ log.info("Initializing RLNBaseContract", { address, rateLimit });
14
23
  this.contract = contract || new ethers.Contract(address, RLN_ABI, signer);
15
24
  this.rateLimit = rateLimit;
25
+ try {
26
+ log.info("Setting up event filters");
27
+ // Initialize event filters
28
+ this._membersFilter = this.contract.filters.MembershipRegistered();
29
+ this._membershipErasedFilter = this.contract.filters.MembershipErased();
30
+ this._membersExpiredFilter = this.contract.filters.MembershipExpired();
31
+ log.info("Event filters initialized successfully");
32
+ }
33
+ catch (error) {
34
+ log.error("Failed to initialize event filters", { error });
35
+ throw new Error("Failed to initialize event filters: " + error.message);
36
+ }
37
+ // Initialize members and subscriptions
38
+ this.fetchMembers()
39
+ .then(() => {
40
+ this.subscribeToMembers();
41
+ })
42
+ .catch((error) => {
43
+ log.error("Failed to initialize members", { error });
44
+ });
45
+ // Validate rate limit asynchronously
46
+ this.validateRateLimit(rateLimit).catch((error) => {
47
+ log.error("Failed to validate initial rate limit", { error });
48
+ });
16
49
  }
17
50
  /**
18
51
  * Gets the current rate limit for this contract instance
@@ -80,221 +113,298 @@ export class RLNBaseContract {
80
113
  * @param newRateLimit The new rate limit to use
81
114
  */
82
115
  async setRateLimit(newRateLimit) {
83
- this.validateRateLimit(newRateLimit);
116
+ await this.validateRateLimit(newRateLimit);
84
117
  this.rateLimit = newRateLimit;
85
118
  }
86
- /**
87
- * Gets all members in the given range
88
- * @param startIndex Start index (inclusive)
89
- * @param endIndex End index (exclusive)
90
- */
91
- async getMembersInRange(startIndex, endIndex) {
92
- try {
93
- // Get all commitments in one call
94
- const idCommitments = await this.contract.getRateCommitmentsInRangeBoundsInclusive(startIndex, endIndex - 1 // -1 because getRateCommitmentsInRangeBoundsInclusive is inclusive
95
- );
96
- // Get membership info for each commitment in a single batch
97
- const membershipPromises = idCommitments.map((idCommitment) => this.contract
98
- .memberships(idCommitment)
99
- .then((info) => ({
100
- idCommitment: idCommitment.toString(),
101
- index: ethers.BigNumber.from(info.index)
102
- }))
103
- .catch(() => null) // Skip invalid members
104
- );
105
- // Wait for all promises to resolve
106
- const members = (await Promise.all(membershipPromises)).filter((m) => m !== null);
107
- return members;
119
+ get members() {
120
+ const sortedMembers = Array.from(this._members.values()).sort((left, right) => left.index.toNumber() - right.index.toNumber());
121
+ return sortedMembers;
122
+ }
123
+ async fetchMembers(options = {}) {
124
+ const registeredMemberEvents = await RLNBaseContract.queryFilter(this.contract, {
125
+ fromBlock: this.deployBlock,
126
+ ...options,
127
+ membersFilter: this.membersFilter
128
+ });
129
+ const removedMemberEvents = await RLNBaseContract.queryFilter(this.contract, {
130
+ fromBlock: this.deployBlock,
131
+ ...options,
132
+ membersFilter: this.membershipErasedFilter
133
+ });
134
+ const expiredMemberEvents = await RLNBaseContract.queryFilter(this.contract, {
135
+ fromBlock: this.deployBlock,
136
+ ...options,
137
+ membersFilter: this.membersExpiredFilter
138
+ });
139
+ const events = [
140
+ ...registeredMemberEvents,
141
+ ...removedMemberEvents,
142
+ ...expiredMemberEvents
143
+ ];
144
+ this.processEvents(events);
145
+ }
146
+ static async queryFilter(contract, options) {
147
+ const FETCH_CHUNK = 5;
148
+ const BLOCK_RANGE = 3000;
149
+ const { fromBlock, membersFilter, fetchRange = BLOCK_RANGE, fetchChunks = FETCH_CHUNK } = options;
150
+ if (fromBlock === undefined) {
151
+ return contract.queryFilter(membersFilter);
108
152
  }
109
- catch (error) {
110
- if (error instanceof Error &&
111
- error.message.includes("InvalidPaginationQuery")) {
112
- throw new RLNContractError(`Invalid pagination range: start=${startIndex}, end=${endIndex}`);
153
+ if (!contract.provider) {
154
+ throw Error("No provider found on the contract.");
155
+ }
156
+ const toBlock = await contract.provider.getBlockNumber();
157
+ if (toBlock - fromBlock < fetchRange) {
158
+ return contract.queryFilter(membersFilter, fromBlock, toBlock);
159
+ }
160
+ const events = [];
161
+ const chunks = RLNBaseContract.splitToChunks(fromBlock, toBlock, fetchRange);
162
+ for (const portion of RLNBaseContract.takeN(chunks, fetchChunks)) {
163
+ const promises = portion.map(([left, right]) => RLNBaseContract.ignoreErrors(contract.queryFilter(membersFilter, left, right), []));
164
+ const fetchedEvents = await Promise.all(promises);
165
+ events.push(fetchedEvents.flatMap((v) => v));
166
+ }
167
+ return events.flatMap((v) => v);
168
+ }
169
+ processEvents(events) {
170
+ const toRemoveTable = new Map();
171
+ const toInsertTable = new Map();
172
+ events.forEach((evt) => {
173
+ if (!evt.args) {
174
+ return;
113
175
  }
114
- throw error;
176
+ if (evt.event === "MembershipErased" ||
177
+ evt.event === "MembershipExpired") {
178
+ let index = evt.args.index;
179
+ if (!index) {
180
+ return;
181
+ }
182
+ if (typeof index === "number" || typeof index === "string") {
183
+ index = ethers.BigNumber.from(index);
184
+ }
185
+ const toRemoveVal = toRemoveTable.get(evt.blockNumber);
186
+ if (toRemoveVal != undefined) {
187
+ toRemoveVal.push(index.toNumber());
188
+ toRemoveTable.set(evt.blockNumber, toRemoveVal);
189
+ }
190
+ else {
191
+ toRemoveTable.set(evt.blockNumber, [index.toNumber()]);
192
+ }
193
+ }
194
+ else if (evt.event === "MembershipRegistered") {
195
+ let eventsPerBlock = toInsertTable.get(evt.blockNumber);
196
+ if (eventsPerBlock == undefined) {
197
+ eventsPerBlock = [];
198
+ }
199
+ eventsPerBlock.push(evt);
200
+ toInsertTable.set(evt.blockNumber, eventsPerBlock);
201
+ }
202
+ });
203
+ }
204
+ static splitToChunks(from, to, step) {
205
+ const chunks = [];
206
+ let left = from;
207
+ while (left < to) {
208
+ const right = left + step < to ? left + step : to;
209
+ chunks.push([left, right]);
210
+ left = right;
115
211
  }
212
+ return chunks;
116
213
  }
117
- /**
118
- * Gets all current members
119
- */
120
- async getAllMembers() {
121
- const nextIndex = (await this.contract.nextFreeIndex()).toNumber();
122
- return this.getMembersInRange(0, nextIndex);
214
+ static *takeN(array, size) {
215
+ let start = 0;
216
+ while (start < array.length) {
217
+ const portion = array.slice(start, start + size);
218
+ yield portion;
219
+ start += size;
220
+ }
123
221
  }
124
- /**
125
- * Gets the member index if it exists, or null if it doesn't
126
- * Throws only on actual errors (invalid input, network issues, etc)
127
- */
128
- async getMemberIndex(idCommitment) {
222
+ static async ignoreErrors(promise, defaultValue) {
129
223
  try {
130
- const isValid = await this.contract.isInMembershipSet(idCommitment);
131
- if (!isValid) {
132
- return null;
133
- }
134
- const membershipInfo = await this.contract.memberships(idCommitment);
135
- return ethers.BigNumber.from(membershipInfo.index);
224
+ return await promise;
136
225
  }
137
- catch (error) {
138
- log.error(`Error getting member index: ${error.message}`);
139
- throw new InvalidMembershipError(idCommitment);
226
+ catch (err) {
227
+ if (err instanceof Error) {
228
+ log.info(`Ignoring an error during query: ${err.message}`);
229
+ }
230
+ else {
231
+ log.info(`Ignoring an unknown error during query`);
232
+ }
233
+ return defaultValue;
140
234
  }
141
235
  }
142
- async getMembershipInfo(idCommitment) {
236
+ subscribeToMembers() {
237
+ this.contract.on(this.membersFilter, (_idCommitment, _membershipRateLimit, _index, event) => {
238
+ this.processEvents([event]);
239
+ });
240
+ this.contract.on(this.membershipErasedFilter, (_idCommitment, _membershipRateLimit, _index, event) => {
241
+ this.processEvents([event]);
242
+ });
243
+ this.contract.on(this.membersExpiredFilter, (_idCommitment, _membershipRateLimit, _index, event) => {
244
+ this.processEvents([event]);
245
+ });
246
+ }
247
+ async getMembershipInfo(idCommitmentBigInt) {
143
248
  try {
144
- const [startBlock, endBlock, rateLimit] = await this.contract.getMembershipInfo(idCommitment);
249
+ const membershipData = await this.contract.memberships(idCommitmentBigInt);
145
250
  const currentBlock = await this.contract.provider.getBlockNumber();
251
+ console.log("membershipData", membershipData);
252
+ const [depositAmount, activeDuration, gracePeriodStartTimestamp, gracePeriodDuration, rateLimit, index, holder, token] = membershipData;
253
+ const gracePeriodEnd = gracePeriodStartTimestamp.add(gracePeriodDuration);
146
254
  let state;
147
- if (currentBlock < startBlock) {
255
+ if (currentBlock < gracePeriodStartTimestamp.toNumber()) {
148
256
  state = MembershipState.Active;
149
257
  }
150
- else if (currentBlock < endBlock) {
258
+ else if (currentBlock < gracePeriodEnd.toNumber()) {
151
259
  state = MembershipState.GracePeriod;
152
260
  }
153
261
  else {
154
262
  state = MembershipState.Expired;
155
263
  }
156
- const index = await this.getMemberIndex(idCommitment);
157
- if (index === null) {
158
- throw new MembershipNotFoundError(idCommitment);
159
- }
160
264
  return {
161
265
  index,
162
- idCommitment,
163
- rateLimit: rateLimit.toNumber(),
164
- startBlock: startBlock.toNumber(),
165
- endBlock: endBlock.toNumber(),
166
- state
266
+ idCommitment: idCommitmentBigInt.toString(),
267
+ rateLimit: Number(rateLimit),
268
+ startBlock: gracePeriodStartTimestamp.toNumber(),
269
+ endBlock: gracePeriodEnd.toNumber(),
270
+ state,
271
+ depositAmount,
272
+ activeDuration,
273
+ gracePeriodDuration,
274
+ holder,
275
+ token
167
276
  };
168
277
  }
169
278
  catch (error) {
170
- if (error instanceof RLNContractError) {
171
- throw error;
172
- }
173
- log.error(`Error getting membership info: ${error.message}`);
174
- throw new InvalidMembershipError(idCommitment);
279
+ console.error("Error in getMembershipInfo:", error);
280
+ return undefined;
175
281
  }
176
282
  }
177
- async extendMembership(idCommitment) {
178
- const tx = await this.contract.extendMemberships([idCommitment]);
179
- await this.confirmTransaction(tx, "MembershipExtended", (event) => ({
180
- idCommitment: event.args.idCommitment,
181
- endBlock: event.args.endBlock
182
- }));
283
+ async extendMembership(idCommitmentBigInt) {
284
+ const tx = await this.contract.extendMemberships([idCommitmentBigInt]);
285
+ await tx.wait();
286
+ return tx;
183
287
  }
184
- async eraseMembership(idCommitment, eraseFromMembershipSet = true) {
185
- const tx = await this.contract.eraseMemberships([idCommitment], eraseFromMembershipSet);
186
- await this.confirmTransaction(tx, "MembershipErased", (event) => ({
187
- idCommitment: event.args.idCommitment,
188
- index: event.args.index
189
- }));
288
+ async eraseMembership(idCommitmentBigInt, eraseFromMembershipSet = true) {
289
+ const tx = await this.contract.eraseMemberships([idCommitmentBigInt], eraseFromMembershipSet);
290
+ await tx.wait();
291
+ return tx;
190
292
  }
191
- async registerMembership(idCommitment, rateLimit = DEFAULT_RATE_LIMIT) {
192
- if (rateLimit < RATE_LIMIT_PARAMS.MIN_RATE ||
193
- rateLimit > RATE_LIMIT_PARAMS.MAX_RATE) {
194
- throw new Error(`Rate limit must be between ${RATE_LIMIT_PARAMS.MIN_RATE} and ${RATE_LIMIT_PARAMS.MAX_RATE}`);
195
- }
196
- const tx = await this.contract.register(idCommitment, rateLimit, []);
197
- await this.confirmTransaction(tx, "MembershipRegistered", (event) => ({
198
- idCommitment: event.args.idCommitment,
199
- membershipRateLimit: event.args.membershipRateLimit,
200
- index: event.args.index
201
- }));
202
- }
203
- async withdraw(token, holder) {
293
+ async withdraw(token, from) {
204
294
  try {
205
- const tx = await this.contract.withdraw(token, { from: holder });
206
- await this.confirmTransaction(tx, "TokenWithdrawn", (event) => ({
207
- token: event.args.token,
208
- holder: event.args.holder,
209
- amount: event.args.amount
210
- }));
295
+ const tx = await this.contract.withdraw(token, from);
296
+ await tx.wait();
211
297
  }
212
298
  catch (error) {
213
299
  log.error(`Error in withdraw: ${error.message}`);
214
- throw error;
215
300
  }
216
301
  }
302
+ async registerMembership(idCommitmentBigInt, rateLimit = DEFAULT_RATE_LIMIT) {
303
+ if (rateLimit < RATE_LIMIT_PARAMS.MIN_RATE ||
304
+ rateLimit > RATE_LIMIT_PARAMS.MAX_RATE) {
305
+ throw new Error(`Rate limit must be between ${RATE_LIMIT_PARAMS.MIN_RATE} and ${RATE_LIMIT_PARAMS.MAX_RATE}`);
306
+ }
307
+ return this.contract.register(idCommitmentBigInt, rateLimit, []);
308
+ }
217
309
  async registerWithIdentity(identity) {
218
310
  try {
219
311
  log.info(`Registering identity with rate limit: ${this.rateLimit} messages/epoch`);
220
312
  // Check if the ID commitment is already registered
221
- const existingIndex = await this.getMemberIndex(identity.IDCommitmentBigInt.toString());
222
- if (existingIndex !== null) {
223
- throw new MembershipExistsError(identity.IDCommitmentBigInt.toString(), existingIndex.toString());
313
+ const existingIndex = await this.getMemberIndex(identity.IDCommitmentBigInt);
314
+ if (existingIndex) {
315
+ throw new Error(`ID commitment is already registered with index ${existingIndex}`);
224
316
  }
225
317
  // Check if there's enough remaining rate limit
226
318
  const remainingRateLimit = await this.getRemainingTotalRateLimit();
227
319
  if (remainingRateLimit < this.rateLimit) {
228
- throw new RateLimitExceededError(this.rateLimit, remainingRateLimit);
320
+ throw new Error(`Not enough remaining rate limit. Requested: ${this.rateLimit}, Available: ${remainingRateLimit}`);
229
321
  }
230
322
  const estimatedGas = await this.contract.estimateGas.register(identity.IDCommitmentBigInt, this.rateLimit, []);
231
323
  const gasLimit = estimatedGas.add(10000);
232
- const tx = await this.contract.register(identity.IDCommitmentBigInt, this.rateLimit, [], { gasLimit });
233
- const decodedData = await this.confirmTransaction(tx, "MembershipRegistered", (event) => ({
234
- idCommitment: event.args.idCommitment,
235
- membershipRateLimit: event.args.membershipRateLimit,
236
- index: event.args.index
237
- }));
324
+ const txRegisterResponse = await this.contract.register(identity.IDCommitmentBigInt, this.rateLimit, [], { gasLimit });
325
+ const txRegisterReceipt = await txRegisterResponse.wait();
326
+ if (txRegisterReceipt.status === 0) {
327
+ throw new Error("Transaction failed on-chain");
328
+ }
329
+ const memberRegistered = txRegisterReceipt.events?.find((event) => event.event === "MembershipRegistered");
330
+ if (!memberRegistered || !memberRegistered.args) {
331
+ log.error("Failed to register membership: No MembershipRegistered event found");
332
+ return undefined;
333
+ }
334
+ const decodedData = {
335
+ idCommitment: memberRegistered.args.idCommitment,
336
+ membershipRateLimit: memberRegistered.args.membershipRateLimit,
337
+ index: memberRegistered.args.index
338
+ };
238
339
  log.info(`Successfully registered membership with index ${decodedData.index} ` +
239
340
  `and rate limit ${decodedData.membershipRateLimit}`);
240
341
  const network = await this.contract.provider.getNetwork();
241
342
  const address = this.contract.address;
242
- const membershipId = decodedData.index.toString();
343
+ const membershipId = Number(decodedData.index);
243
344
  return {
244
345
  identity,
245
346
  membership: {
246
347
  address,
247
- treeIndex: parseInt(membershipId),
348
+ treeIndex: membershipId,
248
349
  chainId: network.chainId.toString(),
249
350
  rateLimit: decodedData.membershipRateLimit.toNumber()
250
351
  }
251
352
  };
252
353
  }
253
354
  catch (error) {
254
- if (error instanceof RLNContractError) {
255
- throw error;
256
- }
257
355
  if (error instanceof Error) {
258
356
  const errorMessage = error.message;
259
357
  log.error("registerWithIdentity - error message:", errorMessage);
260
358
  log.error("registerWithIdentity - error stack:", error.stack);
261
- // Map contract errors to our custom errors
359
+ // Try to extract more specific error information
262
360
  if (errorMessage.includes("CannotExceedMaxTotalRateLimit")) {
263
- throw new RateLimitExceededError(this.rateLimit, await this.getRemainingTotalRateLimit());
361
+ throw new Error("Registration failed: Cannot exceed maximum total rate limit");
264
362
  }
265
363
  else if (errorMessage.includes("InvalidIdCommitment")) {
266
- throw new InvalidMembershipError(identity.IDCommitmentBigInt.toString());
364
+ throw new Error("Registration failed: Invalid ID commitment");
267
365
  }
268
366
  else if (errorMessage.includes("InvalidMembershipRateLimit")) {
269
- throw new InvalidRateLimitError(this.rateLimit, RATE_LIMIT_PARAMS.MIN_RATE, RATE_LIMIT_PARAMS.MAX_RATE);
367
+ throw new Error("Registration failed: Invalid membership rate limit");
270
368
  }
271
369
  else if (errorMessage.includes("execution reverted")) {
272
- throw new TransactionError("Contract execution reverted. Check contract requirements.");
370
+ throw new Error("Contract execution reverted. Check contract requirements.");
371
+ }
372
+ else {
373
+ throw new Error(`Error in registerWithIdentity: ${errorMessage}`);
273
374
  }
274
- throw new RLNContractError(`Error in registerWithIdentity: ${errorMessage}`);
275
375
  }
276
- throw new RLNContractError("Unknown error in registerWithIdentity");
376
+ else {
377
+ throw new Error("Unknown error in registerWithIdentity", {
378
+ cause: error
379
+ });
380
+ }
277
381
  }
278
382
  }
279
383
  async registerWithPermitAndErase(identity, permit, idCommitmentsToErase) {
280
384
  try {
281
385
  log.info(`Registering identity with permit and rate limit: ${this.rateLimit} messages/epoch`);
282
- const tx = await this.contract.registerWithPermit(permit.owner, permit.deadline, permit.v, permit.r, permit.s, identity.IDCommitmentBigInt, this.rateLimit, idCommitmentsToErase.map((id) => ethers.BigNumber.from(id)));
283
- const decodedData = await this.confirmTransaction(tx, "MembershipRegistered", (event) => ({
284
- idCommitment: event.args.idCommitment,
285
- membershipRateLimit: event.args.membershipRateLimit,
286
- index: event.args.index
287
- }));
386
+ const txRegisterResponse = await this.contract.registerWithPermit(permit.owner, permit.deadline, permit.v, permit.r, permit.s, identity.IDCommitmentBigInt, this.rateLimit, idCommitmentsToErase.map((id) => ethers.BigNumber.from(id)));
387
+ const txRegisterReceipt = await txRegisterResponse.wait();
388
+ const memberRegistered = txRegisterReceipt.events?.find((event) => event.event === "MembershipRegistered");
389
+ if (!memberRegistered || !memberRegistered.args) {
390
+ log.error("Failed to register membership with permit: No MembershipRegistered event found");
391
+ return undefined;
392
+ }
393
+ const decodedData = {
394
+ idCommitment: memberRegistered.args.idCommitment,
395
+ membershipRateLimit: memberRegistered.args.membershipRateLimit,
396
+ index: memberRegistered.args.index
397
+ };
288
398
  log.info(`Successfully registered membership with permit. Index: ${decodedData.index}, ` +
289
399
  `Rate limit: ${decodedData.membershipRateLimit}, Erased ${idCommitmentsToErase.length} commitments`);
290
400
  const network = await this.contract.provider.getNetwork();
291
401
  const address = this.contract.address;
292
- const membershipId = decodedData.index.toString();
402
+ const membershipId = Number(decodedData.index);
293
403
  return {
294
404
  identity,
295
405
  membership: {
296
406
  address,
297
- treeIndex: parseInt(membershipId),
407
+ treeIndex: membershipId,
298
408
  chainId: network.chainId.toString(),
299
409
  rateLimit: decodedData.membershipRateLimit.toNumber()
300
410
  }
@@ -302,29 +412,54 @@ export class RLNBaseContract {
302
412
  }
303
413
  catch (error) {
304
414
  log.error(`Error in registerWithPermitAndErase: ${error.message}`);
305
- throw error;
415
+ return undefined;
306
416
  }
307
417
  }
308
418
  /**
309
419
  * Validates that the rate limit is within the allowed range
310
420
  * @throws Error if the rate limit is outside the allowed range
311
421
  */
312
- validateRateLimit(rateLimit) {
313
- if (rateLimit < RATE_LIMIT_PARAMS.MIN_RATE ||
314
- rateLimit > RATE_LIMIT_PARAMS.MAX_RATE) {
315
- throw new InvalidRateLimitError(rateLimit, RATE_LIMIT_PARAMS.MIN_RATE, RATE_LIMIT_PARAMS.MAX_RATE);
422
+ async validateRateLimit(rateLimit) {
423
+ const [minRate, maxRate] = await Promise.all([
424
+ this.contract.minMembershipRateLimit(),
425
+ this.contract.maxMembershipRateLimit()
426
+ ]);
427
+ const minRateNum = ethers.BigNumber.from(minRate).toNumber();
428
+ const maxRateNum = ethers.BigNumber.from(maxRate).toNumber();
429
+ if (rateLimit < minRateNum || rateLimit > maxRateNum) {
430
+ throw new Error(`Rate limit must be between ${minRateNum} and ${maxRateNum} messages per epoch`);
316
431
  }
317
432
  }
318
- /**
319
- * Helper to confirm a transaction and extract event data
320
- */
321
- async confirmTransaction(tx, expectedEvent, transform) {
322
- const receipt = await tx.wait();
323
- const event = receipt.events?.find((e) => e.event === expectedEvent);
324
- if (!event || !event.args) {
325
- throw new TransactionError(`Expected event ${expectedEvent} not found`);
433
+ get membersFilter() {
434
+ if (!this._membersFilter) {
435
+ throw Error("Members filter was not initialized.");
436
+ }
437
+ return this._membersFilter;
438
+ }
439
+ get membershipErasedFilter() {
440
+ if (!this._membershipErasedFilter) {
441
+ throw Error("MembershipErased filter was not initialized.");
442
+ }
443
+ return this._membershipErasedFilter;
444
+ }
445
+ get membersExpiredFilter() {
446
+ if (!this._membersExpiredFilter) {
447
+ throw Error("MembersExpired filter was not initialized.");
448
+ }
449
+ return this._membersExpiredFilter;
450
+ }
451
+ async getMemberIndex(idCommitmentBigInt) {
452
+ try {
453
+ const events = await this.contract.queryFilter(this.contract.filters.MembershipRegistered(idCommitmentBigInt));
454
+ if (events.length === 0)
455
+ return undefined;
456
+ // Get the most recent registration event
457
+ const event = events[events.length - 1];
458
+ return event.args?.index;
459
+ }
460
+ catch (error) {
461
+ return undefined;
326
462
  }
327
- return transform(event);
328
463
  }
329
464
  }
330
465
  //# sourceMappingURL=rln_base_contract.js.map