package torrent

import "github.com/debridmediamanager/zurg/pkg/logutil"

type LibraryState struct {
	TotalCount     int
	ActiveCount    int
	FirstTorrentId string
	log            *logutil.Logger
}

func (ls *LibraryState) Eq(a LibraryState) bool {
	if ls.TotalCount == 0 || ls.FirstTorrentId == "" {
		return false
	} else if a.TotalCount != ls.TotalCount {
		ls.log.Debugf("Detected changes! Total count mismatch: was %d now %d", ls.TotalCount, a.TotalCount)
		return false
	} else if a.ActiveCount != ls.ActiveCount {
		ls.log.Debugf("Detected changes! Active count mismatch: was %d now %d", ls.ActiveCount, a.ActiveCount)
		return false
	} else if a.FirstTorrentId != ls.FirstTorrentId {
		ls.log.Debugf("Detected changes! First torrent id mismatch: was %s now %s", ls.FirstTorrentId, a.FirstTorrentId)
		return false
	}
	return true
}

func (t *TorrentManager) setNewLatestState(checksum LibraryState) {
	t.latestState.ActiveCount = checksum.ActiveCount
	t.latestState.TotalCount = checksum.TotalCount
	t.latestState.FirstTorrentId = checksum.FirstTorrentId
}

// generates a checksum based on the number of torrents, the first torrent id and the number of active torrents
func (t *TorrentManager) getCurrentState() LibraryState {
	var state LibraryState

	torrents, totalCount, err := t.rd.GetTorrents(true)
	if err != nil {
		t.log.Errorf("Checksum API Error (GetTorrents): %v", err)
		state.TotalCount = t.latestState.TotalCount
		state.FirstTorrentId = t.latestState.FirstTorrentId
	} else {
		state.TotalCount = totalCount
		if len(torrents) > 0 {
			state.FirstTorrentId = torrents[0].ID
		}
	}

	count, err := t.rd.GetActiveTorrentCount()
	if err != nil {
		t.log.Errorf("Checksum API Error (GetActiveTorrentCount): %v", err)
		state.ActiveCount = t.latestState.ActiveCount
	} else {
		state.ActiveCount = count.DownloadingCount
	}

	return state
}