status-go/services/incentivisation/contract.go

package incentivisation

import (
	"context"
	"errors"
	"math/big"
	"time"

	"github.com/ethereum/go-ethereum/accounts/abi/bind"
	gethcommon "github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/core/types"
	"github.com/ethereum/go-ethereum/ethclient"

	"github.com/status-im/status-go/mailserver/registry"
)

type Contract interface {
	Vote(opts *bind.TransactOpts, joinNodes []gethcommon.Address, removeNodes []gethcommon.Address) (*types.Transaction, error)
	GetCurrentSession(opts *bind.CallOpts) (*big.Int, error)
	Registered(opts *bind.CallOpts, publicKey []byte) (bool, error)
	RegisterNode(opts *bind.TransactOpts, publicKey []byte, ip uint32, port uint16) (*types.Transaction, error)
	ActiveNodeCount(opts *bind.CallOpts) (*big.Int, error)
	InactiveNodeCount(opts *bind.CallOpts) (*big.Int, error)
	GetNode(opts *bind.CallOpts, index *big.Int) ([]byte, uint32, uint16, uint32, uint32, error)
	GetInactiveNode(opts *bind.CallOpts, index *big.Int) ([]byte, uint32, uint16, uint32, uint32, error)
	VoteSync(opts *bind.TransactOpts, joinNodes []gethcommon.Address, removeNodes []gethcommon.Address) (*types.Transaction, error)
}

type ContractImpl struct {
	registry.NodesV2
	client *ethclient.Client
}

// VoteSync votes on the contract and wait until the transaction has been accepted, returns an error otherwise
func (c *ContractImpl) VoteSync(opts *bind.TransactOpts, joinNodes []gethcommon.Address, removeNodes []gethcommon.Address) (*types.Transaction, error) {
	tx, err := c.Vote(opts, joinNodes, removeNodes)
	if err != nil {
		return nil, err
	}

	for {
		receipt, _ := c.client.TransactionReceipt(context.TODO(), tx.Hash())
		if receipt != nil {
			if receipt.Status == 0 {
				return nil, errors.New("Invalid receipt")
			}
			break
		}
		time.Sleep(500 * time.Millisecond)
	}

	return tx, nil

}

// NewContract creates a new instance of Contract, bound to a specific deployed contract.
func NewContract(address gethcommon.Address, backend bind.ContractBackend, client *ethclient.Client) (Contract, error) {
	contract := &ContractImpl{}
	contract.client = client

	caller, err := registry.NewNodesV2Caller(address, backend)
	if err != nil {
		return nil, err
	}
	contract.NodesV2Caller = *caller

	transactor, err := registry.NewNodesV2Transactor(address, backend)
	if err != nil {
		return nil, err
	}
	contract.NodesV2Transactor = *transactor

	filterer, err := registry.NewNodesV2Filterer(address, backend)
	if err != nil {
		return nil, err
	}
	contract.NodesV2Filterer = *filterer

	return contract, nil
}