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 == "" {
		ls.log.Debugf("Checksum is empty")
		return false
	} else if a.TotalCount != ls.TotalCount {
		ls.log.Debugf("Checksum total count mismatch: %d != %d", a.TotalCount, ls.TotalCount)
		return false
	} else if a.ActiveCount != ls.ActiveCount {
		ls.log.Debugf("Checksum active count mismatch: %d != %d", a.ActiveCount, ls.ActiveCount)
		return false
	} else if a.FirstTorrentId != ls.FirstTorrentId {
		ls.log.Debugf("Checksum first torrent id mismatch: %s != %s", a.FirstTorrentId, ls.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.api.GetTorrents(true)
	if err != nil {
		t.log.Errorf("Checksum API Error (GetTorrents): %v", err)
		return LibraryState{}
	}
	state.TotalCount = totalCount
	if len(torrents) > 0 {
		state.FirstTorrentId = torrents[0].ID
	}

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

	return state
}