downloader/show_debug_info.go
Timon Ringwald 9110cf4076 polishing
2022-08-25 23:25:48 +02:00

85 lines
2.6 KiB
Go

package main
import (
"fmt"
"strings"
"github.com/fatih/color"
"git.milar.in/milarin/slices"
"git.milar.in/milarin/tprint"
"git.milar.in/nyaanime/model"
)
var (
BoldText = color.New(color.Bold)
)
func ShowDebugInfo(
parsedTorrentsByAnimeEp, animeListTorrents, essentialTorrents map[model.AnimeEpisode][]*model.ParsedTorrent,
preferredTorrents map[model.AnimeEpisode]*TorrentPriority,
downloadingEpisodes map[model.AnimeEpisode]struct{},
inCollectionEpisodes map[model.AnimeEpisode]bool,
lowerPrioTorrents map[*model.ParsedTorrent]bool,
downloadedTorrents map[model.AnimeEpisode]*TorrentPriority) {
for animeEp, parsedTorrents := range parsedTorrentsByAnimeEp {
table := tprint.NewTable("id", "resolution", "languages", "subtitles", "seeders", "leechers", "downloads", "trusted", "group", "evaluation")
for _, torrent := range parsedTorrents {
eval := getEvaluation(
slices.Contains(essentialTorrents[animeEp], torrent),
preferredTorrents[animeEp] != nil && preferredTorrents[animeEp].ParsedTorrent == torrent,
lowerPrioTorrents[torrent],
downloadedTorrents[animeEp] != nil && downloadedTorrents[animeEp].ParsedTorrent == torrent,
)
table.AddRow(
torrent.Torrent.ID,
torrent.Resolution,
strings.Join(torrent.Languages, ", "),
strings.Join(torrent.Subtitles, ", "),
torrent.Torrent.Seeders,
torrent.Torrent.Leechers,
torrent.Torrent.Downloads,
torrent.Torrent.Trusted,
torrent.Parser.Identity,
eval,
)
}
var epState string
if _, onList := animeListTorrents[animeEp]; onList {
if _, downloading := downloadingEpisodes[animeEp]; downloading {
epState = color.BlueString("CURRENTLY DOWNLOADING")
} else if inCollectionEpisodes[animeEp] {
epState = color.GreenString("IN COLLECTION")
} else {
epState = color.YellowString("ON LIST")
}
} else {
epState = color.RedString("NOT ON LIST")
}
b := new(strings.Builder)
DebugAnimeEpisodePattern.Execute(b, AnimePathPatternData{Title: animeEp.Anime.Title, Episode: animeEp.Episode})
header := BoldText.Sprintf("%s (%s)", color.MagentaString(b.String()), epState)
fmt.Println(tprint.FormatHeaderTable(header, table))
}
}
func getEvaluation(essential, preferred, lowerPrio, downloaded bool) string {
if downloaded {
return color.GreenString("DOWNLOAD STARTED")
} else if lowerPrio {
return color.GreenString("collection preferred")
} else if preferred {
return color.BlueString("torrent preferred")
} else if essential {
return color.YellowString("torrent considered")
} else {
return color.RedString("torrent ignored")
}
}