keycard-redeem/contracts/NFTBucket.sol

pragma solidity ^0.6.1;
pragma experimental ABIEncoderV2;

import "./Bucket.sol";
import "./erc721/IERC721.sol";
import "./erc721/IERC721Receiver.sol";
import "./erc721/IERC165.sol";

contract NFTBucket is Bucket, IERC165, IERC721Receiver {
  bytes4 private constant _ERC721_RECEIVED = 0x150b7a02; //bytes4(keccak256("onERC721Received(address,address,uint256,bytes)"))

  constructor(
    address _tokenAddress,
    uint256 _startTime,
    uint256 _expirationTime,
    uint256 _maxTxDelayInBlocks) Bucket("KeycardNFTBucket", _tokenAddress, _startTime, _expirationTime, _maxTxDelayInBlocks) public {}

  function transferRedeemable(uint256 data, Redeem memory redeem) internal override {
    IERC721(tokenAddress).safeTransferFrom(address(this), redeem.receiver, data);
  }

  function transferRedeemablesToOwner() internal override {
    IERC721(tokenAddress).setApprovalForAll(owner, true);
    assert(IERC721(tokenAddress).isApprovedForAll(address(this), owner));
  }

  function bucketType() external override returns (uint256) {
    return 721;
  }

  function supportsInterface(bytes4 interfaceID) external override(IERC165) view returns (bool) {
      return interfaceID == _ERC721_RECEIVED;
  }

  function onERC721Received(address _operator, address _from, uint256 _tokenID, bytes calldata _data) external override(IERC721Receiver) returns(bytes4) {
    require(msg.sender == tokenAddress, "only the NFT contract can call this");
    require((_operator == owner) || (_from == owner), "only the owner can create redeemables");
    require(_data.length == 52, "invalid data field");

    bytes memory d = _data;
    bytes32 tmp;
    bytes32 code;

    assembly {
      // tmp is 12 bytes of padding (taken from the array length) + 20 bytes of address
      tmp := mload(add(d, 20))
      code := mload(add(d, 52))
    }

    address recipient = address(uint160(uint256(tmp)));

    Redeemable storage redeemable = redeemables[recipient];
    require(redeemable.recipient == address(0), "recipient already used");

    redeemable.recipient = recipient;
    redeemable.code = code;
    redeemable.data = _tokenID;

    return _ERC721_RECEIVED;
  }
}