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

Track who said what

This commit is contained in:
skiesel 2019-03-21 21:36:11 -04:00
parent 650ccb190d
commit b4b09e74c6
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
plugins/tldr/tldr.go
View File

@ -19,6 +19,7 @@ var (
type TLDRPlugin struct { type TLDRPlugin struct {
Bot bot.Bot Bot bot.Bot
History []string History []string
Users []string
Index int Index int
} }
@ -26,6 +27,7 @@ func New(b bot.Bot) *TLDRPlugin {
plugin := &TLDRPlugin{ plugin := &TLDRPlugin{
Bot: b, Bot: b,
History: []string{}, History: []string{},
Users: []string{},
Index: 0, Index: 0,
} }
b.Register(plugin, bot.Message, plugin.message) b.Register(plugin, bot.Message, plugin.message)
@ -54,6 +56,7 @@ func (p *TLDRPlugin) message(kind bot.Kind, message msg.Message, args ...interfa
bestScores := make([]float64, nTopics) bestScores := make([]float64, nTopics)
bestDocs := make([]string, nTopics) bestDocs := make([]string, nTopics)
bestUsers := make([]string, nTopics)
dr, dc := docsOverTopics.Dims() dr, dc := docsOverTopics.Dims()
for doc := 0; doc < dc; doc++ { for doc := 0; doc < dc; doc++ {
@ -62,6 +65,7 @@ func (p *TLDRPlugin) message(kind bot.Kind, message msg.Message, args ...interfa
if score > bestScores[topic] { if score > bestScores[topic] {
bestScores[topic] = score bestScores[topic] = score
bestDocs[topic] = p.History[doc] bestDocs[topic] = p.History[doc]
bestUsers[topic] = p.Users[doc]
} }
} }
} }
@ -77,17 +81,17 @@ func (p *TLDRPlugin) message(kind bot.Kind, message msg.Message, args ...interfa
response := "Here you go captain 'too good to read backlog':\n" response := "Here you go captain 'too good to read backlog':\n"
for topic := 0; topic < tr; topic++ { for topic := 0; topic < tr; topic++ {
max := -1. bestScore := -1.
best := "" bestTopic := ""
for word := 0; word < tc; word++ { for word := 0; word < tc; word++ {
score := topicsOverWords.At(topic, word) score := topicsOverWords.At(topic, word)
if score > max { if score > bestScore {
max = score bestScore = score
best = vocab[word] bestTopic = vocab[word]
} }
} }
response += fmt.Sprintf("Topic #%d : %s\n", topic, best) response += fmt.Sprintf("Topic #%d : %s\n", topic, bestTopic)
response += fmt.Sprintf("\t%s\n", bestDocs[topic]) response += fmt.Sprintf("\t<%s>%s\n", bestUsers[topic], bestDocs[topic])
} }
p.Bot.Send(bot.Message, message.Channel, response) p.Bot.Send(bot.Message, message.Channel, response)
@ -100,6 +104,7 @@ func (p *TLDRPlugin) message(kind bot.Kind, message msg.Message, args ...interfa
maxHistorySize := p.Bot.Config().GetInt("TLDR.HistorySize", 1000) maxHistorySize := p.Bot.Config().GetInt("TLDR.HistorySize", 1000)
if currentHistorySize < maxHistorySize { if currentHistorySize < maxHistorySize {
p.History = append(p.History, lowercaseMessage) p.History = append(p.History, lowercaseMessage)
p.Users = append(p.Users, message.User.Name)
p.Index = 0 p.Index = 0
} else { } else {
if currentHistorySize > maxHistorySize { if currentHistorySize > maxHistorySize {
@ -112,6 +117,7 @@ func (p *TLDRPlugin) message(kind bot.Kind, message msg.Message, args ...interfa
} }
p.History[p.Index] = lowercaseMessage p.History[p.Index] = lowercaseMessage
p.Users[p.Index] = message.User.Name
p.Index++ p.Index++
} }
} }