forked from TrueCloudLab/frostfs-node
103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
package invoke
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neofs-node/pkg/morph/client"
|
|
)
|
|
|
|
type (
|
|
UpdatePeerArgs struct {
|
|
Key *keys.PublicKey
|
|
Status uint32
|
|
}
|
|
|
|
SetConfigArgs struct {
|
|
Key []byte
|
|
Value []byte
|
|
}
|
|
)
|
|
|
|
const (
|
|
getEpochMethod = "epoch"
|
|
setNewEpochMethod = "newEpoch"
|
|
approvePeerMethod = "addPeer"
|
|
updatePeerStateMethod = "updateState"
|
|
setConfigMethod = "setConfigMethod"
|
|
updateInnerRingMethod = "updateInnerRingMethod"
|
|
)
|
|
|
|
// Epoch return epoch value from contract.
|
|
func Epoch(cli *client.Client, con util.Uint160) (int64, error) {
|
|
if cli == nil {
|
|
return 0, client.ErrNilClient
|
|
}
|
|
|
|
val, err := cli.TestInvoke(con, getEpochMethod)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
epoch, err := client.IntFromStackItem(val[0])
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return epoch, nil
|
|
}
|
|
|
|
// SetNewEpoch invokes newEpoch method.
|
|
func SetNewEpoch(cli *client.Client, con util.Uint160, epoch uint64) error {
|
|
if cli == nil {
|
|
return client.ErrNilClient
|
|
}
|
|
|
|
return cli.Invoke(con, extraFee, setNewEpochMethod, int64(epoch))
|
|
}
|
|
|
|
// ApprovePeer invokes addPeer method.
|
|
func ApprovePeer(cli *client.Client, con util.Uint160, peer []byte) error {
|
|
if cli == nil {
|
|
return client.ErrNilClient
|
|
}
|
|
|
|
return cli.Invoke(con, extraFee, approvePeerMethod, peer)
|
|
}
|
|
|
|
// UpdatePeerState invokes addPeer method.
|
|
func UpdatePeerState(cli *client.Client, con util.Uint160, args *UpdatePeerArgs) error {
|
|
if cli == nil {
|
|
return client.ErrNilClient
|
|
}
|
|
|
|
return cli.Invoke(con, extraFee, updatePeerStateMethod,
|
|
args.Key.Bytes(),
|
|
int64(args.Status),
|
|
)
|
|
}
|
|
|
|
// SetConfig invokes setConfig method.
|
|
func SetConfig(cli *client.Client, con util.Uint160, args *SetConfigArgs) error {
|
|
if cli == nil {
|
|
return client.ErrNilClient
|
|
}
|
|
|
|
return cli.Invoke(con, extraFee, setConfigMethod,
|
|
args.Key,
|
|
args.Value,
|
|
)
|
|
}
|
|
|
|
// UpdateInnerRing invokes updateInnerRing method.
|
|
func UpdateInnerRing(cli *client.Client, con util.Uint160, list []*keys.PublicKey) error {
|
|
if cli == nil {
|
|
return client.ErrNilClient
|
|
}
|
|
|
|
rawKeys := make([][]byte, 0, len(list))
|
|
for i := range list {
|
|
rawKeys = append(rawKeys, list[i].Bytes())
|
|
}
|
|
|
|
return cli.Invoke(con, extraFee, updateInnerRingMethod, rawKeys)
|
|
}
|