You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.2 KiB
85 lines
2.2 KiB
// SPDX-License-Identifier: MIT
|
|
|
|
pragma solidity ^0.8.0;
|
|
|
|
import "../ERC1155.sol";
|
|
|
|
/**
|
|
* @dev Extension of ERC1155 that adds tracking of total supply per id.
|
|
*
|
|
* Useful for scenarios where Fungible and Non-fungible tokens have to be
|
|
* clearly identified. Note: While a totalSupply of 1 might mean the
|
|
* corresponding is an NFT, there is no guarantees that no other token with the
|
|
* same id are not going to be minted.
|
|
*/
|
|
abstract contract ERC1155Supply is ERC1155 {
|
|
mapping(uint256 => uint256) private _totalSupply;
|
|
|
|
/**
|
|
* @dev Total amount of tokens in with a given id.
|
|
*/
|
|
function totalSupply(uint256 id) public view virtual returns (uint256) {
|
|
return _totalSupply[id];
|
|
}
|
|
|
|
/**
|
|
* @dev Indicates weither any token exist with a given id, or not.
|
|
*/
|
|
function exists(uint256 id) public view virtual returns (bool) {
|
|
return ERC1155Supply.totalSupply(id) > 0;
|
|
}
|
|
|
|
/**
|
|
* @dev See {ERC1155-_mint}.
|
|
*/
|
|
function _mint(
|
|
address account,
|
|
uint256 id,
|
|
uint256 amount,
|
|
bytes memory data
|
|
) internal virtual override {
|
|
super._mint(account, id, amount, data);
|
|
_totalSupply[id] += amount;
|
|
}
|
|
|
|
/**
|
|
* @dev See {ERC1155-_mintBatch}.
|
|
*/
|
|
function _mintBatch(
|
|
address to,
|
|
uint256[] memory ids,
|
|
uint256[] memory amounts,
|
|
bytes memory data
|
|
) internal virtual override {
|
|
super._mintBatch(to, ids, amounts, data);
|
|
for (uint256 i = 0; i < ids.length; ++i) {
|
|
_totalSupply[ids[i]] += amounts[i];
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @dev See {ERC1155-_burn}.
|
|
*/
|
|
function _burn(
|
|
address account,
|
|
uint256 id,
|
|
uint256 amount
|
|
) internal virtual override {
|
|
super._burn(account, id, amount);
|
|
_totalSupply[id] -= amount;
|
|
}
|
|
|
|
/**
|
|
* @dev See {ERC1155-_burnBatch}.
|
|
*/
|
|
function _burnBatch(
|
|
address account,
|
|
uint256[] memory ids,
|
|
uint256[] memory amounts
|
|
) internal virtual override {
|
|
super._burnBatch(account, ids, amounts);
|
|
for (uint256 i = 0; i < ids.length; ++i) {
|
|
_totalSupply[ids[i]] -= amounts[i];
|
|
}
|
|
}
|
|
}
|
|
|