feat: blacklist api

Author: simon-ding

db/db.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"os"
 	"polaris/ent"
+	"polaris/ent/blacklist"
 	"polaris/ent/downloadclients"
 	"polaris/ent/episode"
 	"polaris/ent/history"
@@ -742,3 +743,17 @@ func (c *client) GetTmdbApiKey() string {
 	}
 	return k
 }
+
+
+func (c *client) AddTorrent2Blacklist(hash, name string, mediaId int) error  {
+	count := c.ent.Blacklist.Query().Where(blacklist.TorrentHash(hash), blacklist.MediaID(mediaId)).CountX(context.TODO())
+	if count > 0 { //already exist
+		log.Infof("torrent %s already in blacklist", hash)
+		return nil
+	}
+	return c.ent.Blacklist.Create().SetType(blacklist.TypeTorrent).SetTorrentHash(hash).SetTorrentName(name).SetMediaID(mediaId).Exec(context.Background())
+}
+
+func (c *client) GetTorrentBlacklist() (ent.Blacklists, error) {
+	return c.ent.Blacklist.Query().Where(blacklist.TypeEQ(blacklist.TypeTorrent)).All(context.Background())
+}

db/interface.go

@@ -39,6 +39,9 @@ type Database interface {
 	AddNotificationClient(name, service string, setting string, enabled bool) error
 	DeleteNotificationClient(id int) error
 	GetNotificationClient(id int) (*NotificationClient, error)
+
+	AddTorrent2Blacklist(hash, name string, mediaId int) error
+	GetTorrentBlacklist() (ent.Blacklists, error)
 }
 
 type Settings interface {

engine/client.go

@@ -51,6 +51,10 @@ func (c *Engine) Init() {
 	go c.checkW500PosterOnStartup()
 }
 
+
+func (c *Engine) GetTask(id int) (*Task, bool) {
+	return c.tasks.Load(id)
+}
 func (c *Engine) reloadUsingBuildinDownloader(h *ent.History) error{
 	cl, err := buildin.NewDownloader(c.db.GetDownloadDir())
 	if err != nil {