picker: refactor

plugins/picker/picker.go

@@ -26,7 +26,7 @@ func New(b bot.Bot) *PickerPlugin {
 	pp := &PickerPlugin{
 		bot: b,
 	}
-	b.Register(pp, bot.Message, pp.message)
+	b.RegisterRegex(pp, bot.Message, pickRegex, pp.message)
 	b.Register(pp, bot.Help, pp.help)
 	return pp
 }
@@ -34,14 +34,12 @@ func New(b bot.Bot) *PickerPlugin {
 // 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 *PickerPlugin) message(c bot.Connector, kind bot.Kind, message msg.Message, args ...interface{}) bool {
-	if !strings.HasPrefix(message.Body, "pick") || !message.Command {
-		return false
-	}
+func (p *PickerPlugin) message(r bot.Request) bool {
+	message := r.Msg
 
 	n, items, err := p.parse(message.Body)
 	if err != nil {
-		p.bot.Send(c, bot.Message, message.Channel, err.Error())
+		p.bot.Send(r.Conn, bot.Message, message.Channel, err.Error())
 		return true
 	}
 
@@ -50,7 +48,7 @@ func (p *PickerPlugin) message(c bot.Connector, kind bot.Kind, message msg.Messa
 		for _, pref := range preferences {
 			if pref == it {
 				out := fmt.Sprintf("I've chosen %q for you.", strings.TrimSpace(it))
-				p.bot.Send(c, bot.Message, message.Channel, out)
+				p.bot.Send(r.Conn, bot.Message, message.Channel, out)
 				return true
 			}
 		}
@@ -59,7 +57,7 @@ func (p *PickerPlugin) message(c bot.Connector, kind bot.Kind, message msg.Messa
 	if n == 1 {
 		item := items[rand.Intn(len(items))]
 		out := fmt.Sprintf("I've chosen %q for you.", strings.TrimSpace(item))
-		p.bot.Send(c, bot.Message, message.Channel, out)
+		p.bot.Send(r.Conn, bot.Message, message.Channel, out)
 		return true
 	}
 
@@ -75,7 +73,7 @@ func (p *PickerPlugin) message(c bot.Connector, kind bot.Kind, message msg.Messa
 		fmt.Fprintf(&b, ", %q", item)
 	}
 	b.WriteString(" }")
-	p.bot.Send(c, bot.Message, message.Channel, b.String())
+	p.bot.Send(r.Conn, bot.Message, message.Channel, b.String())
 	return true
 }
 

plugins/picker/picker_test.go

@@ -8,24 +8,29 @@ import (
 
 	"github.com/rs/zerolog/log"
 
-	"github.com/velour/catbase/plugins/cli"
-
 	"github.com/stretchr/testify/assert"
 	"github.com/velour/catbase/bot"
 	"github.com/velour/catbase/bot/msg"
 	"github.com/velour/catbase/bot/user"
+	"github.com/velour/catbase/plugins/cli"
 )
 
-func makeMessage(payload string) (bot.Connector, bot.Kind, msg.Message) {
+func makeMessage(payload string) bot.Request {
 	isCmd := strings.HasPrefix(payload, "!")
 	if isCmd {
 		payload = payload[1:]
 	}
-	return &cli.CliPlugin{}, bot.Message, msg.Message{
+	values := bot.ParseValues(pickRegex, payload)
+	return bot.Request{
+		Conn:   &cli.CliPlugin{},
+		Kind:   bot.Message,
+		Values: values,
+		Msg: msg.Message{
 			User:    &user.User{Name: "tester"},
 			Channel: "test",
 			Body:    payload,
 			Command: isCmd,
+		},
 	}
 }
 
@@ -108,10 +113,3 @@ func TestPickIsCommand(t *testing.T) {
 	assert.NotContains(t, mb.Messages[0], "hot picks")
 	log.Debug().Str("resp", mb.Messages[0]).Msg("choose")
 }
-
-func TestPickMustBeCommand(t *testing.T) {
-	mb := bot.NewMockBot()
-	c := New(mb)
-	_ = c.message(makeMessage("pick⌘ { bagel/egg/smoked turkey/butte/cheese ⌘ fuck all that, just have a bagel and cream cheese }"))
-	assert.Len(t, mb.Messages, 0)
-}