jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-06-09 11:39:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: blacklist api

Browse Source
This commit is contained in:
Simon Ding
2025-04-22 18:18:32 +08:00
parent 1a3807acc9
commit fdcf7f487c
16 changed files with 965 additions and 124 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
db/db.go
View File

@@ -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())
}