admin: refactor

This commit is contained in:
Chris Sexton 2021-02-01 06:32:41 -05:00 committed by Chris Sexton
parent 00dae33e8a
commit 8e4129ecb3
1 changed files with 240 additions and 210 deletions

View File

@ -30,16 +30,37 @@ type AdminPlugin struct {
quiet bool quiet bool
} }
// NewAdminPlugin creates a new AdminPlugin with the Plugin interface // New creates a new AdminPlugin with the Plugin interface
func New(b bot.Bot) *AdminPlugin { func New(b bot.Bot) *AdminPlugin {
p := &AdminPlugin{ p := &AdminPlugin{
bot: b, bot: b,
db: b.DB(), db: b.DB(),
cfg: b.Config(), cfg: b.Config(),
} }
b.Register(p, bot.Message, p.message)
b.RegisterRegex(p, bot.Message, comeBackRegex, p.comeBackCmd)
b.RegisterRegexCmd(p, bot.Message, shutupRegex, p.shutupCmd)
b.RegisterRegexCmd(p, bot.Message, addBlacklistRegex, p.isAdmin(p.addBlacklistCmd))
b.RegisterRegexCmd(p, bot.Message, rmBlacklistRegex, p.isAdmin(p.rmBlacklistCmd))
b.RegisterRegexCmd(p, bot.Message, addWhitelistRegex, p.isAdmin(p.addWhitelistCmd))
b.RegisterRegexCmd(p, bot.Message, rmWhitelistRegex, p.isAdmin(p.rmWhitelistCmd))
b.RegisterRegexCmd(p, bot.Message, allWhitelistRegex, p.isAdmin(p.allWhitelistCmd))
b.RegisterRegexCmd(p, bot.Message, allUnWhitelistRegex, p.isAdmin(p.allUnWhitelistCmd))
b.RegisterRegexCmd(p, bot.Message, getWhitelistRegex, p.isAdmin(p.getWhitelistCmd))
b.RegisterRegexCmd(p, bot.Message, getPluginsRegex, p.isAdmin(p.getPluginsCmd))
b.RegisterRegexCmd(p, bot.Message, rebootRegex, p.isAdmin(p.rebootCmd))
b.RegisterRegexCmd(p, bot.Message, passwordRegex, p.passwordCmd)
b.RegisterRegex(p, bot.Message, variableSetRegex, p.variableSetCmd)
b.RegisterRegex(p, bot.Message, variableUnSetRegex, p.variableUnSetCmd)
b.RegisterRegexCmd(p, bot.Message, unsetConfigRegex, p.isAdmin(p.unsetConfigCmd))
b.RegisterRegexCmd(p, bot.Message, setConfigRegex, p.isAdmin(p.setConfigCmd))
b.RegisterRegexCmd(p, bot.Message, pushConfigRegex, p.isAdmin(p.pushConfigCmd))
b.RegisterRegexCmd(p, bot.Message, setKeyConfigRegex, p.isAdmin(p.setKeyConfigCmd))
b.RegisterRegexCmd(p, bot.Message, getConfigRegex, p.isAdmin(p.getConfigCmd))
b.Register(p, bot.Help, p.help) b.Register(p, bot.Help, p.help)
p.registerWeb() p.registerWeb()
return p return p
} }
@ -51,270 +72,279 @@ var forbiddenKeys = map[string]bool{
"meme.memes": true, "meme.memes": true,
} }
var addBlacklist = regexp.MustCompile(`(?i)disable plugin (.*)`) var shutupRegex = regexp.MustCompile(`(?i)^shut up$`)
var rmBlacklist = regexp.MustCompile(`(?i)enable plugin (.*)`) var comeBackRegex = regexp.MustCompile(`(?i)^come back$`)
var addWhitelist = regexp.MustCompile(`(?i)^whitelist plugin (.*)`) var addBlacklistRegex = regexp.MustCompile(`(?i)disable plugin (?P<plugin>.*)$`)
var rmWhitelist = regexp.MustCompile(`(?i)^unwhitelist plugin (.*)`) var rmBlacklistRegex = regexp.MustCompile(`(?i)enable plugin (?P<plugin>.*)$`)
var allWhitelist = regexp.MustCompile(`(?i)^whitelist all`)
var allUnwhitelist = regexp.MustCompile(`(?i)^unwhitelist all`)
var getWhitelist = regexp.MustCompile(`(?i)^list whitelist`)
var getPlugins = regexp.MustCompile(`(?i)^list plugins`)
// Message responds to the bot hook on recieving messages. var addWhitelistRegex = regexp.MustCompile(`(?i)^whitelist plugin (?P<plugin>.*)$`)
// This function returns true if the plugin responds in a meaningful way to the users message. var rmWhitelistRegex = regexp.MustCompile(`(?i)^unwhitelist plugin (?P<plugin>.*)$`)
// Otherwise, the function returns false and the bot continues execution of other plugins. var allWhitelistRegex = regexp.MustCompile(`(?i)^whitelist all$`)
func (p *AdminPlugin) message(conn bot.Connector, k bot.Kind, message msg.Message, args ...interface{}) bool { var allUnWhitelistRegex = regexp.MustCompile(`(?i)^unwhitelist all$`)
body := message.Body var getWhitelistRegex = regexp.MustCompile(`(?i)^list whitelist$`)
var getPluginsRegex = regexp.MustCompile(`(?i)^list plugins$`)
if p.quiet && message.Body != "come back" { var rebootRegex = regexp.MustCompile(`^reboot$`)
return true var passwordRegex = regexp.MustCompile(`(?i)^password$`)
}
if len(body) > 0 && body[0] == '$' { var variableSetRegex = regexp.MustCompile(`(?i)^\$(?P<var>\S+)\s?=\s?(?P<value>\S+)$`)
return p.handleVariables(conn, message) var variableUnSetRegex = regexp.MustCompile(`(?i)^\$(?P<var>\S+)\s?!=\s?(?P<value>\S+)$`)
}
if !message.Command { var unsetConfigRegex = regexp.MustCompile(`(?i)^unset (?P<key>\S+)$`)
return false var setConfigRegex = regexp.MustCompile(`(?i)^set (?P<key>\S+) (?P<value>.*)$`)
} var pushConfigRegex = regexp.MustCompile(`(?i)^push (?P<key>\S+) (?P<value>.*)$`)
var setKeyConfigRegex = regexp.MustCompile(`(?i)^setkey (?P<key>\S+) (?P<name>\S+) (?P<value>.*)$`)
var getConfigRegex = regexp.MustCompile(`(?i)^get (?P<key>\S+)$`)
if p.quiet && message.Body == "come back" { func (p *AdminPlugin) isAdmin(rh bot.ResponseHandler) bot.ResponseHandler {
p.quiet = false return func(r bot.Request) bool {
p.bot.SetQuiet(false) if !p.bot.CheckAdmin(r.Msg.User.Name) {
backMsg := p.bot.Config().Get("admin.comeback", "Okay, I'm back.") log.Debug().Msgf("User %s is not an admin", r.Msg.User.Name)
p.bot.Send(conn, bot.Message, message.Channel, backMsg) return false
return true
}
if strings.ToLower(body) == "shut up" {
dur := time.Duration(p.cfg.GetInt("quietDuration", 5)) * time.Minute
log.Info().Msgf("Going to sleep for %v, %v", dur, time.Now().Add(dur))
leaveMsg := p.bot.Config().Get("admin.shutup", "Okay. I'll be back later.")
p.bot.Send(conn, bot.Message, message.Channel, leaveMsg)
p.quiet = true
p.bot.SetQuiet(true)
go func() {
select {
case <-time.After(dur):
p.quiet = false
p.bot.SetQuiet(false)
log.Info().Msg("Waking up from nap.")
backMsg := p.bot.Config().Get("admin.backmsg", "I'm back, bitches.")
p.bot.Send(conn, bot.Message, message.Channel, backMsg)
}
}()
return true
}
if !p.bot.CheckAdmin(message.User.Name) {
log.Debug().Msgf("User %s is not an admin", message.User.Name)
return false
}
if strings.ToLower(body) == "reboot" {
p.bot.Send(conn, bot.Message, message.Channel, "brb")
log.Info().Msgf("Got reboot command")
os.Exit(0)
}
if addBlacklist.MatchString(body) {
submatches := addBlacklist.FindStringSubmatch(message.Body)
plugin := submatches[1]
if err := p.addBlacklist(message.Channel, plugin); err != nil {
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("I couldn't add that item: %s", err))
log.Error().Err(err).Msgf("error adding blacklist item")
return true
} }
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("%s disabled. Use `!enable plugin %s` to re-enable it.", plugin, plugin)) return rh(r)
return true
} }
}
if rmBlacklist.MatchString(body) { func (p *AdminPlugin) shutupCmd(r bot.Request) bool {
submatches := rmBlacklist.FindStringSubmatch(message.Body) dur := time.Duration(p.cfg.GetInt("quietDuration", 5)) * time.Minute
plugin := submatches[1] log.Info().Msgf("Going to sleep for %v, %v", dur, time.Now().Add(dur))
if err := p.rmBlacklist(message.Channel, plugin); err != nil { leaveMsg := p.bot.Config().Get("admin.shutup", "Okay. I'll be back later.")
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("I couldn't remove that item: %s", err)) p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, leaveMsg)
log.Error().Err(err).Msgf("error removing blacklist item") p.quiet = true
return true p.bot.SetQuiet(true)
go func() {
select {
case <-time.After(dur):
p.quiet = false
p.bot.SetQuiet(false)
log.Info().Msg("Waking up from nap.")
backMsg := p.bot.Config().Get("admin.backmsg", "I'm back, bitches.")
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, backMsg)
} }
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("%s enabled. Use `!disable plugin %s` to disable it.", plugin, plugin)) }()
return true
}
func (p *AdminPlugin) comeBackCmd(r bot.Request) bool {
p.quiet = false
p.bot.SetQuiet(false)
backMsg := p.bot.Config().Get("admin.comeback", "Okay, I'm back.")
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, backMsg)
return true
}
func (p *AdminPlugin) addBlacklistCmd(r bot.Request) bool {
plugin := r.Values["plugin"]
if err := p.addBlacklist(r.Msg.Channel, plugin); err != nil {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("I couldn't add that item: %s", err))
log.Error().Err(err).Msgf("error adding blacklist item")
return true return true
} }
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("%s disabled. Use `!enable plugin %s` to re-enable it.", plugin, plugin))
return true
}
if allWhitelist.MatchString(body) { func (p *AdminPlugin) rmBlacklistCmd(r bot.Request) bool {
plugins := p.bot.GetPluginNames() plugin := r.Values["plugin"]
for _, plugin := range plugins { if err := p.rmBlacklist(r.Msg.Channel, plugin); err != nil {
if err := p.addWhitelist(plugin); err != nil { p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("I couldn't remove that item: %s", err))
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("I couldn't whitelist that item: %s", err)) log.Error().Err(err).Msgf("error removing blacklist item")
log.Error().Err(err).Msgf("error adding whitelist item")
return true
}
}
p.bot.Send(conn, bot.Message, message.Channel, "Enabled all plugins")
return true return true
} }
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("%s enabled. Use `!disable plugin %s` to disable it.", plugin, plugin))
return true
}
if allUnwhitelist.MatchString(body) { func (p *AdminPlugin) addWhitelistCmd(r bot.Request) bool {
plugins := p.bot.GetPluginNames() plugin := r.Values["plugin"]
for _, plugin := range plugins { if err := p.addWhitelist(plugin); err != nil {
if plugin == "admin" { p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("I couldn't whitelist that item: %s", err))
continue log.Error().Err(err).Msgf("error adding whitelist item")
}
if err := p.rmWhitelist(plugin); err != nil {
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("I couldn't unwhitelist that item: %s", err))
log.Error().Err(err).Msgf("error removing whitelist item")
return true
}
}
p.bot.Send(conn, bot.Message, message.Channel, "Disabled all plugins")
return true return true
} }
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("%s enabled. Use `!unwhitelist plugin %s` to disable it.", plugin, plugin))
return true
}
if addWhitelist.MatchString(body) { func (p *AdminPlugin) rmWhitelistCmd(r bot.Request) bool {
submatches := addWhitelist.FindStringSubmatch(message.Body) plugin := r.Values["plugin"]
plugin := submatches[1] if err := p.rmWhitelist(plugin); err != nil {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("I couldn't unwhitelist that item: %s", err))
log.Error().Err(err).Msgf("error removing whitelist item")
return true
}
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("%s disabled. Use `!whitelist plugin %s` to enable it.", plugin, plugin))
return true
}
func (p *AdminPlugin) allWhitelistCmd(r bot.Request) bool {
plugins := p.bot.GetPluginNames()
for _, plugin := range plugins {
if err := p.addWhitelist(plugin); err != nil { if err := p.addWhitelist(plugin); err != nil {
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("I couldn't whitelist that item: %s", err)) p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("I couldn't whitelist that item: %s", err))
log.Error().Err(err).Msgf("error adding whitelist item") log.Error().Err(err).Msgf("error adding whitelist item")
return true return true
} }
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("%s enabled. Use `!unwhitelist plugin %s` to disable it.", plugin, plugin))
return true
} }
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "Enabled all plugins")
return true
}
if rmWhitelist.MatchString(body) { func (p *AdminPlugin) allUnWhitelistCmd(r bot.Request) bool {
submatches := rmWhitelist.FindStringSubmatch(message.Body) plugins := p.bot.GetPluginNames()
plugin := submatches[1] for _, plugin := range plugins {
if plugin == "admin" {
continue
}
if err := p.rmWhitelist(plugin); err != nil { if err := p.rmWhitelist(plugin); err != nil {
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("I couldn't unwhitelist that item: %s", err)) p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("I couldn't unwhitelist that item: %s", err))
log.Error().Err(err).Msgf("error removing whitelist item") log.Error().Err(err).Msgf("error removing whitelist item")
return true return true
} }
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("%s disabled. Use `!whitelist plugin %s` to enable it.", plugin, plugin))
return true
} }
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "Disabled all plugins")
if getWhitelist.MatchString(body) { return true
list := p.bot.GetWhitelist()
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Whitelist: %v", list))
return true
}
if getPlugins.MatchString(body) {
plugins := p.bot.GetPluginNames()
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Plugins: %v", plugins))
return true
}
if strings.ToLower(body) == "password" {
p.bot.Send(conn, bot.Message, message.Channel, p.bot.GetPassword())
return true
}
verbs := map[string]bool{
"set": true,
"push": true,
"setkey": true,
}
parts := strings.Split(body, " ")
if verbs[parts[0]] && len(parts) > 2 && forbiddenKeys[parts[1]] {
p.bot.Send(conn, bot.Message, message.Channel, "You cannot access that key")
return true
} else if parts[0] == "unset" && len(parts) > 1 {
if err := p.cfg.Unset(parts[1]); err != nil {
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Unset error: %s", err))
return true
}
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Unset %s", parts[1]))
return true
} else if parts[0] == "set" && len(parts) > 2 {
if err := p.cfg.Set(parts[1], strings.Join(parts[2:], " ")); err != nil {
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Set error: %s", err))
return true
}
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Set %s", parts[1]))
return true
} else if parts[0] == "push" && len(parts) > 2 {
items := p.cfg.GetArray(parts[1], []string{})
items = append(items, strings.Join(parts[2:], ""))
if err := p.cfg.Set(parts[1], strings.Join(items, ";;")); err != nil {
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Set error: %s", err))
return true
}
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Set %s", parts[1]))
return true
} else if parts[0] == "setkey" && len(parts) > 3 {
items := p.cfg.GetMap(parts[1], map[string]string{})
items[parts[2]] = strings.Join(parts[3:], " ")
if err := p.cfg.SetMap(parts[1], items); err != nil {
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Set error: %s", err))
return true
}
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("Set %s", parts[1]))
return true
}
if parts[0] == "get" && len(parts) == 2 && forbiddenKeys[parts[1]] {
p.bot.Send(conn, bot.Message, message.Channel, "You cannot access that key")
return true
} else if parts[0] == "get" && len(parts) == 2 {
v := p.cfg.Get(parts[1], "<unknown>")
p.bot.Send(conn, bot.Message, message.Channel, fmt.Sprintf("%s: %s", parts[1], v))
return true
}
return false
} }
func (p *AdminPlugin) handleVariables(conn bot.Connector, message msg.Message) bool { func (p *AdminPlugin) getWhitelistCmd(r bot.Request) bool {
if parts := strings.SplitN(message.Body, "!=", 2); len(parts) == 2 { list := p.bot.GetWhitelist()
variable := strings.ToLower(strings.TrimSpace(parts[0])) p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Whitelist: %v", list))
value := strings.TrimSpace(parts[1]) return true
}
_, err := p.db.Exec(`delete from variables where name=? and value=?`, variable, value) func (p *AdminPlugin) getPluginsCmd(r bot.Request) bool {
if err != nil { plugins := p.bot.GetPluginNames()
p.bot.Send(conn, bot.Message, message.Channel, "I'm broke and need attention in my variable creation code.") p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Plugins: %v", plugins))
log.Error().Err(err) return true
} else { }
p.bot.Send(conn, bot.Message, message.Channel, "Removed.")
}
return true func (p *AdminPlugin) rebootCmd(r bot.Request) bool {
} p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "brb")
log.Info().Msgf("Got reboot command")
os.Exit(0)
return true
}
parts := strings.SplitN(message.Body, "=", 2) func (p *AdminPlugin) passwordCmd(r bot.Request) bool {
if len(parts) != 2 { p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, p.bot.GetPassword())
return false return true
} }
variable := strings.ToLower(strings.TrimSpace(parts[0])) func (p *AdminPlugin) variableSetCmd(r bot.Request) bool {
value := strings.TrimSpace(parts[1]) variable := strings.ToLower(r.Values["var"])
value := r.Values["value"]
var count int64 var count int64
row := p.db.QueryRow(`select count(*) from variables where value = ?`, variable, value) row := p.db.QueryRow(`select count(*) from variables where value = ?`, variable, value)
err := row.Scan(&count) err := row.Scan(&count)
if err != nil { if err != nil {
p.bot.Send(conn, bot.Message, message.Channel, "I'm broke and need attention in my variable creation code.") p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "I'm broke and need attention in my variable creation code.")
log.Error().Err(err) log.Error().Err(err)
return true return true
} }
if count > 0 { if count > 0 {
p.bot.Send(conn, bot.Message, message.Channel, "I've already got that one.") p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "I've already got that one.")
} else { } else {
_, err := p.db.Exec(`INSERT INTO variables (name, value) VALUES (?, ?)`, variable, value) _, err := p.db.Exec(`INSERT INTO variables (name, value) VALUES (?, ?)`, variable, value)
if err != nil { if err != nil {
p.bot.Send(conn, bot.Message, message.Channel, "I'm broke and need attention in my variable creation code.") p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "I'm broke and need attention in my variable creation code.")
log.Error().Err(err) log.Error().Err(err)
return true return true
} }
p.bot.Send(conn, bot.Message, message.Channel, "Added.") p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "Added.")
} }
return true return true
} }
func (p *AdminPlugin) variableUnSetCmd(r bot.Request) bool {
variable := strings.ToLower(r.Values["var"])
value := r.Values["value"]
_, err := p.db.Exec(`delete from variables where name=? and value=?`, variable, value)
if err != nil {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "I'm broke and need attention in my variable creation code.")
log.Error().Err(err)
} else {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "Removed.")
}
return true
}
func (p *AdminPlugin) unsetConfigCmd(r bot.Request) bool {
key := r.Values["key"]
if forbiddenKeys[key] {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "You cannot access that key")
return true
}
if err := p.cfg.Unset(key); err != nil {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Unset error: %s", err))
return true
}
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Unset %s", key))
return true
}
func (p *AdminPlugin) setConfigCmd(r bot.Request) bool {
key, value := r.Values["key"], r.Values["value"]
if forbiddenKeys[key] {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "You cannot access that key")
return true
}
if err := p.cfg.Set(key, value); err != nil {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Set error: %s", err))
return true
}
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Set %s", key))
return true
}
func (p *AdminPlugin) pushConfigCmd(r bot.Request) bool {
key, value := r.Values["key"], r.Values["value"]
if forbiddenKeys[key] {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "You cannot access that key")
return true
}
items := p.cfg.GetArray(key, []string{})
items = append(items, value)
if err := p.cfg.Set(key, strings.Join(items, ";;")); err != nil {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Set error: %s", err))
return true
}
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Set %s", key))
return true
}
func (p *AdminPlugin) setKeyConfigCmd(r bot.Request) bool {
key, name, value := r.Values["key"], r.Values["name"], r.Values["value"]
if forbiddenKeys[key] {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "You cannot access that key")
return true
}
items := p.cfg.GetMap(key, map[string]string{})
items[name] = value
if err := p.cfg.SetMap(key, items); err != nil {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Set error: %s", err))
return true
}
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Set %s", key))
return true
}
func (p *AdminPlugin) getConfigCmd(r bot.Request) bool {
key := r.Values["key"]
if forbiddenKeys[key] {
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "You cannot access that key")
return true
}
v := p.cfg.Get(key, "<unknown>")
p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("%s: %s", key, v))
return true
}
// Help responds to help requests. Every plugin must implement a help function. // Help responds to help requests. Every plugin must implement a help function.
func (p *AdminPlugin) help(conn bot.Connector, kind bot.Kind, m msg.Message, args ...interface{}) bool { func (p *AdminPlugin) help(conn bot.Connector, kind bot.Kind, m msg.Message, args ...interface{}) bool {
p.bot.Send(conn, bot.Message, m.Channel, "This does super secret things that you're not allowed to know about.") p.bot.Send(conn, bot.Message, m.Channel, "This does super secret things that you're not allowed to know about.")