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