Main/Features: Add configuration management commands.

cmd/bot_config.json

@@ -4,10 +4,9 @@
     "guild_id": "506629366659153951",
 
     "status_text": ".help",
-    "welcome_text": "Hi! Welcome to the Hak5 Discord Server. Please be sure to check out the #rules channel and say hello in #general!",
-
+    "welcome_text": "Hi! Welcome to the Hak5 Discord Server. Please be sure to check out the #rules channel and say hello in #general! Remember to use the correct channels and grab a nickname color from #bots!",
     "admin_roles": ["Admin", "Moderator"],
-    "join_role_ids": ["697929841638637778"],
+    "join_role_ids": [],
 
     "command_restrictions": [
         { "command": "color", "mode": "white", "channels": ["698567678025138277", "698519835532984470"] },

cmd/main.go

@@ -55,6 +55,7 @@ func main() {
 	if err != nil {
 		log.Fatal("[!] Error: " + err.Error())
 	}
+	Config.ConfigPath = ConfigPath
 
 	// Instantiate Bot
 	bot, err := discordgo.New("Bot " + Token)

features/handlers.go

@@ -45,6 +45,10 @@ func (f *Features) RegisterHandlers() {
 	f.RegisterCommand("purge", f.handlePurgeChannel)
 	f.RegisterCommand("kick", f.handleKickUser)
 	f.RegisterCommand("ban", f.handleBanUser)
+	f.RegisterCommand("setconfig", f.handleSetConfig)
+	f.RegisterCommand("getconfig", f.handleGetConfig)
+	f.RegisterCommand("saveconfig", f.handleSaveConfig)
+	f.RegisterCommand("reloadconfig", f.handleReloadConfig)
 }
 
 func (f *Features) ProcessCommand(s *discordgo.Session, m *discordgo.MessageCreate) error {
@@ -186,9 +190,8 @@ func (f *Features) ProcessMessage(s *discordgo.Session, m interface{}) {
 		// Handle new member (Welcome message, etc)
 		err := f.ProcessUserJoin(s, m.(*discordgo.GuildMemberAdd))
 		if err != nil {
-			log.Println("[!] Error: " + err.Error())
+			log.Println("[!] Error (Guild Member Joined): " + err.Error())
 		}
 		break
 	}
-
 }

features/misc.go

@@ -1,15 +1,195 @@
 package features
 
 import (
+	"encoding/json"
 	"errors"
 	"fmt"
 	"github.com/bwmarrin/discord.go"
 	"github.com/foxtrot/scuzzy/models"
+	"io/ioutil"
+	"os"
+	"reflect"
 	"strconv"
 	"strings"
 	"time"
 )
 
+func (f *Features) handleSetConfig(s *discordgo.Session, m *discordgo.MessageCreate) error {
+	if !f.Auth.CheckAdminRole(m.Member) {
+		return errors.New("You do not have permissions to use that command.")
+	}
+
+	configArgs := strings.Split(m.Content, " ")
+
+	if len(configArgs) != 3 {
+		return errors.New("Invalid arguments supplied. Usage: " + f.Config.CommandKey + "setconfig <key> <value>")
+	}
+
+	configKey := configArgs[1]
+	configVal := configArgs[2]
+
+	rt := reflect.TypeOf(f.Config)
+	for i := 0; i < rt.NumField(); i++ {
+		x := rt.Field(i)
+		tagVal := strings.Split(x.Tag.Get("json"), ",")[0]
+		tagName := x.Name
+
+		if tagVal == configKey {
+			prop := reflect.ValueOf(&f.Config).Elem().FieldByName(tagName)
+
+			switch prop.Interface().(type) {
+			case string:
+				prop.SetString(configVal)
+				break
+			case int:
+				intVal, err := strconv.ParseInt(configVal, 10, 64)
+				if err != nil {
+					return err
+				}
+				prop.SetInt(intVal)
+				break
+			case float64:
+				floatVal, err := strconv.ParseFloat(configVal, 64)
+				if err != nil {
+					return err
+				}
+				prop.SetFloat(floatVal)
+				break
+			case bool:
+				boolVal, err := strconv.ParseBool(configVal)
+				if err != nil {
+					return err
+				}
+				prop.SetBool(boolVal)
+				break
+			default:
+				return errors.New("Unsupported key value type")
+			}
+
+			msgE := f.CreateDefinedEmbed("Set Configuration", "Successfully set property '"+configKey+"'!", "success", m.Author)
+			_, err := s.ChannelMessageSendEmbed(m.ChannelID, msgE)
+			if err != nil {
+				return err
+			}
+
+			return nil
+		}
+	}
+
+	return errors.New("Unknown key specified")
+}
+
+func (f *Features) handleGetConfig(s *discordgo.Session, m *discordgo.MessageCreate) error {
+	//TODO: Handle printing of slices (check the Type, loop accordingly)
+
+	if !f.Auth.CheckAdminRole(m.Member) {
+		return errors.New("You do not have permissions to use that command.")
+	}
+
+	configArgs := strings.Split(m.Content, " ")
+	configKey := "all"
+	if len(configArgs) == 2 {
+		configKey = configArgs[1]
+	}
+
+	msg := ""
+
+	rt := reflect.TypeOf(f.Config)
+	for i := 0; i < rt.NumField(); i++ {
+		x := rt.Field(i)
+		tagVal := strings.Split(x.Tag.Get("json"), ",")[0]
+		tagName := x.Name
+		prop := reflect.ValueOf(&f.Config).Elem().FieldByName(tagName)
+
+		if configKey == "all" {
+			switch prop.Interface().(type) {
+			case string:
+				if len(prop.String()) > 256 {
+					// Truncate large values.
+					msg += "`" + tagName + "` - " + "Truncated...\n"
+				} else {
+					msg += "`" + tagName + "` - `" + prop.String() + "`\n"
+				}
+				break
+			default:
+				// Ignore non strings for now...
+				msg += "`" + tagName + "` - Skipped Value\n"
+				continue
+			}
+		} else {
+			if tagVal == configKey {
+				switch prop.Interface().(type) {
+				case string:
+					msg += "`" + tagName + "` - `" + prop.String() + "`\n"
+				default:
+					// Ignore non strings for now...
+					msg += "`" + tagName + "` - Skipped Value\n"
+				}
+
+				eMsg := f.CreateDefinedEmbed("Get Configuration", msg, "success", m.Author)
+				_, err := s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
+				if err != nil {
+					return err
+				}
+
+				return nil
+			}
+		}
+	}
+
+	if msg == "" {
+		return errors.New("Unknown key specified")
+	}
+
+	eMsg := f.CreateDefinedEmbed("Get Configuration", msg, "success", m.Author)
+	_, err := s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (f *Features) handleReloadConfig(s *discordgo.Session, m *discordgo.MessageCreate) error {
+	fBuf, err := ioutil.ReadFile(f.Config.ConfigPath)
+	if err != nil {
+		return err
+	}
+
+	err = json.Unmarshal(fBuf, &f.Config)
+	if err != nil {
+		return err
+	}
+
+	eMsg := f.CreateDefinedEmbed("Reload Configuration", "Successfully reloaded configuration from disk", "success", m.Author)
+	_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (f *Features) handleSaveConfig(s *discordgo.Session, m *discordgo.MessageCreate) error {
+	j, err := json.Marshal(f.Config)
+	if err != nil {
+		return err
+	}
+
+	err = ioutil.WriteFile(f.Config.ConfigPath, j, os.ModePerm)
+	if err != nil {
+		return err
+	}
+
+	eMsg := f.CreateDefinedEmbed("Save Configuration", "Saved runtime configuration successfully", "success", m.Author)
+	_, err = s.ChannelMessageSendEmbed(m.ChannelID, eMsg)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
 func (f *Features) handleCat(s *discordgo.Session, m *discordgo.MessageCreate) error {
 	_, err := s.ChannelMessageSend(m.ChannelID, "https://giphy.com/gifs/cat-cute-no-rCxogJBzaeZuU")
 	if err != nil {
@@ -124,6 +304,8 @@ func (f *Features) handleHelp(s *discordgo.Session, m *discordgo.MessageCreate)
 		desc += "`purge` - Purge channel messages\n"
 		desc += "`kick` - Kick a specified user\n"
 		desc += "`ban` - Ban a specified user\n"
+		desc += "`setconfig` - Manage the runtime configuration\n"
+		desc += "`getconfig` - View the runtime configuration\n"
 	}
 
 	desc += "\n\nAll commands are prefixed with `" + f.Config.CommandKey + "`\n"

models/config.go

@@ -1,6 +1,6 @@
 package models
 
-import "github.com/bwmarrin/discord.go"
+import discordgo "github.com/bwmarrin/discord.go"
 
 type ColorRole struct {
 	Name string `json:"color"`
@@ -29,7 +29,8 @@ type Configuration struct {
 
 	ColorRoles []ColorRole `json:"color_roles"`
 
-	Guild *discordgo.Guild `json:"reserved_guild"`
-
 	LoggingChannel string `json:"logging_channel"`
+
+	Guild      *discordgo.Guild
+	ConfigPath string
 }