Support Whitelist

config.yml

@@ -15,10 +15,11 @@ db:
 guild:
   prefix: p.
   lang: japanese
-domain_blacklist:
-  - "google.com"
-  - "letmegooglethat.com"
-  - "discord.com"
+domain:
+  type: white
+  list:
+    - "twitter.com"
+    - "youtube.com"
 user_blacklist:
   - "0000000000000000000" # Add here to ignore user at all
 channel_blacklist:

lib/config/config.go

@@ -25,18 +25,23 @@ type Config struct {
 		Parameter   string
 		Tableprefix string
 	}
-	Guild           Guild
-	DomainBlacklist []string `yaml:"domain_blacklist"`
-	UserBlacklist   []string `yaml:"user_blacklist"`
+	Guild  Guild
+	Domain struct {
+		Type     string
+		YamlList []string `yaml:"list"`
+	}
+	UserBlacklist    []string `yaml:"user_blacklist"`
 	ChannelBlacklist []string `yaml:"channel_blacklist"`
-	LogPeriod       int64    `yaml:"log_period"`
+	LogPeriod        int64    `yaml:"log_period"`
 }
 
 type Guild struct {
 	Prefix string
 	Lang   string
 }
 
+var ListMap map[string]bool
+
 const configFile = "./config.yml"
 
 var CurrentConfig Config
@@ -61,8 +66,17 @@ func init() {
 	if CurrentConfig.Db.Tableprefix == "" {
 		log.Fatal("Tableprefix is empty")
 	}
+	if CurrentConfig.Domain.Type != "white" && CurrentConfig.Domain.Type != "black" {
+		log.Fatal("Domain type is invalid")
+	}
 	if CurrentConfig.LogPeriod == 0 {
-		log.Print("LogPeriod is empty, setting 31536000")
+		CurrentConfig.LogPeriod = 2592000
+		log.Print("LogPeriod is empty, setting to 2592000")
+	}
+
+	ListMap = make(map[string]bool)
+	for _, item := range CurrentConfig.Domain.YamlList {
+		ListMap[item] = true
 	}
 
 	loadLang()

lib/handler/url_check.go

@@ -64,8 +64,14 @@ url_loop:
 		if err != nil {
 			log.Panic("Invalid URL: ", rawUrl)
 		}
-		for _, v := range config.CurrentConfig.DomainBlacklist {
-			if strings.Contains(Url.Host, v) {
+		list := config.ListMap
+		switch config.CurrentConfig.Domain.Type {
+		case "black":
+			if list[Url.Host] {
+				continue url_loop
+			}
+		case "white":
+			if !list[Url.Host] {
 				continue url_loop
 			}
 		}