05a351d344
Make `PeerID` type to be the wrapper over v2 `PeerID` message structure. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
93 lines
2.2 KiB
Go
93 lines
2.2 KiB
Go
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 reputation.PeerID
|
|
|
|
// NewPeerID creates and returns blank PeerID.
|
|
func NewPeerID() *PeerID {
|
|
return PeerIDFromV2(new(reputation.PeerID))
|
|
}
|
|
|
|
// PeerIDFromV2 converts NeoFS API v2 reputation.PeerID message to PeerID.
|
|
func PeerIDFromV2(id *reputation.PeerID) *PeerID {
|
|
return (*PeerID)(id)
|
|
}
|
|
|
|
// SetPublicKey sets peer ID as a compressed public key.
|
|
func (x *PeerID) SetPublicKey(v [crypto.PublicKeyCompressedSize]byte) {
|
|
(*reputation.PeerID)(x).
|
|
SetValue(v[:])
|
|
}
|
|
|
|
// ToV2 converts PeerID to NeoFS API v2 reputation.PeerID message.
|
|
func (x *PeerID) ToV2() *reputation.PeerID {
|
|
return (*reputation.PeerID)(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)
|
|
}
|