0408e51ae6
* Merge ERC20Detailed into ERC20, make derived contracts abstract * Fix Create2 tests * Fix failing test * Default decimals to 18 * Add tests for setupDecimals * Add changelog entry * Update CHANGELOG.md * Update CHANGELOG.md * Replace isConstructor for !isContract * Update CHANGELOG.md Co-Authored-By: Francisco Giordano <frangio.1@gmail.com> Co-authored-by: Francisco Giordano <frangio.1@gmail.com>
42 lines
1.1 KiB
Solidity
42 lines
1.1 KiB
Solidity
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");
|
|
}
|
|
}
|
|
}
|