Human friendly flag byte counts

flagbytes.go

import (
	"flag"
	"strings"

	humanize "github.com/dustin/go-humanize"
)

type BytesFlagVar struct {
	Bytes *uint64
}

func (f BytesFlagVar) String() string {
	if f.Bytes == nil {
		return ""
	}
	return strings.Replace(humanize.IBytes(*f.Bytes), " ", "", -1)
}

func (f *BytesFlagVar) Set(x string) error {
	bytes, err := humanize.ParseBytes(x)
	if err != nil {
		return err
	}
	*f.Bytes = bytes
	return nil
}

func (f BytesFlagVar) Get() interface{} {
	return *f.Bytes
}

/*
var fooBytes uint64 = 1024
flag.Var(&BytesFlagVar{Bytes: &fooBytes}, "fooBytes", "Foo bytes")
*/