feat: support for torrent with multi episodes

2026-06-09 03:27:39 +08:00 · 2024-11-15 15:43:07 +08:00
parent c433ccaa0e
commit fbfee65a50
14 changed files with 649 additions and 91 deletions
--- a/server/core/resources.go
+++ b/server/core/resources.go
@@ -1,6 +1,7 @@
 package core

 import (
+	"bytes"
 	"fmt"
 	"polaris/ent"
 	"polaris/ent/episode"
@@ -14,7 +15,7 @@ import (
 	"github.com/pkg/errors"
 )

-func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum, episodeNum int) (*string, error) {
+func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum int, episodeNums ...int) (*string, error) {
 	trc, dlc, err := c.GetDownloadClient()
 	if err != nil {
 		return nil, errors.Wrap(err, "connect transmission")
@@ -32,19 +33,6 @@ func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum,
 		return nil, errors.New("no enough space")
 	}

-	var ep *ent.Episode
-	if episodeNum > 0 {
-		for _, e := range series.Episodes {
-			if e.SeasonNumber == seasonNum && e.EpisodeNumber == episodeNum {
-				ep = e
-			}
-		}
-		if ep == nil {
-			return nil, errors.Errorf("no episode of season %d episode %d", seasonNum, episodeNum)
-		}
-	} else { //season package download
-		ep = &ent.Episode{}
-	}
 	magnet, err := utils.Link2Magnet(r1.Link)
 	if err != nil {
 		return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", r1.Link, err)
@@ -58,9 +46,31 @@ func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum,

 	dir := fmt.Sprintf("%s/Season %02d/", series.TargetDir, seasonNum)

+	if len(episodeNums) > 0 {
+		for _, epNum := range episodeNums {
+			var ep *ent.Episode
+			for _, e := range series.Episodes {
+				if e.SeasonNumber == seasonNum && e.EpisodeNumber == epNum {
+					ep = e
+				}
+			}
+			if ep == nil {
+				return nil, errors.Errorf("no episode of season %d episode %d", seasonNum, epNum)
+			}
+
+			if ep.Status == episode.StatusMissing {
+				c.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)
+			}
+
+		}
+	} else { //season package download
+		c.db.SetSeasonAllEpisodeStatus(seriesId, seasonNum, episode.StatusDownloading)
+
+	}
 	history, err := c.db.SaveHistoryRecord(ent.History{
 		MediaID:     seriesId,
-		EpisodeID:   ep.ID,
+		EpisodeNums: episodeNums,
+		SeasonNum:   seasonNum,
 		SourceTitle: r1.Name,
 		TargetDir:   dir,
 		Status:      history.StatusRunning,
@@ -73,72 +83,112 @@ func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum,
 	if err != nil {
 		return nil, errors.Wrap(err, "save record")
 	}
-	if episodeNum > 0 {
-		if ep.Status == episode.StatusMissing {
-			c.db.SetEpisodeStatus(ep.ID, episode.StatusDownloading)
-		}
-	} else {
-		c.db.SetSeasonAllEpisodeStatus(seriesId, seasonNum, episode.StatusDownloading)
-	}

 	c.tasks[history.ID] = &Task{Torrent: torrent}
-	c.sendMsg(fmt.Sprintf(message.BeginDownload, r1.Name))
+	name := r1.Name
+
+	if len(episodeNums) > 0 {
+		buff := &bytes.Buffer{}
+		for i, ep := range episodeNums {
+			if i != 0 {
+				buff.WriteString(",")
+
+			}
+			buff.WriteString(fmt.Sprint(ep))
+		}
+		name = fmt.Sprintf("第%s集 (%s)", buff.String(), name)
+	} else {
+		name = fmt.Sprintf("全集 (%s)", name)
+	}
+
+	c.sendMsg(fmt.Sprintf(message.BeginDownload, name))

 	log.Infof("success add %s to download task", r1.Name)
 	return &r1.Name, nil
-
 }
-func (c *Client) SearchAndDownload(seriesId, seasonNum, episodeNum int) (*string, error) {
-	var episodes []int
-	if episodeNum > 0 {
-		episodes = append(episodes, episodeNum)
+
+/*
+tmdb 校验获取的资源名，如果用资源名在tmdb搜索出来的结果能匹配上想要的资源，则认为资源有效，否则无效
+解决名称过于简单的影视会匹配过多资源的问题， 例如：梦魇绝镇 FROM
+*/
+func (c *Client) checkBtReourceWithTmdb(r *torznab.Result, seriesId int) bool {
+	m := metadata.ParseTv(r.Name)
+	se, err := c.MustTMDB().SearchMedia(m.NameEn, "", 1)
+	if err != nil {
+		log.Warnf("tmdb search error, consider this torrent ok: ", err)
+		return true
+	} else {
+		if len(se.Results) == 0 {
+			log.Debugf("tmdb search no result, consider this torrent ok: %s", r.Name) //because tv name parse is not accurate
+			return true
+		}
+		series := c.db.GetMediaDetails(seriesId)
+
+		se0 := se.Results[0]
+		if se0.ID != int64(series.TmdbID) {
+			log.Warnf("bt reosurce name not match tmdb id: %s", r.Name)
+			return false
+		} else { //resource tmdb id match
+			return true
+		}
 	}
+}
+
+func (c *Client) SearchAndDownload(seriesId, seasonNum int, episodeNums ...int) ([]string, error) {
+
 	res, err := SearchTvSeries(c.db, &SearchParam{
 		MediaId:         seriesId,
 		SeasonNum:       seasonNum,
-		Episodes:        episodes,
+		Episodes:        episodeNums,
 		CheckFileSize:   true,
 		CheckResolution: true,
 	})
 	if err != nil {
 		return nil, err
 	}
-	/*
-		tmdb 校验获取的资源名，如果用资源名在tmdb搜索出来的结果能匹配上想要的资源，则认为资源有效，否则无效
-		解决名称过于简单的影视会匹配过多资源的问题， 例如：梦魇绝镇 FROM
-	*/
-	var r1 *torznab.Result
-	for _, r := range res { //
+	wanted := make(map[int]bool, len(episodeNums))
+	for _, ep := range episodeNums {
+		wanted[ep] = true
+	}
+	var torrentNames []string
+lo:
+	for _, r := range res {
+		if !c.checkBtReourceWithTmdb(&r, seriesId) {
+			continue
+		}
 		m := metadata.ParseTv(r.Name)
-		se, err := c.MustTMDB().SearchMedia(m.NameEn, "", 1)
-		if err != nil {
-			log.Warnf("tmdb search error, consider this torrent ok: ", err)
-			r1 = &r
-			break
-		} else {
-			if len(se.Results) == 0 {
-				log.Debugf("tmdb search no result, consider this torrent ok: %s", r.Name) //because tv name parse is not accurate
-				r1 = &r
-				break
+		m.ParseExtraDescription(r.Description)
+		if len(episodeNums) == 0 { //want season pack
+			if m.IsSeasonPack {
+				name, err := c.DownloadEpisodeTorrent(r, seriesId, seasonNum)
+				if err != nil {
+					return nil, err
+				}
+				torrentNames = append(torrentNames, *name)
 			}
-			series := c.db.GetMediaDetails(seriesId)
+		} else {
+			torrentEpisodes := make([]int, 0)
+			for i := m.StartEpisode; i <= m.EndEpisode; i++ {
+				if !wanted[i] { //torrent has episode not wanted
+					continue lo
+				}
+				torrentEpisodes = append(torrentEpisodes, i)
+			}
+			name, err := c.DownloadEpisodeTorrent(r, seriesId, seasonNum, torrentEpisodes...)
+			if err != nil {
+				return nil, err
+			}
+			torrentNames = append(torrentNames, *name)

-			se0 := se.Results[0]
-			if se0.ID != int64(series.TmdbID) {
-				log.Warnf("bt reosurce name not match tmdb id: %s", r.Name)
-				continue
-			} else { //resource tmdb id match
-				r1 = &r
+			for _, num := range torrentEpisodes {
+				delete(wanted, num) //delete downloaded episode from wanted
 			}
 		}
 	}
-	if r1 != nil {
-		log.Infof("found resource to download: %+v", r1)
-		return c.DownloadEpisodeTorrent(*r1, seriesId, seasonNum, episodeNum)
-
-	} else {
-		return nil, errors.Errorf("no resource")
+	if len(wanted) > 0 {
+		log.Warnf("still wanted but not downloaded episodes: %v", wanted)
 	}
+	return torrentNames, nil
 }

 func (c *Client) DownloadMovie(m *ent.Media, link, name string, size int, indexerID int) (*string, error) {