37 lines
903 B
Solidity
37 lines
903 B
Solidity
// SPDX-License-Identifier: MIT
|
|
|
|
pragma solidity ^0.8.0;
|
|
|
|
import "../ERC20.sol";
|
|
|
|
/**
|
|
* @dev Extension of {ERC20} that adds a cap to the supply of tokens.
|
|
*/
|
|
abstract contract ERC20Capped is ERC20 {
|
|
uint256 immutable private _cap;
|
|
|
|
/**
|
|
* @dev Sets the value of the `cap`. This value is immutable, it can only be
|
|
* set once during construction.
|
|
*/
|
|
constructor (uint256 cap_) {
|
|
require(cap_ > 0, "ERC20Capped: cap is 0");
|
|
_cap = cap_;
|
|
}
|
|
|
|
/**
|
|
* @dev Returns the cap on the token's total supply.
|
|
*/
|
|
function cap() public view virtual returns (uint256) {
|
|
return _cap;
|
|
}
|
|
|
|
/**
|
|
* @dev See {ERC20-_mint}.
|
|
*/
|
|
function _mint(address account, uint256 amount) internal virtual override {
|
|
require(ERC20.totalSupply() + amount <= cap(), "ERC20Capped: cap exceeded");
|
|
super._mint(account, amount);
|
|
}
|
|
}
|