mirror of
https://github.com/rclone/rclone.git
synced 2024-11-28 11:16:45 +08:00
2925e1384c
Includes adding support for additional size input suffix Mi and MiB, treated equivalent to M. Extends binary suffix output with letter i, e.g. Ki and Mi. Centralizes creation of bit/byte unit strings.
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package size
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"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"
|
|
)
|
|
|
|
var jsonOutput bool
|
|
|
|
func init() {
|
|
cmd.Root.AddCommand(commandDefinition)
|
|
cmdFlags := commandDefinition.Flags()
|
|
flags.BoolVarP(cmdFlags, &jsonOutput, "json", "", false, "format output as JSON")
|
|
}
|
|
|
|
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 {
|
|
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
|
|
}
|
|
|
|
if jsonOutput {
|
|
return json.NewEncoder(os.Stdout).Encode(results)
|
|
}
|
|
|
|
fmt.Printf("Total objects: %d\n", results.Count)
|
|
fmt.Printf("Total size: %s (%d bytes)\n", fs.SizeSuffix(results.Bytes).ByteUnit(), results.Bytes)
|
|
|
|
return nil
|
|
})
|
|
},
|
|
}
|