[#265] pkg/reputation: Change PeerID implementation

Make `PeerID` type to be the wrapper over v2 `PeerID` message structure.

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>

pkg/reputation/peer.go

@@ -1,29 +1,93 @@
 package reputation
 
+import (
+	"bytes"
+
+	"github.com/mr-tron/base58"
+	"github.com/nspcc-dev/neofs-api-go/v2/reputation"
+	crypto "github.com/nspcc-dev/neofs-crypto"
+)
+
 // PeerID represents peer ID compatible with NeoFS API v2.
-type PeerID []byte
+type PeerID reputation.PeerID
 
 // NewPeerID creates and returns blank PeerID.
 func NewPeerID() *PeerID {
-	return PeerIDFromV2(nil)
+	return PeerIDFromV2(new(reputation.PeerID))
 }
 
-// PeerIDFromV2 converts bytes slice to PeerID.
-func PeerIDFromV2(data []byte) *PeerID {
-	return (*PeerID)(&data)
+// PeerIDFromV2 converts NeoFS API v2 reputation.PeerID message to PeerID.
+func PeerIDFromV2(id *reputation.PeerID) *PeerID {
+	return (*PeerID)(id)
 }
 
-// SetBytes sets bytes of peer ID.
-func (x *PeerID) SetBytes(v []byte) {
-	*x = v
+// SetPublicKey sets peer ID as a compressed public key.
+func (x *PeerID) SetPublicKey(v [crypto.PublicKeyCompressedSize]byte) {
+	(*reputation.PeerID)(x).
+		SetValue(v[:])
 }
 
-// Bytes returns bytes of peer ID.
-func (x PeerID) Bytes() []byte {
-	return x
+// ToV2 converts PeerID to NeoFS API v2 reputation.PeerID message.
+func (x *PeerID) ToV2() *reputation.PeerID {
+	return (*reputation.PeerID)(x)
 }
 
-// ToV2 converts PeerID to byte slice.
-func (x PeerID) ToV2() []byte {
-	return x
+// Equal returns true if identifiers are identical.
+func (x *PeerID) Equal(x2 *PeerID) bool {
+	return bytes.Equal(
+		(*reputation.PeerID)(x).GetValue(),
+		(*reputation.PeerID)(x2).GetValue(),
+	)
+}
+
+// Parse parses PeerID from base58 string.
+func (x *PeerID) Parse(s string) error {
+	data, err := base58.Decode(s)
+	if err != nil {
+		return err
+	}
+
+	(*reputation.PeerID)(x).SetValue(data)
+
+	return nil
+}
+
+// String returns base58 string representation of PeerID.
+func (x *PeerID) String() string {
+	return base58.Encode(
+		(*reputation.PeerID)(x).
+			GetValue(),
+	)
+}
+
+// Marshal marshals PeerID into a protobuf binary form.
+//
+// Buffer is allocated when the argument is empty.
+// Otherwise, the first buffer is used.
+func (x *PeerID) Marshal(b ...[]byte) ([]byte, error) {
+	var buf []byte
+	if len(b) > 0 {
+		buf = b[0]
+	}
+
+	return (*reputation.PeerID)(x).
+		StableMarshal(buf)
+}
+
+// Unmarshal unmarshals protobuf binary representation of PeerID.
+func (x *PeerID) Unmarshal(data []byte) error {
+	return (*reputation.PeerID)(x).
+		Unmarshal(data)
+}
+
+// MarshalJSON encodes PeerID to protobuf JSON format.
+func (x *PeerID) MarshalJSON() ([]byte, error) {
+	return (*reputation.PeerID)(x).
+		MarshalJSON()
+}
+
+// UnmarshalJSON decodes PeerID from protobuf JSON format.
+func (x *PeerID) UnmarshalJSON(data []byte) error {
+	return (*reputation.PeerID)(x).
+		UnmarshalJSON(data)
 }