From 1759ff57f11b42620fe946f56740d99fc8800e4d Mon Sep 17 00:00:00 2001
From: Alex Vanin <alexey@nspcc.ru>
Date: Tue, 22 Sep 2020 14:02:32 +0300
Subject: [PATCH] [#36] Parse attributes from node config

Well-known attributes set up with explicit configuration
values, other attributes set up from chain of attributes.

Chain of attributes is a string, that contains keys-value pairs,
divided by semicolon. Pairs itself divided by slash.

E.g. "StorageType:HDD/RPM:7200"

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
---
 cmd/neofs-node/config.go | 35 +++++++++++++++++++++++++++++++----
 cmd/neofs-node/morph.go  | 11 +++++++++--
 2 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/cmd/neofs-node/config.go b/cmd/neofs-node/config.go
index 52e95ddce..bebe0164b 100644
--- a/cmd/neofs-node/config.go
+++ b/cmd/neofs-node/config.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"crypto/ecdsa"
 	"net"
+	"strconv"
 	"strings"
 	"sync"
 
@@ -18,8 +19,11 @@ import (
 
 const (
 	// config keys for cfgNodeInfo
-	cfgNodeKey          = "node.key"
-	cfgBootstrapAddress = "node.address"
+	cfgNodeKey             = "node.key"
+	cfgBootstrapAddress    = "node.address"
+	cfgNodeAttributePrefix = "node.attribute"
+	cfgNodeCapacity        = "node.capacity"
+	cfgNodePrice           = "node.price"
 
 	// config keys for cfgGRPC
 	cfgListenAddress = "grpc.endpoint"
@@ -95,7 +99,10 @@ type cfgNetmap struct {
 type BootstrapType uint32
 
 type cfgNodeInfo struct {
-	bootType BootstrapType
+	bootType   BootstrapType
+	attributes []string
+	capacity   uint64
+	price      uint64
 }
 
 const (
@@ -140,7 +147,10 @@ func initCfg(path string) *cfg {
 			fee:        util.Fixed8(viperCfg.GetInt(cfgNetmapFee)),
 		},
 		cfgNodeInfo: cfgNodeInfo{
-			bootType: StorageNode,
+			bootType:   StorageNode,
+			attributes: readAttributes(viperCfg),
+			capacity:   viperCfg.GetUint64(cfgNodeCapacity),
+			price:      viperCfg.GetUint64(cfgNodePrice),
 		},
 	}
 }
@@ -169,6 +179,8 @@ func initViper(path string) *viper.Viper {
 func defaultConfiguration(v *viper.Viper) {
 	// fixme: all hardcoded private keys must be removed
 	v.SetDefault(cfgNodeKey, "Kwk6k2eC3L3QuPvD8aiaNyoSXgQ2YL1bwS5CP1oKoA9waeAze97s")
+	v.SetDefault(cfgNodeCapacity, "10")
+	v.SetDefault(cfgNodePrice, "10")
 	v.SetDefault(cfgBootstrapAddress, "") // address to bootstrap with
 
 	v.SetDefault(cfgMorphRPCAddress, "http://morph_chain.localtest.nspcc.ru:30333/")
@@ -183,3 +195,18 @@ func defaultConfiguration(v *viper.Viper) {
 	v.SetDefault(cfgNetmapContract, "75194459637323ea8837d2afe8225ec74a5658c3")
 	v.SetDefault(cfgNetmapFee, "1")
 }
+
+func readAttributes(v *viper.Viper) (attrs []string) {
+	const maxAttributes = 100
+
+	for i := 0; i < maxAttributes; i++ {
+		attr := v.GetString(cfgNodeAttributePrefix + "_" + strconv.Itoa(i))
+		if attr == "" {
+			return
+		} else {
+			attrs = append(attrs, attr)
+		}
+	}
+
+	return attrs
+}
diff --git a/cmd/neofs-node/morph.go b/cmd/neofs-node/morph.go
index 54b916cb8..50396e3cb 100644
--- a/cmd/neofs-node/morph.go
+++ b/cmd/neofs-node/morph.go
@@ -1,10 +1,12 @@
 package main
 
 import (
+	v2netmap "github.com/nspcc-dev/neofs-api-go/v2/netmap"
 	crypto "github.com/nspcc-dev/neofs-crypto"
 	"github.com/nspcc-dev/neofs-node/pkg/morph/client"
 	"github.com/nspcc-dev/neofs-node/pkg/morph/client/netmap"
 	"github.com/nspcc-dev/neofs-node/pkg/morph/client/netmap/wrapper"
+	"github.com/nspcc-dev/neofs-node/pkg/util/attributes"
 	"github.com/pkg/errors"
 )
 
@@ -30,10 +32,15 @@ func bootstrapNode(c *cfg) {
 		cliWrapper, err := wrapper.New(cli)
 		fatalOnErr(errors.Wrap(err, "bootstrap error"))
 
-		peerInfo := new(netmap.NodeInfo)
+		attrs, err := attributes.ParseV2Attributes(c.cfgNodeInfo.attributes, nil)
+		if err != nil {
+			fatalOnErr(errors.Wrap(err, "bootstrap attribute error"))
+		}
+
+		peerInfo := new(v2netmap.NodeInfo)
 		peerInfo.SetAddress(c.viper.GetString(cfgBootstrapAddress))
 		peerInfo.SetPublicKey(crypto.MarshalPublicKey(&c.key.PublicKey))
-		// todo: add attributes as opts
+		peerInfo.SetAttributes(attrs)
 
 		err = cliWrapper.AddPeer(peerInfo)
 		fatalOnErr(errors.Wrap(err, "bootstrap error"))