// SPDX-License-Identifier: MIT
pragma solidity ^0.8.24;

import {ERC20} from "@openzeppelin/contracts/token/ERC20/ERC20.sol";
import {ERC20Burnable} from "@openzeppelin/contracts/token/ERC20/extensions/ERC20Burnable.sol";
import {ERC20Permit} from "@openzeppelin/contracts/token/ERC20/extensions/ERC20Permit.sol";
import {AccessControl} from "@openzeppelin/contracts/access/AccessControl.sol";
import {Pausable} from "@openzeppelin/contracts/utils/Pausable.sol";

/// @title ProjectToken
/// @notice ERC-20 token with role-based minting, burning, and emergency pause
/// @dev Uses OpenZeppelin v5 contracts — no custom crypto
contract ProjectToken is ERC20, ERC20Burnable, ERC20Permit, AccessControl, Pausable {
    bytes32 public constant MINTER_ROLE = keccak256("MINTER_ROLE");
    bytes32 public constant PAUSER_ROLE = keccak256("PAUSER_ROLE");

    uint256 public immutable MAX_SUPPLY;

    error MaxSupplyExceeded(uint256 requested, uint256 available);

    constructor(
        string memory name_,
        string memory symbol_,
        uint256 maxSupply_
    ) ERC20(name_, symbol_) ERC20Permit(name_) {
        MAX_SUPPLY = maxSupply_;

        _grantRole(DEFAULT_ADMIN_ROLE, msg.sender);
        _grantRole(MINTER_ROLE, msg.sender);
        _grantRole(PAUSER_ROLE, msg.sender);
    }

    /// @notice Mint tokens to a recipient
    /// @param to Recipient address
    /// @param amount Amount of tokens to mint (in wei)
    function mint(address to, uint256 amount) external onlyRole(MINTER_ROLE) {
        if (totalSupply() + amount > MAX_SUPPLY) {
            revert MaxSupplyExceeded(amount, MAX_SUPPLY - totalSupply());
        }
        _mint(to, amount);
    }

    function pause() external onlyRole(PAUSER_ROLE) {
        _pause();
    }

    function unpause() external onlyRole(PAUSER_ROLE) {
        _unpause();
    }

    function _update(
        address from,
        address to,
        uint256 value
    ) internal override whenNotPaused {
        super._update(from, to, value);
    }
}

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.24;

import {UUPSUpgradeable} from "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
import {OwnableUpgradeable} from "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
import {ReentrancyGuardUpgradeable} from "@openzeppelin/contracts-upgradeable/utils/ReentrancyGuardUpgradeable.sol";
import {PausableUpgradeable} from "@openzeppelin/contracts-upgradeable/utils/PausableUpgradeable.sol";
import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
import {SafeERC20} from "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";

