-
Notifications
You must be signed in to change notification settings - Fork 7
/
Deposit.sol
110 lines (90 loc) · 4.18 KB
/
Deposit.sol
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
// SPDX-License-Identifier: UNLICENSED
pragma solidity 0.8.13;
import {ERC20} from "solmate/tokens/ERC20.sol";
/// @title Deposit
/// @author kulkarohan
/// @notice Mock contract to transfer ERC-20 tokens with a signed approval
contract Deposit {
/// ///
/// EVENTS ///
/// ///
/// @notice Emitted after an ERC-20 token deposit
/// @param user The depositooor
/// @param tokenContract The token address
/// @param amount The number of tokens deposited
event TokenDeposit(address user, address tokenContract, uint256 amount);
/// @notice Emitted after an ERC-20 token withdraw
/// @param user The withdrawooor
/// @param tokenContract The token address
/// @param amount The number of tokens withdrawn
event TokenWithdraw(address user, address tokenContract, uint256 amount);
/// ///
/// STORAGE ///
/// ///
/// @notice The number of tokens stored for a given user and token contract
/// @dev User => ERC-20 address => Number of tokens deposited
mapping(address => mapping(address => uint256)) public userDeposits;
/// ///
/// DEPOSIT ///
/// ///
/// @notice Deposits ERC-20 tokens (requires pre-approval)
/// @param _tokenContract The ERC-20 token address
/// @param _amount The number of tokens
function deposit(address _tokenContract, uint256 _amount) external {
ERC20(_tokenContract).transferFrom(msg.sender, address(this), _amount);
userDeposits[msg.sender][_tokenContract] += _amount;
emit TokenDeposit(msg.sender, _tokenContract, _amount);
}
/// ///
/// DEPOSIT w/ PERMIT ///
/// ///
/// @notice Deposits ERC-20 tokens with a signed approval
/// @param _tokenContract The ERC-20 token address
/// @param _amount The number of tokens to transfer
/// @param _owner The user signing the approval
/// @param _spender The user to transfer the tokens (ie this contract)
/// @param _value The number of tokens to appprove the spender
/// @param _deadline The timestamp the permit expires
/// @param _v The 129th byte and chain id of the signature
/// @param _r The first 64 bytes of the signature
/// @param _s Bytes 64-128 of the signature
function depositWithPermit(
address _tokenContract,
uint256 _amount,
address _owner,
address _spender,
uint256 _value,
uint256 _deadline,
uint8 _v,
bytes32 _r,
bytes32 _s
) external {
ERC20(_tokenContract).permit(
_owner,
_spender,
_value,
_deadline,
_v,
_r,
_s
);
ERC20(_tokenContract).transferFrom(_owner, address(this), _amount);
userDeposits[_owner][_tokenContract] += _amount;
emit TokenDeposit(_owner, _tokenContract, _amount);
}
/// ///
/// WITHDRAW ///
/// ///
/// @notice Withdraws deposited ERC-20 tokens
/// @param _tokenContract The ERC-20 token address
/// @param _amount The number of tokens
function withdraw(address _tokenContract, uint256 _amount) external {
require(
_amount <= userDeposits[msg.sender][_tokenContract],
"INVALID_WITHDRAW"
);
userDeposits[msg.sender][_tokenContract] -= _amount;
ERC20(_tokenContract).transfer(msg.sender, _amount);
emit TokenWithdraw(msg.sender, _tokenContract, _amount);
}
}