mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-21 10:27:46 +08:00
149 lines
3.0 KiB
Go
149 lines
3.0 KiB
Go
package node
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"encoding/base64"
|
|
"errors"
|
|
|
|
"go.uber.org/fx"
|
|
|
|
"github.com/ipfs/go-ipfs/core/node/helpers"
|
|
"github.com/ipfs/go-ipfs/core/node/libp2p"
|
|
"github.com/ipfs/go-ipfs/repo"
|
|
|
|
ds "github.com/ipfs/go-datastore"
|
|
dsync "github.com/ipfs/go-datastore/sync"
|
|
cfg "github.com/ipfs/go-ipfs-config"
|
|
logging "github.com/ipfs/go-log"
|
|
ci "github.com/libp2p/go-libp2p-crypto"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
)
|
|
|
|
var log = logging.Logger("node")
|
|
|
|
type BuildCfg struct {
|
|
// If online is set, the node will have networking enabled
|
|
Online bool
|
|
|
|
// ExtraOpts is a map of extra options used to configure the ipfs nodes creation
|
|
ExtraOpts map[string]bool
|
|
|
|
// If permanent then node should run more expensive processes
|
|
// that will improve performance in long run
|
|
Permanent bool
|
|
|
|
// DisableEncryptedConnections disables connection encryption *entirely*.
|
|
// DO NOT SET THIS UNLESS YOU'RE TESTING.
|
|
DisableEncryptedConnections bool
|
|
|
|
// If NilRepo is set, a Repo backed by a nil datastore will be constructed
|
|
NilRepo bool
|
|
|
|
Routing libp2p.RoutingOption
|
|
Host libp2p.HostOption
|
|
Repo repo.Repo
|
|
}
|
|
|
|
func (cfg *BuildCfg) getOpt(key string) bool {
|
|
if cfg.ExtraOpts == nil {
|
|
return false
|
|
}
|
|
|
|
return cfg.ExtraOpts[key]
|
|
}
|
|
|
|
func (cfg *BuildCfg) fillDefaults() error {
|
|
if cfg.Repo != nil && cfg.NilRepo {
|
|
return errors.New("cannot set a Repo and specify nilrepo at the same time")
|
|
}
|
|
|
|
if cfg.Repo == nil {
|
|
var d ds.Datastore
|
|
if cfg.NilRepo {
|
|
d = ds.NewNullDatastore()
|
|
} else {
|
|
d = ds.NewMapDatastore()
|
|
}
|
|
r, err := defaultRepo(dsync.MutexWrap(d))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
cfg.Repo = r
|
|
}
|
|
|
|
if cfg.Routing == nil {
|
|
cfg.Routing = libp2p.DHTOption
|
|
}
|
|
|
|
if cfg.Host == nil {
|
|
cfg.Host = libp2p.DefaultHostOption
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (cfg *BuildCfg) options(ctx context.Context) fx.Option {
|
|
err := cfg.fillDefaults()
|
|
if err != nil {
|
|
return fx.Error(err)
|
|
}
|
|
|
|
repoOption := fx.Provide(func(lc fx.Lifecycle) repo.Repo {
|
|
lc.Append(fx.Hook{
|
|
OnStop: func(ctx context.Context) error {
|
|
return cfg.Repo.Close()
|
|
},
|
|
})
|
|
|
|
return cfg.Repo
|
|
})
|
|
|
|
metricsCtx := fx.Provide(func() helpers.MetricsCtx {
|
|
return helpers.MetricsCtx(ctx)
|
|
})
|
|
|
|
hostOption := fx.Provide(func() libp2p.HostOption {
|
|
return cfg.Host
|
|
})
|
|
|
|
routingOption := fx.Provide(func() libp2p.RoutingOption {
|
|
return cfg.Routing
|
|
})
|
|
|
|
return fx.Options(
|
|
repoOption,
|
|
hostOption,
|
|
routingOption,
|
|
metricsCtx,
|
|
)
|
|
}
|
|
|
|
func defaultRepo(dstore repo.Datastore) (repo.Repo, error) {
|
|
c := cfg.Config{}
|
|
priv, pub, err := ci.GenerateKeyPairWithReader(ci.RSA, 1024, rand.Reader)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
pid, err := peer.IDFromPublicKey(pub)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
privkeyb, err := priv.Bytes()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
c.Bootstrap = cfg.DefaultBootstrapAddresses
|
|
c.Addresses.Swarm = []string{"/ip4/0.0.0.0/tcp/4001"}
|
|
c.Identity.PeerID = pid.Pretty()
|
|
c.Identity.PrivKey = base64.StdEncoding.EncodeToString(privkeyb)
|
|
|
|
return &repo.Mock{
|
|
D: dstore,
|
|
C: c,
|
|
}, nil
|
|
}
|