Change to use the lua config…

This commit is contained in:
Steve McCoy 2017-12-03 13:04:55 -05:00
parent 85b649bc98
commit 5794c73efd
3 changed files with 19 additions and 4 deletions

View File

@ -91,6 +91,7 @@ type Config struct {
} }
Emojify struct { Emojify struct {
Chance float64 Chance float64
Scoreless []string
} }
Reaction struct { Reaction struct {
GeneralChance float64 GeneralChance float64

View File

@ -29,7 +29,11 @@ config = {
YourChance = 0.4 YourChance = 0.4
}, },
Emojify = { Emojify = {
Chance = 0.02 Chance = 0.02,
Scoreless = {
"a",
"it"
}
}, },
DB = { DB = {
File = "catbase.db", File = "catbase.db",

View File

@ -67,11 +67,12 @@ func (p *EmojifyMePlugin) Message(message msg.Message) bool {
} }
} }
inertTokens := p.Bot.Config().Emojify.Scoreless
emojied := 0.0 emojied := 0.0
tokens := strings.Fields(strings.ToLower(message.Body)) tokens := strings.Fields(strings.ToLower(message.Body))
for i, token := range tokens { for i, token := range tokens {
if _, ok := p.Emoji[token]; ok { if _, ok := p.Emoji[token]; ok {
if token != "a" && token != "it" { if !stringsContain(inertTokens, token) {
emojied++ emojied++
} }
tokens[i] = ":" + token + ":" tokens[i] = ":" + token + ":"
@ -79,7 +80,7 @@ func (p *EmojifyMePlugin) Message(message msg.Message) bool {
//Check to see if we can strip the trailing "es" off and get an emoji //Check to see if we can strip the trailing "es" off and get an emoji
temp := strings.TrimSuffix(token, "s") temp := strings.TrimSuffix(token, "s")
if _, ok := p.Emoji[temp]; ok { if _, ok := p.Emoji[temp]; ok {
if temp != "a" && temp != "it" { if !stringsContain(inertTokens, temp) {
emojied++ emojied++
} }
tokens[i] = ":" + temp + ":s" tokens[i] = ":" + temp + ":s"
@ -87,7 +88,7 @@ func (p *EmojifyMePlugin) Message(message msg.Message) bool {
//Check to see if we can strip the trailing "es" off and get an emoji //Check to see if we can strip the trailing "es" off and get an emoji
temp := strings.TrimSuffix(token, "es") temp := strings.TrimSuffix(token, "es")
if _, ok := p.Emoji[temp]; ok { if _, ok := p.Emoji[temp]; ok {
if temp != "a" && temp != "it" { if !stringsContain(inertTokens, temp) {
emojied++ emojied++
} }
tokens[i] = ":" + temp + ":es" tokens[i] = ":" + temp + ":es"
@ -120,3 +121,12 @@ func (p *EmojifyMePlugin) RegisterWeb() *string {
} }
func (p *EmojifyMePlugin) ReplyMessage(message msg.Message, identifier string) bool { return false } func (p *EmojifyMePlugin) ReplyMessage(message msg.Message, identifier string) bool { return false }
func stringsContain(haystack []string, needle string) bool {
for _, s := range haystack {
if s == needle {
return true
}
}
return false
}