Add finish raffle

Author: VorobevSA

contracts/RaffleFactory.sol

@@ -12,7 +12,7 @@ contract RaffleFactory is Initializable, UUPSUpgradeable, AccessControlUpgradeab
     bytes32 public constant MANAGER_ROLE = keccak256("MANAGER_ROLE");
     address[] public raffles;
 
-    event RaffleCreated(address indexed raffleAddress, address indexed creator);
+    event RaffleCreated(uint256 indexed raffleId, address indexed raffleAddress, address indexed creator);
 
     /// @custom:oz-upgrades-unsafe-allow constructor
     constructor() {
@@ -38,7 +38,7 @@ contract RaffleFactory is Initializable, UUPSUpgradeable, AccessControlUpgradeab
         _;
     }
 
-    // Reentrancy guarded via `nonReentrant` modifier
+    // Reentrancy is guarded via the `nonReentrant` modifier
     // slither-disable-next-line reentrancy-benign
     function createRaffle(
         string memory name_,
@@ -61,8 +61,18 @@ contract RaffleFactory is Initializable, UUPSUpgradeable, AccessControlUpgradeab
             "Approve to raffle failed"
         );
         // Start the raffle (transfer tokens to the contract)
-        raffle.startRaffle(prizeToken, amount);
-        emit RaffleCreated(address(raffle), msg.sender);
+        raffle.start(prizeToken, amount);
+        uint256 raffleId = raffles.length - 1;
+        emit RaffleCreated(raffleId, address(raffle), msg.sender);
         return address(raffle);
     }
+
+    function finishRaffle(uint256 raffleId) external onlyManager nonReentrant {
+        require(raffleId < raffles.length, "Raffle does not exist");
+        address raffleAddr = raffles[raffleId];
+        RaffleNFT raffle = RaffleNFT(raffleAddr);
+        require(raffle.winnerAddress() == address(0), "Winner already determined");
+        require(raffle.started(), "Raffle not started");
+        raffle.finish();
+    }
 }

contracts/RaffleNFT.sol

@@ -4,26 +4,34 @@ pragma solidity ^0.8.28;
 import "@openzeppelin/contracts/token/ERC721/ERC721.sol";
 import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
 import "@openzeppelin/contracts/utils/ReentrancyGuard.sol";
+import "@openzeppelin/contracts/access/Ownable.sol";
 
-contract RaffleNFT is ERC721, ReentrancyGuard {
+contract RaffleNFT is ERC721, ReentrancyGuard, Ownable {
     address public prizeToken;
     uint256 public amount;
     bool public started;
     string private baseTokenURI;
+    uint256 private _tokenIdCounter;
+    mapping(uint256 => address) public tokenOwners;
+    mapping(address => uint256) public lastMintTime;
+    address public winnerAddress;
+
+    event Start(address indexed prizeToken, uint256 amount);
+    event Finish(address indexed prizeToken, uint256 amount, address indexed winner);
 
     constructor(
         string memory name_,
         string memory symbol_,
         string memory tokenURI_
-    ) ERC721(name_, symbol_) {
+    ) ERC721(name_, symbol_) Ownable(msg.sender) {
         baseTokenURI = tokenURI_;
         started = false;
     }
 
-    function startRaffle(
+    function start(
         address prizeToken_,
         uint256 amount_
-    ) external nonReentrant {
+    ) external onlyOwner nonReentrant {
         require(prizeToken_ != address(0), "Zero address");
         require(!started, "Already started");
         started = true;
@@ -33,9 +41,65 @@ contract RaffleNFT is ERC721, ReentrancyGuard {
             IERC20(prizeToken).transferFrom(msg.sender, address(this), amount),
             "ERC20 transfer failed"
         );
+        emit Start(prizeToken_, amount_);
+    }
+
+    function finish() external onlyOwner nonReentrant {
+        require(started, "Raffle not started");
+        require(_tokenIdCounter > 0, "No participants");
+        require(amount > 0, "No prize");
+
+        uint256 winnerTokenId = uint256(
+            keccak256(abi.encodePacked(
+                block.prevrandao, 
+                address(this),
+                tx.gasprice,
+                gasleft(),
+                _tokenIdCounter
+                ))
+        ) % _tokenIdCounter;
+        address winner = ownerOf(winnerTokenId);
+        require(winner != address(0), "Winner not found");
+        winnerAddress = winner;
+
+        uint256 prize = amount;
+        amount = 0;
+        started = false;
+        require(
+            IERC20(prizeToken).transfer(winner, prize),
+            "ERC20 transfer to winner failed"
+        );
+        emit Finish(prizeToken, prize, winner);
+    }
+
+    function mint() public {
+        // We intentionally use block.timestamp to limit NFT minting to once every 24 hours per address.
+        // This is acceptable for our use case, as precise cryptographic unpredictability is not required here.
+        // slither-disable-next-line timestamp
+        require(
+            block.timestamp - lastMintTime[msg.sender] >= 1 days,
+            "You can only mint once every 24 hours"
+        );
+        uint256 tokenId = _tokenIdCounter;
+        _mint(msg.sender, tokenId);
+        tokenOwners[tokenId] = msg.sender;
+        _tokenIdCounter++;
+        lastMintTime[msg.sender] = block.timestamp;
     }
 
     function tokenURI(uint256) public view override returns (string memory) {
         return baseTokenURI;
     }
+
+    function updateTokenURI  (string memory tokenURI_) external onlyOwner {
+        baseTokenURI = tokenURI_;
+    }
+
+    function _update(address to, uint256 tokenId, address auth) internal override returns (address) {
+        address from = super._update(to, tokenId, auth);
+        if (from != to && to != address(0)) {
+            tokenOwners[tokenId] = to;
+        }
+        return from;
+    }
 }

hardhat.config.js

@@ -1,5 +1,6 @@
 require("@nomicfoundation/hardhat-toolbox");
 require('@openzeppelin/hardhat-upgrades');
+require("dotenv").config();
 
 /** @type import('hardhat/config').HardhatUserConfig */
 module.exports = {
@@ -12,4 +13,17 @@ module.exports = {
       },
     },
   },
+  networks: {
+    hardhat: {
+      forking: {
+        url: "https://rpc.eth.haqq.network",
+      },
+    },
+    'haqq-testedge2': {
+      url: `https://rpc.eth.testedge2.haqq.network`,
+    },
+    haqq: {
+      url: "https://rpc.eth.haqq.network",
+    },
+  },
 };

package-lock.json

@@ -13,6 +13,7 @@
     "@nomicfoundation/hardhat-toolbox": "^6.0.0",
     "@openzeppelin/contracts-upgradeable": "^5.3.0",
     "@openzeppelin/hardhat-upgrades": "^3.9.1",
+    "dotenv": "^17.2.0",
     "hardhat": "^2.25.0"
   }
 }