raffle-contracts/lib/erc721a/contracts/mocks/ERC721ATransferCounterMock.sol

// SPDX-License-Identifier: MIT
// ERC721A Contracts v4.2.3
// Creators: Chiru Labs

pragma solidity ^0.8.4;

import './ERC721AMock.sol';

contract ERC721ATransferCounterMock is ERC721AMock {
    constructor(string memory name_, string memory symbol_) ERC721AMock(name_, symbol_) {}

    function _extraData(
        address from,
        address to,
        uint24 previousExtraData
    ) internal view virtual override returns (uint24) {
        if (from == address(0)) {
            return 42;
        }
        if (to == address(0)) {
            return 1337;
        }
        return previousExtraData + 1;
    }

    function setExtraDataAt(uint256 index, uint24 extraData) public {
        _setExtraDataAt(index, extraData);
    }
}