5fe5af5bf9
Implement `owner.ID.Equal` method which defines a comparison relationship between two identifiers. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
115 lines
2.6 KiB
Go
115 lines
2.6 KiB
Go
package owner
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/mr-tron/base58"
|
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
|
"github.com/nspcc-dev/neofs-api-go/v2/refs"
|
|
)
|
|
|
|
// ID represents v2-compatible owner identifier.
|
|
type ID refs.OwnerID
|
|
|
|
var errInvalidIDString = errors.New("incorrect format of the string owner ID")
|
|
|
|
// NewIDFromV2 wraps v2 OwnerID message to ID.
|
|
func NewIDFromV2(idV2 *refs.OwnerID) *ID {
|
|
return (*ID)(idV2)
|
|
}
|
|
|
|
// NewID creates and initializes blank ID.
|
|
//
|
|
// Works similar as NewIDFromV2(new(OwnerID)).
|
|
func NewID() *ID {
|
|
return NewIDFromV2(new(refs.OwnerID))
|
|
}
|
|
|
|
// SetNeo3Wallet sets owner identifier value to NEO3 wallet address.
|
|
func (id *ID) SetNeo3Wallet(v *NEO3Wallet) {
|
|
(*refs.OwnerID)(id).SetValue(v.Bytes())
|
|
}
|
|
|
|
// ToV2 returns the v2 owner ID message.
|
|
func (id *ID) ToV2() *refs.OwnerID {
|
|
return (*refs.OwnerID)(id)
|
|
}
|
|
|
|
func (id *ID) String() string {
|
|
return base58.Encode((*refs.OwnerID)(id).GetValue())
|
|
}
|
|
|
|
// Equal defines a comparison relation on ID's.
|
|
//
|
|
// ID's are equal if they have the same binary representation.
|
|
func (id *ID) Equal(id2 *ID) bool {
|
|
return bytes.Equal(
|
|
(*refs.ObjectID)(id).GetValue(),
|
|
(*refs.ObjectID)(id2).GetValue(),
|
|
)
|
|
}
|
|
|
|
func ScriptHashBE(id fmt.Stringer) ([]byte, error) {
|
|
addr, err := address.StringToUint160(id.String())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return addr.BytesBE(), nil
|
|
}
|
|
|
|
// NewIDFromNeo3Wallet creates new owner identity from 25-byte neo wallet.
|
|
func NewIDFromNeo3Wallet(v *NEO3Wallet) *ID {
|
|
id := NewID()
|
|
id.SetNeo3Wallet(v)
|
|
|
|
return id
|
|
}
|
|
|
|
// Parse converts base58 string representation into ID.
|
|
func (id *ID) Parse(s string) error {
|
|
data, err := base58.Decode(s)
|
|
if err != nil {
|
|
return fmt.Errorf("could not parse owner.ID from string: %w", err)
|
|
} else if len(data) != NEO3WalletSize {
|
|
return errInvalidIDString
|
|
}
|
|
|
|
(*refs.OwnerID)(id).SetValue(data)
|
|
|
|
return nil
|
|
}
|
|
|
|
// Marshal marshals ID into a protobuf binary form.
|
|
//
|
|
// Buffer is allocated when the argument is empty.
|
|
// Otherwise, the first buffer is used.
|
|
func (id *ID) Marshal(b ...[]byte) ([]byte, error) {
|
|
var buf []byte
|
|
if len(b) > 0 {
|
|
buf = b[0]
|
|
}
|
|
|
|
return (*refs.OwnerID)(id).
|
|
StableMarshal(buf)
|
|
}
|
|
|
|
// Unmarshal unmarshals protobuf binary representation of ID.
|
|
func (id *ID) Unmarshal(data []byte) error {
|
|
return (*refs.OwnerID)(id).
|
|
Unmarshal(data)
|
|
}
|
|
|
|
// MarshalJSON encodes ID to protobuf JSON format.
|
|
func (id *ID) MarshalJSON() ([]byte, error) {
|
|
return (*refs.OwnerID)(id).
|
|
MarshalJSON()
|
|
}
|
|
|
|
// UnmarshalJSON decodes ID from protobuf JSON format.
|
|
func (id *ID) UnmarshalJSON(data []byte) error {
|
|
return (*refs.OwnerID)(id).
|
|
UnmarshalJSON(data)
|
|
}
|