f474bb6179
clingy changed some and this is just fixes for that Change-Id: I729aed6329fe0988fcb9b4407f16966a753b3204
81 lines
1.6 KiB
Go
81 lines
1.6 KiB
Go
// Copyright (C) 2021 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/zeebo/clingy"
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/storj/cmd/uplinkng/ulloc"
|
|
)
|
|
|
|
type cmdRm struct {
|
|
projectProvider
|
|
|
|
recursive bool
|
|
encrypted bool
|
|
|
|
location ulloc.Location
|
|
}
|
|
|
|
func (c *cmdRm) Setup(params clingy.Parameters) {
|
|
c.projectProvider.Setup(params)
|
|
|
|
c.recursive = params.Flag("recursive", "Remove recursively", false,
|
|
clingy.Short('r'),
|
|
clingy.Transform(strconv.ParseBool),
|
|
).(bool)
|
|
c.encrypted = params.Flag("encrypted", "Interprets keys base64 encoded without decrypting", false,
|
|
clingy.Transform(strconv.ParseBool),
|
|
).(bool)
|
|
|
|
c.location = params.Arg("location", "Location to remove (sj://BUCKET[/KEY])",
|
|
clingy.Transform(ulloc.Parse),
|
|
).(ulloc.Location)
|
|
}
|
|
|
|
func (c *cmdRm) Execute(ctx clingy.Context) error {
|
|
fs, err := c.OpenFilesystem(ctx, bypassEncryption(c.encrypted))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() { _ = fs.Close() }()
|
|
|
|
if !c.recursive {
|
|
if err := fs.Remove(ctx, c.location); err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Fprintln(ctx.Stdout(), "removed", c.location)
|
|
return nil
|
|
}
|
|
|
|
iter, err := fs.ListObjects(ctx, c.location, c.recursive)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
anyFailed := false
|
|
for iter.Next() {
|
|
loc := iter.Item().Loc
|
|
|
|
if err := fs.Remove(ctx, loc); err != nil {
|
|
fmt.Fprintln(ctx.Stderr(), "remove", loc, "failed:", err.Error())
|
|
anyFailed = true
|
|
} else {
|
|
fmt.Fprintln(ctx.Stdout(), "removed", loc)
|
|
}
|
|
}
|
|
|
|
if err := iter.Err(); err != nil {
|
|
return errs.Wrap(err)
|
|
} else if anyFailed {
|
|
return errs.New("some removals failed")
|
|
}
|
|
return nil
|
|
}
|