forked from TrueCloudLab/frostfs-node
114 lines
2.8 KiB
Go
114 lines
2.8 KiB
Go
package morphsubnet
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/TrueCloudLab/frostfs-node/pkg/morph/client"
|
|
)
|
|
|
|
// UserAllowedPrm groups parameters of UserAllowed method of Subnet contract.
|
|
type UserAllowedPrm struct {
|
|
args [2]any
|
|
}
|
|
|
|
// SetID sets identifier of the subnet in a binary FrostFS 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 FrostFS API protocol format.
|
|
func (x *UserAllowedPrm) SetClient(id []byte) {
|
|
x.args[1] = id
|
|
}
|
|
|
|
// UserAllowedRes groups the 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(userAllowedMethod)
|
|
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
|
|
}
|
|
|
|
// ManageClientsPrm groups parameters of client management in Subnet contract.
|
|
//
|
|
// Zero value adds subnet client. Subnet, group and client ID must be specified via setters.
|
|
type ManageClientsPrm struct {
|
|
// remove or add client
|
|
rm bool
|
|
|
|
args [3]any
|
|
}
|
|
|
|
// SetRemove marks client to be removed. By default, client is added.
|
|
func (x *ManageClientsPrm) SetRemove() {
|
|
x.rm = true
|
|
}
|
|
|
|
// SetSubnet sets identifier of the subnet in a binary FrostFS API protocol format.
|
|
func (x *ManageClientsPrm) SetSubnet(id []byte) {
|
|
x.args[0] = id
|
|
}
|
|
|
|
// SetGroup sets identifier of the client group in a binary FrostFS API protocol format.
|
|
func (x *ManageClientsPrm) SetGroup(id []byte) {
|
|
x.args[1] = id
|
|
}
|
|
|
|
// SetClient sets client's user ID in a binary FrostFS API protocol format.
|
|
func (x *ManageClientsPrm) SetClient(id []byte) {
|
|
x.args[2] = id
|
|
}
|
|
|
|
// ManageClientsRes groups the resulting values of client management methods of Subnet contract.
|
|
type ManageClientsRes struct{}
|
|
|
|
// ManageClients manages client list of the FrostFS subnet through Subnet contract calls.
|
|
func (x Client) ManageClients(prm ManageClientsPrm) (*ManageClientsRes, error) {
|
|
var method string
|
|
|
|
if prm.rm {
|
|
method = removeUserMethod
|
|
} else {
|
|
method = addUserMethod
|
|
}
|
|
|
|
var prmInvoke client.InvokePrm
|
|
|
|
prmInvoke.SetMethod(method)
|
|
prmInvoke.SetArgs(prm.args[:]...)
|
|
|
|
err := x.client.Invoke(prmInvoke)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return new(ManageClientsRes), nil
|
|
}
|