mirror of https://github.com/velour/catbase.git
74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
// © 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(b bot.Bot) *CSWPlugin {
|
|
csw := &CSWPlugin{
|
|
Bot: b,
|
|
}
|
|
b.Register(csw, bot.Message, csw.message)
|
|
return csw
|
|
}
|
|
|
|
func (p *CSWPlugin) message(kind bot.Kind, message msg.Message, args ...interface{}) 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.Send(bot.Message, message.Channel, responses[rand.Intn(len(responses))])
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|