feat: implement seed ratio check logic

db/db.go

@@ -275,6 +275,19 @@ func (c *Client) DeleteTorznab(id int) {
 	c.ent.Indexers.Delete().Where(indexers.ID(id)).Exec(context.TODO())
 }
 
+func (c *Client) GetIndexer(id int) (*TorznabInfo, error) {
+	res, err := c.ent.Indexers.Query().Where(indexers.ID(id)).First(context.TODO())
+	if err != nil {
+		return nil, err
+	}
+	var ss TorznabSetting
+	err = json.Unmarshal([]byte(res.Settings), &ss)
+	if err != nil {
+		
+		return nil, fmt.Errorf("unmarshal torznab %s error: %v", res.Name, err)
+	}
+	return &TorznabInfo{Indexers: res, TorznabSetting: ss}, nil
+}
 type TorznabInfo struct {
 	*ent.Indexers
 	TorznabSetting

pkg/doc.go

@@ -8,6 +8,7 @@ type Torrent interface {
 	Remove() error
 	Save() string
 	Exists() bool
+	SeedRatio() *float64
 }
 
 

pkg/transmission/transmission.go

@@ -130,6 +130,9 @@ func (t *Torrent) Exists() bool {
 }
 
 func (t *Torrent) Name() string {
+	if !t.Exists() {
+		return ""
+	}
 	return *t.getTorrent().Name
 }
 
@@ -155,6 +158,10 @@ func (t *Torrent) Stop() error {
 	return t.c.TorrentStopIDs(context.TODO(), []int64{t.ID})
 }
 
+func (t *Torrent) SeedRatio() *float64 {
+	return t.getTorrent().UploadRatio
+}
+
 func (t *Torrent) Start() error {
 	return t.c.TorrentStartIDs(context.TODO(), []int64{t.ID})
 }

server/scheduler.go

@@ -44,6 +44,24 @@ func (s *Server) checkTasks() {
 		}
 		log.Infof("task (%s) percentage done: %d%%", t.Name(), t.Progress())
 		if t.Progress() == 100 {
+			r := s.db.GetHistory(id)
+			if r.Status == history.StatusSuccess {
+				//task already success, check seed ratio
+				torrent := s.tasks[id]
+				ok, err := s.isSeedRatioLimitReached(r.IndexerID, torrent)
+				if err != nil {
+					log.Warnf("getting torrent seed ratio : %v", err)
+					ok = false
+				}
+				if ok {
+					log.Infof("torrent file seed ratio reached, remove: %v", torrent.Name())
+					torrent.Remove()
+					delete(s.tasks, id)
+				} else {
+					log.Infof("torrent file still sedding: %v", torrent.Name())
+				}
+				continue
+			}
 			log.Infof("task is done: %v", t.Name())
 			s.sendMsg(fmt.Sprintf(message.DownloadComplete, t.Name()))
 			go func() {
@@ -88,7 +106,7 @@ func (s *Server) moveCompletedTask(id int) (err1 error) {
 			if downloadclient.RemoveFailedDownloads {
 				log.Debugf("task failed, remove failed torrent and files related")
 				delete(s.tasks, r.ID)
-				torrent.Remove()		
+				torrent.Remove()
 			}
 		}
 	}()
@@ -114,7 +132,6 @@ func (s *Server) moveCompletedTask(id int) (err1 error) {
 		log.Errorf("create .plexmatch file error: %v", err)
 	}
 
-	
 	s.db.SetHistoryStatus(r.ID, history.StatusSuccess)
 	if r.EpisodeID != 0 {
 		s.db.SetEpisodeStatus(r.EpisodeID, episode.StatusDownloaded)
@@ -124,13 +141,17 @@ func (s *Server) moveCompletedTask(id int) (err1 error) {
 	s.sendMsg(fmt.Sprintf(message.ProcessingComplete, torrentName))
 
 	//判断是否需要删除本地文件
-	if downloadclient.RemoveCompletedDownloads {
+	ok, err := s.isSeedRatioLimitReached(r.IndexerID, torrent) 
+	if err != nil {
+		log.Warnf("getting torrent seed ratio %s: %v", torrent.Name(), err)
+		ok = false
+	}
+	if downloadclient.RemoveCompletedDownloads && ok {
 		log.Debugf("download complete,remove torrent and files related")
 		delete(s.tasks, r.ID)
 		torrent.Remove()
 	}
 
-
 	log.Infof("move downloaded files to target dir success, file: %v, target dir: %v", torrentName, r.TargetDir)
 	return nil
 }
@@ -330,3 +351,15 @@ func (s *Server) checkSeiesNewSeason(media *ent.Media) error {
 	}
 	return nil
 }
+
+func (s *Server) isSeedRatioLimitReached(indexId int, t pkg.Torrent) (bool, error) {
+	indexer, err := s.db.GetIndexer(indexId)
+	if err != nil {
+		return false, err
+	}
+	currentRatio := t.SeedRatio()
+	if currentRatio == nil {
+		return false, errors.New("seed ratio not exist")
+	}
+	return *currentRatio >= float64(indexer.SeedRatio), nil
+}

server/server.go

@@ -143,17 +143,17 @@ func (s *Server) MustTMDB() *tmdb.Client {
 }
 
 func (s *Server) reloadTasks() {
-	runningTasks := s.db.GetRunningHistories()
-	if len(runningTasks) == 0 {
-		return
-	}
-	for _, t := range runningTasks {
-		log.Infof("reloading task: %d %s", t.ID, t.SourceTitle)
+	allTasks := s.db.GetHistories()
+	for _, t := range allTasks {
 		torrent, err := transmission.ReloadTorrent(t.Saved)
 		if err != nil {
 			log.Errorf("relaod task %s failed: %v", t.SourceTitle, err)
 			continue
 		}
+		if !torrent.Exists() { //只要种子还存在于客户端中，就重新加载，有可能是还在做种中
+			continue
+		}
+		log.Infof("reloading task: %d %s", t.ID, t.SourceTitle)
 		s.tasks[t.ID] = &Task{Torrent: torrent}
 	}
 }