rclone/cmd/size/size.go

51 lines
1.2 KiB
Go
Raw Normal View History

package size
import (
"context"
2018-04-12 14:45:50 +00:00
"encoding/json"
"fmt"
2018-04-12 14:45:50 +00:00
"os"
"github.com/rclone/rclone/cmd"
"github.com/rclone/rclone/fs"
"github.com/rclone/rclone/fs/config/flags"
"github.com/rclone/rclone/fs/operations"
"github.com/spf13/cobra"
)
2018-04-12 14:45:50 +00:00
var jsonOutput bool
func init() {
2018-04-12 14:45:50 +00:00
cmd.Root.AddCommand(commandDefinition)
cmdFlags := commandDefinition.Flags()
2021-08-16 09:30:01 +00:00
flags.BoolVarP(cmdFlags, &jsonOutput, "json", "", false, "Format output as JSON")
}
2018-04-12 14:45:50 +00:00
var commandDefinition = &cobra.Command{
Use: "size remote:path",
Short: `Prints the total size and number of objects in remote:path.`,
Run: func(command *cobra.Command, args []string) {
cmd.CheckArgs(1, 1, command, args)
fsrc := cmd.NewFsSrc(args)
cmd.Run(false, false, command, func() error {
2018-04-12 14:45:50 +00:00
var err error
var results struct {
Count int64 `json:"count"`
Bytes int64 `json:"bytes"`
}
results.Count, results.Bytes, err = operations.Count(context.Background(), fsrc)
if err != nil {
return err
}
2018-04-12 14:45:50 +00:00
if jsonOutput {
return json.NewEncoder(os.Stdout).Encode(results)
}
2021-04-02 18:15:56 +00:00
fmt.Printf("Total objects: %s (%d)\n", fs.CountSuffix(results.Count), results.Count)
fmt.Printf("Total size: %s (%d Byte)\n", fs.SizeSuffix(results.Bytes).ByteUnit(), results.Bytes)
return nil
})
},
}