plugin_web.go

package commands

import (
	"github.com/jonas747/discordgo"
	"github.com/jonas747/yagpdb/common"
	"github.com/jonas747/yagpdb/web"
	"goji.io"
	"goji.io/pat"
	"html/template"
	"net/http"
	"strings"
)

func (p *Plugin) InitWeb() {
	web.Templates = template.Must(web.Templates.Parse(FSMustString(false, "/assets/commands.html")))

	subMux := goji.SubMux()
	web.CPMux.Handle(pat.New("/commands/settings"), subMux)
	web.CPMux.Handle(pat.New("/commands/settings/*"), subMux)

	subMux.Use(web.RequireGuildChannelsMiddleware)
	subMux.Use(web.RequireFullGuildMW)

	subMux.Handle(pat.Get(""), web.RenderHandler(HandleCommands, "cp_commands"))
	subMux.Handle(pat.Get("/"), web.RenderHandler(HandleCommands, "cp_commands"))
	subMux.Handle(pat.Post("/"), web.RenderHandler(HandlePostCommands, "cp_commands"))
}

// Servers the command page with current config
func HandleCommands(w http.ResponseWriter, r *http.Request) interface{} {
	ctx := r.Context()
	client, activeGuild, templateData := web.GetBaseCPContextData(ctx)
	channels := ctx.Value(common.ContextKeyGuildChannels).([]*discordgo.Channel)
	templateData["CommandConfig"] = GetConfig(client, activeGuild.ID, channels)
	return templateData
}

// Handles the updating of global and per channel command settings
func HandlePostCommands(w http.ResponseWriter, r *http.Request) interface{} {
	ctx := r.Context()
	client, activeGuild, templateData := web.GetBaseCPContextData(ctx)
	templateData["VisibleURL"] = "/manage/" + discordgo.StrID(activeGuild.ID) + "/commands/settings"

	newPrefix := strings.TrimSpace(r.FormValue("prefix"))
	if len(newPrefix) > 100 {
		return templateData.AddAlerts(web.ErrorAlert("Command prefix is too long (max 100)"))
	}

	err := client.Cmd("SET", "command_prefix:"+discordgo.StrID(activeGuild.ID), newPrefix).Err
	web.CheckErr(templateData, err, "Failed saving prefix", web.CtxLogger(r.Context()).Error)

	channels := ctx.Value(common.ContextKeyGuildChannels).([]*discordgo.Channel)

	config := GetConfig(client, activeGuild.ID, channels)

	// Update all the overrides
	for _, channel := range channels {
		// Find the override
		var override *ChannelOverride
		for _, r := range config.ChannelOverrides {
			if r.Channel == discordgo.StrID(channel.ID) {
				override = r
				break
			}
		}

		// A new channel was created in the meantime maybe?
		if override == nil {
			continue
		}

		// Update all the command settings for the override
		for _, overrideCmd := range override.Settings {
			overrideCmd.CommandEnabled = r.FormValue(discordgo.StrID(channel.ID)+"_enabled_"+overrideCmd.Cmd) == "on"
			overrideCmd.AutoDelete = r.FormValue(discordgo.StrID(channel.ID)+"_autodelete_"+overrideCmd.Cmd) == "on"
			overrideCmd.RequiredRole = r.FormValue(discordgo.StrID(channel.ID) + "_required_role_" + overrideCmd.Cmd)
		}

		override.OverrideEnabled = r.FormValue(discordgo.StrID(channel.ID)+"_override_enabled") == "on"
	}

	// Update the global settings
	for _, cmd := range config.Global {
		// Check for custom switch
		if cmd.Info.Key == "" {
			cmd.CommandEnabled = r.FormValue("global_enabled_"+cmd.Cmd) == "on"
		}
		cmd.RequiredRole = r.FormValue("global_required_role_" + cmd.Cmd)
		cmd.AutoDelete = r.FormValue("global_autodelete_"+cmd.Cmd) == "on"
	}

	err = common.SetRedisJson(client, "commands_settings:"+discordgo.StrID(activeGuild.ID), config)
	if web.CheckErr(templateData, err, "Failed saving item :'(", web.CtxLogger(ctx).Error) {
		return templateData
	}

	user := ctx.Value(common.ContextKeyUser).(*discordgo.User)
	go common.AddCPLogEntry(user, activeGuild.ID, "Updated command settings")

	templateData["CommandConfig"] = GetConfig(client, activeGuild.ID, channels)

	return templateData
}