fix nexugenerator

Author: workwithhim

contracts/NexusGenerator.sol

@@ -2,10 +2,11 @@
 pragma solidity ^0.8.12;
 
 import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
+import "@openzeppelin/contracts/security/ReentrancyGuard.sol";
 import "./NexusToken.sol";
 import "./Utils/MultiOwnable.sol";
 
-contract NexusGenerator is MultiOwnable {
+contract NexusGenerator is MultiOwnable, ReentrancyGuard {
     using SafeMath for uint256;
 
     using SafeERC20 for IERC20;
@@ -444,7 +445,7 @@ contract NexusGenerator is MultiOwnable {
     }
 
     // Deposit LP tokens to NexusGenerator for NXS allocation.
-    function depositLP(uint256 _pid, uint256 _amount) public {
+    function depositLP(uint256 _pid, uint256 _amount) public nonReentrant {
         PoolInfo storage pool = poolInfo[_pid];
         UserInfo storage user = userInfo[_pid][msg.sender];
         uint256 rtLen = rewardTokenInfo[_pid].length;
@@ -478,7 +479,7 @@ contract NexusGenerator is MultiOwnable {
         uint256 _pid,
         IERC20 _rewardToken,
         uint256 _depositAmount
-    ) public {
+    ) public nonReentrant {
         require(_pid < poolLength(), "Non Exist Pool");
         uint256 _rid = getRewardTokenId(_pid, _rewardToken);
         require(_rid < rewardTokenInfo[_pid].length, "No reward token setting");
@@ -505,7 +506,7 @@ contract NexusGenerator is MultiOwnable {
     }
 
     // Withdraw LP tokens from NexusGenerator.
-    function withdraw(uint256 _pid, uint256 _amount) public {
+    function withdraw(uint256 _pid, uint256 _amount) public nonReentrant {
         PoolInfo storage pool = poolInfo[_pid];
         UserInfo storage user = userInfo[_pid][msg.sender];
         uint256 rtLen = rewardTokenInfo[_pid].length;
@@ -533,7 +534,7 @@ contract NexusGenerator is MultiOwnable {
     }
 
     // Withdraw without caring about rewards. EMERGENCY ONLY.
-    function emergencyWithdrawLP(uint256 _pid) public {
+    function emergencyWithdrawLP(uint256 _pid) public nonReentrant {
         PoolInfo storage pool = poolInfo[_pid];
         UserInfo storage user = userInfo[_pid][msg.sender];
         pool.lpToken.safeTransfer(address(msg.sender), user.amount);

scripts/mainnet/extra_deploy.ts

@@ -0,0 +1,22 @@
+import * as dotenv from "dotenv";
+import { deployContract } from '../utils';
+import { parseEther } from "ethers/lib/utils";
+
+dotenv.config();
+
+const nexusTokenaddress = "0x6DaF228391e388B05BBc682FEA3CB1Cc3E38c44E"
+const multiStakingDistributor = "0xa96d4Dfb9EED8824cae29fA0b7e62c72b5e51018";
+const nexusPerBlock = parseEther("0");
+const startBlock = 0
+const bonusEndBlock = 0
+
+
+async function main() {
+    // await deployContract("Multicall2", [], "Mocks");
+    const nexusGenerator = await deployContract("NexusGenerator", [nexusTokenaddress, multiStakingDistributor, nexusPerBlock, startBlock, bonusEndBlock], "")
+}
+
+main().catch((error) => {
+    console.error(error);
+    process.exitCode = 1;
+});

scripts/result/NexusGenerator_res.result

@@ -1 +1 @@
-NexusGenerator : 0x8Ad2D1A537fe16d1C619fD877a26FA584798107f
+NexusGenerator : 0x265287f794e956cFAF66297515BFF34631bB625c

scripts/utils.ts

@@ -15,6 +15,8 @@ export const deployContract = async function (name: string, params: any, periphe
     let address = factory.address
     await contract.deployTransaction.wait(5);
 
+    console.log(name, address)
+
     fs.mkdirSync(`scripts/params`, { recursive: true });
     fs.mkdirSync(`scripts/result`, { recursive: true });
     const outputCodes = `module.exports = ${JSON.stringify(params)}`