Add a VestingWallet (#2748)
Co-authored-by: Francisco Giordano <frangio.1@gmail.com>
This commit is contained in:
@ -3,8 +3,18 @@
|
||||
[.readme-notice]
|
||||
NOTE: This document is better viewed at https://docs.openzeppelin.com/contracts/api/finance
|
||||
|
||||
This directory includes primitives for financial systems. We currently only offer the {PaymentSplitter} contract, but we want to grow this directory so we welcome ideas.
|
||||
This directory includes primitives for financial systems:
|
||||
|
||||
== PaymentSplitter
|
||||
- {PaymentSplitter} allows to split Ether and ERC20 payments among a group of accounts. The sender does not need to be
|
||||
aware that the assets will be split in this way, since it is handled transparently by the contract. The split can be
|
||||
in equal parts or in any other arbitrary proportion.
|
||||
|
||||
- {VestingWallet} handles the vesting of Ether and ERC20 tokens for a given beneficiary. Custody of multiple tokens can
|
||||
be given to this contract, which will release the token to the beneficiary following a given, customizable, vesting
|
||||
schedule.
|
||||
|
||||
== Contracts
|
||||
|
||||
{{PaymentSplitter}}
|
||||
|
||||
{{VestingWallet}}
|
||||
|
||||
134
contracts/finance/VestingWallet.sol
Normal file
134
contracts/finance/VestingWallet.sol
Normal file
@ -0,0 +1,134 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
pragma solidity ^0.8.0;
|
||||
|
||||
import "../token/ERC20/utils/SafeERC20.sol";
|
||||
import "../utils/Address.sol";
|
||||
import "../utils/Context.sol";
|
||||
import "../utils/math/Math.sol";
|
||||
|
||||
/**
|
||||
* @title VestingWallet
|
||||
* @dev This contract handles the vesting of Eth and ERC20 tokens for a given beneficiary. Custody of multiple tokens
|
||||
* can be given to this contract, which will release the token to the beneficiary following a given vesting schedule.
|
||||
* The vesting schedule is customizable through the {vestedAmount} function.
|
||||
*
|
||||
* Any token transferred to this contract will follow the vesting schedule as if they were locked from the beginning.
|
||||
* Consequently, if the vesting has already started, any amount of tokens sent to this contract will (at least partly)
|
||||
* be immediately releasable.
|
||||
*/
|
||||
contract VestingWallet is Context {
|
||||
event EtherReleased(uint256 amount);
|
||||
event ERC20Released(address token, uint256 amount);
|
||||
|
||||
uint256 private _released;
|
||||
mapping(address => uint256) private _erc20Released;
|
||||
address private immutable _beneficiary;
|
||||
uint64 private immutable _start;
|
||||
uint64 private immutable _duration;
|
||||
|
||||
/**
|
||||
* @dev Set the beneficiary, start timestamp and vesting duration of the vesting wallet.
|
||||
*/
|
||||
constructor(
|
||||
address beneficiaryAddress,
|
||||
uint64 startTimestamp,
|
||||
uint64 durationSeconds
|
||||
) {
|
||||
require(beneficiaryAddress != address(0), "VestingWallet: beneficiary is zero address");
|
||||
_beneficiary = beneficiaryAddress;
|
||||
_start = startTimestamp;
|
||||
_duration = durationSeconds;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev The contract should be able to receive Eth.
|
||||
*/
|
||||
receive() external payable virtual {}
|
||||
|
||||
/**
|
||||
* @dev Getter for the beneficiary address.
|
||||
*/
|
||||
function beneficiary() public view virtual returns (address) {
|
||||
return _beneficiary;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Getter for the start timestamp.
|
||||
*/
|
||||
function start() public view virtual returns (uint256) {
|
||||
return _start;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Getter for the vesting duration.
|
||||
*/
|
||||
function duration() public view virtual returns (uint256) {
|
||||
return _duration;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Amount of eth already released
|
||||
*/
|
||||
function released() public view virtual returns (uint256) {
|
||||
return _released;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Amount of token already released
|
||||
*/
|
||||
function released(address token) public view virtual returns (uint256) {
|
||||
return _erc20Released[token];
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Release the native token (ether) that have already vested.
|
||||
*
|
||||
* Emits a {TokensReleased} event.
|
||||
*/
|
||||
function release() public virtual {
|
||||
uint256 releasable = vestedAmount(uint64(block.timestamp)) - released();
|
||||
_released += releasable;
|
||||
emit EtherReleased(releasable);
|
||||
Address.sendValue(payable(beneficiary()), releasable);
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Release the tokens that have already vested.
|
||||
*
|
||||
* Emits a {TokensReleased} event.
|
||||
*/
|
||||
function release(address token) public virtual {
|
||||
uint256 releasable = vestedAmount(token, uint64(block.timestamp)) - released(token);
|
||||
_erc20Released[token] += releasable;
|
||||
emit ERC20Released(token, releasable);
|
||||
SafeERC20.safeTransfer(IERC20(token), beneficiary(), releasable);
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Calculates the amount of ether that has already vested. Default implementation is a linear vesting curve.
|
||||
*/
|
||||
function vestedAmount(uint64 timestamp) public view virtual returns (uint256) {
|
||||
return _vestingSchedule(address(this).balance + released(), timestamp);
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Calculates the amount of tokens that has already vested. Default implementation is a linear vesting curve.
|
||||
*/
|
||||
function vestedAmount(address token, uint64 timestamp) public view virtual returns (uint256) {
|
||||
return _vestingSchedule(IERC20(token).balanceOf(address(this)) + released(token), timestamp);
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Virtual implementation of the vesting formula. This returns the amout vested, as a function of time, for
|
||||
* an asset given its total historical allocation.
|
||||
*/
|
||||
function _vestingSchedule(uint256 totalAllocation, uint64 timestamp) internal view virtual returns (uint256) {
|
||||
if (timestamp < start()) {
|
||||
return 0;
|
||||
} else if (timestamp > start() + duration()) {
|
||||
return totalAllocation;
|
||||
} else {
|
||||
return (totalAllocation * (timestamp - start())) / duration();
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user