-
Notifications
You must be signed in to change notification settings - Fork 9
/
Copy pathAirdrop.sol
195 lines (147 loc) · 5.97 KB
/
Airdrop.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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
pragma solidity ^0.8.4;
// SPDX-License-Identifier: Unlicensed
abstract contract Context {
function _msgSender() internal view virtual returns (address payable) {
return payable(msg.sender);
}
function _msgData() internal view virtual returns (bytes memory) {
this; // silence state mutability warning without generating bytecode - see https://github.com/ethereum/solidity/issues/2691
return msg.data;
}
}
interface IERC20 {
function totalSupply() external view returns (uint256);
function balanceOf(address account) external view returns (uint256);
function transfer(address recipient, uint256 amount) external returns (bool);
function allowance(address owner, address spender) external view returns (uint256);
function approve(address spender, uint256 amount) external returns (bool);
function transferFrom(address sender, address recipient, uint256 amount) external returns (bool);
event Transfer(address indexed from, address indexed to, uint256 value);
event Approval(address indexed owner, address indexed spender, uint256 value);
}
abstract contract ReentrancyGuard {
bool private _notEntered;
constructor () {
_notEntered = true;
}
modifier nonReentrant() {
// On the first call to nonReentrant, _notEntered will be true
require(_notEntered, "ReentrancyGuard: reentrant call");
// Any calls to nonReentrant after this point will fail
_notEntered = false;
_;
_notEntered = true;
}
}
library SafeMath {
function add(uint256 a, uint256 b) internal pure returns (uint256) {
uint256 c = a + b;
require(c >= a, "SafeMath: addition overflow");
return c;
}
function sub(uint256 a, uint256 b) internal pure returns (uint256) {
return sub(a, b, "SafeMath: subtraction overflow");
}
function sub(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
require(b <= a, errorMessage);
uint256 c = a - b;
return c;
}
function mul(uint256 a, uint256 b) internal pure returns (uint256) {
if (a == 0) {
return 0;
}
uint256 c = a * b;
require(c / a == b, "SafeMath: multiplication overflow");
return c;
}
function div(uint256 a, uint256 b) internal pure returns (uint256) {
return div(a, b, "SafeMath: division by zero");
}
function div(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
require(b > 0, errorMessage);
uint256 c = a / b;
// assert(a == b * c + a % b); // There is no case in which this doesn't hold
return c;
}
function mod(uint256 a, uint256 b) internal pure returns (uint256) {
return mod(a, b, "SafeMath: modulo by zero");
}
function mod(uint256 a, uint256 b, string memory errorMessage) internal pure returns (uint256) {
require(b != 0, errorMessage);
return a % b;
}
}
contract Ownable is Context {
address private _owner;
address private _previousOwner;
event OwnershipTransferred(address indexed previousOwner, address indexed newOwner);
constructor () {
address msgSender = _msgSender();
_owner = msgSender;
emit OwnershipTransferred(address(0), msgSender);
}
function owner() public view returns (address) {
return _owner;
}
modifier onlyOwner() {
require(_owner == _msgSender(), "Ownable: caller is not the owner");
_;
}
function renounceOwnership() public virtual onlyOwner {
emit OwnershipTransferred(_owner, address(0));
_owner = address(0);
}
function transferOwnership(address newOwner) public virtual onlyOwner {
require(newOwner != address(0), "Ownable: new owner is the zero address");
emit OwnershipTransferred(_owner, newOwner);
_owner = newOwner;
}
}
contract Airdrop is ReentrancyGuard, Context, Ownable{
using SafeMath for uint256;
mapping(address => bool) private Claimed;
mapping(address => bool) private _isWhitelist;
mapping(address => uint256) private _valDrop;
IERC20 private _token;
bool airdropLive = false;
event AirdropClaimed(address receiver, uint256 amount);
event WhitelistSetted(address[] recipient, uint256[] amount);
receive () external payable {
claimTokens();
}
//Start Airdrop
function startAirdrop(IERC20 tokenAddress) public onlyOwner{
require(airdropLive == false, 'Airdrop already started');
_token = tokenAddress;
airdropLive = true;
}
function setWhitelist(address[] calldata recipients, uint256[] calldata amount) external onlyOwner{
for(uint i = 0; i< recipients.length; i++){
require(recipients[i] != address(0));
_valDrop[recipients[i]] = amount[i];
}
emit WhitelistSetted(recipients, amount);
}
function claimTokens() public nonReentrant payable {
require(airdropLive == true, 'Airdrop not started yet');
require(Claimed[msg.sender] == false, 'Airdrop already claimed!');
if(_token.balanceOf(address(this)) == 0) { airdropLive = false; return;}
Claimed[msg.sender] = true;
uint256 amount = _valDrop[msg.sender].mul(10**9);
_token.transfer(msg.sender, amount);
emit AirdropClaimed(msg.sender, amount);
}
function withdraw() external onlyOwner {
require(address(this).balance > 0, 'Contract has no money');
address payable wallet = payable(msg.sender);
wallet.transfer(address(this).balance);
}
function takeTokens(IERC20 tokenAddress) public onlyOwner{
IERC20 tokenBEP = tokenAddress;
uint256 tokenAmt = tokenBEP.balanceOf(address(this));
require(tokenAmt > 0, 'BEP-20 balance is 0');
address payable wallet = payable(msg.sender);
tokenBEP.transfer(wallet, tokenAmt);
}
}