package utils import ( "io/fs" "path/filepath" "strconv" "strings" "github.com/pkg/errors" ) var sizeSuffixes = [...]string{"B", "KiB", "MiB", "GiB", "TiB"} // ConvertFileSize converts size in bytes down to biggest units it represents. // Returns converted size, unit and, a concatenation of size and unit func ConvertFileSize(size int64) (float64, string, string) { var idx int var fSize float64 = float64(size) for idx = 0; fSize >= 1024; fSize /= 1024 { idx++ } fSizeStr := strconv.FormatFloat(fSize, 'f', 3, 64) fSizeStr = strings.TrimRight(fSizeStr, "0") fSizeStr = strings.TrimSuffix(fSizeStr, ".") return fSize, sizeSuffixes[idx], strings.Join([]string{fSizeStr, sizeSuffixes[idx]}, " ") } func DirectorySize(path string) (dirSz int64, err error) { err = filepath.Walk(path, func(_ string, info fs.FileInfo, err error) error { if err != nil { return errors.Wrapf(err, "failed to compute %s directory size", path) } dirSz += info.Size() return nil }) if err != nil { return 0, err } return dirSz, nil }