Add blacklist-fanart feature

Author: JustHumanz

config.toml.template

@@ -7,6 +7,9 @@ DonationLink = "#DonationLink"
 Domain = "humanz.tech"
 LowResources = true 
 
+[BannFanartAccount]
+Twitter = ["VtuberAnlysBot"]
+
 YoutubeToken = [
   "#GoogleAPI",
   "#GoogleAPI"

pkg/config/config.go

@@ -151,19 +151,24 @@ const (
 
 //ConfigFile config file struct for config.toml
 type ConfigFile struct {
-	Discord        string `toml:"Discord"`
-	BiliSess       string `toml:"BiliSess"`
-	PixivSession   string `toml:"PixivSess"`
-	SauceAPI       string `toml:"SauceAPI"`
-	InviteLog      string `toml:"InviteLog"`
-	PilotReporting string `toml:"PilotReporting"`
-	MultiTOR       string `toml:"Multitor"`
-	DonationLink   string `toml:"DonationLink"`
-	TopGG          string `toml:"TOPGG"`
-	Domain         string `toml:"Domain"`
-	PrometheusURL  string `toml:"PrometheusURL"`
-	Metric         bool   `toml:"Metric"`
-	Twitch         struct {
+	Discord           string `toml:"Discord"`
+	BiliSess          string `toml:"BiliSess"`
+	PixivSession      string `toml:"PixivSess"`
+	SauceAPI          string `toml:"SauceAPI"`
+	InviteLog         string `toml:"InviteLog"`
+	PilotReporting    string `toml:"PilotReporting"`
+	MultiTOR          string `toml:"Multitor"`
+	DonationLink      string `toml:"DonationLink"`
+	TopGG             string `toml:"TOPGG"`
+	Domain            string `toml:"Domain"`
+	PrometheusURL     string `toml:"PrometheusURL"`
+	Metric            bool   `toml:"Metric"`
+	BannFanartAccount struct {
+		Twitter  []string `toml:"Twitter"`
+		BiliBili []string `toml:"BiliBili"`
+		Pixiv    []string `toml:"Pixiv"`
+	} `toml:"BannFanartAccount"`
+	Twitch struct {
 		ClientID     string `toml:"ClientID"`
 		ClientSecret string `toml:"ClientSecret"`
 	} `toml:"Twitch"`

pkg/database/fanart.go

@@ -555,6 +555,12 @@ func (Member Member) ScrapTwitterFanart(Scraper *twitterscraper.Scraper, Lewd bo
 			continue
 		}
 
+		for _, Ban := range config.GoSimpConf.BannFanartAccount.Twitter {
+			if strings.EqualFold(Ban, tweet.Username) {
+				continue
+			}
+		}
+
 		for _, TweetHashtag := range tweet.Hashtags {
 			if (strings.EqualFold("#"+TweetHashtag, Member.TwitterHashtag) || strings.EqualFold("#"+TweetHashtag, Member.TwitterLewd)) && !tweet.IsQuoted && !tweet.IsReply && len(tweet.Photos) > 0 {
 				TweetArt := DataFanart{
@@ -590,6 +596,8 @@ func (Member Member) ScrapTwitterFanart(Scraper *twitterscraper.Scraper, Lewd bo
 				if New {
 					FanartList = append(FanartList, TweetArt)
 				}
+
+				break
 			}
 		}
 	}

service/fanart/bilibili/bilibili.go

@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"flag"
 	"net/url"
+	"strings"
 	"sync"
 	"time"
 
@@ -216,6 +217,13 @@ func (k *checkBlJob) Run() {
 								"Vtuber": Member.Name,
 							}).Error(err)
 						}
+
+						for _, Ban := range config.GoSimpConf.BannFanartAccount.BiliBili {
+							if strings.EqualFold(Ban, v.Desc.UserProfile.Info.Uname) {
+								continue
+							}
+						}
+
 						if STB.Item.Pictures != nil && v.Desc.Type == 2 { //type 2 is picture post (prob,heheheh)
 							for _, pic := range STB.Item.Pictures {
 								img = append(img, pic.ImgSrc)

service/fanart/pixiv/pixiv.go

@@ -212,6 +212,12 @@ func Pixiv(p string, FixFanArt *database.DataFanart, l bool) error {
 				TextFix string
 			)
 
+			for _, Ban := range config.GoSimpConf.BannFanartAccount.Pixiv {
+				if strings.EqualFold(Ban, v2["userName"].(string)) {
+					continue
+				}
+			}
+
 			if IsVtuber {
 				if v2["xRestrict"].(float64) == 0 && !l {
 					illusbyte, err := network.Curl(config.PixivIllustsEnd+v2["id"].(string), nil)