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

do not learn from commands to the babbler

This commit is contained in:
skiesel 2016-05-11 12:11:21 -04:00
parent e6324ad5a2
commit ac3e1da9e5
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
plugins/babbler/babbler.go
View File

@ -80,9 +80,14 @@ func (p *BabblerPlugin) Message(message msg.Message) bool {
lowercase := strings.ToLower(message.Body)
tokens := strings.Fields(lowercase)
addToMarkovChain(p.babblers[message.User.Name], lowercase)
if len(tokens) == 4 && strings.Contains(lowercase, "initialize babbler for ") {
if len(tokens) == 2 && tokens[1] == "says" {
saying := p.babble(tokens[0])
if saying == "" {
p.Bot.SendMessage(message.Channel, "Ze ain't said nothin'")
}
p.Bot.SendMessage(message.Channel, saying)
return true
} else if len(tokens) == 4 && strings.Contains(lowercase, "initialize babbler for ") {
who := tokens[len(tokens)-1]
if _, ok := p.babblers[who]; !ok {
babbler, err := getMarkovChain(p.db, who)
@ -94,16 +99,11 @@ func (p *BabblerPlugin) Message(message msg.Message) bool {
p.Bot.SendMessage(message.Channel, "Okay.")
return true
}
} else {
addToMarkovChain(p.babblers[message.User.Name], lowercase)
}
if len(tokens) == 2 && tokens[1] == "says" {
saying := p.babble(tokens[0])
if saying == "" {
p.Bot.SendMessage(message.Channel, "Ze ain't said nothin'")
}
p.Bot.SendMessage(message.Channel, saying)
return true
}
return false
}