feat: support prowlarr connection

server/core/torrent.go

@@ -6,6 +6,7 @@ import (
 	"polaris/ent/media"
 	"polaris/log"
 	"polaris/pkg/metadata"
+	"polaris/pkg/prowlarr"
 	"polaris/pkg/torznab"
 	"slices"
 	"sort"
@@ -136,7 +137,7 @@ func torrentSizeOk(detail *db.MediaDetails, torrentSize int, param *SearchParam)
 			}
 		}
 	}
-	return torrentSize > defaultMinSize * multiplier
+	return torrentSize > defaultMinSize*multiplier
 }
 
 func seasonEpisodeCount(detail *db.MediaDetails, seasonNum int) int {
@@ -230,6 +231,17 @@ func searchWithTorznab(db *db.Client, queries ...string) []torznab.Result {
 
 	var res []torznab.Result
 	allTorznab := db.GetAllTorznabInfo()
+
+	p, err := db.GetProwlarrSetting()
+	if err == nil { //prowlarr exists
+		c := prowlarr.New(p.ApiKey, p.URL)
+		all, err := c.GetIndexers()
+		if err != nil {
+			log.Warnf("get prowlarr all indexer error: %v", err)
+		} else {
+			allTorznab = append(allTorznab, all...)
+		}
+	}
 	resChan := make(chan []torznab.Result)
 	var wg sync.WaitGroup
 

server/server.go

@@ -70,6 +70,8 @@ func (s *Server) Serve() error {
 		setting.POST("/parse/movie", HttpHandler(s.ParseMovie))
 		setting.POST("/monitoring", HttpHandler(s.ChangeEpisodeMonitoring))
 		setting.POST("/cron/trigger", HttpHandler(s.TriggerCronJob))
+		setting.GET("/prowlarr", HttpHandler(s.GetProwlarrSetting))
+		setting.POST("/prowlarr", HttpHandler(s.SaveProwlarrSetting))
 	}
 	activity := api.Group("/activity")
 	{

server/setting.go

@@ -8,6 +8,7 @@ import (
 	"polaris/ent"
 	"polaris/ent/downloadclients"
 	"polaris/log"
+	"polaris/pkg/prowlarr"
 	"polaris/pkg/qbittorrent"
 	"polaris/pkg/torznab"
 	"polaris/pkg/transmission"
@@ -303,3 +304,26 @@ func (s *Server) TriggerCronJob(c *gin.Context) (interface{}, error) {
 	}
 	return "success", nil
 }
+
+func (s *Server) GetProwlarrSetting(c *gin.Context) (interface{}, error) {
+	se, err :=s.db.GetProwlarrSetting()
+	if err != nil {
+		return &db.ProwlarrSetting{}, nil
+	}
+	return se, nil
+}
+func (s *Server) SaveProwlarrSetting(c *gin.Context) (interface{}, error) {
+	var in db.ProwlarrSetting
+	if err := c.ShouldBindJSON(&in); err != nil {
+		return nil, err
+	}
+	client := prowlarr.New(in.ApiKey, in.URL)
+	if _, err := client.GetIndexers(); err != nil {
+		return nil, errors.Wrap(err, "connect to prowlarr error")
+	}
+	err := s.db.SaveProwlarrSetting(&in)
+	if err != nil {
+		return nil, err
+	}
+	return "success", nil
+}