-
Notifications
You must be signed in to change notification settings - Fork 37
/
CustodialWalletFactoryV2.sol
79 lines (64 loc) · 3.17 KB
/
CustodialWalletFactoryV2.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
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.7;
import "../../access/Ownable.sol";
import "./CustodialWallet.sol";
import "../../proxy/Clones.sol";
contract CustodialWalletFactoryV2 {
using Clones for CustodialWalletFactoryV2;
uint256 private constant _MAX_ARRAY_BOUNDS = 2000;
uint256 private constant _MAX_ARRAY_CALCULATE_BOUNDS = 10_000;
CustodialWallet private rawWallet;
mapping(bytes32 => address) public wallets;
event WalletDetails(address addr, address owner, uint256 index);
event Created(address addr);
event CreateFailed(address addr, address owner, string reason);
constructor () {
rawWallet = new CustodialWallet();
}
function getWallet(address owner, uint256 index) public view returns (address addr, bool exists, bytes32 salt) {
salt = keccak256(abi.encodePacked(owner, index));
addr = Clones.predictDeterministicAddress(address(rawWallet), salt);
exists = wallets[salt] != address(0);
}
function getWallets(address owner, uint256[] memory index) external view returns (address[] memory, bool[] memory, bytes32[] memory) {
require(index.length <= _MAX_ARRAY_CALCULATE_BOUNDS, "Maximum allowable size of array has been exceeded");
address[] memory addr = new address[](index.length);
bool[] memory exists = new bool[](index.length);
bytes32[] memory salt = new bytes32[](index.length);
for (uint256 i = 0; i < index.length; i++) {
salt[i] = keccak256(abi.encodePacked(owner, index[i]));
addr[i] = Clones.predictDeterministicAddress(address(rawWallet), salt[i]);
exists[i] = wallets[salt[i]] != address(0);
}
return (addr, exists, salt);
}
function createBatch(address owner, uint256[] memory index) external {
require(index.length <= _MAX_ARRAY_BOUNDS, "Maximum allowable size of array has been exceeded");
for (uint256 i = 0; i < index.length; i++) {
(address calculatedAddress, bool exists, bytes32 salt) = getWallet(owner, index[i]);
if(exists) {
emit CreateFailed(calculatedAddress, owner, "Wallet already exists");
continue;
}
address addr = Clones.cloneDeterministic(address(rawWallet), salt);
if(addr != calculatedAddress) {
emit CreateFailed(calculatedAddress, owner, "Address doesnt match with predicted address.");
continue;
}
wallets[salt] = addr;
CustodialWallet(payable(addr)).init(owner);
emit Created(addr);
emit WalletDetails(addr, owner, index[i]);
}
}
function create(address owner, uint256 index) external {
(address calculatedAddress, bool exists, bytes32 salt) = getWallet(owner, index);
require(!exists, "Wallet already exists");
address addr = Clones.cloneDeterministic(address(rawWallet), salt);
require(addr == calculatedAddress, "Address doesnt match with predicted address.");
wallets[salt] = addr;
CustodialWallet(payable(addr)).init(owner);
emit Created(addr);
emit WalletDetails(addr, owner, index);
}
}