Add Inner Ring code

cmd/neofs-node/modules/morph/netmap.go

@@ -0,0 +1,94 @@
+package morph
+
+import (
+	contract "github.com/nspcc-dev/neofs-node/pkg/morph/client/netmap"
+	clientWrapper "github.com/nspcc-dev/neofs-node/pkg/morph/client/netmap/wrapper"
+	"github.com/nspcc-dev/neofs-node/pkg/network/bootstrap"
+	"github.com/pkg/errors"
+	"go.uber.org/dig"
+)
+
+type netmapContractResult struct {
+	dig.Out
+
+	Client *clientWrapper.Wrapper
+
+	NodeRegisterer *bootstrap.Registerer
+}
+
+const (
+	// NetmapContractName is a Netmap contract's config section name.
+	NetmapContractName = "netmap"
+
+	netmapContractAddPeerOpt = "add_peer_method"
+
+	netmapContractNewEpochOpt = "new_epoch_method"
+
+	netmapContractNetmapOpt = "netmap_method"
+
+	netmapContractUpdStateOpt = "update_state_method"
+
+	netmapContractIRListOpt = "ir_list_method"
+)
+
+// NetmapContractAddPeerOptPath returns the config path to add peer method of Netmap contract.
+func NetmapContractAddPeerOptPath() string {
+	return optPath(prefix, NetmapContractName, netmapContractAddPeerOpt)
+}
+
+// NetmapContractNewEpochOptPath returns the config path to new epoch method of Netmap contract.
+func NetmapContractNewEpochOptPath() string {
+	return optPath(prefix, NetmapContractName, netmapContractNewEpochOpt)
+}
+
+// NetmapContractNetmapOptPath returns the config path to get netmap method of Netmap contract.
+func NetmapContractNetmapOptPath() string {
+	return optPath(prefix, NetmapContractName, netmapContractNetmapOpt)
+}
+
+// NetmapContractUpdateStateOptPath returns the config path to update state method of Netmap contract.
+func NetmapContractUpdateStateOptPath() string {
+	return optPath(prefix, NetmapContractName, netmapContractUpdStateOpt)
+}
+
+// NetmapContractIRListOptPath returns the config path to inner ring list method of Netmap contract.
+func NetmapContractIRListOptPath() string {
+	return optPath(prefix, NetmapContractName, netmapContractIRListOpt)
+}
+
+func newNetmapContract(p contractParams) (res netmapContractResult, err error) {
+	client, ok := p.MorphContracts[NetmapContractName]
+	if !ok {
+		err = errors.Errorf("missing %s contract client", NetmapContractName)
+		return
+	}
+
+	var (
+		addPeerMethod  = p.Viper.GetString(NetmapContractAddPeerOptPath())
+		newEpochMethod = p.Viper.GetString(NetmapContractNewEpochOptPath())
+		netmapMethod   = p.Viper.GetString(NetmapContractNetmapOptPath())
+		updStateMethod = p.Viper.GetString(NetmapContractUpdateStateOptPath())
+		irListMethod   = p.Viper.GetString(NetmapContractIRListOptPath())
+	)
+
+	var c *contract.Client
+	if c, err = contract.New(client,
+		contract.WithAddPeerMethod(addPeerMethod),
+		contract.WithNewEpochMethod(newEpochMethod),
+		contract.WithNetMapMethod(netmapMethod),
+		contract.WithUpdateStateMethod(updStateMethod),
+		contract.WithInnerRingListMethod(irListMethod),
+	); err != nil {
+		return
+	}
+
+	if res.Client, err = clientWrapper.New(c); err != nil {
+		return
+	}
+
+	if res.NodeRegisterer, err = bootstrap.New(res.Client, p.NodeInfo); err != nil {
+		return
+	}
+
+	return res, nil
+}