package geth import ( "errors" "math/big" "github.com/ethereum/go-ethereum" "github.com/ethereum/go-ethereum/common" "golang.org/x/net/context" "github.com/ethereum/go-ethereum/common/hexutil" "github.com/ethereum/go-ethereum/core/types" "github.com/vulcanize/vulcanizedb/pkg/core" vulcCommon "github.com/vulcanize/vulcanizedb/pkg/geth/converters/common" ) var ErrEmptyHeader = errors.New("empty header returned over RPC") type BlockChain struct { blockConverter vulcCommon.BlockConverter ethClient core.EthClient headerConverter vulcCommon.HeaderConverter node core.Node rpcClient core.RpcClient } func NewBlockChain(ethClient core.EthClient, rpcClient core.RpcClient, node core.Node, converter vulcCommon.TransactionConverter) *BlockChain { return &BlockChain{ blockConverter: vulcCommon.NewBlockConverter(converter), ethClient: ethClient, headerConverter: vulcCommon.HeaderConverter{}, node: node, rpcClient: rpcClient, } } func (blockChain *BlockChain) GetBlockByNumber(blockNumber int64) (block core.Block, err error) { gethBlock, err := blockChain.ethClient.BlockByNumber(context.Background(), big.NewInt(blockNumber)) if err != nil { return block, err } return blockChain.blockConverter.ToCoreBlock(gethBlock) } func (blockChain *BlockChain) GetHeaderByNumber(blockNumber int64) (header core.Header, err error) { if blockChain.node.NetworkID == core.KOVAN_NETWORK_ID { return blockChain.getPOAHeader(blockNumber) } return blockChain.getPOWHeader(blockNumber) } func (blockChain *BlockChain) getPOWHeader(blockNumber int64) (header core.Header, err error) { gethHeader, err := blockChain.ethClient.HeaderByNumber(context.Background(), big.NewInt(blockNumber)) if err != nil { return header, err } return blockChain.headerConverter.Convert(gethHeader) } func (blockChain *BlockChain) getPOAHeader(blockNumber int64) (header core.Header, err error) { var POAHeader core.POAHeader blockNumberArg := hexutil.EncodeBig(big.NewInt(blockNumber)) includeTransactions := false err = blockChain.rpcClient.CallContext(context.Background(), &POAHeader, "eth_getBlockByNumber", blockNumberArg, includeTransactions) if err != nil { return header, err } if POAHeader.Number == nil { return header, ErrEmptyHeader } return blockChain.headerConverter.Convert(&types.Header{ ParentHash: POAHeader.ParentHash, UncleHash: POAHeader.UncleHash, Coinbase: POAHeader.Coinbase, Root: POAHeader.Root, TxHash: POAHeader.TxHash, ReceiptHash: POAHeader.ReceiptHash, Bloom: POAHeader.Bloom, Difficulty: POAHeader.Difficulty.ToInt(), Number: POAHeader.Number.ToInt(), GasLimit: uint64(POAHeader.GasLimit), GasUsed: uint64(POAHeader.GasUsed), Time: POAHeader.Time.ToInt(), Extra: POAHeader.Extra, }) } func (blockChain *BlockChain) GetLogs(contract core.Contract, startingBlockNumber, endingBlockNumber *big.Int) ([]core.Log, error) { if endingBlockNumber == nil { endingBlockNumber = startingBlockNumber } contractAddress := common.HexToAddress(contract.Hash) fc := ethereum.FilterQuery{ FromBlock: startingBlockNumber, ToBlock: endingBlockNumber, Addresses: []common.Address{contractAddress}, Topics: nil, } gethLogs, err := blockChain.GetEthLogsWithCustomQuery(fc) if err != nil { return []core.Log{}, err } logs := vulcCommon.ToCoreLogs(gethLogs) return logs, nil } func (blockChain *BlockChain) GetEthLogsWithCustomQuery(query ethereum.FilterQuery) ([]types.Log, error) { gethLogs, err := blockChain.ethClient.FilterLogs(context.Background(), query) if err != nil { return []types.Log{}, err } return gethLogs, nil } func (blockChain *BlockChain) LastBlock() *big.Int { block, _ := blockChain.ethClient.HeaderByNumber(context.Background(), nil) return block.Number } func (blockChain *BlockChain) Node() core.Node { return blockChain.node }