package requests

import (
	"context"
	"fmt"
	"sync"
	"time"

	"github.com/jedib0t/go-pretty/v6/progress"
)

// streamProgress streams the progress of a task to the console using a progress bar.
// It listens for increments on the provided channel and updates the progress bar accordingly.
//
// The function will stop and mark the progress as errored if the context is cancelled.
// It will also stop and mark the progress as done when the total number of increments is reached.
func streamProgress(
	ctx context.Context,
	wg *sync.WaitGroup,
	total int64,
	message string,
	increase <-chan int64,
) {
	defer wg.Done()
	pw := progress.NewWriter()
	pw.SetTrackerPosition(progress.PositionRight)
	pw.SetStyle(progress.StyleBlocks)
	pw.SetTrackerLength(40)
	pw.SetUpdateFrequency(time.Millisecond * 250)
	go pw.Render()
	dodosTracker := progress.Tracker{
		Message: message,
		Total:   total,
	}
	pw.AppendTracker(&dodosTracker)
	for {
		select {
		case <-ctx.Done():
			if ctx.Err() != context.Canceled {
				dodosTracker.MarkAsErrored()
			}
			fmt.Printf("\r")
			time.Sleep(time.Millisecond * 500)
			pw.Stop()
			return

		case value := <-increase:
			dodosTracker.Increment(value)
		}
	}
}