133 lines
3.4 KiB
Go
133 lines
3.4 KiB
Go
package node
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/filecoin-project/go-lotus/api"
|
|
"github.com/filecoin-project/go-lotus/build"
|
|
"github.com/filecoin-project/go-lotus/chain"
|
|
"github.com/filecoin-project/go-lotus/chain/address"
|
|
"github.com/filecoin-project/go-lotus/chain/types"
|
|
"github.com/filecoin-project/go-lotus/miner"
|
|
"github.com/filecoin-project/go-lotus/node/client"
|
|
|
|
"github.com/libp2p/go-libp2p-core/host"
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
|
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
type API struct {
|
|
client.LocalStorage
|
|
|
|
Host host.Host
|
|
Chain *chain.ChainStore
|
|
PubSub *pubsub.PubSub
|
|
Mpool *chain.MessagePool
|
|
Wallet *chain.Wallet
|
|
}
|
|
|
|
func (a *API) ChainSubmitBlock(ctx context.Context, blk *chain.BlockMsg) error {
|
|
if err := a.Chain.AddBlock(blk.Header); err != nil {
|
|
return err
|
|
}
|
|
|
|
b, err := blk.Serialize()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// TODO: anything else to do here?
|
|
return a.PubSub.Publish("/fil/blocks", b)
|
|
}
|
|
|
|
func (a *API) ChainHead(context.Context) (*chain.TipSet, error) {
|
|
return a.Chain.GetHeaviestTipSet(), nil
|
|
}
|
|
|
|
func (a *API) ChainGetRandomness(ctx context.Context, pts *chain.TipSet) ([]byte, error) {
|
|
// TODO: this needs to look back in the chain for the right random beacon value
|
|
return []byte("foo bar random"), nil
|
|
}
|
|
|
|
func (a *API) ID(context.Context) (peer.ID, error) {
|
|
return a.Host.ID(), nil
|
|
}
|
|
|
|
func (a *API) Version(context.Context) (api.Version, error) {
|
|
return api.Version{
|
|
Version: build.Version,
|
|
}, nil
|
|
}
|
|
|
|
func (a *API) MpoolPending(ctx context.Context, ts *chain.TipSet) ([]*chain.SignedMessage, error) {
|
|
// TODO: need to make sure we don't return messages that were already included in the referenced chain
|
|
// also need to accept ts == nil just fine, assume nil == chain.Head()
|
|
return a.Mpool.Pending(), nil
|
|
}
|
|
|
|
func (a *API) MinerStart(ctx context.Context, addr address.Address) error {
|
|
// hrm...
|
|
m := miner.NewMiner(a, addr)
|
|
|
|
go m.Mine(context.TODO())
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *API) MinerCreateBlock(ctx context.Context, addr address.Address, parents *chain.TipSet, tickets []chain.Ticket, proof chain.ElectionProof, msgs []*chain.SignedMessage) (*chain.BlockMsg, error) {
|
|
fblk, err := chain.MinerCreateBlock(a.Chain, addr, parents, tickets, proof, msgs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var out chain.BlockMsg
|
|
out.Header = fblk.Header
|
|
for _, msg := range fblk.Messages {
|
|
out.Messages = append(out.Messages, msg.Cid())
|
|
}
|
|
|
|
return &out, nil
|
|
}
|
|
|
|
func (a *API) NetPeers(context.Context) ([]peer.AddrInfo, error) {
|
|
conns := a.Host.Network().Conns()
|
|
out := make([]peer.AddrInfo, len(conns))
|
|
|
|
for i, conn := range conns {
|
|
out[i] = peer.AddrInfo{
|
|
ID: conn.RemotePeer(),
|
|
Addrs: []ma.Multiaddr{
|
|
conn.RemoteMultiaddr(),
|
|
},
|
|
}
|
|
}
|
|
|
|
return out, nil
|
|
}
|
|
|
|
func (a *API) WalletNew(ctx context.Context, typ string) (address.Address, error) {
|
|
return a.Wallet.GenerateKey(typ)
|
|
}
|
|
|
|
func (a *API) WalletList(ctx context.Context) ([]address.Address, error) {
|
|
return a.Wallet.ListAddrs()
|
|
}
|
|
|
|
func (a *API) WalletBalance(ctx context.Context, addr address.Address) (types.BigInt, error) {
|
|
return a.Chain.GetBalance(addr)
|
|
}
|
|
|
|
func (a *API) NetConnect(ctx context.Context, p peer.AddrInfo) error {
|
|
return a.Host.Connect(ctx, p)
|
|
}
|
|
|
|
func (a *API) NetAddrsListen(context.Context) (peer.AddrInfo, error) {
|
|
return peer.AddrInfo{
|
|
ID: a.Host.ID(),
|
|
Addrs: a.Host.Addrs(),
|
|
}, nil
|
|
}
|
|
|
|
var _ api.API = &API{}
|