mirror of https://github.com/velour/catbase.git
This might be entertaining on occasion
This commit is contained in:
parent
c639d460ac
commit
f03fdb4d4e
2
main.go
2
main.go
|
@ -13,6 +13,7 @@ import (
|
||||||
"github.com/velour/catbase/plugins/babbler"
|
"github.com/velour/catbase/plugins/babbler"
|
||||||
"github.com/velour/catbase/plugins/beers"
|
"github.com/velour/catbase/plugins/beers"
|
||||||
"github.com/velour/catbase/plugins/counter"
|
"github.com/velour/catbase/plugins/counter"
|
||||||
|
"github.com/velour/catbase/plugins/couldashouldwoulda"
|
||||||
"github.com/velour/catbase/plugins/db"
|
"github.com/velour/catbase/plugins/db"
|
||||||
"github.com/velour/catbase/plugins/dice"
|
"github.com/velour/catbase/plugins/dice"
|
||||||
"github.com/velour/catbase/plugins/emojifyme"
|
"github.com/velour/catbase/plugins/emojifyme"
|
||||||
|
@ -78,6 +79,7 @@ func main() {
|
||||||
b.AddHandler("rpgORdie", rpgORdie.New(b))
|
b.AddHandler("rpgORdie", rpgORdie.New(b))
|
||||||
b.AddHandler("sisyphus", sisyphus.New(b))
|
b.AddHandler("sisyphus", sisyphus.New(b))
|
||||||
b.AddHandler("tell", tell.New(b))
|
b.AddHandler("tell", tell.New(b))
|
||||||
|
b.AddHandler("couldashouldwoulda", couldashouldwoulda.New(b))
|
||||||
// catches anything left, will always return true
|
// catches anything left, will always return true
|
||||||
b.AddHandler("factoid", fact.New(b))
|
b.AddHandler("factoid", fact.New(b))
|
||||||
b.AddHandler("db", db.New(b))
|
b.AddHandler("db", db.New(b))
|
||||||
|
|
|
@ -0,0 +1,85 @@
|
||||||
|
// © 2013 the CatBase Authors under the WTFPL. See AUTHORS for the list of authors.
|
||||||
|
|
||||||
|
package couldashouldawoulda
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"math/rand"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/velour/catbase/bot"
|
||||||
|
"github.com/velour/catbase/bot/msg"
|
||||||
|
"github.com/velour/catbase/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CSWPlugin struct {
|
||||||
|
Bot bot.Bot
|
||||||
|
Config *config.Config
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(bot bot.Bot) *CSWPlugin {
|
||||||
|
return &CSWPlugin{
|
||||||
|
Bot: bot,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *CSWPlugin) Message(message msg.Message) bool {
|
||||||
|
if !message.Command {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
lowercase := strings.ToLower(message.Body)
|
||||||
|
tokens := strings.Fields(lowercase)
|
||||||
|
|
||||||
|
if len(tokens) < 3 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
could := tokens[0] == "could"
|
||||||
|
should := tokens[0] == "should"
|
||||||
|
would := tokens[0] == "would"
|
||||||
|
|
||||||
|
if could || should || would {
|
||||||
|
ors := strings.Count(lowercase, " or ")
|
||||||
|
var responses []string
|
||||||
|
if ors == 0 || could {
|
||||||
|
responses = []string{"Yes.", "No.", "Maybe.", "For fucks sake, how should I know?"}
|
||||||
|
} else if ors == 1 {
|
||||||
|
responses = []string{"The former.", "The latter.", "Obviously the former.", "Clearly the latter.", "Can't it be both?"}
|
||||||
|
} else {
|
||||||
|
responses = make([]string, ors*2)
|
||||||
|
for i := 0; i < ors; i++ {
|
||||||
|
responses[i*2] = fmt.Sprintf("I'd say option %d.", i+1)
|
||||||
|
responses[i*2+1] = "You'd be an idiot not to choose the "
|
||||||
|
if i == 0 {
|
||||||
|
responses[i*2+1] += " 1st"
|
||||||
|
} else if i == 1 {
|
||||||
|
responses[i*2+1] += " 2nd"
|
||||||
|
} else if i == 2 {
|
||||||
|
responses[i*2+1] += " 3rd"
|
||||||
|
} else {
|
||||||
|
responses[i*2+1] += fmt.Sprintf(" %dth", i+1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
p.Bot.SendMessage(message.Channel, responses[rand.Intn(len(responses))])
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *CSWPlugin) Help(channel string, parts []string) {}
|
||||||
|
|
||||||
|
func (p *CSWPlugin) Event(kind string, message msg.Message) bool {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *CSWPlugin) BotMessage(message msg.Message) bool {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *CSWPlugin) RegisterWeb() *string {
|
||||||
|
return nil
|
||||||
|
}
|
|
@ -0,0 +1,134 @@
|
||||||
|
// © 2013 the CatBase Authors under the WTFPL. See AUTHORS for the list of authors.
|
||||||
|
|
||||||
|
package couldashouldawoulda
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/velour/catbase/bot"
|
||||||
|
"github.com/velour/catbase/bot/msg"
|
||||||
|
"github.com/velour/catbase/bot/user"
|
||||||
|
)
|
||||||
|
|
||||||
|
func makeMessage(payload string) msg.Message {
|
||||||
|
isCmd := strings.HasPrefix(payload, "!")
|
||||||
|
if isCmd {
|
||||||
|
payload = payload[1:]
|
||||||
|
}
|
||||||
|
return msg.Message{
|
||||||
|
User: &user.User{Name: "tester"},
|
||||||
|
Channel: "test",
|
||||||
|
Body: payload,
|
||||||
|
Command: isCmd,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test0(t *testing.T) {
|
||||||
|
mb := bot.NewMockBot()
|
||||||
|
c := New(mb)
|
||||||
|
assert.NotNil(t, c)
|
||||||
|
res := c.Message(makeMessage("!should I drink a beer?"))
|
||||||
|
assert.Len(t, mb.Messages, 1)
|
||||||
|
assert.True(t, res)
|
||||||
|
possibilities := []string{"Yes.", "No.", "Maybe.", "For fucks sake, how should I know?"}
|
||||||
|
match := false
|
||||||
|
for _, possibility := range possibilities {
|
||||||
|
if strings.Contains(mb.Messages[0], possibility) {
|
||||||
|
match = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assert.True(t, match)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test1(t *testing.T) {
|
||||||
|
mb := bot.NewMockBot()
|
||||||
|
c := New(mb)
|
||||||
|
assert.NotNil(t, c)
|
||||||
|
res := c.Message(makeMessage("!should I drink a beer or a bourbon?"))
|
||||||
|
assert.Len(t, mb.Messages, 1)
|
||||||
|
assert.True(t, res)
|
||||||
|
possibilities := []string{"The former.", "The latter.", "Obviously the former.", "Clearly the latter.", "Can't it be both?"}
|
||||||
|
match := false
|
||||||
|
for _, possibility := range possibilities {
|
||||||
|
if strings.Contains(mb.Messages[0], possibility) {
|
||||||
|
match = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assert.True(t, match)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test2(t *testing.T) {
|
||||||
|
mb := bot.NewMockBot()
|
||||||
|
c := New(mb)
|
||||||
|
assert.NotNil(t, c)
|
||||||
|
res := c.Message(makeMessage("!could I drink a beer or a bourbon?"))
|
||||||
|
assert.Len(t, mb.Messages, 1)
|
||||||
|
assert.True(t, res)
|
||||||
|
possibilities := []string{"Yes.", "No.", "Maybe.", "For fucks sake, how should I know?"}
|
||||||
|
match := false
|
||||||
|
for _, possibility := range possibilities {
|
||||||
|
if strings.Contains(mb.Messages[0], possibility) {
|
||||||
|
match = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assert.True(t, match)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test3(t *testing.T) {
|
||||||
|
mb := bot.NewMockBot()
|
||||||
|
c := New(mb)
|
||||||
|
assert.NotNil(t, c)
|
||||||
|
res := c.Message(makeMessage("!would I die if I drank too much bourbon?"))
|
||||||
|
assert.Len(t, mb.Messages, 1)
|
||||||
|
assert.True(t, res)
|
||||||
|
possibilities := []string{"Yes.", "No.", "Maybe.", "For fucks sake, how should I know?"}
|
||||||
|
match := false
|
||||||
|
for _, possibility := range possibilities {
|
||||||
|
if strings.Contains(mb.Messages[0], possibility) {
|
||||||
|
match = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assert.True(t, match)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test4(t *testing.T) {
|
||||||
|
mb := bot.NewMockBot()
|
||||||
|
c := New(mb)
|
||||||
|
assert.NotNil(t, c)
|
||||||
|
res := c.Message(makeMessage("!would I die or be sick if I drank all the bourbon?"))
|
||||||
|
assert.Len(t, mb.Messages, 1)
|
||||||
|
assert.True(t, res)
|
||||||
|
possibilities := []string{"The former.", "The latter.", "Obviously the former.", "Clearly the latter.", "Can't it be both?"}
|
||||||
|
match := false
|
||||||
|
for _, possibility := range possibilities {
|
||||||
|
if strings.Contains(mb.Messages[0], possibility) {
|
||||||
|
match = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assert.True(t, match)
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test5(t *testing.T) {
|
||||||
|
mb := bot.NewMockBot()
|
||||||
|
c := New(mb)
|
||||||
|
assert.NotNil(t, c)
|
||||||
|
res := c.Message(makeMessage("!should I have another beer or bourbon or tequila?"))
|
||||||
|
assert.Len(t, mb.Messages, 1)
|
||||||
|
assert.True(t, res)
|
||||||
|
possibilities := []string{"I'd say option", "You'd be an idiot not to choose the"}
|
||||||
|
match := false
|
||||||
|
for _, possibility := range possibilities {
|
||||||
|
if strings.Contains(mb.Messages[0], possibility) {
|
||||||
|
match = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assert.True(t, match)
|
||||||
|
}
|
Loading…
Reference in New Issue