// © 2013 the CatBase Authors under the WTFPL. See AUTHORS for the list of authors. package plugins import "github.com/velour/catbase/bot" import ( "fmt" "math/rand" "strconv" "strings" "time" ) // This is a dice plugin to serve as an example and quick copy/paste for new plugins. type DicePlugin struct { Bot *bot.Bot } // NewDicePlugin creates a new DicePlugin with the Plugin interface func NewDicePlugin(bot *bot.Bot) *DicePlugin { return &DicePlugin{ Bot: bot, } } func rollDie(sides int) int { return rand.Intn(sides) + 1 } // Message responds to the bot hook on recieving messages. // This function returns true if the plugin responds in a meaningful way to the users message. // Otherwise, the function returns false and the bot continues execution of other plugins. func (p *DicePlugin) Message(message bot.Message) bool { channel := message.Channel parts := strings.Fields(message.Body) if (len(parts) == 2 || len(parts) == 1) && message.Command { var dice []string if len(parts) == 1 { dice = strings.Split(parts[0], "d") } else { dice = strings.Split(parts[1], "d") } if len(dice) == 2 { // We actually have a die roll. nDice, err := strconv.Atoi(dice[0]) if err != nil { return false } sides, err := strconv.Atoi(dice[1]) if err != nil { return false } if sides < 2 || nDice < 1 || nDice > 20 { p.Bot.SendMessage(channel, "You're a dick.") return true } rolls := fmt.Sprintf("%s, you rolled: ", message.User.Name) for i := 0; i < nDice; i++ { rolls = fmt.Sprintf("%s %d", rolls, rollDie(sides)) if i != nDice-1 { rolls = fmt.Sprintf("%s,", rolls) } else { rolls = fmt.Sprintf("%s.", rolls) } } p.Bot.SendMessage(channel, rolls) return true } } return false } // LoadData imports any configuration data into the plugin. This is not strictly necessary other // than the fact that the Plugin interface demands it exist. This may be deprecated at a later // date. func (p *DicePlugin) LoadData() { rand.Seed(time.Now().Unix()) } // Help responds to help requests. Every plugin must implement a help function. func (p *DicePlugin) Help(channel string, parts []string) { p.Bot.SendMessage(channel, "Roll dice using notation XdY. Try \"3d20\".") } // Empty event handler because this plugin does not do anything on event recv func (p *DicePlugin) Event(kind string, message bot.Message) bool { return false } // Handler for bot's own messages func (p *DicePlugin) BotMessage(message bot.Message) bool { return false } // Register any web URLs desired func (p *DicePlugin) RegisterWeb() *string { return nil }