From c80bce9438c4ecfd7fb2bce1c76ff2362bf63098 Mon Sep 17 00:00:00 2001
From: Alex Vanin <alexey@nspcc.ru>
Date: Thu, 29 Oct 2020 17:27:13 +0300
Subject: [PATCH] [#132] Implement netmap snapshot getter for inner ring

Later it should be merged with morph/client getters.

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
---
 go.sum                         | Bin 59509 -> 59592 bytes
 pkg/innerring/invoke/netmap.go |  68 ++++++++++++++++++++++++++++++---
 2 files changed, 62 insertions(+), 6 deletions(-)

diff --git a/go.sum b/go.sum
index d0aa1269742e40b87eec082d3f328449ffa80245..20bbac32ef40d59d2b5e862fc2f05ab896381f79 100644
GIT binary patch
delta 76
zcmV-S0JHz~&;!WP1F*5(lSR`Ilg---H8CzSH7+n9XfZlPYIAFGD^qz;Z)i(bD=~Fx
iY%oenO>}KYLuX1*N-uF&RCsbYad0bAG-I=^-N*{4pB;<<

delta 14
WcmX?ck@@Qd<_#O~Zf3c6gbM&a!3YKb

diff --git a/pkg/innerring/invoke/netmap.go b/pkg/innerring/invoke/netmap.go
index 7f71321f2..73bbdb74d 100644
--- a/pkg/innerring/invoke/netmap.go
+++ b/pkg/innerring/invoke/netmap.go
@@ -3,7 +3,11 @@ package invoke
 import (
 	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
 	"github.com/nspcc-dev/neo-go/pkg/util"
+	"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
+	netmap "github.com/nspcc-dev/neofs-api-go/v2/netmap/grpc"
 	"github.com/nspcc-dev/neofs-node/pkg/morph/client"
+	"github.com/pkg/errors"
+	"google.golang.org/protobuf/proto"
 )
 
 type (
@@ -19,12 +23,13 @@ type (
 )
 
 const (
-	getEpochMethod        = "epoch"
-	setNewEpochMethod     = "newEpoch"
-	approvePeerMethod     = "addPeer"
-	updatePeerStateMethod = "updateState"
-	setConfigMethod       = "setConfigMethod"
-	updateInnerRingMethod = "updateInnerRingMethod"
+	getEpochMethod          = "epoch"
+	setNewEpochMethod       = "newEpoch"
+	approvePeerMethod       = "addPeer"
+	updatePeerStateMethod   = "updateState"
+	setConfigMethod         = "setConfigMethod"
+	updateInnerRingMethod   = "updateInnerRingMethod"
+	getNetmapSnapshotMethod = "netmap"
 )
 
 // Epoch return epoch value from contract.
@@ -101,3 +106,54 @@ func UpdateInnerRing(cli *client.Client, con util.Uint160, list []*keys.PublicKe
 
 	return cli.Invoke(con, extraFee, updateInnerRingMethod, rawKeys)
 }
+
+// NetmapSnapshot returns current netmap node infos.
+func NetmapSnapshot(cli *client.Client, con util.Uint160) ([]netmap.NodeInfo, error) {
+	if cli == nil {
+		return nil, client.ErrNilClient
+	}
+
+	rawNetmapStack, err := cli.TestInvoke(con, getNetmapSnapshotMethod)
+	if err != nil {
+		return nil, err
+	}
+
+	if ln := len(rawNetmapStack); ln != 1 {
+		return nil, errors.New("invalid RPC response")
+	}
+
+	rawNodeInfos, err := client.ArrayFromStackItem(rawNetmapStack[0])
+	if err != nil {
+		return nil, err
+	}
+
+	result := make([]netmap.NodeInfo, 0, len(rawNodeInfos))
+
+	for i := range rawNodeInfos {
+		nodeInfo, err := peerInfoFromStackItem(rawNodeInfos[i])
+		if err != nil {
+			return nil, errors.Wrap(err, "invalid RPC response")
+		}
+
+		result = append(result, *nodeInfo)
+	}
+
+	return result, nil
+}
+
+func peerInfoFromStackItem(prm stackitem.Item) (*netmap.NodeInfo, error) {
+	node := new(netmap.NodeInfo)
+
+	subItems, err := client.ArrayFromStackItem(prm)
+	if err != nil {
+		return nil, err
+	} else if ln := len(subItems); ln != 1 {
+		return nil, errors.New("invalid RPC response")
+	} else if rawNodeInfo, err := client.BytesFromStackItem(subItems[0]); err != nil {
+		return nil, err
+	} else if err = proto.Unmarshal(rawNodeInfo, node); err != nil {
+		return nil, err
+	}
+
+	return node, nil
+}