feat: filter resources that is qiangban

server/core/scheduler.go

@@ -3,6 +3,7 @@ package core
 import (
 	"fmt"
 	"path/filepath"
+	"polaris/db"
 	"polaris/ent"
 	"polaris/ent/episode"
 	"polaris/ent/history"
@@ -276,11 +277,17 @@ func (c *Client) downloadMovieSingleEpisode(ep *ent.Episode) error {
 	if err != nil {
 		return errors.Wrap(err, "connect transmission")
 	}
+	qiangban := c.db.GetSetting(db.SettingAllowQiangban)
+	allowQiangban := false
+	if qiangban == "true" {
+		allowQiangban = false
+	}
 
 	res, err := SearchMovie(c.db, &SearchParam{
 		MediaId:         ep.MediaID,
 		CheckFileSize:   true,
 		CheckResolution: true,
+		FilterQiangban:  !allowQiangban,
 	})
 	if err != nil {
 

server/core/torrent.go

@@ -23,6 +23,7 @@ type SearchParam struct {
 	Episodes        []int //for tv
 	CheckResolution bool
 	CheckFileSize   bool
+	FilterQiangban  bool //for movie, 是否过滤枪版电影
 }
 
 func SearchTvSeries(db1 *db.Client, param *SearchParam) ([]torznab.Result, error) {
@@ -150,6 +151,9 @@ func SearchMovie(db1 *db.Client, param *SearchParam) ([]torznab.Result, error) {
 		if !torrentSizeOk(movieDetail, r.Size, param) {
 			continue
 		}
+		if param.FilterQiangban && meta.IsQingban { //过滤枪版电影
+			continue
+		}
 
 		ss := strings.Split(movieDetail.AirDate, "-")[0]
 		year, _ := strconv.Atoi(ss)

server/resources.go

@@ -2,6 +2,7 @@ package server
 
 import (
 	"fmt"
+	"polaris/db"
 	"polaris/ent/media"
 	"polaris/log"
 	"polaris/pkg/torznab"
@@ -77,8 +78,15 @@ func (s *Server) SearchAvailableTorrents(c *gin.Context) (interface{}, error) {
 		}
 	} else {
 		log.Info("search movie %d", in.ID)
+		qiangban := s.db.GetSetting(db.SettingAllowQiangban)
+		allowQiangban := false
+		if qiangban == "true" {
+			allowQiangban = true
+		}
+
 		res, err = core.SearchMovie(s.db, &core.SearchParam{
-			MediaId: in.ID,
+			MediaId:        in.ID,
+			FilterQiangban: !allowQiangban,
 		})
 		if err != nil {
 			if err.Error() == "no resource found" {

server/setting.go

@@ -19,6 +19,7 @@ type GeneralSettings struct {
 	LogLevel        string `json:"log_level"`
 	Proxy           string `json:"proxy"`
 	EnablePlexmatch bool   `json:"enable_plexmatch"`
+	AllowQiangban   bool   `json:"allow_qiangban"`
 }
 
 func (s *Server) SetSetting(c *gin.Context) (interface{}, error) {
@@ -54,6 +55,13 @@ func (s *Server) SetSetting(c *gin.Context) (interface{}, error) {
 	}
 
 	s.db.SetSetting(db.SettingProxy, in.Proxy)
+
+	if in.AllowQiangban {
+		s.db.SetSetting(db.SettingAllowQiangban, "true")
+	} else {
+		s.db.SetSetting(db.SettingAllowQiangban, "false")
+	}
+
 	return nil, nil
 }
 
@@ -62,12 +70,14 @@ func (s *Server) GetSetting(c *gin.Context) (interface{}, error) {
 	downloadDir := s.db.GetSetting(db.SettingDownloadDir)
 	logLevel := s.db.GetSetting(db.SettingLogLevel)
 	plexmatchEnabled := s.db.GetSetting(db.SettingPlexMatchEnabled)
+	allowQiangban := s.db.GetSetting(db.SettingAllowQiangban)
 	return &GeneralSettings{
 		TmdbApiKey:      tmdb,
 		DownloadDir:     downloadDir,
 		LogLevel:        logLevel,
 		Proxy:           s.db.GetSetting(db.SettingProxy),
 		EnablePlexmatch: plexmatchEnabled == "true",
+		AllowQiangban:   allowQiangban == "true",
 	}, nil
 }
 
@@ -218,4 +228,4 @@ func (s *Server) EditMediaMetadata(c *gin.Context) (interface{}, error) {
 		return nil, errors.Wrap(err, "save db")
 	}
 	return "success", nil
-}
+}