f474bb6179
clingy changed some and this is just fixes for that Change-Id: I729aed6329fe0988fcb9b4407f16966a753b3204
61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
// Copyright (C) 2021 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/zeebo/clingy"
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/uplink"
|
|
)
|
|
|
|
type cmdAccessSave struct {
|
|
access string
|
|
name string
|
|
force bool
|
|
use bool
|
|
}
|
|
|
|
func (c *cmdAccessSave) Setup(params clingy.Parameters) {
|
|
c.access = params.Flag("access", "Access to save (prompted if unspecified)", "").(string)
|
|
c.name = params.Flag("name", "Name to save the access grant under", "default").(string)
|
|
|
|
c.force = params.Flag("force", "Force overwrite an existing saved access grant", false,
|
|
clingy.Short('f'),
|
|
clingy.Transform(strconv.ParseBool),
|
|
).(bool)
|
|
c.use = params.Flag("use", "Set the saved access to be the one used by default", false,
|
|
clingy.Transform(strconv.ParseBool),
|
|
).(bool)
|
|
}
|
|
|
|
func (c *cmdAccessSave) Execute(ctx clingy.Context) error {
|
|
accessDefault, accesses, err := gf.GetAccessInfo(false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if c.access == "" {
|
|
c.access, err = gf.PromptInput(ctx, "Access:")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if _, err := uplink.ParseAccess(c.access); err != nil {
|
|
return err
|
|
}
|
|
if _, ok := accesses[c.name]; ok && !c.force {
|
|
return errs.New("Access %q already exists. Overwrite by specifying --force or choose a new name with --name", c.name)
|
|
}
|
|
|
|
accesses[c.name] = c.access
|
|
if c.use || accessDefault == "" {
|
|
accessDefault = c.name
|
|
}
|
|
|
|
return gf.SaveAccessInfo(accessDefault, accesses)
|
|
}
|