/// @title StakingVault
/// @notice Upgradeable staking vault with timelock withdrawals
/// @dev UUPS proxy pattern — upgrade logic lives in implementation
contract StakingVault is
    UUPSUpgradeable,
    OwnableUpgradeable,
    ReentrancyGuardUpgradeable,
    PausableUpgradeable
{
    using SafeERC20 for IERC20;

    struct StakeInfo {
        uint128 amount;       // Packed: 128 bits
        uint64 stakeTime;     // Packed: 64 bits — good until year 584 billion
        uint64 lockEndTime;   // Packed: 64 bits — same slot as above
    }

    IERC20 public stakingToken;
    uint256 public lockDuration;
    uint256 public totalStaked;
    mapping(address => StakeInfo) public stakes;

    event Staked(address indexed user, uint256 amount, uint256 lockEndTime);
    event Withdrawn(address indexed user, uint256 amount);
    event LockDurationUpdated(uint256 oldDuration, uint256 newDuration);

    error ZeroAmount();
    error LockNotExpired(uint256 lockEndTime, uint256 currentTime);
    error NoStake();

    /// @custom:oz-upgrades-unsafe-allow constructor
    constructor() {
        _disableInitializers();
    }

    function initialize(
        address stakingToken_,
        uint256 lockDuration_,
        address owner_
    ) external initializer {
        __UUPSUpgradeable_init();
        __Ownable_init(owner_);
        __ReentrancyGuard_init();
        __Pausable_init();

        stakingToken = IERC20(stakingToken_);
        lockDuration = lockDuration_;
    }

    /// @notice Stake tokens into the vault
    /// @param amount Amount of tokens to stake
    function stake(uint256 amount) external nonReentrant whenNotPaused {
        if (amount == 0) revert ZeroAmount();

        // Effects before interactions
        StakeInfo storage info = stakes[msg.sender];
        info.amount += uint128(amount);
        info.stakeTime = uint64(block.timestamp);
        info.lockEndTime = uint64(block.timestamp + lockDuration);
        totalStaked += amount;

        emit Staked(msg.sender, amount, info.lockEndTime);

        // Interaction last — SafeERC20 handles non-standard returns
        stakingToken.safeTransferFrom(msg.sender, address(this), amount);
    }

    /// @notice Withdraw staked tokens after lock period
    function withdraw() external nonReentrant {
        StakeInfo storage info = stakes[msg.sender];
        uint256 amount = info.amount;

        if (amount == 0) revert NoStake();
        if (block.timestamp < info.lockEndTime) {
            revert LockNotExpired(info.lockEndTime, block.timestamp);
        }

        // Effects before interactions
        info.amount = 0;
        info.stakeTime = 0;
        info.lockEndTime = 0;
        totalStaked -= amount;

        emit Withdrawn(msg.sender, amount);

        // Interaction last
        stakingToken.safeTransfer(msg.sender, amount);
    }

    function setLockDuration(uint256 newDuration) external onlyOwner {
        emit LockDurationUpdated(lockDuration, newDuration);
        lockDuration = newDuration;
    }

    function pause() external onlyOwner { _pause(); }
    function unpause() external onlyOwner { _unpause(); }

    /// @dev Only owner can authorize upgrades
    function _authorizeUpgrade(address) internal override onlyOwner {}
}

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.24;

import {Test, console2} from "forge-std/Test.sol";
import {StakingVault} from "../src/StakingVault.sol";
import {ERC1967Proxy} from "@openzeppelin/contracts/proxy/ERC1967/ERC1967Proxy.sol";
import {MockERC20} from "./mocks/MockERC20.sol";

contract StakingVaultTest is Test {
    StakingVault public vault;
    MockERC20 public token;
    address public owner = makeAddr("owner");
    address public alice = makeAddr("alice");
    address public bob = makeAddr("bob");

    uint256 constant LOCK_DURATION = 7 days;
    uint256 constant STAKE_AMOUNT = 1000e18;

    function setUp() public {
        token = new MockERC20("Stake Token", "STK");

        // Deploy behind UUPS proxy
        StakingVault impl = new StakingVault();
        bytes memory initData = abi.encodeCall(
            StakingVault.initialize,
            (address(token), LOCK_DURATION, owner)
        );
        ERC1967Proxy proxy = new ERC1967Proxy(address(impl), initData);
        vault = StakingVault(address(proxy));

        // Fund test accounts
        token.mint(alice, 10_000e18);
        token.mint(bob, 10_000e18);

        vm.prank(alice);
        token.approve(address(vault), type(uint256).max);
        vm.prank(bob);
        token.approve(address(vault), type(uint256).max);
    }

    function test_stake_updatesBalance() public {
        vm.prank(alice);
        vault.stake(STAKE_AMOUNT);

        (uint128 amount,,) = vault.stakes(alice);
        assertEq(amount, STAKE_AMOUNT);
        assertEq(vault.totalStaked(), STAKE_AMOUNT);
        assertEq(token.balanceOf(address(vault)), STAKE_AMOUNT);
    }

    function test_withdraw_revertsBeforeLock() public {
        vm.prank(alice);
        vault.stake(STAKE_AMOUNT);

        vm.prank(alice);
        vm.expectRevert();
        vault.withdraw();
    }

    function test_withdraw_succeedsAfterLock() public {
        vm.prank(alice);
        vault.stake(STAKE_AMOUNT);

        vm.warp(block.timestamp + LOCK_DURATION + 1);

        vm.prank(alice);
        vault.withdraw();

        (uint128 amount,,) = vault.stakes(alice);
        assertEq(amount, 0);
        assertEq(token.balanceOf(alice), 10_000e18);
    }

    function test_stake_revertsWhenPaused() public {
        vm.prank(owner);
        vault.pause();

        vm.prank(alice);
        vm.expectRevert();
        vault.stake(STAKE_AMOUNT);
    }

    function testFuzz_stake_arbitraryAmount(uint128 amount) public {
        vm.assume(amount > 0 && amount <= 10_000e18);

        vm.prank(alice);
        vault.stake(amount);

        (uint128 staked,,) = vault.stakes(alice);
        assertEq(staked, amount);
    }
}

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.24;

