downloader/torrent_parse.go

package main

import (
	"errors"
	"fmt"

	"git.milar.in/milarin/adverr"
	"git.milar.in/nyaanime/logic"
	"git.milar.in/nyaanime/model"
	"git.milar.in/nyaanime/parsers"
)

func ParseTorrent(torrent *model.Torrent) (*model.ParsedTorrent, error) {
	for _, parser := range parsers.Parsers {
		parsedTorrent, ok := parser.TorrentParser(&parser, torrent)
		if ok {
			anime, err := logic.SearchAnimeByTitle(parsedTorrent.OriginalAnimeTitle)
			if err != nil {
				return parsedTorrent, ErrTorrentParseFailed.Wrap(err, torrent.ID, parser.String())
			}

			parsedTorrent.Anime = anime
			return parsedTorrent, nil
		}
	}

	return nil, ErrNoSuitableParser.New(torrent.ID)
}

func ParseTorrentsByAnimeEp(torrents []model.Torrent) map[model.AnimeEpisode][]*model.ParsedTorrent {
	torrentsByAnimeEp := map[model.AnimeEpisode][]*model.ParsedTorrent{}

	for _, torrent := range torrents {
		torrent := torrent
		parsedTorrent, err := ParseTorrent(&torrent)
		if err != nil {
			if errors.Is(err, ErrAnimeNotFound) {
				fmt.Printf("anime not found: \"%s\" (torrent ID: %s)\n", parsedTorrent.OriginalAnimeTitle, torrent.ID)
			} else if !errors.Is(err, ErrNoSuitableParser) {
				adverr.Println(err)
			}
			continue
		}

		animeEpisode := model.AnimeEpisode{
			Anime:   parsedTorrent.Anime,
			Episode: parsedTorrent.Episode,
		}

		torrentsByAnimeEp[animeEpisode] = append(torrentsByAnimeEp[animeEpisode], parsedTorrent)
	}

	return torrentsByAnimeEp
}