catbase/plugins/beers.go

package plugins

import (
	"bitbucket.org/phlyingpenguin/godeepintir/bot"
	"fmt"
	"strconv"
	"strings"
	"time"
)

// This is a skeleton plugin to serve as an example and quick copy/paste for new plugins.

type BeersPlugin struct {
	Bot *bot.Bot
}

// NewBeersPlugin creates a new BeersPlugin with the Plugin interface
func NewBeersPlugin(bot *bot.Bot) *BeersPlugin {
	return &BeersPlugin{
		Bot: bot,
	}
}

type UserBeers struct {
	Nick      string
	berrcount int
	lastdrunk time.Time
	momentum  float64
}

// 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 *BeersPlugin) Message(message bot.Message) bool {
	parts := strings.Fields(message.Body)

	if len(parts) == 0 {
		return false
	}

	channel := message.Channel
	user := message.User
	nick := user.Name

	// respond to the beers type of queries
	if parts[0] == "beers" {
		if len(parts) == 3 {

			// try to get a count out of parts[2]
			count, err := strconv.Atoi(parts[2])
			if err != nil {
				// if it's not a number, maybe it's a nick!
				if p.doIKnow(parts[2]) {
					p.reportCount(parts[2], false)
				} else {
					msg := fmt.Sprintf("Sorry, I don't know %s.", parts[2])
					p.Bot.SendMessage(channel, msg)
				}
			}

			if count < 0 {
				// you can't be negative
				msg := fmt.Sprintf("Sorry %s, you can't have negative beers!", nick)
				p.Bot.SendMessage(channel, msg)
			}
			if parts[1] == "+=" {
				p.setBeers(user, p.getBeers(user)+count)
			} else if parts[1] == "=" {
				if count == 0 {
					p.puke(user)
				} else {
					p.setBeers(user, count)
					p.reportCount(nick, true)
				}
			} else {
				p.Bot.SendMessage(channel, "I don't know your math.")
			}
		}

		// no matter what, if we're in here, then we've responded
		return true
	}

	if message.Command && parts[0] == "imbibe" {
		p.setBeers(user, p.getBeers(user)+1)
		p.reportCount(nick, true)
		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 *BeersPlugin) LoadData() {
	// This bot has no data to load
}

// Help responds to help requests. Every plugin must implement a help function.
func (p *BeersPlugin) Help(channel string, parts []string) {
	p.Bot.SendMessage(channel, "Sorry, Beers does not do a goddamn thing.")
}

func (p *BeersPlugin) setBeers(user *bot.User, amount int) {
}

func (p *BeersPlugin) addBeers(user *bot.User) {
}

func (p *BeersPlugin) getBeers(user *bot.User) int {
	return 0
}

func (p *BeersPlugin) hasBeers(user *bot.User) {
}

func (p *BeersPlugin) reportCount(user string, himself bool) {
}

func (p *BeersPlugin) puke(user *bot.User) {
}

func (p *BeersPlugin) doIKnow(user string) bool {
	return false
}