Reorganize the repo structure (#2503)
Co-authored-by: Francisco Giordano <frangio.1@gmail.com>
This commit is contained in:
24
contracts/utils/escrow/ConditionalEscrow.sol
Normal file
24
contracts/utils/escrow/ConditionalEscrow.sol
Normal file
@ -0,0 +1,24 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
pragma solidity ^0.8.0;
|
||||
|
||||
import "./Escrow.sol";
|
||||
|
||||
/**
|
||||
* @title ConditionalEscrow
|
||||
* @dev Base abstract escrow to only allow withdrawal if a condition is met.
|
||||
* @dev Intended usage: See {Escrow}. Same usage guidelines apply here.
|
||||
*/
|
||||
abstract contract ConditionalEscrow is Escrow {
|
||||
/**
|
||||
* @dev Returns whether an address is allowed to withdraw their funds. To be
|
||||
* implemented by derived contracts.
|
||||
* @param payee The destination address of the funds.
|
||||
*/
|
||||
function withdrawalAllowed(address payee) public view virtual returns (bool);
|
||||
|
||||
function withdraw(address payable payee) public virtual override {
|
||||
require(withdrawalAllowed(payee), "ConditionalEscrow: payee is not allowed to withdraw");
|
||||
super.withdraw(payee);
|
||||
}
|
||||
}
|
||||
63
contracts/utils/escrow/Escrow.sol
Normal file
63
contracts/utils/escrow/Escrow.sol
Normal file
@ -0,0 +1,63 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
pragma solidity ^0.8.0;
|
||||
|
||||
import "../../access/Ownable.sol";
|
||||
import "../Address.sol";
|
||||
|
||||
/**
|
||||
* @title Escrow
|
||||
* @dev Base escrow contract, holds funds designated for a payee until they
|
||||
* withdraw them.
|
||||
*
|
||||
* Intended usage: This contract (and derived escrow contracts) should be a
|
||||
* standalone contract, that only interacts with the contract that instantiated
|
||||
* it. That way, it is guaranteed that all Ether will be handled according to
|
||||
* the `Escrow` rules, and there is no need to check for payable functions or
|
||||
* transfers in the inheritance tree. The contract that uses the escrow as its
|
||||
* payment method should be its owner, and provide public methods redirecting
|
||||
* to the escrow's deposit and withdraw.
|
||||
*/
|
||||
contract Escrow is Ownable {
|
||||
using Address for address payable;
|
||||
|
||||
event Deposited(address indexed payee, uint256 weiAmount);
|
||||
event Withdrawn(address indexed payee, uint256 weiAmount);
|
||||
|
||||
mapping(address => uint256) private _deposits;
|
||||
|
||||
function depositsOf(address payee) public view returns (uint256) {
|
||||
return _deposits[payee];
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Stores the sent amount as credit to be withdrawn.
|
||||
* @param payee The destination address of the funds.
|
||||
*/
|
||||
function deposit(address payee) public payable virtual onlyOwner {
|
||||
uint256 amount = msg.value;
|
||||
_deposits[payee] = _deposits[payee] + amount;
|
||||
|
||||
emit Deposited(payee, amount);
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Withdraw accumulated balance for a payee, forwarding all gas to the
|
||||
* recipient.
|
||||
*
|
||||
* WARNING: Forwarding all gas opens the door to reentrancy vulnerabilities.
|
||||
* Make sure you trust the recipient, or are either following the
|
||||
* checks-effects-interactions pattern or using {ReentrancyGuard}.
|
||||
*
|
||||
* @param payee The address whose funds will be withdrawn and transferred to.
|
||||
*/
|
||||
function withdraw(address payable payee) public virtual onlyOwner {
|
||||
uint256 payment = _deposits[payee];
|
||||
|
||||
_deposits[payee] = 0;
|
||||
|
||||
payee.sendValue(payment);
|
||||
|
||||
emit Withdrawn(payee, payment);
|
||||
}
|
||||
}
|
||||
95
contracts/utils/escrow/RefundEscrow.sol
Normal file
95
contracts/utils/escrow/RefundEscrow.sol
Normal file
@ -0,0 +1,95 @@
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
pragma solidity ^0.8.0;
|
||||
|
||||
import "./ConditionalEscrow.sol";
|
||||
|
||||
/**
|
||||
* @title RefundEscrow
|
||||
* @dev Escrow that holds funds for a beneficiary, deposited from multiple
|
||||
* parties.
|
||||
* @dev Intended usage: See {Escrow}. Same usage guidelines apply here.
|
||||
* @dev The owner account (that is, the contract that instantiates this
|
||||
* contract) may deposit, close the deposit period, and allow for either
|
||||
* withdrawal by the beneficiary, or refunds to the depositors. All interactions
|
||||
* with `RefundEscrow` will be made through the owner contract.
|
||||
*/
|
||||
contract RefundEscrow is ConditionalEscrow {
|
||||
using Address for address payable;
|
||||
|
||||
enum State { Active, Refunding, Closed }
|
||||
|
||||
event RefundsClosed();
|
||||
event RefundsEnabled();
|
||||
|
||||
State private _state;
|
||||
address payable immutable private _beneficiary;
|
||||
|
||||
/**
|
||||
* @dev Constructor.
|
||||
* @param beneficiary_ The beneficiary of the deposits.
|
||||
*/
|
||||
constructor (address payable beneficiary_) {
|
||||
require(beneficiary_ != address(0), "RefundEscrow: beneficiary is the zero address");
|
||||
_beneficiary = beneficiary_;
|
||||
_state = State.Active;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return The current state of the escrow.
|
||||
*/
|
||||
function state() public view virtual returns (State) {
|
||||
return _state;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return The beneficiary of the escrow.
|
||||
*/
|
||||
function beneficiary() public view virtual returns (address payable) {
|
||||
return _beneficiary;
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Stores funds that may later be refunded.
|
||||
* @param refundee The address funds will be sent to if a refund occurs.
|
||||
*/
|
||||
function deposit(address refundee) public payable virtual override {
|
||||
require(state() == State.Active, "RefundEscrow: can only deposit while active");
|
||||
super.deposit(refundee);
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Allows for the beneficiary to withdraw their funds, rejecting
|
||||
* further deposits.
|
||||
*/
|
||||
function close() public virtual onlyOwner {
|
||||
require(state() == State.Active, "RefundEscrow: can only close while active");
|
||||
_state = State.Closed;
|
||||
emit RefundsClosed();
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Allows for refunds to take place, rejecting further deposits.
|
||||
*/
|
||||
function enableRefunds() public onlyOwner virtual {
|
||||
require(state() == State.Active, "RefundEscrow: can only enable refunds while active");
|
||||
_state = State.Refunding;
|
||||
emit RefundsEnabled();
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Withdraws the beneficiary's funds.
|
||||
*/
|
||||
function beneficiaryWithdraw() public virtual {
|
||||
require(state() == State.Closed, "RefundEscrow: beneficiary can only withdraw while closed");
|
||||
beneficiary().sendValue(address(this).balance);
|
||||
}
|
||||
|
||||
/**
|
||||
* @dev Returns whether refundees can withdraw their deposits (be refunded). The overridden function receives a
|
||||
* 'payee' argument, but we ignore it here since the condition is global, not per-payee.
|
||||
*/
|
||||
function withdrawalAllowed(address) public view override returns (bool) {
|
||||
return state() == State.Refunding;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user