alexvanin/frostfs-node
1
0
Fork 0
forked from TrueCloudLab/frostfs-node
Code Issues Pull requests Projects Releases Packages Wiki Activity

[#372] cli/sg: Support bearer token in storagegroup command

Browse source 
Add `--bearer` flag to storagegroup command similar to object command.
Attach parsed bearer token to all API requests of storagegroup command.

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
Leonard Lyubich 2021-02-10 00:55:38 +03:00 committed by Alex Vanin
parent 687af30c05
commit c6411e4f87
1 changed files with 32 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
cmd/neofs-cli/modules/storagegroup.go
View file

@ -52,6 +52,7 @@ var sgDelCmd = &cobra.Command{
const ( const (
sgMembersFlag = "members" sgMembersFlag = "members"
sgIDFlag = "id" sgIDFlag = "id"
sgBearerFlag = "bearer"
) )
var ( var (
@ -62,6 +63,9 @@ var (
func init() { func init() {
rootCmd.AddCommand(storagegroupCmd) rootCmd.AddCommand(storagegroupCmd)
storagegroupCmd.PersistentFlags().String(sgBearerFlag, "",
"File with signed JSON or binary encoded bearer token")
storagegroupCmd.AddCommand(sgPutCmd) storagegroupCmd.AddCommand(sgPutCmd)
sgPutCmd.Flags().String("cid", "", "Container ID") sgPutCmd.Flags().String("cid", "", "Container ID")
_ = sgPutCmd.MarkFlagRequired("cid") _ = sgPutCmd.MarkFlagRequired("cid")
@ -115,6 +119,10 @@ func (c *sgHeadReceiver) Head(addr *objectSDK.Address) (interface{}, error) {
} }
} }
func sgBearerToken(cmd *cobra.Command) (*token.BearerToken, error) {
return getBearerToken(cmd, sgBearerFlag)
}
func putSG(cmd *cobra.Command, _ []string) error { func putSG(cmd *cobra.Command, _ []string) error {
ownerID, err := getOwnerID() ownerID, err := getOwnerID()
if err != nil { if err != nil {
@ -137,6 +145,11 @@ func putSG(cmd *cobra.Command, _ []string) error {
members = append(members, id) members = append(members, id)
} }
bearerToken, err := sgBearerToken(cmd)
if err != nil {
return err
}
ctx := context.Background() ctx := context.Background()
cli, tok, err := initSession(ctx) cli, tok, err := initSession(ctx)
@ -169,6 +182,7 @@ func putSG(cmd *cobra.Command, _ []string) error {
WithObject(obj.Object()), WithObject(obj.Object()),
append(globalCallOptions(), append(globalCallOptions(),
client.WithSession(tok), client.WithSession(tok),
client.WithBearer(bearerToken),
)..., )...,
) )
if err != nil { if err != nil {
@ -199,6 +213,11 @@ func getSG(cmd *cobra.Command, _ []string) error {
return err return err
} }
bearerToken, err := sgBearerToken(cmd)
if err != nil {
return err
}
addr := objectSDK.NewAddress() addr := objectSDK.NewAddress()
addr.SetContainerID(cid) addr.SetContainerID(cid)
addr.SetObjectID(id) addr.SetObjectID(id)
@ -215,6 +234,7 @@ func getSG(cmd *cobra.Command, _ []string) error {
WithAddress(addr), WithAddress(addr),
append(globalCallOptions(), append(globalCallOptions(),
client.WithSession(tok), client.WithSession(tok),
client.WithBearer(bearerToken),
)..., )...,
) )
if err != nil { if err != nil {
@ -247,6 +267,11 @@ func listSG(cmd *cobra.Command, _ []string) error {
return err return err
} }
bearerToken, err := sgBearerToken(cmd)
if err != nil {
return err
}
ctx := context.Background() ctx := context.Background()
cli, tok, err := initSession(ctx) cli, tok, err := initSession(ctx)
@ -260,6 +285,7 @@ func listSG(cmd *cobra.Command, _ []string) error {
WithSearchFilters(storagegroup.SearchQuery()), WithSearchFilters(storagegroup.SearchQuery()),
append(globalCallOptions(), append(globalCallOptions(),
client.WithSession(tok), client.WithSession(tok),
client.WithBearer(bearerToken),
)..., )...,
) )
if err != nil { if err != nil {
@ -286,6 +312,11 @@ func delSG(cmd *cobra.Command, _ []string) error {
return err return err
} }
bearerToken, err := sgBearerToken(cmd)
if err != nil {
return err
}
ctx := context.Background() ctx := context.Background()
cli, tok, err := initSession(ctx) cli, tok, err := initSession(ctx)
@ -302,6 +333,7 @@ func delSG(cmd *cobra.Command, _ []string) error {
WithAddress(addr), WithAddress(addr),
append(globalCallOptions(), append(globalCallOptions(),
client.WithSession(tok), client.WithSession(tok),
client.WithBearer(bearerToken),
)..., )...,
) )
if err != nil { if err != nil {