forked from ChainSafe/ChainBridge
-
Notifications
You must be signed in to change notification settings - Fork 0
/
deploy.go
159 lines (122 loc) 路 4.32 KB
/
deploy.go
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
// Copyright 2020 ChainSafe Systems
// SPDX-License-Identifier: LGPL-3.0-only
package utils
import (
"context"
"math/big"
"github.com/egorgripasov/ChainBridge/bindings/GenericHandler"
"github.com/ethereum/go-ethereum/common"
bridge "github.com/egorgripasov/ChainBridge/bindings/Bridge"
erc20Handler "github.com/egorgripasov/ChainBridge/bindings/ERC20Handler"
erc721Handler "github.com/egorgripasov/ChainBridge/bindings/ERC721Handler"
"github.com/egorgripasov/chainbridge-utils/keystore"
)
var (
RelayerAddresses = []common.Address{
common.HexToAddress(keystore.TestKeyRing.EthereumKeys[keystore.AliceKey].Address()),
common.HexToAddress(keystore.TestKeyRing.EthereumKeys[keystore.BobKey].Address()),
common.HexToAddress(keystore.TestKeyRing.EthereumKeys[keystore.CharlieKey].Address()),
common.HexToAddress(keystore.TestKeyRing.EthereumKeys[keystore.DaveKey].Address()),
common.HexToAddress(keystore.TestKeyRing.EthereumKeys[keystore.EveKey].Address()),
}
ZeroAddress = common.HexToAddress("0x0000000000000000000000000000000000000000")
)
type DeployedContracts struct {
BridgeAddress common.Address
ERC20HandlerAddress common.Address
ERC721HandlerAddress common.Address
GenericHandlerAddress common.Address
}
// DeployContracts deploys Bridge, Relayer, ERC20Handler, ERC721Handler and CentrifugeAssetHandler and returns the addresses
func DeployContracts(client *Client, chainID uint8, initialRelayerThreshold *big.Int) (*DeployedContracts, error) {
bridgeAddr, err := deployBridge(client, chainID, RelayerAddresses, initialRelayerThreshold)
if err != nil {
return nil, err
}
erc20HandlerAddr, err := deployERC20Handler(client, bridgeAddr)
if err != nil {
return nil, err
}
erc721HandlerAddr, err := deployERC721Handler(client, bridgeAddr)
if err != nil {
return nil, err
}
genericHandlerAddr, err := deployGenericHandler(client, bridgeAddr)
if err != nil {
return nil, err
}
deployedContracts := DeployedContracts{bridgeAddr, erc20HandlerAddr, erc721HandlerAddr, genericHandlerAddr}
return &deployedContracts, nil
}
func UpdateNonce(client *Client) error {
newNonce, err := client.Client.PendingNonceAt(context.Background(), client.CallOpts.From)
if err != nil {
return err
}
client.Opts.Nonce = big.NewInt(int64(newNonce))
return nil
}
func deployBridge(client *Client, chainID uint8, relayerAddrs []common.Address, initialRelayerThreshold *big.Int) (common.Address, error) {
err := client.LockNonceAndUpdate()
if err != nil {
return ZeroAddress, err
}
bridgeAddr, tx, _, err := bridge.DeployBridge(client.Opts, client.Client, chainID, relayerAddrs, initialRelayerThreshold, big.NewInt(0), big.NewInt(100))
if err != nil {
return ZeroAddress, err
}
err = WaitForTx(client, tx)
if err != nil {
return ZeroAddress, err
}
client.UnlockNonce()
return bridgeAddr, nil
}
func deployERC20Handler(client *Client, bridgeAddress common.Address) (common.Address, error) {
err := client.LockNonceAndUpdate()
if err != nil {
return ZeroAddress, err
}
erc20HandlerAddr, tx, _, err := erc20Handler.DeployERC20Handler(client.Opts, client.Client, bridgeAddress, [][32]byte{}, []common.Address{}, []common.Address{})
if err != nil {
return ZeroAddress, err
}
err = WaitForTx(client, tx)
if err != nil {
return ZeroAddress, err
}
client.UnlockNonce()
return erc20HandlerAddr, nil
}
func deployERC721Handler(client *Client, bridgeAddress common.Address) (common.Address, error) {
err := client.LockNonceAndUpdate()
if err != nil {
return ZeroAddress, err
}
erc721HandlerAddr, tx, _, err := erc721Handler.DeployERC721Handler(client.Opts, client.Client, bridgeAddress, [][32]byte{}, []common.Address{}, []common.Address{})
if err != nil {
return ZeroAddress, err
}
err = WaitForTx(client, tx)
if err != nil {
return ZeroAddress, err
}
client.UnlockNonce()
return erc721HandlerAddr, nil
}
func deployGenericHandler(client *Client, bridgeAddress common.Address) (common.Address, error) {
err := client.LockNonceAndUpdate()
if err != nil {
return ZeroAddress, err
}
addr, tx, _, err := GenericHandler.DeployGenericHandler(client.Opts, client.Client, bridgeAddress, [][32]byte{}, []common.Address{}, [][4]byte{}, [][4]byte{})
if err != nil {
return ZeroAddress, err
}
err = WaitForTx(client, tx)
if err != nil {
return ZeroAddress, err
}
client.UnlockNonce()
return addr, nil
}