mirror of
https://github.com/kopia/kopia.git
synced 2025-12-23 22:57:50 -05:00
d37de8316e
Generalize a couple of functions in the units package using generics. This allows removing duplicate code and simplifying callers by removing unnecessary integer conversions. Additional cleanups: - make "/s" part of the Printf format string ; - simplify setSizeMBParameter; - generalize cli.maybeHumanReadable*` helpers; - remove unneeded receiver in commandRepositorySetParameters helpers.
121 lines
2.3 KiB
Go
121 lines
2.3 KiB
Go
package cli
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"compress/gzip"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/exp/constraints"
|
|
|
|
"github.com/kopia/kopia/internal/iocopy"
|
|
"github.com/kopia/kopia/internal/units"
|
|
)
|
|
|
|
const oneHundredPercent = 100.0
|
|
|
|
// TODO - remove this global.
|
|
var timeZone = "local" //nolint:gochecknoglobals
|
|
|
|
func showContentWithFlags(w io.Writer, rd io.Reader, unzip, indentJSON bool) error {
|
|
if unzip {
|
|
gz, err := gzip.NewReader(rd)
|
|
if err != nil {
|
|
return errors.Wrap(err, "unable to open gzip stream")
|
|
}
|
|
|
|
rd = gz
|
|
}
|
|
|
|
var buf1, buf2 bytes.Buffer
|
|
|
|
if indentJSON {
|
|
if err := iocopy.JustCopy(&buf1, rd); err != nil {
|
|
return errors.Wrap(err, "error copying data")
|
|
}
|
|
|
|
if err := json.Indent(&buf2, buf1.Bytes(), "", " "); err != nil {
|
|
return errors.Wrap(err, "errors indenting JSON")
|
|
}
|
|
|
|
rd = io.NopCloser(&buf2)
|
|
}
|
|
|
|
if err := iocopy.JustCopy(w, rd); err != nil {
|
|
return errors.Wrap(err, "error copying data")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func maybeHumanReadableBytes[I constraints.Integer](enable bool, value I) string {
|
|
if enable {
|
|
return units.BytesString(value)
|
|
}
|
|
|
|
return strconv.FormatInt(int64(value), 10)
|
|
}
|
|
|
|
func maybeHumanReadableCount[I constraints.Integer](enable bool, value I) string {
|
|
if enable {
|
|
return units.Count(value)
|
|
}
|
|
|
|
return strconv.FormatInt(int64(value), 10)
|
|
}
|
|
|
|
func formatTimestamp(ts time.Time) string {
|
|
return convertTimezone(ts).Format("2006-01-02 15:04:05 MST")
|
|
}
|
|
|
|
func formatTimestampPrecise(ts time.Time) string {
|
|
return convertTimezone(ts).Format("2006-01-02 15:04:05.000 MST")
|
|
}
|
|
|
|
func convertTimezone(ts time.Time) time.Time {
|
|
switch timeZone {
|
|
case "local":
|
|
return ts.Local()
|
|
case "utc":
|
|
return ts.UTC()
|
|
case "original":
|
|
return ts
|
|
default:
|
|
loc, err := time.LoadLocation(timeZone)
|
|
if err == nil {
|
|
return ts.In(loc)
|
|
}
|
|
|
|
return ts
|
|
}
|
|
}
|
|
|
|
func formatCompressionPercentage(original, compressed int64) string {
|
|
if compressed >= original {
|
|
return "0%"
|
|
}
|
|
|
|
if original == 0 {
|
|
return "0%"
|
|
}
|
|
|
|
return fmt.Sprintf("%.1f%%", oneHundredPercent*(1-float64(compressed)/float64(original)))
|
|
}
|
|
|
|
func indentMultilineString(l, prefix string) string {
|
|
var lines []string
|
|
|
|
s := bufio.NewScanner(strings.NewReader(l))
|
|
for s.Scan() {
|
|
lines = append(lines, prefix+s.Text())
|
|
}
|
|
|
|
return strings.Join(lines, "\n")
|
|
}
|