package ProgressBar

import (
	"github.com/vbauerster/mpb"
	"github.com/vbauerster/mpb/decor"
)

var (
	P = mpb.New()
)

func InitBar(name string, total int) *mpb.Bar {
	var (
		bar *mpb.Bar
	)

	bar = P.AddBar(int64(total),
		mpb.PrependDecorators(
			decor.Name(name),
			decor.Percentage(decor.WCSyncSpace),
		),
		mpb.AppendDecorators(
			// replace ETA decorator with "done" message, OnComplete event
			decor.OnComplete(
				decor.AverageETA(decor.ET_STYLE_GO, decor.WC{W: 4}), "done",
			),
		),
	)

	return bar
}

func CloseBar(bar *mpb.Bar) {
	bar.Abort(false)
}