mirror of https://github.com/velour/catbase.git
94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
package plugins
|
|
|
|
import "bitbucket.org/phlyingpenguin/godeepintir/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")
|
|
fmt.Println()
|
|
} else {
|
|
dice = strings.Split(parts[1], "d")
|
|
}
|
|
|
|
if dice < 1 || sides < 2 {
|
|
p.Bot.SendMessage(channel, "You're a dick.")
|
|
}
|
|
|
|
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
|
|
}
|
|
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
|
|
}
|