395 lines
13 KiB
Go
395 lines
13 KiB
Go
// Copyright 2024 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package v2
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/ethereum/go-ethereum"
|
|
"github.com/ethereum/go-ethereum/accounts/abi"
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/event"
|
|
)
|
|
|
|
// ContractDeployParams represents state needed to deploy a contract:
|
|
// the metdata and constructor input (which can be nil if no input is specified).
|
|
type ContractDeployParams struct {
|
|
Meta *bind.MetaData
|
|
// Input is the ABI-encoded constructor input for the contract deployment.
|
|
Input []byte
|
|
}
|
|
|
|
// DeploymentParams represents parameters needed to deploy a
|
|
// set of contracts, their dependency libraries. It takes an optional override
|
|
// list to specify libraries that have already been deployed on-chain.
|
|
type DeploymentParams struct {
|
|
Contracts []*bind.MetaData
|
|
Inputs map[string][]byte
|
|
// Overrides is an optional map of pattern to deployment address.
|
|
// Contracts/libraries that refer to dependencies in the override
|
|
// set are linked to the provided address (an already-deployed contract).
|
|
Overrides map[string]common.Address
|
|
}
|
|
|
|
// DeploymentResult contains the relevant information from the deployment of
|
|
// multiple contracts: their deployment txs and addresses.
|
|
type DeploymentResult struct {
|
|
// map of contract library pattern -> deploy transaction
|
|
Txs map[string]*types.Transaction
|
|
// map of contract library pattern -> deployed address
|
|
Addrs map[string]common.Address
|
|
}
|
|
|
|
func (d *DeploymentResult) Accumulate(other *DeploymentResult) {
|
|
for pattern, tx := range other.Txs {
|
|
d.Txs[pattern] = tx
|
|
}
|
|
for pattern, addr := range other.Addrs {
|
|
d.Addrs[pattern] = addr
|
|
}
|
|
}
|
|
|
|
// depTreeBuilder turns a set of unlinked contracts and their dependent libraries into a collection of trees
|
|
// representing the relation of their dependencies.
|
|
type depTreeBuilder struct {
|
|
overrides map[string]common.Address
|
|
// map of pattern to unlinked contract bytecode (for libraries or contracts)
|
|
contracts map[string]string
|
|
// map of pattern to subtree represented by contract
|
|
subtrees map[string]*depTreeNode
|
|
// map of nodes that aren't referenced by other dependencies (these can be libraries too if user is doing lib-only deployment)
|
|
roots map[string]struct{}
|
|
}
|
|
|
|
// depTreeNode represents a node (contract) in a dependency tree. it contains its unlinked code, and references to any
|
|
// library contracts that it requires. If it is specified as an override, it contains the address where it has already
|
|
// been deployed at.
|
|
type depTreeNode struct {
|
|
pattern string
|
|
unlinkedCode string
|
|
nodes []*depTreeNode
|
|
overrideAddr *common.Address
|
|
}
|
|
|
|
func (d *depTreeBuilder) buildDepTrees(pattern, contract string) {
|
|
// if the node is in the subtree set already, it has already been fully recursed/built so we can bail out.
|
|
if _, ok := d.subtrees[pattern]; ok {
|
|
return
|
|
}
|
|
node := &depTreeNode{
|
|
pattern: pattern,
|
|
unlinkedCode: contract,
|
|
}
|
|
if addr, ok := d.overrides[pattern]; ok {
|
|
node.overrideAddr = &addr
|
|
}
|
|
// iterate each referenced library in the unlinked code, recurse and built its subtree.
|
|
reMatchSpecificPattern, err := regexp.Compile(`__\$([a-f0-9]+)\$__`)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
for _, match := range reMatchSpecificPattern.FindAllStringSubmatch(contract, -1) {
|
|
depPattern := match[1]
|
|
d.buildDepTrees(depPattern, d.contracts[depPattern])
|
|
node.nodes = append(node.nodes, d.subtrees[depPattern])
|
|
|
|
// this library can't be a root dependency if it is referenced by other contracts.
|
|
delete(d.roots, depPattern)
|
|
}
|
|
d.subtrees[pattern] = node
|
|
}
|
|
|
|
// BuildDepTrees will compute a set of dependency trees from a set of unlinked contracts. The root of each tree
|
|
// corresponds to a contract/library that is not referenced as a dependency anywhere else. Children of each node are
|
|
// its library dependencies.
|
|
func (d *depTreeBuilder) BuildDepTrees() (roots []*depTreeNode) {
|
|
// before the trees of dependencies are known, consider that any provided contract could be a root.
|
|
for pattern, _ := range d.contracts {
|
|
d.roots[pattern] = struct{}{}
|
|
}
|
|
|
|
// recursively build each part of the dependency subtree by starting at
|
|
for pattern, contract := range d.contracts {
|
|
d.buildDepTrees(pattern, contract)
|
|
}
|
|
for pattern, _ := range d.roots {
|
|
roots = append(roots, d.subtrees[pattern])
|
|
}
|
|
return roots
|
|
}
|
|
|
|
func newDepTreeBuilder(overrides map[string]common.Address, contracts map[string]string) *depTreeBuilder {
|
|
return &depTreeBuilder{
|
|
overrides: overrides,
|
|
contracts: contracts,
|
|
subtrees: make(map[string]*depTreeNode),
|
|
roots: make(map[string]struct{}),
|
|
}
|
|
}
|
|
|
|
type deployFn func(input, deployer []byte) (common.Address, *types.Transaction, error)
|
|
|
|
// depTreeDeployer is responsible for taking a dependency, deploying-and-linking its components in the proper
|
|
// order. A depTreeDeployer cannot be used after calling LinkAndDeploy other than to retrieve the deployment result.
|
|
type depTreeDeployer struct {
|
|
deployedAddrs map[string]common.Address
|
|
deployerTxs map[string]*types.Transaction
|
|
input map[string][]byte // map of the root contract pattern to the constructor input (if there is any)
|
|
deploy deployFn
|
|
err error
|
|
}
|
|
|
|
// linkAndDeploy recursively deploys a contract/library: starting by linking/deploying its dependencies.
|
|
// The deployment result (deploy addresses/txs or an error) is stored in the depTreeDeployer object.
|
|
func (d *depTreeDeployer) linkAndDeploy(node *depTreeNode) {
|
|
// short-circuit further deployment of contracts if a previous deployment encountered an error.
|
|
if d.err != nil {
|
|
return
|
|
}
|
|
|
|
// don't deploy contracts specified as overrides. don't deploy their dependencies.
|
|
if node.overrideAddr != nil {
|
|
return
|
|
}
|
|
|
|
// if this contract/library depends on other libraries deploy them (and their dependencies) first
|
|
for _, childNode := range node.nodes {
|
|
d.linkAndDeploy(childNode)
|
|
}
|
|
// if we just deployed any prerequisite contracts, link their deployed addresses into the bytecode to produce
|
|
// a deployer bytecode for this contract.
|
|
deployerCode := node.unlinkedCode
|
|
for _, child := range node.nodes {
|
|
var linkAddr common.Address
|
|
if child.overrideAddr != nil {
|
|
linkAddr = *child.overrideAddr
|
|
} else {
|
|
linkAddr = d.deployedAddrs[child.pattern]
|
|
}
|
|
deployerCode = strings.ReplaceAll(deployerCode, "__$"+child.pattern+"$__", strings.ToLower(linkAddr.String()[2:]))
|
|
}
|
|
|
|
// Finally, deploy the contract.
|
|
addr, tx, err := d.deploy(d.input[node.pattern], common.Hex2Bytes(deployerCode))
|
|
if err != nil {
|
|
d.err = err
|
|
} else {
|
|
d.deployedAddrs[node.pattern] = addr
|
|
d.deployerTxs[node.pattern] = tx
|
|
}
|
|
}
|
|
|
|
// result returns a result for this deployment, or an error if it failed.
|
|
func (d *depTreeDeployer) result() (*DeploymentResult, error) {
|
|
if d.err != nil {
|
|
return nil, d.err
|
|
}
|
|
return &DeploymentResult{
|
|
Txs: d.deployerTxs,
|
|
Addrs: d.deployedAddrs,
|
|
}, nil
|
|
}
|
|
|
|
func newDepTreeDeployer(deploy deployFn) *depTreeDeployer {
|
|
return &depTreeDeployer{
|
|
deploy: deploy,
|
|
deployedAddrs: make(map[string]common.Address),
|
|
deployerTxs: make(map[string]*types.Transaction)}
|
|
}
|
|
|
|
// LinkAndDeploy deploys a specified set of contracts and their dependent
|
|
// libraries. If an error occurs, only contracts which were successfully
|
|
// deployed are returned in the result.
|
|
func LinkAndDeploy(deployParams DeploymentParams, deploy deployFn) (res *DeploymentResult, err error) {
|
|
unlinkedContracts := make(map[string]string)
|
|
accumRes := &DeploymentResult{
|
|
Txs: make(map[string]*types.Transaction),
|
|
Addrs: make(map[string]common.Address),
|
|
}
|
|
for _, meta := range deployParams.Contracts {
|
|
unlinkedContracts[meta.Pattern] = meta.Bin[2:]
|
|
}
|
|
treeBuilder := newDepTreeBuilder(deployParams.Overrides, unlinkedContracts)
|
|
deps := treeBuilder.BuildDepTrees()
|
|
|
|
for _, tr := range deps {
|
|
deployer := newDepTreeDeployer(deploy)
|
|
if deployParams.Inputs != nil {
|
|
deployer.input = map[string][]byte{tr.pattern: deployParams.Inputs[tr.pattern]}
|
|
}
|
|
deployer.linkAndDeploy(tr)
|
|
res, err := deployer.result()
|
|
if err != nil {
|
|
return accumRes, err
|
|
}
|
|
accumRes.Accumulate(res)
|
|
}
|
|
return accumRes, nil
|
|
}
|
|
|
|
// TODO: this will be generated as part of the bindings, contain the ABI (or metadata object?) and errors
|
|
type ContractInstance struct {
|
|
Address common.Address
|
|
Backend bind.ContractBackend
|
|
}
|
|
|
|
// TODO: adding docs soon (jwasinger)
|
|
func FilterEvents[T any](instance *ContractInstance, opts *bind.FilterOpts, eventID common.Hash, unpack func(*types.Log) (*T, error), topics ...[]any) (*EventIterator[T], error) {
|
|
backend := instance.Backend
|
|
c := bind.NewBoundContract(instance.Address, abi.ABI{}, backend, backend, backend)
|
|
logs, sub, err := c.FilterLogsById(opts, eventID, topics...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &EventIterator[T]{unpack: unpack, logs: logs, sub: sub}, nil
|
|
}
|
|
|
|
// WatchEvents causes logs emitted with a given event id from a specified
|
|
// contract to be intercepted, unpacked, and forwarded to sink. If
|
|
// unpack returns an error, the returned subscription is closed with the
|
|
// error.
|
|
func WatchEvents[T any](instance *ContractInstance, opts *bind.WatchOpts, eventID common.Hash, unpack func(*types.Log) (*T, error), sink chan<- *T, topics ...[]any) (event.Subscription, error) {
|
|
backend := instance.Backend
|
|
c := bind.NewBoundContract(instance.Address, abi.ABI{}, backend, backend, backend)
|
|
logs, sub, err := c.WatchLogsForId(opts, eventID, topics...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return event.NewSubscription(func(quit <-chan struct{}) error {
|
|
defer sub.Unsubscribe()
|
|
for {
|
|
select {
|
|
case log := <-logs:
|
|
// New log arrived, parse the event and forward to the user
|
|
ev, err := unpack(&log)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
select {
|
|
case sink <- ev:
|
|
case err := <-sub.Err():
|
|
return err
|
|
case <-quit:
|
|
return nil
|
|
}
|
|
case err := <-sub.Err():
|
|
return err
|
|
case <-quit:
|
|
return nil
|
|
}
|
|
}
|
|
}), nil
|
|
}
|
|
|
|
// EventIterator is returned from FilterLogs and is used to iterate over the raw logs and unpacked data for events.
|
|
type EventIterator[T any] struct {
|
|
event *T // event containing the contract specifics and raw log
|
|
|
|
unpack func(*types.Log) (*T, error) // Unpack function for the event
|
|
|
|
logs <-chan types.Log // Log channel receiving the found contract events
|
|
sub ethereum.Subscription // Subscription for solc_errors, completion and termination
|
|
done bool // Whether the subscription completed delivering logs
|
|
fail error // Occurred error to stop iteration
|
|
}
|
|
|
|
// Value returns the current value of the iterator, or nil if there isn't one.
|
|
func (it *EventIterator[T]) Value() *T {
|
|
return it.event
|
|
}
|
|
|
|
// Next advances the iterator to the subsequent event, returning whether there
|
|
// are any more events found. In case of a retrieval or parsing error, false is
|
|
// returned and Error() can be queried for the exact failure.
|
|
func (it *EventIterator[T]) Next() bool {
|
|
// If the iterator failed, stop iterating
|
|
if it.fail != nil {
|
|
return false
|
|
}
|
|
// If the iterator completed, deliver directly whatever's available
|
|
if it.done {
|
|
select {
|
|
case log := <-it.logs:
|
|
res, err := it.unpack(&log)
|
|
if err != nil {
|
|
it.fail = err
|
|
return false
|
|
}
|
|
it.event = res
|
|
return true
|
|
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
// Iterator still in progress, wait for either a data or an error event
|
|
select {
|
|
case log := <-it.logs:
|
|
res, err := it.unpack(&log)
|
|
if err != nil {
|
|
it.fail = err
|
|
return false
|
|
}
|
|
it.event = res
|
|
return true
|
|
|
|
case err := <-it.sub.Err():
|
|
it.done = true
|
|
it.fail = err
|
|
return it.Next()
|
|
}
|
|
}
|
|
|
|
// Error returns any retrieval or parsing error occurred during filtering.
|
|
func (it *EventIterator[T]) Error() error {
|
|
return it.fail
|
|
}
|
|
|
|
// Close terminates the iteration process, releasing any pending underlying
|
|
// resources.
|
|
func (it *EventIterator[T]) Close() error {
|
|
it.sub.Unsubscribe()
|
|
return nil
|
|
}
|
|
|
|
// Transact creates and submits a transaction to the bound contract instance
|
|
// using the provided abi-encoded input (or nil).
|
|
func Transact(instance *ContractInstance, opts *bind.TransactOpts, input []byte) (*types.Transaction, error) {
|
|
var (
|
|
addr = instance.Address
|
|
backend = instance.Backend
|
|
)
|
|
c := bind.NewBoundContract(addr, abi.ABI{}, backend, backend, backend)
|
|
return c.RawTransact(opts, input)
|
|
}
|
|
|
|
// Call performs an eth_call on the given bound contract instance, using the
|
|
// provided abi-encoded input (or nil).
|
|
func Call[T any](instance *ContractInstance, opts *bind.CallOpts, packedInput []byte, unpack func([]byte) (*T, error)) (*T, error) {
|
|
backend := instance.Backend
|
|
c := bind.NewBoundContract(instance.Address, abi.ABI{}, backend, backend, backend)
|
|
packedOutput, err := c.CallRaw(opts, packedInput)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return unpack(packedOutput)
|
|
}
|