cws
/
catbase
mirror of https://github.com/velour/catbase.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Adding ~= and =~ equality for factoids

This commit is contained in:
Chris Sexton 2013-03-18 17:44:49 -04:00
parent 340b7ee1ae
commit ee1962b71c
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
plugins/factoid.go
View File

@ -217,9 +217,14 @@ func (p *FactoidPlugin) learnAction(message bot.Message, action string) bool {
return true
}
// Checks body for the ~= operator
func changeOperator(body string) bool {
return strings.Contains(body, "=~")
// Checks body for the ~= operator returns it
func changeOperator(body string) string {
if strings.Contains(body, "=~") {
return "=~"
} else if strings.Contains(body, "~=") {
return "~="
}
return ""
}
// If the user requesting forget is either the owner of the last learned fact or
@ -247,7 +252,8 @@ func (p *FactoidPlugin) forgetLastFact(message bot.Message) bool {
// Allow users to change facts with a simple regexp
func (p *FactoidPlugin) changeFact(message bot.Message) bool {
parts := strings.SplitN(message.Body, "=~", 2)
oper := changeOperator(message.Body)
parts := strings.SplitN(message.Body, oper, 2)
userexp := strings.TrimSpace(parts[1])
trigger := strings.TrimSpace(parts[0])
@ -342,7 +348,7 @@ func (p *FactoidPlugin) Message(message bot.Message) bool {
return p.forgetLastFact(message)
}
if changeOperator(message.Body) {
if changeOperator(message.Body) != "" {
return p.changeFact(message)
}