catbase/plugins/dice.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 len(dice) == 2 {
			// We actually have a die roll.
			nDice, err := strconv.Atoi(dice[0])
			if err != nil || nDice < 1 {
				p.Bot.SendMessage(channel, "You're a dick.")
				return true
			}
			sides, err := strconv.Atoi(dice[1])
			if err != nil || sides < 2 {
				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
}