44 lines
1.1 KiB
Solidity
44 lines
1.1 KiB
Solidity
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
pragma solidity ^0.6.0;
|
||
|
|
|
||
|
|
import "./ERC20.sol";
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dev Extension of {ERC20} that adds a cap to the supply of tokens.
|
||
|
|
*/
|
||
|
|
abstract contract ERC20Capped is ERC20 {
|
||
|
|
uint256 private _cap;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dev Sets the value of the `cap`. This value is immutable, it can only be
|
||
|
|
* set once during construction.
|
||
|
|
*/
|
||
|
|
constructor (uint256 cap) public {
|
||
|
|
require(cap > 0, "ERC20Capped: cap is 0");
|
||
|
|
_cap = cap;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dev Returns the cap on the token's total supply.
|
||
|
|
*/
|
||
|
|
function cap() public view returns (uint256) {
|
||
|
|
return _cap;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dev See {ERC20-_beforeTokenTransfer}.
|
||
|
|
*
|
||
|
|
* Requirements:
|
||
|
|
*
|
||
|
|
* - minted tokens must not cause the total supply to go over the cap.
|
||
|
|
*/
|
||
|
|
function _beforeTokenTransfer(address from, address to, uint256 amount) internal virtual override {
|
||
|
|
super._beforeTokenTransfer(from, to, amount);
|
||
|
|
|
||
|
|
if (from == address(0)) { // When minting tokens
|
||
|
|
require(totalSupply().add(amount) <= _cap, "ERC20Capped: cap exceeded");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|