Add support for a file with guild bans

Author: riking

config.go

@@ -76,6 +76,10 @@ type Config struct {
 	DonorBacklogLimit  int `yaml:"backlog_limit_donor"`
 }
 
+type BansFile struct {
+	Guilds []string `yaml:"guilds"`
+}
+
 type ManagedChannelMarshal struct {
 	ID      string `yaml:"id"`
 	GuildID string `yaml:"guild_id"`

oauth.go

@@ -57,9 +57,27 @@ func (b *Bot) HTTPOAuthCallback(w http.ResponseWriter, r *http.Request) {
 		return
 	} else if err != nil {
 		fmt.Printf("%T %v", err, err)
-		http.Error(w, "bad token", http.StatusUnprocessableEntity)
+		http.Error(w, "An error occured and the bot could not join the server.", http.StatusUnprocessableEntity)
 		return
 	}
+	if guildInfo, ok := t.Extra("guild").(map[string]interface{}); ok {
+		if guildID, ok := guildInfo["id"].(string); ok {
+			if banned, err := b.storage.IsBanned(guildID); banned {
+				b.s.GuildLeave(guildID)
+				http.Error(w, "AutoDelete is not available on this server.", http.StatusForbidden)
+				fmt.Printf("[INFO] join attempt for banned server %s\n", guildID)
+				return
+			} else if err != nil {
+				fmt.Printf("[ERR] Could not check banlist: %T %v", err, err)
+				http.Error(w, "An error occured and the bot may not have joined the server.", http.StatusUnprocessableEntity)
+				return
+			}
+		} else {
+			fmt.Printf("[ERR] Unexpected type for guild.id: got %T\n", guildInfo["id"])
+		}
+	} else {
+		fmt.Printf("[ERR] Unexpected type for guild: got %T\n", t.Extra("guild"))
+	}
 
 	fmt.Println(t)
 	w.WriteHeader(200)

storage.go

@@ -17,6 +17,9 @@ type Storage interface {
 	GetChannel(id string) (ManagedChannelMarshal, error)
 	SaveChannel(conf ManagedChannelMarshal) error
 	DeleteChannel(id string) error
+
+	IsBanned(guildID string) (bool, error)
+	AddBan(guildID string) (error)
 }
 
 /******************
@@ -29,6 +32,7 @@ type DiskStorage struct {
 
 const pathChannelConfDir = "./data"
 const pathChannelConfig = "./data/%s.yml"
+const pathBanList = "./data/bans.yml"
 
 func (s *DiskStorage) ListChannels() ([]string, error) {
 	files, err := ioutil.ReadDir(pathChannelConfDir)
@@ -41,6 +45,9 @@ func (s *DiskStorage) ListChannels() ([]string, error) {
 		if !strings.HasSuffix(n, ".yml") {
 			continue
 		}
+		if strings.HasPrefix(n, "bans.yml") {
+			continue
+		}
 		chID := strings.TrimSuffix(n, ".yml")
 		channelIDs = append(channelIDs, chID)
 	}
@@ -96,3 +103,29 @@ func (s *DiskStorage) DeleteChannel(id string) error {
 	}
 	return nil
 }
+
+func (s *DiskStorage) IsBanned(guildID string) (bool, error) {
+	by, err := ioutil.ReadFile(pathBanList)
+	if os.IsNotExist(err) {
+		return false, nil
+	} else if err != nil {
+		return false, err
+	}
+
+	var conf BansFile
+	err = yaml.Unmarshal(by, &conf)
+	if err != nil {
+		return false, err
+	}
+
+	for _, v := range conf.Guilds {
+		if v == guildID {
+			return true, nil
+		}
+	}
+	return false, nil
+}
+
+func (s *DiskStorage) AddBan(guildID string) (error) {
+	return fmt.Errorf("unimplemented!")
+}