2020-08-17 15:27:29 +00:00
|
|
|
package commands
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2020-09-26 21:01:11 +00:00
|
|
|
"fmt"
|
2020-09-15 16:44:28 +00:00
|
|
|
"strconv"
|
|
|
|
|
2020-08-29 07:24:50 +00:00
|
|
|
"github.com/ledgerwatch/turbo-geth/common/hexutil"
|
2020-08-17 15:27:29 +00:00
|
|
|
|
|
|
|
"github.com/ledgerwatch/turbo-geth/ethdb"
|
|
|
|
)
|
|
|
|
|
2020-09-26 21:01:11 +00:00
|
|
|
// NetAPI the interface for the net_ RPC commands
|
2020-08-17 15:27:29 +00:00
|
|
|
type NetAPI interface {
|
2020-09-15 16:44:28 +00:00
|
|
|
Listening(_ context.Context) (bool, error)
|
|
|
|
Version(_ context.Context) (string, error)
|
|
|
|
PeerCount(_ context.Context) (hexutil.Uint, error)
|
2020-08-17 15:27:29 +00:00
|
|
|
}
|
|
|
|
|
2020-09-26 21:01:11 +00:00
|
|
|
// NetAPIImpl data structure to store things needed for net_ commands
|
2020-08-17 15:27:29 +00:00
|
|
|
type NetAPIImpl struct {
|
|
|
|
ethBackend ethdb.Backend
|
|
|
|
}
|
|
|
|
|
2020-09-26 21:01:11 +00:00
|
|
|
// NewNetAPIImpl returns NetAPIImplImpl instance
|
2020-08-17 15:27:29 +00:00
|
|
|
func NewNetAPIImpl(eth ethdb.Backend) *NetAPIImpl {
|
|
|
|
return &NetAPIImpl{
|
|
|
|
ethBackend: eth,
|
|
|
|
}
|
|
|
|
}
|
2020-08-29 07:24:50 +00:00
|
|
|
|
2020-09-26 21:01:11 +00:00
|
|
|
// Listening implements RPC call for net_listening
|
2020-09-15 16:44:28 +00:00
|
|
|
// TODO(tjayrush) remove hard coded value
|
|
|
|
func (api *NetAPIImpl) Listening(_ context.Context) (bool, error) {
|
|
|
|
return true, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Version implements RPC call for net_version
|
|
|
|
func (api *NetAPIImpl) Version(_ context.Context) (string, error) {
|
2020-09-26 21:01:11 +00:00
|
|
|
if api.ethBackend == nil {
|
|
|
|
// We're running in --chaindata mode or otherwise cannot get the backend
|
|
|
|
return "", fmt.Errorf(NotAvailableChainData, "net_version")
|
|
|
|
}
|
|
|
|
|
2020-09-15 16:44:28 +00:00
|
|
|
res, err := api.ethBackend.NetVersion()
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
2020-09-26 21:01:11 +00:00
|
|
|
|
2020-09-15 16:44:28 +00:00
|
|
|
return strconv.FormatUint(res, 10), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// PeerCount implements RPC call for net_peerCount
|
|
|
|
// TODO(tjayrush) remove hard coded value
|
2020-08-29 07:24:50 +00:00
|
|
|
func (api *NetAPIImpl) PeerCount(_ context.Context) (hexutil.Uint, error) {
|
|
|
|
return hexutil.Uint(25), nil
|
|
|
|
}
|