downloader/torrent_sort.go

package main

import (
	"sort"

	"git.milar.in/milarin/slices"
	"git.milar.in/nyaanime/model"
)

func SortTorrentsByPreferredProperties(torrents map[model.AnimeEpisode][]*model.ParsedTorrent) map[model.AnimeEpisode][]*TorrentPriority {
	torrentsWithPrio := map[model.AnimeEpisode][]*TorrentPriority{}

	for animeEp, torrentList := range torrents {
		torrentPrioList := slices.Map(torrentList, NewTorrentPriority)
		sort.Slice(torrentPrioList, func(i, j int) bool { return torrentPrioList[i].Priority > torrentPrioList[j].Priority })
		torrentsWithPrio[animeEp] = torrentPrioList
	}

	return torrentsWithPrio
}

func DeterminePriority(props model.PropertyHolder) (priority int, preferredProperties map[string]int) {
	preferredProperties = map[string]int{}

	for _, lang := range props.GetLanguages() {
		if langPriority, ok := PreferredLanguages[lang]; ok {
			priority += langPriority
			preferredProperties["lang/"+lang] = langPriority
		}
	}

	for _, sub := range props.GetSubtitles() {
		if subPriority, ok := PreferredSubtitles[sub]; ok {
			priority += subPriority
			preferredProperties["sub/"+sub] = subPriority
		}
	}

	if prefRes, ok := PreferredResolutions[props.GetResolution()]; ok {
		priority += prefRes
		preferredProperties["res/"+props.GetResolution().String()] = prefRes
	}

	return
}