/// @title GasOptimizationPatterns
/// @notice Reference patterns for minimizing gas consumption
contract GasOptimizationPatterns {
    // PATTERN 1: Storage packing — fit multiple values in one 32-byte slot
    // Bad: 3 slots (96 bytes)
    // uint256 id;      // slot 0
    // uint256 amount;  // slot 1
    // address owner;   // slot 2

    // Good: 2 slots (64 bytes)
    struct PackedData {
        uint128 id;       // slot 0 (16 bytes)
        uint128 amount;   // slot 0 (16 bytes) — same slot!
        address owner;    // slot 1 (20 bytes)
        uint96 timestamp; // slot 1 (12 bytes) — same slot!
    }

    // PATTERN 2: Custom errors save ~50 gas per revert vs require strings
    error Unauthorized(address caller);
    error InsufficientBalance(uint256 requested, uint256 available);

    // PATTERN 3: Use mappings over arrays for lookups — O(1) vs O(n)
    mapping(address => uint256) public balances;

    // PATTERN 4: Cache storage reads in memory
    function optimizedTransfer(address to, uint256 amount) external {
        uint256 senderBalance = balances[msg.sender]; // 1 SLOAD
        if (senderBalance < amount) {
            revert InsufficientBalance(amount, senderBalance);
        }
        unchecked {
            // Safe because of the check above
            balances[msg.sender] = senderBalance - amount;
        }
        balances[to] += amount;
    }

    // PATTERN 5: Use calldata for read-only external array params
    function processIds(uint256[] calldata ids) external pure returns (uint256 sum) {
        uint256 len = ids.length; // Cache length
        for (uint256 i; i < len;) {
            sum += ids[i];
            unchecked { ++i; } // Save gas on increment — cannot overflow
        }
    }

    // PATTERN 6: Prefer uint256 / int256 — the EVM operates on 32-byte words
    // Smaller types (uint8, uint16) cost extra gas for masking UNLESS packed in storage
}

import { ethers, upgrades } from "hardhat";

async function main() {
  const [deployer] = await ethers.getSigners();
  console.log("Deploying with:", deployer.address);

  // 1. Deploy token
  const Token = await ethers.getContractFactory("ProjectToken");
  const token = await Token.deploy(
    "Protocol Token",
    "PTK",
    ethers.parseEther("1000000000") // 1B max supply
  );
  await token.waitForDeployment();
  console.log("Token deployed to:", await token.getAddress());

  // 2. Deploy vault behind UUPS proxy
  const Vault = await ethers.getContractFactory("StakingVault");
  const vault = await upgrades.deployProxy(
    Vault,
    [await token.getAddress(), 7 * 24 * 60 * 60, deployer.address],
    { kind: "uups" }
  );
  await vault.waitForDeployment();
  console.log("Vault proxy deployed to:", await vault.getAddress());

  // 3. Grant minter role to vault if needed
  // const MINTER_ROLE = await token.MINTER_ROLE();
  // await token.grantRole(MINTER_ROLE, await vault.getAddress());
}

main().catch((error) => {
  console.error(error);
  process.exitCode = 1;
});