From 2245bf85d869776ab33891055fa78423adec7a38 Mon Sep 17 00:00:00 2001 From: Pavel Karpy Date: Wed, 1 Dec 2021 15:27:29 +0300 Subject: [PATCH] [#1002] morph: Add `UserAllowed` method to subnetwork client Signed-off-by: Pavel Karpy --- pkg/morph/client/subnet/clients.go | 58 ++++++++++++++++++++++++++++++ 1 file changed, 58 insertions(+) create mode 100644 pkg/morph/client/subnet/clients.go diff --git a/pkg/morph/client/subnet/clients.go b/pkg/morph/client/subnet/clients.go new file mode 100644 index 000000000..e7ff8494a --- /dev/null +++ b/pkg/morph/client/subnet/clients.go @@ -0,0 +1,58 @@ +package morphsubnet + +import ( + "fmt" + + "github.com/nspcc-dev/neofs-node/pkg/morph/client" +) + +// UserAllowedPrm groups parameters of UserAllowed method of Subnet contract. +type UserAllowedPrm struct { + args [2]interface{} +} + +// SetID sets identifier of the subnet in a binary NeoFS API protocol format. +func (x *UserAllowedPrm) SetID(id []byte) { + x.args[0] = id +} + +// SetClient sets owner ID of the client that is being checked in a binary NeoFS API protocol format. +func (x *UserAllowedPrm) SetClient(id []byte) { + x.args[1] = id +} + +// UserAllowedRes groups resulting values of UserAllowed method of Subnet contract. +type UserAllowedRes struct { + result bool +} + +// Allowed returns true iff the client is allowed to create containers in the subnet. +func (x UserAllowedRes) Allowed() bool { + return x.result +} + +// UserAllowed checks if the user has access to the subnetwork. +func (x *Client) UserAllowed(prm UserAllowedPrm) (*UserAllowedRes, error) { + args := client.TestInvokePrm{} + + args.SetMethod("userAllowed") + args.SetArgs(prm.args[:]...) + + res, err := x.client.TestInvoke(args) + if err != nil { + return nil, fmt.Errorf("could not make test invoke: %w", err) + } + + if len(res) == 0 { + return nil, errEmptyResponse + } + + result, err := client.BoolFromStackItem(res[0]) + if err != nil { + return nil, err + } + + return &UserAllowedRes{ + result: result, + }, nil +}