codex-contracts-eth/contracts/StateRetrieval.sol

42 lines
1.3 KiB
Solidity

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.8;
import "@openzeppelin/contracts/utils/structs/EnumerableSet.sol";
import "./Requests.sol";
contract StateRetrieval {
using EnumerableSet for EnumerableSet.Bytes32Set;
using Requests for bytes32[];
mapping(address => EnumerableSet.Bytes32Set) private _requestsPerClient;
mapping(address => EnumerableSet.Bytes32Set) private _slotsPerHost;
function myRequests() public view returns (RequestId[] memory) {
return _requestsPerClient[msg.sender].values().toRequestIds();
}
function mySlots() public view returns (SlotId[] memory) {
return _slotsPerHost[msg.sender].values().toSlotIds();
}
function _hasSlots(address host) internal view returns (bool) {
return _slotsPerHost[host].length() > 0;
}
function _addToMyRequests(address client, RequestId requestId) internal {
_requestsPerClient[client].add(RequestId.unwrap(requestId));
}
function _addToMySlots(address host, SlotId slotId) internal {
_slotsPerHost[host].add(SlotId.unwrap(slotId));
}
function _removeFromMyRequests(address client, RequestId requestId) internal {
_requestsPerClient[client].remove(RequestId.unwrap(requestId));
}
function _removeFromMySlots(address host, SlotId slotId) internal {
_slotsPerHost[host].remove(SlotId.unwrap(slotId));
}
}