// SPDX-License-Identifier: MIT
// OpenZeppelin Confidential Contracts (last updated v0.5.0) (utils/structs/CheckpointsConfidential.sol)
// This file was procedurally generated from scripts/generate/templates/CheckpointsConfidential.js.

pragma solidity ^0.8.24;

import {euint32, euint64} from "@fhevm/solidity/lib/FHE.sol";
import {Checkpoints} from "@openzeppelin/contracts/utils/structs/Checkpoints.sol";

/**
 * @dev This library defines the `Trace*` struct, for checkpointing values as they change at different points in
 * time, and later looking up past values by block number.
 *
 * To create a history of checkpoints, define a variable type `CheckpointsConfidential.Trace*` in your contract, and store a new
 * checkpoint for the current transaction block using the {push} function.
 */
library CheckpointsConfidential {
    using Checkpoints for Checkpoints.Trace256;

    struct TraceEuint32 {
        Checkpoints.Trace256 _inner;
    }

    /**
     * @dev Pushes a (`key`, `value`) pair into a TraceEuint32 so that it is stored as the checkpoint.
     *
     * Returns previous value and new value.
     *
     * IMPORTANT: Never accept `key` as a user input, since an arbitrary `type(uint256).max` key set will disable the
     * library.
     */
    function push(
        TraceEuint32 storage self,
        uint256 key,
        euint32 value
    ) internal returns (euint32 oldValue, euint32 newValue) {
        (uint256 oldValueAsUint256, uint256 newValueAsUint256) = self._inner.push(key, uint256(euint32.unwrap(value)));
        oldValue = euint32.wrap(bytes32(oldValueAsUint256));
        newValue = euint32.wrap(bytes32(newValueAsUint256));
    }

    /**
     * @dev Returns the value in the first (oldest) checkpoint with key greater or equal than the search key, or zero if
     * there is none.
     */
    function lowerLookup(TraceEuint32 storage self, uint256 key) internal view returns (euint32) {
        return euint32.wrap(bytes32(self._inner.lowerLookup(key)));
    }

    /**
     * @dev Returns the value in the last (most recent) checkpoint with key lower or equal than the search key, or zero
     * if there is none.
     */
    function upperLookup(TraceEuint32 storage self, uint256 key) internal view returns (euint32) {
        return euint32.wrap(bytes32(self._inner.upperLookup(key)));
    }

    /**
     * @dev Returns the value in the last (most recent) checkpoint with key lower or equal than the search key, or zero
     * if there is none.
     *
     * NOTE: This is a variant of {upperLookup} that is optimized to find "recent" checkpoint (checkpoints with high
     * keys).
     */
    function upperLookupRecent(TraceEuint32 storage self, uint256 key) internal view returns (euint32) {
        return euint32.wrap(bytes32(self._inner.upperLookupRecent(key)));
    }

    /**
     * @dev Returns the value in the most recent checkpoint, or zero if there are no checkpoints.
     */
    function latest(TraceEuint32 storage self) internal view returns (euint32) {
        return euint32.wrap(bytes32(self._inner.latest()));
    }

    /**
     * @dev Returns whether there is a checkpoint in the structure (i.e. it is not empty), and if so the key and value
     * in the most recent checkpoint.
     */
    function latestCheckpoint(
        TraceEuint32 storage self
    ) internal view returns (bool exists, uint256 key, euint32 value) {
        uint256 valueAsUint256;
        (exists, key, valueAsUint256) = self._inner.latestCheckpoint();
        value = euint32.wrap(bytes32(valueAsUint256));
    }

    /**
     * @dev Returns the number of checkpoints.
     */
    function length(TraceEuint32 storage self) internal view returns (uint256) {
        return self._inner.length();
    }

    /**
     * @dev Returns checkpoint at given position.
     */
    function at(TraceEuint32 storage self, uint32 pos) internal view returns (uint256 key, euint32 value) {
        Checkpoints.Checkpoint256 memory checkpoint = self._inner.at(pos);
        key = checkpoint._key;
        value = euint32.wrap(bytes32(checkpoint._value));
    }

    struct TraceEuint64 {
        Checkpoints.Trace256 _inner;
    }

    /**
     * @dev Pushes a (`key`, `value`) pair into a TraceEuint64 so that it is stored as the checkpoint.
     *
     * Returns previous value and new value.
     *
     * IMPORTANT: Never accept `key` as a user input, since an arbitrary `type(uint256).max` key set will disable the
     * library.
     */
    function push(
        TraceEuint64 storage self,
        uint256 key,
        euint64 value
    ) internal returns (euint64 oldValue, euint64 newValue) {
        (uint256 oldValueAsUint256, uint256 newValueAsUint256) = self._inner.push(key, uint256(euint64.unwrap(value)));
        oldValue = euint64.wrap(bytes32(oldValueAsUint256));
        newValue = euint64.wrap(bytes32(newValueAsUint256));
    }

    /**
     * @dev Returns the value in the first (oldest) checkpoint with key greater or equal than the search key, or zero if
     * there is none.
     */
    function lowerLookup(TraceEuint64 storage self, uint256 key) internal view returns (euint64) {
        return euint64.wrap(bytes32(self._inner.lowerLookup(key)));
    }

    /**
     * @dev Returns the value in the last (most recent) checkpoint with key lower or equal than the search key, or zero
     * if there is none.
     */
    function upperLookup(TraceEuint64 storage self, uint256 key) internal view returns (euint64) {
        return euint64.wrap(bytes32(self._inner.upperLookup(key)));
    }

    /**
     * @dev Returns the value in the last (most recent) checkpoint with key lower or equal than the search key, or zero
     * if there is none.
     *
     * NOTE: This is a variant of {upperLookup} that is optimized to find "recent" checkpoint (checkpoints with high
     * keys).
     */
    function upperLookupRecent(TraceEuint64 storage self, uint256 key) internal view returns (euint64) {
        return euint64.wrap(bytes32(self._inner.upperLookupRecent(key)));
    }

    /**
     * @dev Returns the value in the most recent checkpoint, or zero if there are no checkpoints.
     */
    function latest(TraceEuint64 storage self) internal view returns (euint64) {
        return euint64.wrap(bytes32(self._inner.latest()));
    }

    /**
     * @dev Returns whether there is a checkpoint in the structure (i.e. it is not empty), and if so the key and value
     * in the most recent checkpoint.
     */
    function latestCheckpoint(
        TraceEuint64 storage self
    ) internal view returns (bool exists, uint256 key, euint64 value) {
        uint256 valueAsUint256;
        (exists, key, valueAsUint256) = self._inner.latestCheckpoint();
        value = euint64.wrap(bytes32(valueAsUint256));
    }

    /**
     * @dev Returns the number of checkpoints.
     */
    function length(TraceEuint64 storage self) internal view returns (uint256) {
        return self._inner.length();
    }

    /**
     * @dev Returns checkpoint at given position.
     */
    function at(TraceEuint64 storage self, uint32 pos) internal view returns (uint256 key, euint64 value) {
        Checkpoints.Checkpoint256 memory checkpoint = self._inner.at(pos);
        key = checkpoint._key;
        value = euint64.wrap(bytes32(checkpoint._value));
    }
}