package main

import (
	"io"
	"math/rand"
	"time"

	"github.com/vbauerster/mpb/v6"
	"github.com/vbauerster/mpb/v6/decor"
)

func init() {
	rand.Seed(time.Now().UnixNano())
}

func main() {
	p := mpb.New(mpb.WithWidth(64))

	var total int64
	// new bar with 'trigger complete event' disabled, because total is zero
	bar := p.AddBar(total,
		mpb.PrependDecorators(decor.Counters(decor.UnitKiB, "% .1f / % .1f")),
		mpb.AppendDecorators(decor.Percentage()),
	)

	maxSleep := 100 * time.Millisecond
	read := makeStream(200)
	for {
		n, err := read()
		total += int64(n)
		if err == io.EOF {
			break
		}
		// while total is unknown,
		// set it to a positive number which is greater than current total,
		// to make sure no complete event is triggered by next IncrBy call.
		bar.SetTotal(total+2048, false)
		bar.IncrBy(n)
		time.Sleep(time.Duration(rand.Intn(10)+1) * maxSleep / 10)
	}

	// force bar complete event, note true flag
	bar.SetTotal(total, true)

	p.Wait()
}

func makeStream(limit int) func() (int, error) {
	return func() (int, error) {
		if limit <= 0 {
			return 0, io.EOF
		}
		limit--
		return rand.Intn(1024) + 1, nil
	}
}