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

Changing factoid random a bit to improve efficiency, added 'factoid' command for random fact

This commit is contained in:
Chris Sexton 2013-06-01 16:46:16 -04:00
parent 11feafbee0
commit 79bec9fc21
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
plugins/factoid.go
View File

@ -347,6 +347,15 @@ func (p *FactoidPlugin) Message(message bot.Message) bool {
return p.trigger(message)
}
if strings.ToLower(message.Body) == "factoid" {
if fact := p.randomFact(); fact != nil {
p.sayFact(message, *fact)
return true
} else {
log.Println("Got a nil fact.")
}
}
if strings.ToLower(message.Body) == "forget that" {
return p.forgetLastFact(message)
}
@ -391,19 +400,17 @@ func (p *FactoidPlugin) Event(kind string, message bot.Message) bool {
// Pull a fact at random from the database
func (p *FactoidPlugin) randomFact() *Factoid {
var results []Factoid
iter := p.Coll.Find(bson.M{}).Iter()
err := iter.All(&results)
if err != nil {
panic(err)
}
var fact Factoid
nfacts := len(results)
if nfacts == 0 {
nFacts, err := p.Coll.Count()
if err != nil {
return nil
}
fact := results[rand.Intn(nfacts)]
if err := p.Coll.Find(nil).Skip(rand.Intn(nFacts)).One(&fact); err != nil {
log.Println("Couldn't get next...")
}
return &fact
}