achievements: refactor

Refs #346

bot/handlers.go

@@ -41,7 +41,7 @@ RET:
 	return true
 }
 
-func parseValues(r *regexp.Regexp, body string) RegexValues {
+func ParseValues(r *regexp.Regexp, body string) RegexValues {
 	out := RegexValues{}
 	subs := r.FindStringSubmatch(body)
 	if len(subs) == 0 {
@@ -62,7 +62,7 @@ func (b *bot) runCallback(conn Connector, plugin Plugin, evt Kind, message msg.M
 					Conn:   conn,
 					Kind:   evt,
 					Msg:    message,
-					Values: parseValues(r, message.Body),
+					Values: ParseValues(r, message.Body),
 					Args:   args,
 				}
 				if cb(resp) {

plugins/achievements/achievements.go

@@ -31,7 +31,11 @@ func New(b bot.Bot) *AchievementsPlugin {
 	if err != nil {
 		log.Fatal().Err(err).Msg("unable to create achievements tables")
 	}
-	b.Register(ap, bot.Message, ap.message)
+
+	b.RegisterRegexCmd(ap, bot.Message, grantRegex, ap.grantCmd)
+	b.RegisterRegexCmd(ap, bot.Message, createRegex, ap.createCmd)
+	b.RegisterRegexCmd(ap, bot.Message, greatRegex, ap.greatCmd)
+
 	b.Register(ap, bot.Help, ap.help)
 	return ap
 }
@@ -80,77 +84,74 @@ var grantRegex = regexp.MustCompile(`(?i)grant (?P<emojy>(?::[[:word:][:punct:]]
 var createRegex = regexp.MustCompile(`(?i)create trophy (?P<emojy>(?::[[:word:][:punct:]]+:\s?)+) (?P<description>.+)`)
 var greatRegex = regexp.MustCompile(`(?i)how great (?:am i|is :?(?P<who>[[:word:]]+))[[:punct:]]*`)
 
-func (p *AchievementsPlugin) message(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {
-	nick := message.User.Name
+func (p *AchievementsPlugin) grantCmd(r bot.Request) bool {
+	nick := r.Msg.User.Name
+	emojy := r.Values["emojy"]
+	receiver := r.Values["nick"]
+	trophy, err := p.FindTrophy(emojy)
+	if err != nil {
+		log.Error().Err(err).Msg("could not find trophy")
+		msg := fmt.Sprintf("The %s award doesn't exist.", emojy)
+		p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, msg)
+		return true
+	}
+	if nick == trophy.Creator {
+		a, err := p.Grant(receiver, emojy)
+		if err != nil {
+			log.Error().Err(err).Msg("could not award trophy")
+		}
+		msg := fmt.Sprintf("Congrats %s. You just got the %s award for %s.",
+			receiver, emojy, a.Description)
+		p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, msg)
+	} else {
+		msg := fmt.Sprintf("Sorry, %s. %s owns that trophy.", nick, trophy.Creator)
+		p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, msg)
+	}
+	return true
+}
 
-	if greatRegex.MatchString(message.Body) {
-		submatches := greatRegex.FindAllStringSubmatch(message.Body, -1)
-		who := submatches[0][1]
-		if who == "" {
-			who = nick
-		}
-		awards := p.GetAwards(who)
-		if len(awards) == 0 {
-			m := fmt.Sprintf("%s has no achievements to their name. They really suck.", who)
-			if who == nick {
-				m = fmt.Sprintf("You have no achievements to your name. "+
-					"You are a sad and terrible specimen of the human condition, %s.", who)
-			}
-			p.bot.Send(c, bot.Message, message.Channel, m)
-		} else {
-			m := fmt.Sprintf("Wow, let's all clap for %s. Look at these awards:", who)
-			for _, a := range awards {
-				m += fmt.Sprintf("\n%s - %s", a.Emojy, a.Description)
-			}
-			p.bot.Send(c, bot.Message, message.Channel, m)
-		}
-		return true
-	}
-	if message.Command && grantRegex.MatchString(message.Body) {
-		submatches := grantRegex.FindAllStringSubmatch(message.Body, -1)
-		emojy := submatches[0][1]
-		receiver := submatches[0][2]
-		trophy, err := p.FindTrophy(emojy)
-		if err != nil {
-			log.Error().Err(err).Msg("could not find trophy")
-			msg := fmt.Sprintf("The %s award doesn't exist.", emojy)
-			p.bot.Send(c, bot.Message, message.Channel, msg)
+func (p *AchievementsPlugin) createCmd(r bot.Request) bool {
+	nick := r.Msg.User.Name
+	emojy := r.Values["emojy"]
+	description := r.Values["description"]
+	t, err := p.Create(emojy, description, nick)
+	if err != nil {
+		log.Error().Err(err).Msg("could not create trophy")
+		if strings.Contains(err.Error(), "exists") {
+			p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, err.Error())
 			return true
 		}
-		if nick == trophy.Creator {
-			a, err := p.Grant(receiver, emojy)
-			if err != nil {
-				log.Error().Err(err).Msg("could not award trophy")
-			}
-			msg := fmt.Sprintf("Congrats %s. You just got the %s award for %s.",
-				receiver, emojy, a.Description)
-			p.bot.Send(c, bot.Message, message.Channel, msg)
-		} else {
-			msg := fmt.Sprintf("Sorry, %s. %s owns that trophy.", nick, trophy.Creator)
-			p.bot.Send(c, bot.Message, message.Channel, msg)
-		}
+		p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, "I'm too humble to ever award that trophy")
 		return true
 	}
-	if message.Command && createRegex.MatchString(message.Body) {
-		submatches := createRegex.FindAllStringSubmatch(message.Body, -1)
-		emojy := submatches[0][1]
-		description := submatches[0][2]
-		t, err := p.Create(emojy, description, nick)
-		if err != nil {
-			log.Error().Err(err).Msg("could not create trophy")
-			if strings.Contains(err.Error(), "exists") {
-				p.bot.Send(c, bot.Message, message.Channel, err.Error())
-				return true
-			}
-			p.bot.Send(c, bot.Message, message.Channel, "I'm too humble to ever award that trophy")
-			return true
-		}
-		resp := fmt.Sprintf("Okay %s. I have crafted a one-of-a-kind %s trophy to give for %s",
-			nick, t.Emojy, t.Description)
-		p.bot.Send(c, bot.Message, message.Channel, resp)
-		return true
+	resp := fmt.Sprintf("Okay %s. I have crafted a one-of-a-kind %s trophy to give for %s",
+		nick, t.Emojy, t.Description)
+	p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, resp)
+	return true
+}
+
+func (p *AchievementsPlugin) greatCmd(r bot.Request) bool {
+	nick := r.Msg.User.Name
+	who := r.Values["who"]
+	if who == "" {
+		who = nick
 	}
-	return false
+	awards := p.GetAwards(who)
+	if len(awards) == 0 {
+		m := fmt.Sprintf("%s has no achievements to their name. They really suck.", who)
+		if who == nick {
+			m = fmt.Sprintf("You have no achievements to your name. "+
+				"You are a sad and terrible specimen of the human condition, %s.", who)
+		}
+		p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, m)
+	} else {
+		m := fmt.Sprintf("Wow, let's all clap for %s. Look at these awards:", who)
+		for _, a := range awards {
+			m += fmt.Sprintf("\n%s - %s", a.Emojy, a.Description)
+		}
+		p.bot.Send(r.Conn, bot.Message, r.Msg.Channel, m)
+	}
+	return true
 }
 
 func (p *AchievementsPlugin) help(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {

plugins/achievements/achievements_test.go

@@ -1 +0,0 @@
-package achievements

plugins/counter/counter.go

@@ -20,8 +20,6 @@ import (
 
 // This is a counter plugin to count arbitrary things.
 
-var teaMatcher = regexp.MustCompile("(?i)^([^.]+)\\. [^.]*\\. ([^.]*\\.?)+$")
-
 type CounterPlugin struct {
 	Bot bot.Bot
 	DB  *sqlx.DB
@@ -262,7 +260,20 @@ func New(b bot.Bot) *CounterPlugin {
 		Bot: b,
 		DB:  b.DB(),
 	}
-	b.Register(cp, bot.Message, cp.message)
+
+	b.RegisterRegexCmd(cp, bot.Message, mkAliasRegex, cp.mkAliasCmd)
+	b.RegisterRegexCmd(cp, bot.Message, rmAliasRegex, cp.rmAliasCmd)
+	b.RegisterRegexCmd(cp, bot.Message, leaderboardRegex, cp.leaderboardCmd)
+	b.RegisterRegex(cp, bot.Message, teaRegex, cp.teaMatchCmd)
+	b.RegisterRegexCmd(cp, bot.Message, resetRegex, cp.resetCmd)
+	b.RegisterRegexCmd(cp, bot.Message, inspectRegex, cp.inspectCmd)
+	b.RegisterRegexCmd(cp, bot.Message, clearRegex, cp.clearCmd)
+	b.RegisterRegexCmd(cp, bot.Message, countRegex, cp.countCmd)
+	b.RegisterRegex(cp, bot.Message, incrementRegex, cp.incrementCmd)
+	b.RegisterRegex(cp, bot.Message, decrementRegex, cp.decrementCmd)
+	b.RegisterRegex(cp, bot.Message, addToRegex, cp.addToCmd)
+	b.RegisterRegex(cp, bot.Message, removeFromRegex, cp.removeFromCmd)
+
 	b.Register(cp, bot.Help, cp.help)
 	cp.registerWeb()
 
@@ -273,299 +284,322 @@ func trimUnicode(s string) string {
 	return strings.Trim(s, string(rune(0xFE0F)))
 }
 
-// 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 *CounterPlugin) message(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {
-	// This bot does not reply to anything
-	nick := message.User.Name
-	channel := message.Channel
-	parts := strings.Fields(message.Body)
+var mkAliasRegex = regexp.MustCompile(`(?i)^mkalias (?P<what>\S+) (?P<to>\S+)$`)
+var rmAliasRegex = regexp.MustCompile(`(?i)^rmalias (?P<what>\S+)$`)
+var leaderboardRegex = regexp.MustCompile(`(?i)^leaderboard\s?(?P<what>\S+)?$`)
+var teaRegex = regexp.MustCompile("(?i)^([^.]+)\\. [^.]*\\. ([^.]*\\.?)+$")
+var resetRegex = regexp.MustCompile(`(?i)^reset me$`)
+var inspectRegex = regexp.MustCompile(`(?i)^inspect (?P<who>\S+)$`)
+var clearRegex = regexp.MustCompile(`(?i)^clear (?P<what>\S+)$`)
+var countRegex = regexp.MustCompile(`(?i)^count (?P<who>\S+)\s?(?P<what>\S+)?$`)
+var incrementRegex = regexp.MustCompile(`(?i)^(?P<who>[^.\t\n\f\r ]+\s?\.)?(?P<thing>\S+)\s?\+\+$`)
+var decrementRegex = regexp.MustCompile(`(?i)^(?P<who>[^.\t\n\f\r ]+\s?\.)?(?P<thing>\S+)\s?--$`)
+var addToRegex = regexp.MustCompile(`(?i)^(?P<who>[^.\t\n\f\r ]+\s?\.)?(?P<thing>\S+)\s+\+=\s+(?P<amount>\d+)$`)
+var removeFromRegex = regexp.MustCompile(`(?i)^(?P<who>[^.\t\n\f\r ]+\s?\.)?(?P<thing>\S+)\s+-=\s+(?P<amount>\d+)$`)
 
-	if len(parts) == 0 {
+func (p *CounterPlugin) mkAliasCmd(r bot.Request) bool {
+	what := r.Values["what"]
+	to := r.Values["to"]
+	if what == "" || to == "" {
+		p.Bot.Send(r.Conn, bot.Message, fmt.Sprintf("You must provide all fields for an alias: %s", mkAliasRegex))
+		return true
+	}
+	if _, err := MkAlias(p.DB, what, to); err != nil {
+		log.Error().Err(err).Msg("Could not mkalias")
+		p.Bot.Send(r.Conn, bot.Message, r.Msg.Channel, "We're gonna need too much DB space to make an alias for your mom.")
+		return true
+	}
+	p.Bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("Created alias %s -> %s",
+		what, to))
+	return true
+}
+
+func (p *CounterPlugin) rmAliasCmd(r bot.Request) bool {
+	what := r.Values["what"]
+	if what == "" {
+		p.Bot.Send(r.Conn, bot.Message, r.Msg.Channel, "You must specify an alias to remove.")
+		return true
+	}
+	if err := RmAlias(p.DB, what); err != nil {
+		log.Error().Err(err).Msg("could not RmAlias")
+		p.Bot.Send(r.Conn, bot.Message, r.Msg.Channel, "`sudo rm your mom` => Nope, she's staying with me.")
+		return true
+	}
+	p.Bot.Send(r.Conn, bot.Message, r.Msg.Channel, "`sudo rm your mom`")
+	return true
+}
+
+func (p *CounterPlugin) leaderboardCmd(r bot.Request) bool {
+	var cmd func() ([]Item, error)
+	itNameTxt := ""
+	what := r.Values["what"]
+
+	if what == "" {
+		cmd = func() ([]Item, error) { return LeaderAll(p.DB) }
+	} else {
+		itNameTxt = fmt.Sprintf(" for %s", what)
+		cmd = func() ([]Item, error) { return Leader(p.DB, what) }
+	}
+
+	its, err := cmd()
+	if err != nil {
+		log.Error().Err(err).Msg("Error with leaderboard")
+		return false
+	} else if len(its) == 0 {
+		p.Bot.Send(r.Conn, bot.Message, r.Msg.Channel, "There are not enough entries for a leaderboard.")
+		return true
+	}
+
+	out := fmt.Sprintf("Leaderboard%s:\n", itNameTxt)
+	for _, it := range its {
+		out += fmt.Sprintf("%s with %d %s\n",
+			it.Nick,
+			it.Count,
+			it.Item,
+		)
+	}
+	p.Bot.Send(r.Conn, bot.Message, r.Msg.Channel, out)
+	return true
+}
+
+func (p *CounterPlugin) resetCmd(r bot.Request) bool {
+	nick := r.Msg.User.Name
+	channel := r.Msg.Channel
+
+	items, err := GetItems(p.DB, strings.ToLower(nick))
+	if err != nil {
+		log.Error().
+			Err(err).
+			Str("nick", nick).
+			Msg("Error getting items to reset")
+		p.Bot.Send(r.Conn, bot.Message, channel, "Something is technically wrong with your counters.")
+		return true
+	}
+	log.Debug().Msgf("Items: %+v", items)
+	for _, item := range items {
+		item.Delete()
+	}
+	p.Bot.Send(r.Conn, bot.Message, channel, fmt.Sprintf("%s, you are as new, my son.", nick))
+	return true
+}
+
+func (p *CounterPlugin) inspectCmd(r bot.Request) bool {
+	var subject string
+	subject = r.Values["who"]
+	channel := r.Msg.Channel
+	c := r.Conn
+
+	if subject == "me" {
+		subject = strings.ToLower(r.Msg.User.Name)
+	} else {
+		subject = strings.ToLower(subject)
+	}
+
+	log.Debug().
+		Str("subject", subject).
+		Msg("Getting counter")
+	// pull all of the items associated with "subject"
+	items, err := GetItems(p.DB, subject)
+	if err != nil {
+		log.Error().
+			Err(err).
+			Str("subject", subject).
+			Msg("Error retrieving items")
+		p.Bot.Send(c, bot.Message, channel, "Something went wrong finding that counter;")
+		return true
+	}
+
+	resp := fmt.Sprintf("%s has the following counters:", subject)
+	count := 0
+	for _, it := range items {
+		count += 1
+		if count > 1 {
+			resp += ","
+		}
+		resp += fmt.Sprintf(" %s: %d", it.Item, it.Count)
+		if count > 20 {
+			resp += ", and a few others"
+			break
+		}
+	}
+	resp += "."
+
+	if count == 0 {
+		p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("%s has no counters.", subject))
+		return true
+	}
+
+	p.Bot.Send(c, bot.Message, channel, resp)
+	return true
+}
+
+func (p *CounterPlugin) clearCmd(r bot.Request) bool {
+	subject := strings.ToLower(r.Msg.User.Name)
+	itemName := strings.ToLower(r.Values["what"])
+	channel := r.Msg.Channel
+	c := r.Conn
+
+	it, err := GetUserItem(p.DB, subject, itemName)
+	if err != nil {
+		log.Error().
+			Err(err).
+			Str("subject", subject).
+			Str("itemName", itemName).
+			Msg("Error getting item to remove")
+		p.Bot.Send(c, bot.Message, channel, "Something went wrong removing that counter;")
+		return true
+	}
+	err = it.Delete()
+	if err != nil {
+		log.Error().
+			Err(err).
+			Str("subject", subject).
+			Str("itemName", itemName).
+			Msg("Error removing item")
+		p.Bot.Send(c, bot.Message, channel, "Something went wrong removing that counter;")
+		return true
+	}
+
+	p.Bot.Send(c, bot.Action, channel, fmt.Sprintf("chops a few %s out of his brain",
+		itemName))
+	return true
+}
+
+func (p *CounterPlugin) countCmd(r bot.Request) bool {
+	var subject string
+	var itemName string
+
+	if r.Values["what"] == "" && r.Values["who"] == "" {
+		return false
+	} else if r.Values["what"] != "" {
+		subject = strings.ToLower(r.Values["who"])
+		itemName = strings.ToLower(r.Values["what"])
+	} else {
+		subject = strings.ToLower(r.Msg.User.Name)
+		itemName = strings.ToLower(r.Values["who"])
+	}
+
+	var item Item
+	item, err := GetUserItem(p.DB, subject, itemName)
+	switch {
+	case err == sql.ErrNoRows:
+		p.Bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("I don't think %s has any %s.",
+			subject, itemName))
+		return true
+	case err != nil:
+		log.Error().
+			Err(err).
+			Str("subject", subject).
+			Str("itemName", itemName).
+			Msg("Error retrieving item count")
+		return true
+	}
+
+	p.Bot.Send(r.Conn, bot.Message, r.Msg.Channel, fmt.Sprintf("%s has %d %s.", subject, item.Count,
+		itemName))
+
+	return true
+}
+
+func (p *CounterPlugin) incrementCmd(r bot.Request) bool {
+	subject := r.Msg.User.Name
+	if r.Values["who"] != "" {
+		subject = strings.TrimSuffix(r.Values["who"], ".")
+	}
+	itemName := r.Values["thing"]
+	channel := r.Msg.Channel
+	// ++ those fuckers
+	item, err := GetUserItem(p.DB, subject, itemName)
+	if err != nil {
+		log.Error().
+			Err(err).
+			Str("subject", subject).
+			Str("itemName", itemName).
+			Msg("error finding item")
+		// Item ain't there, I guess
 		return false
 	}
+	log.Debug().Msgf("About to update item: %#v", item)
+	item.UpdateDelta(1)
+	p.Bot.Send(r.Conn, bot.Message, channel, fmt.Sprintf("%s has %d %s.", subject,
+		item.Count, item.Item))
+	return true
+}
 
-	if len(parts) == 3 && strings.ToLower(parts[0]) == "mkalias" {
-		if _, err := MkAlias(p.DB, parts[1], parts[2]); err != nil {
-			log.Error().Err(err).Msg("Could not mkalias")
-			p.Bot.Send(c, bot.Message, channel, "We're gonna need too much DB space to make an alias for your mom.")
-			return true
-		}
-		p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("Created alias %s -> %s",
-			parts[1], parts[2]))
-		return true
-	} else if len(parts) == 2 && strings.ToLower(parts[0]) == "rmalias" {
-		if err := RmAlias(p.DB, parts[1]); err != nil {
-			log.Error().Err(err).Msg("could not RmAlias")
-			p.Bot.Send(c, bot.Message, channel, "`sudo rm your mom` => Nope, she's staying with me.")
-			return true
-		}
-		p.Bot.Send(c, bot.Message, channel, "`sudo rm your mom`")
-		return true
-	} else if strings.ToLower(parts[0]) == "leaderboard" {
-		var cmd func() ([]Item, error)
-		itNameTxt := ""
-
-		if len(parts) == 1 {
-			cmd = func() ([]Item, error) { return LeaderAll(p.DB) }
-		} else {
-			itNameTxt = fmt.Sprintf(" for %s", parts[1])
-			cmd = func() ([]Item, error) { return Leader(p.DB, parts[1]) }
-		}
-
-		its, err := cmd()
-		if err != nil {
-			log.Error().Err(err).Msg("Error with leaderboard")
-			return false
-		} else if len(its) == 0 {
-			return false
-		}
-
-		out := fmt.Sprintf("Leaderboard%s:\n", itNameTxt)
-		for _, it := range its {
-			out += fmt.Sprintf("%s with %d %s\n",
-				it.Nick,
-				it.Count,
-				it.Item,
-			)
-		}
-		p.Bot.Send(c, bot.Message, channel, out)
-		return true
-	} else if match := teaMatcher.MatchString(message.Body); match {
-		// check for tea match TTT
-		return p.checkMatch(c, message)
-	} else if message.Command && message.Body == "reset me" {
-		items, err := GetItems(p.DB, strings.ToLower(nick))
-		if err != nil {
-			log.Error().
-				Err(err).
-				Str("nick", nick).
-				Msg("Error getting items to reset")
-			p.Bot.Send(c, bot.Message, channel, "Something is technically wrong with your counters.")
-			return true
-		}
-		log.Debug().Msgf("Items: %+v", items)
-		for _, item := range items {
-			item.Delete()
-		}
-		p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("%s, you are as new, my son.", nick))
-		return true
-	} else if message.Command && parts[0] == "inspect" && len(parts) == 2 {
-		var subject string
-
-		if parts[1] == "me" {
-			subject = strings.ToLower(nick)
-		} else {
-			subject = strings.ToLower(parts[1])
-		}
-
-		log.Debug().
-			Str("subject", subject).
-			Msg("Getting counter")
-		// pull all of the items associated with "subject"
-		items, err := GetItems(p.DB, subject)
-		if err != nil {
-			log.Error().
-				Err(err).
-				Str("subject", subject).
-				Msg("Error retrieving items")
-			p.Bot.Send(c, bot.Message, channel, "Something went wrong finding that counter;")
-			return true
-		}
-
-		resp := fmt.Sprintf("%s has the following counters:", subject)
-		count := 0
-		for _, it := range items {
-			count += 1
-			if count > 1 {
-				resp += ","
-			}
-			resp += fmt.Sprintf(" %s: %d", it.Item, it.Count)
-			if count > 20 {
-				resp += ", and a few others"
-				break
-			}
-		}
-		resp += "."
-
-		if count == 0 {
-			p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("%s has no counters.", subject))
-			return true
-		}
-
-		p.Bot.Send(c, bot.Message, channel, resp)
-		return true
-	} else if message.Command && len(parts) == 2 && parts[0] == "clear" {
-		subject := strings.ToLower(nick)
-		itemName := strings.ToLower(parts[1])
-
-		it, err := GetUserItem(p.DB, subject, itemName)
-		if err != nil {
-			log.Error().
-				Err(err).
-				Str("subject", subject).
-				Str("itemName", itemName).
-				Msg("Error getting item to remove")
-			p.Bot.Send(c, bot.Message, channel, "Something went wrong removing that counter;")
-			return true
-		}
-		err = it.Delete()
-		if err != nil {
-			log.Error().
-				Err(err).
-				Str("subject", subject).
-				Str("itemName", itemName).
-				Msg("Error removing item")
-			p.Bot.Send(c, bot.Message, channel, "Something went wrong removing that counter;")
-			return true
-		}
-
-		p.Bot.Send(c, bot.Action, channel, fmt.Sprintf("chops a few %s out of his brain",
-			itemName))
-		return true
-
-	} else if message.Command && parts[0] == "count" {
-		var subject string
-		var itemName string
-
-		if len(parts) == 3 {
-			// report count for parts[1]
-			subject = strings.ToLower(parts[1])
-			itemName = strings.ToLower(parts[2])
-		} else if len(parts) == 2 {
-			subject = strings.ToLower(nick)
-			itemName = strings.ToLower(parts[1])
-		} else {
-			return false
-		}
-
-		var item Item
-		item, err := GetUserItem(p.DB, subject, itemName)
-		switch {
-		case err == sql.ErrNoRows:
-			p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("I don't think %s has any %s.",
-				subject, itemName))
-			return true
-		case err != nil:
-			log.Error().
-				Err(err).
-				Str("subject", subject).
-				Str("itemName", itemName).
-				Msg("Error retrieving item count")
-			return true
-		}
-
-		p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("%s has %d %s.", subject, item.Count,
-			itemName))
-
-		return true
-	} else if len(parts) <= 2 {
-		if (len(parts) == 2) && (parts[1] == "++" || parts[1] == "--") {
-			parts = []string{parts[0] + parts[1]}
-		}
-		// Need to have at least 3 characters to ++ or --
-		if len(parts[0]) < 3 {
-			return false
-		}
-
-		subject := strings.ToLower(nick)
-		itemName := strings.ToLower(parts[0])[:len(parts[0])-2]
-
-		if nameParts := strings.SplitN(itemName, ".", 2); len(nameParts) == 2 {
-			subject = nameParts[0]
-			itemName = nameParts[1]
-		}
-
-		if strings.HasSuffix(parts[0], "++") {
-			// ++ those fuckers
-			item, err := GetUserItem(p.DB, subject, itemName)
-			if err != nil {
-				log.Error().
-					Err(err).
-					Str("subject", subject).
-					Str("itemName", itemName).
-					Msg("error finding item")
-				// Item ain't there, I guess
-				return false
-			}
-			log.Debug().Msgf("About to update item: %#v", item)
-			item.UpdateDelta(1)
-			p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("%s has %d %s.", subject,
-				item.Count, item.Item))
-			return true
-		} else if strings.HasSuffix(parts[0], "--") {
-			// -- those fuckers
-			item, err := GetUserItem(p.DB, subject, itemName)
-			if err != nil {
-				log.Error().
-					Err(err).
-					Str("subject", subject).
-					Str("itemName", itemName).
-					Msg("Error finding item")
-				// Item ain't there, I guess
-				return false
-			}
-			item.UpdateDelta(-1)
-			p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("%s has %d %s.", subject,
-				item.Count, item.Item))
-			return true
-		}
-	} else if len(parts) == 3 {
-		// Need to have at least 3 characters to ++ or --
-		if len(parts[0]) < 3 {
-			return false
-		}
-
-		subject := strings.ToLower(nick)
-		itemName := strings.ToLower(parts[0])
-
-		if nameParts := strings.SplitN(itemName, ".", 2); len(nameParts) == 2 {
-			subject = nameParts[0]
-			itemName = nameParts[1]
-		}
-
-		if parts[1] == "+=" {
-			// += those fuckers
-			item, err := GetUserItem(p.DB, subject, itemName)
-			if err != nil {
-				log.Error().
-					Err(err).
-					Str("subject", subject).
-					Str("itemName", itemName).
-					Msg("Error finding item")
-				// Item ain't there, I guess
-				return false
-			}
-			n, _ := strconv.Atoi(parts[2])
-			log.Debug().Msgf("About to update item by %d: %#v", n, item)
-			item.UpdateDelta(n)
-			p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("%s has %d %s.", subject,
-				item.Count, item.Item))
-			return true
-		} else if parts[1] == "-=" {
-			// -= those fuckers
-			item, err := GetUserItem(p.DB, subject, itemName)
-			if err != nil {
-				log.Error().
-					Err(err).
-					Str("subject", subject).
-					Str("itemName", itemName).
-					Msg("Error finding item")
-				// Item ain't there, I guess
-				return false
-			}
-			n, _ := strconv.Atoi(parts[2])
-			log.Debug().Msgf("About to update item by -%d: %#v", n, item)
-			item.UpdateDelta(-n)
-			p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("%s has %d %s.", subject,
-				item.Count, item.Item))
-			return true
-		}
+func (p *CounterPlugin) decrementCmd(r bot.Request) bool {
+	subject := r.Msg.User.Name
+	if r.Values["who"] != "" {
+		subject = strings.TrimSuffix(r.Values["who"], ".")
 	}
+	itemName := r.Values["thing"]
+	channel := r.Msg.Channel
+	// -- those fuckers
+	item, err := GetUserItem(p.DB, subject, itemName)
+	if err != nil {
+		log.Error().
+			Err(err).
+			Str("subject", subject).
+			Str("itemName", itemName).
+			Msg("Error finding item")
+		// Item ain't there, I guess
+		return false
+	}
+	item.UpdateDelta(-1)
+	p.Bot.Send(r.Conn, bot.Message, channel, fmt.Sprintf("%s has %d %s.", subject,
+		item.Count, item.Item))
+	return true
+}
 
-	return false
+func (p *CounterPlugin) addToCmd(r bot.Request) bool {
+	subject := r.Msg.User.Name
+	if r.Values["who"] != "" {
+		subject = strings.TrimSuffix(r.Values["who"], ".")
+	}
+	itemName := r.Values["thing"]
+	channel := r.Msg.Channel
+	// += those fuckers
+	item, err := GetUserItem(p.DB, subject, itemName)
+	if err != nil {
+		log.Error().
+			Err(err).
+			Str("subject", subject).
+			Str("itemName", itemName).
+			Msg("Error finding item")
+		// Item ain't there, I guess
+		return false
+	}
+	n, _ := strconv.Atoi(r.Values["amount"])
+	log.Debug().Msgf("About to update item by %d: %#v", n, item)
+	item.UpdateDelta(n)
+	p.Bot.Send(r.Conn, bot.Message, channel, fmt.Sprintf("%s has %d %s.", subject,
+		item.Count, item.Item))
+	return true
+}
+
+func (p *CounterPlugin) removeFromCmd(r bot.Request) bool {
+	subject := r.Msg.User.Name
+	if r.Values["who"] != "" {
+		subject = strings.TrimSuffix(r.Values["who"], ".")
+	}
+	itemName := r.Values["thing"]
+	channel := r.Msg.Channel
+	// -= those fuckers
+	item, err := GetUserItem(p.DB, subject, itemName)
+	if err != nil {
+		log.Error().
+			Err(err).
+			Str("subject", subject).
+			Str("itemName", itemName).
+			Msg("Error finding item")
+		// Item ain't there, I guess
+		return false
+	}
+	n, _ := strconv.Atoi(r.Values["amount"])
+	log.Debug().Msgf("About to update item by -%d: %#v", n, item)
+	item.UpdateDelta(-n)
+	p.Bot.Send(r.Conn, bot.Message, channel, fmt.Sprintf("%s has %d %s.", subject,
+		item.Count, item.Item))
+	return true
 }
 
 // Help responds to help requests. Every plugin must implement a help function.
@@ -579,11 +613,11 @@ func (p *CounterPlugin) help(c bot.Connector, kind bot.Kind, message msg.Message
 	return true
 }
 
-func (p *CounterPlugin) checkMatch(c bot.Connector, message msg.Message) bool {
-	nick := message.User.Name
-	channel := message.Channel
+func (p *CounterPlugin) teaMatchCmd(r bot.Request) bool {
+	nick := r.Msg.User.Name
+	channel := r.Msg.Channel
 
-	submatches := teaMatcher.FindStringSubmatch(message.Body)
+	submatches := teaRegex.FindStringSubmatch(r.Msg.Body)
 	if len(submatches) <= 1 {
 		return false
 	}
@@ -605,7 +639,7 @@ func (p *CounterPlugin) checkMatch(c bot.Connector, message msg.Message) bool {
 		delta = -1
 	}
 	item.UpdateDelta(delta)
-	p.Bot.Send(c, bot.Message, channel, fmt.Sprintf("%s... %s has %d %s",
+	p.Bot.Send(r.Conn, bot.Message, channel, fmt.Sprintf("%s... %s has %d %s",
 		strings.Join(everyDayImShuffling([]string{"bleep", "bloop", "blop"}), "-"), nick, item.Count, itemName))
 	return true
 }

plugins/counter/counter_test.go

@@ -4,6 +4,7 @@ package counter
 
 import (
 	"fmt"
+	"regexp"
 	"strings"
 	"testing"
 
@@ -25,24 +26,28 @@ func setup(t *testing.T) (*bot.MockBot, *CounterPlugin) {
 	return mb, c
 }
 
-func makeMessage(payload string) (bot.Connector, bot.Kind, msg.Message) {
+func makeMessage(payload string, r *regexp.Regexp) bot.Request {
 	isCmd := strings.HasPrefix(payload, "!")
 	if isCmd {
 		payload = payload[1:]
 	}
-	return &cli.CliPlugin{}, bot.Message, msg.Message{
-		User:    &user.User{Name: "tester"},
-		Channel: "test",
-		Body:    payload,
-		Command: isCmd,
+	values := bot.ParseValues(r, payload)
+	return bot.Request{
+		Conn: nil,
+		Msg: msg.Message{
+			User:    &user.User{Name: "tester"},
+			Body:    payload,
+			Command: isCmd,
+		},
+		Values: values,
 	}
 }
 
 func TestMkAlias(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("mkalias fuck mornings"))
-	c.message(makeMessage("fuck++"))
+	c.mkAliasCmd(makeMessage("mkalias fuck mornings", mkAliasRegex))
+	c.incrementCmd(makeMessage("fuck++", incrementRegex))
 	item, err := GetUserItem(mb.DB(), "tester", "mornings")
 	assert.Nil(t, err)
 	assert.Equal(t, 1, item.Count)
@@ -51,9 +56,9 @@ func TestMkAlias(t *testing.T) {
 func TestRmAlias(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("mkalias fuck mornings"))
-	c.message(makeMessage("rmalias fuck"))
-	c.message(makeMessage("fuck++"))
+	c.mkAliasCmd(makeMessage("mkalias fuck mornings", mkAliasRegex))
+	c.rmAliasCmd(makeMessage("rmalias fuck", rmAliasRegex))
+	c.incrementCmd(makeMessage("fuck++", incrementRegex))
 	item, err := GetUserItem(mb.DB(), "tester", "mornings")
 	assert.Nil(t, err)
 	assert.Equal(t, 0, item.Count)
@@ -62,8 +67,8 @@ func TestRmAlias(t *testing.T) {
 func TestThreeSentencesExists(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage(":beer:++"))
-	c.message(makeMessage(":beer:. Earl Grey. Hot."))
+	c.incrementCmd(makeMessage(":beer:++", incrementRegex))
+	c.teaMatchCmd(makeMessage(":beer:. Earl Grey. Hot.", teaRegex))
 	item, err := GetUserItem(mb.DB(), "tester", ":beer:")
 	assert.Nil(t, err)
 	assert.Equal(t, 2, item.Count)
@@ -73,7 +78,7 @@ func TestThreeSentencesNotExists(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
 	item, err := GetUserItem(mb.DB(), "tester", ":beer:")
-	c.message(makeMessage(":beer:. Earl Grey. Hot."))
+	c.teaMatchCmd(makeMessage(":beer:. Earl Grey. Hot.", teaRegex))
 	item, err = GetUserItem(mb.DB(), "tester", ":beer:")
 	assert.Nil(t, err)
 	assert.Equal(t, 0, item.Count)
@@ -82,8 +87,8 @@ func TestThreeSentencesNotExists(t *testing.T) {
 func TestTeaEarlGreyHot(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("Tea. Earl Grey. Hot."))
-	c.message(makeMessage("Tea. Earl Grey. Hot."))
+	c.teaMatchCmd(makeMessage("Tea. Earl Grey. Hot.", teaRegex))
+	c.teaMatchCmd(makeMessage("Tea. Earl Grey. Hot.", teaRegex))
 	item, err := GetUserItem(mb.DB(), "tester", ":tea:")
 	assert.Nil(t, err)
 	assert.Equal(t, 2, item.Count)
@@ -92,8 +97,8 @@ func TestTeaEarlGreyHot(t *testing.T) {
 func TestTeaTwoPeriods(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("Tea. Earl Grey."))
-	c.message(makeMessage("Tea. Earl Grey."))
+	c.teaMatchCmd(makeMessage("Tea. Earl Grey.", teaRegex))
+	c.teaMatchCmd(makeMessage("Tea. Earl Grey.", teaRegex))
 	item, err := GetUserItem(mb.DB(), "tester", ":tea:")
 	assert.Nil(t, err)
 	assert.Equal(t, 0, item.Count)
@@ -102,8 +107,8 @@ func TestTeaTwoPeriods(t *testing.T) {
 func TestTeaMultiplePeriods(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("Tea. Earl Grey. Spiked. Hot."))
-	c.message(makeMessage("Tea. Earl Grey. Spiked. Hot."))
+	c.teaMatchCmd(makeMessage("Tea. Earl Grey. Spiked. Hot.", teaRegex))
+	c.teaMatchCmd(makeMessage("Tea. Earl Grey. Spiked. Hot.", teaRegex))
 	item, err := GetUserItem(mb.DB(), "tester", ":tea:")
 	assert.Nil(t, err)
 	assert.Equal(t, 2, item.Count)
@@ -112,9 +117,9 @@ func TestTeaMultiplePeriods(t *testing.T) {
 func TestTeaGreenHot(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("Tea. Green. Hot."))
-	c.message(makeMessage("Tea. Green. Hot"))
-	c.message(makeMessage("Tea. Green. Iced."))
+	c.teaMatchCmd(makeMessage("Tea. Green. Hot.", teaRegex))
+	c.teaMatchCmd(makeMessage("Tea. Green. Hot", teaRegex))
+	c.teaMatchCmd(makeMessage("Tea. Green. Iced.", teaRegex))
 	item, err := GetUserItem(mb.DB(), "tester", ":tea:")
 	assert.Nil(t, err)
 	assert.Equal(t, 3, item.Count)
@@ -123,8 +128,8 @@ func TestTeaGreenHot(t *testing.T) {
 func TestTeaUnrelated(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("Tea."))
-	c.message(makeMessage("Tea. It's great."))
+	c.teaMatchCmd(makeMessage("Tea.", teaRegex))
+	c.teaMatchCmd(makeMessage("Tea. It's great.", teaRegex))
 	item, err := GetUserItem(mb.DB(), "tester", ":tea:")
 	assert.Nil(t, err)
 	assert.Equal(t, 0, item.Count)
@@ -133,7 +138,7 @@ func TestTeaUnrelated(t *testing.T) {
 func TestTeaSkieselQuote(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("blah, this is a whole page of explanation where \"we did local search and used a tabu list\" would have sufficed"))
+	c.teaMatchCmd(makeMessage("blah, this is a whole page of explanation where \"we did local search and used a tabu list\" would have sufficed", teaRegex))
 	item, err := GetUserItem(mb.DB(), "tester", ":tea:")
 	assert.Nil(t, err)
 	assert.Equal(t, 0, item.Count)
@@ -141,7 +146,7 @@ func TestTeaSkieselQuote(t *testing.T) {
 func TestTeaUnicodeJapanese(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("Tea. おちや. Hot."))
+	c.teaMatchCmd(makeMessage("Tea. おちや. Hot.", teaRegex))
 	item, err := GetUserItem(mb.DB(), "tester", ":tea:")
 	assert.Nil(t, err)
 	assert.Equal(t, 1, item.Count)
@@ -150,8 +155,8 @@ func TestTeaUnicodeJapanese(t *testing.T) {
 func TestResetMe(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("test++"))
-	c.message(makeMessage("!reset me"))
+	c.incrementCmd(makeMessage("test++", incrementRegex))
+	c.resetCmd(makeMessage("!reset me", resetRegex))
 	items, err := GetItems(mb.DB(), "tester")
 	assert.Nil(t, err)
 	assert.Len(t, items, 0)
@@ -160,7 +165,7 @@ func TestResetMe(t *testing.T) {
 func TestCounterOne(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage("test++"))
+	c.incrementCmd(makeMessage("test++", incrementRegex))
 	assert.Len(t, mb.Messages, 1)
 	assert.Equal(t, mb.Messages[0], "tester has 1 test.")
 }
@@ -168,7 +173,7 @@ func TestCounterOne(t *testing.T) {
 func TestCounterOneWithSpace(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
-	c.message(makeMessage(":test: ++"))
+	c.incrementCmd(makeMessage(":test: ++", incrementRegex))
 	assert.Len(t, mb.Messages, 1)
 	assert.Equal(t, mb.Messages[0], "tester has 1 :test:.")
 }
@@ -177,7 +182,7 @@ func TestCounterFour(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
 	for i := 0; i < 4; i++ {
-		c.message(makeMessage("test++"))
+		c.incrementCmd(makeMessage("test++", incrementRegex))
 	}
 	assert.Len(t, mb.Messages, 4)
 	assert.Equal(t, mb.Messages[3], "tester has 4 test.")
@@ -187,10 +192,10 @@ func TestCounterDecrement(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
 	for i := 0; i < 4; i++ {
-		c.message(makeMessage("test++"))
+		c.incrementCmd(makeMessage("test++", incrementRegex))
 		assert.Equal(t, mb.Messages[i], fmt.Sprintf("tester has %d test.", i+1))
 	}
-	c.message(makeMessage("test--"))
+	c.decrementCmd(makeMessage("test--", decrementRegex))
 	assert.Len(t, mb.Messages, 5)
 	assert.Equal(t, mb.Messages[4], "tester has 3 test.")
 }
@@ -199,10 +204,10 @@ func TestFriendCounterDecrement(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
 	for i := 0; i < 4; i++ {
-		c.message(makeMessage("other.test++"))
+		c.incrementCmd(makeMessage("other.test++", incrementRegex))
 		assert.Equal(t, mb.Messages[i], fmt.Sprintf("other has %d test.", i+1))
 	}
-	c.message(makeMessage("other.test--"))
+	c.decrementCmd(makeMessage("other.test--", decrementRegex))
 	assert.Len(t, mb.Messages, 5)
 	assert.Equal(t, mb.Messages[4], "other has 3 test.")
 }
@@ -211,12 +216,12 @@ func TestDecrementZero(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
 	for i := 0; i < 4; i++ {
-		c.message(makeMessage("test++"))
+		c.incrementCmd(makeMessage("test++", incrementRegex))
 		assert.Equal(t, mb.Messages[i], fmt.Sprintf("tester has %d test.", i+1))
 	}
 	j := 4
 	for i := 4; i > 0; i-- {
-		c.message(makeMessage("test--"))
+		c.decrementCmd(makeMessage("test--", decrementRegex))
 		assert.Equal(t, mb.Messages[j], fmt.Sprintf("tester has %d test.", i-1))
 		j++
 	}
@@ -228,10 +233,10 @@ func TestClear(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
 	for i := 0; i < 4; i++ {
-		c.message(makeMessage("test++"))
+		c.incrementCmd(makeMessage("test++", incrementRegex))
 		assert.Equal(t, mb.Messages[i], fmt.Sprintf("tester has %d test.", i+1))
 	}
-	res := c.message(makeMessage("!clear test"))
+	res := c.clearCmd(makeMessage("!clear test", clearRegex))
 	assert.True(t, res)
 	assert.Len(t, mb.Actions, 1)
 	assert.Equal(t, mb.Actions[0], "chops a few test out of his brain")
@@ -241,10 +246,10 @@ func TestCount(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
 	for i := 0; i < 4; i++ {
-		c.message(makeMessage("test++"))
+		c.incrementCmd(makeMessage("test++", incrementRegex))
 		assert.Equal(t, mb.Messages[i], fmt.Sprintf("tester has %d test.", i+1))
 	}
-	res := c.message(makeMessage("!count test"))
+	res := c.countCmd(makeMessage("!count test", countRegex))
 	assert.True(t, res)
 	assert.Len(t, mb.Messages, 5)
 	assert.Equal(t, mb.Messages[4], "tester has 4 test.")
@@ -254,18 +259,18 @@ func TestInspectMe(t *testing.T) {
 	mb, c := setup(t)
 	assert.NotNil(t, c)
 	for i := 0; i < 4; i++ {
-		c.message(makeMessage("test++"))
+		c.incrementCmd(makeMessage("test++", incrementRegex))
 		assert.Equal(t, mb.Messages[i], fmt.Sprintf("tester has %d test.", i+1))
 	}
 	for i := 0; i < 2; i++ {
-		c.message(makeMessage("fucks++"))
+		c.incrementCmd(makeMessage("fucks++", incrementRegex))
 		assert.Equal(t, mb.Messages[i+4], fmt.Sprintf("tester has %d fucks.", i+1))
 	}
 	for i := 0; i < 20; i++ {
-		c.message(makeMessage("cheese++"))
+		c.incrementCmd(makeMessage("cheese++", incrementRegex))
 		assert.Equal(t, mb.Messages[i+6], fmt.Sprintf("tester has %d cheese.", i+1))
 	}
-	res := c.message(makeMessage("!inspect me"))
+	res := c.inspectCmd(makeMessage("!inspect me", inspectRegex))
 	assert.True(t, res)
 	assert.Len(t, mb.Messages, 27)
 	assert.Equal(t, mb.Messages[26], "tester has the following counters: test: 4, fucks: 2, cheese: 20.")