logos-messaging/waku-rlnv1-contract
1
0
Fork 0
mirror of https://github.com/logos-messaging/waku-rlnv1-contract.git synced 2026-01-07 08:43:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: membership TTL

Browse Source
This commit is contained in:
Richard Ramos 2024-04-22 10:20:29 -04:00
parent 1f9661038d
commit 9e086ab2c6
No known key found for this signature in database
GPG Key ID: 1CE87DB518195760
1 changed files with 55 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
contracts/WakuRln.sol
View File

@ -7,25 +7,46 @@ import {Ownable} from "openzeppelin-contracts/contracts/access/Ownable.sol";
error NotImplemented();
error InvalidIdCommitmentIndex(uint256 idCommitment, uint256 index);
error MembershipStillActive(uint256 idCommitment);
contract WakuRln is Ownable, RlnBase {
uint16 public immutable contractIndex;
/// @notice The default TTL period in seconds for a membership
uint256 public immutable MEMBERSHIP_TTL;
/// @notice The expiry timestamp of a membership
/// maps from idCommitment to a timestamp
mapping(uint256 => uint256) public membershipExpiry;
constructor(
address _poseidonHasher,
uint16 _contractIndex
uint16 _contractIndex,
uint256 _ttl
) Ownable() RlnBase(0, 20, _poseidonHasher, address(0)) {
contractIndex = _contractIndex;
MEMBERSHIP_TTL = _ttl;
}
function _setCommitment(uint256 idCommitment, uint256 index) internal {
_validateRegistration(idCommitment);
members[idCommitment] = index;
memberExists[idCommitment] = true;
membershipExpiry[idCommitment] = block.timestamp + MEMBERSHIP_TTL;
emit MemberRegistered(idCommitment, index);
}
/// Registers a member
/// @param idCommitment The idCommitment of the member
function _register(uint256 idCommitment) internal {
_validateRegistration(idCommitment);
if (idCommitmentIndex >= SET_SIZE) revert FullTree();
members[idCommitment] = idCommitmentIndex;
memberExists[idCommitment] = true;
_setCommitment(idCommitment, idCommitmentIndex);
emit MemberRegistered(idCommitment, idCommitmentIndex);
idCommitmentIndex += 1;
}
@ -39,6 +60,27 @@ contract WakuRln is Ownable, RlnBase {
}
}
/// Register a member at specific index
/// @param idCommitment The idCommitment of the member
/// @param index The index in which to register the member
function registerAtIndex(uint256 idCommitment, uint256 index) external onlyOwner {
_validateIndexExpiration(idCommitment, index);
_setCommitment(idCommitment, index);
}
/// Renew membership credentials
function renew(uint256 idCommitment /*, uint256 periods*/) external onlyOwner {
// TODO: should this function be payable, and also, accept renewals for more than 1 period?
// if (msg.value != periods * MEMBERSHIP_DEPOSIT) revert InvalidTransactionValue()
// TODO: should we allow renewals using a grace period?
uint256 expiry = membershipExpiry[idCommitment];
if (expiry == 0) revert InvalidIdCommitment(idCommitment);
membershipExpiry[idCommitment] += MEMBERSHIP_TTL /* * periods */;
}
function register(uint256 idCommitment) external payable override {
revert NotImplemented();
}
@ -57,7 +99,14 @@ contract WakuRln is Ownable, RlnBase {
if (!isValidCommitment(idCommitment))
revert InvalidIdCommitment(idCommitment);
if (memberExists[idCommitment] == true) revert DuplicateIdCommitment();
if (idCommitmentIndex >= SET_SIZE) revert FullTree();
}
function _validateIndexExpiration(
uint256 idCommitment,
uint256 index
) internal view {
if (index >= SET_SIZE) revert InvalidIdCommitmentIndex(idCommitment, index);
if (membershipExpiry[idCommitment] > block.timestamp) revert MembershipStillActive(idCommitment);
}
function _validateSlash(