package processor

import (
	log "github.com/sirupsen/logrus"
	"sync"
	"task-processor/internal/appState"
	"task-processor/internal/shared"
)

func (p *Processor) StartWork(receivedTasks []shared.TaskResponse) {
	log.Info("Starting work...")
	p.state.ResetCounters()

	if len(receivedTasks) == 0 {
		p.state.SetStatus(appState.StatusIdle)
		return
	}

	p.state.SetStatus(appState.StatusWorkInProgress)
	p.state.SetTasksReceived(len(receivedTasks))

	sorted := p.sortTasks(receivedTasks)

	var wg sync.WaitGroup
	for origin, tasks := range sorted {
		wg.Add(1)
		go func(origin string, tasks []shared.Task) {
			defer wg.Done()
			p.handlers[origin].HandleTasks(tasks, p.out, p.state)
		}(origin, tasks)
	}
	wg.Wait()

	log.Debug("All goroutines finished")
	p.state.SetLastCheck()
	p.state.SetStatus(appState.StatusIdle)
	log.Debugf("State | %+v", p.state)
}

func (p *Processor) sortTasks(receivedTasks []shared.TaskResponse) map[string][]shared.Task {
	sorted := make(map[string][]shared.Task, len(receivedTasks))
	for _, task := range receivedTasks {
		switch {
		case task.OriginSurugayaLink != "":
			sorted[shared.OriginSurugaya] = append(sorted[shared.OriginSurugaya], shared.Task{
				MerchUuid: task.MerchUuid,
				Origin:    shared.OriginSurugaya,
				Link:      task.OriginSurugayaLink,
			})
		case task.OriginMandarakeLink != "":
			sorted[shared.OriginMandarake] = append(sorted[shared.OriginMandarake], shared.Task{
				MerchUuid:  task.MerchUuid,
				Origin:     shared.OriginMandarake,
				Link:       task.OriginMandarakeLink,
				RetryCount: 3,
			})
		}
	}
	return sorted
}