feat(display): add progress

utils/wget.go

@@ -12,6 +12,20 @@ import (
 	"strings"
 )
 
+type ProgressReader struct {
+	io.Reader
+	Total   int64
+	Current int64
+}
+
+func (r *ProgressReader) Read(p []byte) (n int, err error) {
+	n, err = r.Reader.Read(p)
+
+	r.Current += int64(n)
+	fmt.Printf("\rdownloading percent %.2f%%", float64(r.Current*10000/r.Total)/100)
+	return
+}
+
 func WgetBinary(url string, downloadTo string, headers map[string]string) error {
 	out, err := os.Create(downloadTo)
 	if err != nil {
@@ -43,8 +57,12 @@ func WgetBinary(url string, downloadTo string, headers map[string]string) error
 	if len(resp.Header["Content-Type"]) > 0 && strings.Contains(resp.Header["Content-Type"][0], "text") {
 		return errors.New("target is not a binary")
 	}
-
-	_, err = io.Copy(out, resp.Body)
+	reader := &ProgressReader{
+		Reader: resp.Body,
+		Total:  resp.ContentLength,
+	}
+	_, err = io.Copy(out, reader)
+	fmt.Printf("\n")
 	if err != nil {
 		return err
 	}