package torrent

import (
	"time"

	"github.com/debridmediamanager/zurg/pkg/realdebrid"
)

type LibraryState struct {
	TotalCount         int
	ActiveCount        int
	FirstTorrent       realdebrid.Torrent
	FirstActiveTorrent realdebrid.Torrent
}

func (ls LibraryState) equal(a LibraryState) bool {
	return a.TotalCount == ls.TotalCount &&
		a.ActiveCount == ls.ActiveCount &&
		a.FirstTorrent.ID == ls.FirstTorrent.ID &&
		a.FirstTorrent.Status == ls.FirstTorrent.Status &&
		a.FirstActiveTorrent.ID == ls.FirstActiveTorrent.ID &&
		a.FirstActiveTorrent.Status == ls.FirstActiveTorrent.Status
}

func EmptyState() LibraryState {
	empty := realdebrid.Torrent{
		ID:    "",
		Added: (time.Time{}).Format(time.RFC3339),
	}
	return LibraryState{
		TotalCount:         0,
		ActiveCount:        0,
		FirstTorrent:       empty,
		FirstActiveTorrent: empty,
	}
}

func (t *TorrentManager) SetNewLatestState(checksum LibraryState) {
	t.latestState.ActiveCount = checksum.ActiveCount
	t.latestState.TotalCount = checksum.TotalCount
	t.latestState.FirstTorrent = checksum.FirstTorrent
	t.latestState.FirstActiveTorrent = checksum.FirstActiveTorrent
}

type torrentsResp struct {
	torrents   []realdebrid.Torrent
	totalCount int
}

// generates a checksum based on the number of torrents, the first torrent id and the number of active torrents
func (t *TorrentManager) getCurrentState() LibraryState {
	torrentChan := make(chan torrentsResp, 1)
	activeChan := make(chan torrentsResp, 1)
	countChan := make(chan int, 1)
	errChan := make(chan error, 3) // accommodate errors from both goroutines
	defer close(torrentChan)
	defer close(activeChan)
	defer close(countChan)
	defer close(errChan)

	_ = t.workerPool.Submit(func() {
		torrents, totalCount, err := t.Api.GetTorrents(1, false)
		if err != nil {
			errChan <- err
			return
		}
		torrentChan <- torrentsResp{torrents: torrents, totalCount: totalCount}
	})

	_ = t.workerPool.Submit(func() {
		torrents, totalCount, err := t.Api.GetTorrents(1, true)
		if err != nil {
			errChan <- err
			return
		}
		activeChan <- torrentsResp{torrents: torrents, totalCount: totalCount}
	})

	_ = t.workerPool.Submit(func() {
		count, err := t.Api.GetActiveTorrentCount()
		if err != nil {
			errChan <- err
			return
		}
		countChan <- count.DownloadingCount
	})

	// Existing goroutines for GetTorrents and GetActiveTorrentCount
	var first realdebrid.Torrent
	var active realdebrid.Torrent
	var totalCount, count int

	for i := 0; i < 3; i++ {
		select {
		case firstResp := <-torrentChan:
			first = firstResp.torrents[0]
			totalCount = firstResp.totalCount
		case activeResp := <-activeChan:
			active = activeResp.torrents[0]
		case count = <-countChan:
		case err := <-errChan:
			t.log.Warnf("Checksum API Error: %v", err)
			return EmptyState()
		}
	}

	return LibraryState{
		TotalCount:         totalCount,
		ActiveCount:        count,
		FirstTorrent:       first,
		FirstActiveTorrent: active,
	}
}