-
Notifications
You must be signed in to change notification settings - Fork 25
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
37c0bc3
commit 2b0ca89
Showing
4 changed files
with
89 additions
and
21 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,70 @@ | ||
| // SPDX-License-Identifier: MIT | ||
| pragma solidity 0.8.25; | ||
|
|
||
| import {Checkpoints} from "@openzeppelin/contracts/utils/structs/Checkpoints.sol"; | ||
| import {Time} from "@openzeppelin/contracts/utils/types/Time.sol"; | ||
|
|
||
| import {EnumerableMapV3} from "./EnumerableMapV3.sol"; | ||
|
|
||
| library OperatorMapWithTimeV3 { | ||
| using EnumerableMapV3 for EnumerableMapV3.OperatorMap; | ||
|
|
||
| error AlreadyAdded(); | ||
| error NotEnabled(); | ||
| error AlreadyEnabled(); | ||
|
|
||
| uint256 private constant ENABLED_TIME_MASK = 0xFFFFFFFFFFFFFFFFFFFFFFFF; | ||
| uint256 private constant DISABLED_TIME_MASK = 0xFFFFFFFFFFFFFFFFFFFFFFFF << 48; | ||
|
|
||
| function add(EnumerableMapV3.OperatorMap storage self, address addr) internal { | ||
| if (!self.set(addr, EnumerableMapV3.Operator("", address(0), 0))) { | ||
| revert AlreadyAdded(); | ||
| } | ||
| } | ||
|
|
||
| function disable(EnumerableMapV3.OperatorMap storage self, address addr) internal { | ||
| EnumerableMapV3.Operator memory operator = self.get(addr); | ||
| uint256 value = operator.timestamp; | ||
|
|
||
| if (uint48(value) == 0 || uint48(value >> 48) != 0) { | ||
| revert NotEnabled(); | ||
| } | ||
|
|
||
| value |= uint256(Time.timestamp()) << 48; | ||
| operator.timestamp = value; | ||
| self.set(addr, operator); | ||
| } | ||
|
|
||
| function enable(EnumerableMapV3.OperatorMap storage self, address addr) internal { | ||
| EnumerableMapV3.Operator memory operator = self.get(addr); | ||
| uint256 value = operator.timestamp; | ||
|
|
||
| if (uint48(value) != 0 && uint48(value >> 48) == 0) { | ||
| revert AlreadyEnabled(); | ||
| } | ||
|
|
||
| value = uint256(Time.timestamp()); | ||
| operator.timestamp = value; | ||
| self.set(addr, operator); | ||
| } | ||
|
|
||
| function atWithTimes( | ||
| EnumerableMapV3.OperatorMap storage self, | ||
| uint256 idx | ||
| ) internal view returns (address key, uint48 enabledTime, uint48 disabledTime) { | ||
| EnumerableMapV3.Operator memory value; | ||
| (key, value) = self.at(idx); | ||
| uint256 timestamp = value.timestamp; | ||
| enabledTime = uint48(timestamp); | ||
| disabledTime = uint48(timestamp >> 48); | ||
| } | ||
|
|
||
| function getTimes( | ||
| EnumerableMapV3.OperatorMap storage self, | ||
| address addr | ||
| ) internal view returns (uint48 enabledTime, uint48 disabledTime) { | ||
| EnumerableMapV3.Operator memory value = self.get(addr); | ||
| enabledTime = uint48(value.timestamp); | ||
| disabledTime = uint48(value.timestamp >> 48); | ||
| } | ||
| } |