luzifer/twitch-bot
1
0
Fork 0
mirror of https://github.com/Luzifer/twitch-bot.git synced 2024-11-09 08:40:01 +00:00
Code Issues Projects Releases Wiki Activity

Reorganize template functions, add recentGame

Browse source 
Signed-off-by: Knut Ahlers <knut@ahlers.me>
This commit is contained in:
Knut Ahlers 2020-12-24 13:19:58 +01:00
parent e19fe57ec7
commit b3defa44fd
Signed by: luzifer
GPG key ID: 0066F03ED215AD7D
1 changed files with 22 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
msgformatter.go
View file

@ -12,15 +12,12 @@ import (
"github.com/pkg/errors"
)
func formatMessage(tplString string, m *irc.Message, fields map[string]interface{}) (string, error) {
fm := korvike.GetFunctionMap()
fm["getArg"] = tplGetMessageArg
fm["getCounterValue"] = tplGetCounterValue
fm["getTag"] = tplGetTagFromMessage
var messageFunctions = korvike.GetFunctionMap()
func formatMessage(tplString string, m *irc.Message, fields map[string]interface{}) (string, error) {
tpl, err := template.
New(tplString).
Funcs(fm).
Funcs(messageFunctions).
Parse(tplString)
if err != nil {
return "", errors.Wrap(err, "parse template")
@ -40,20 +37,27 @@ func formatMessage(tplString string, m *irc.Message, fields map[string]interface
return buf.String(), errors.Wrap(err, "execute template")
}
func tplGetCounterValue(name string, _ ...string) int64 {
return store.GetCounterValue(name)
}
func init() {
messageFunctions["getArg"] = func(m *irc.Message, params ...int) (string, error) {
msgParts := strings.Split(m.Trailing(), " ")
if len(msgParts) < params[0]+1 {
return "", errors.New("argument not found")
}
func tplGetMessageArg(m *irc.Message, params ...int) (string, error) {
msgParts := strings.Split(m.Trailing(), " ")
if len(msgParts) < params[0]+1 {
return "", errors.New("argument not found")
return msgParts[params[0]], nil
}
return msgParts[params[0]], nil
}
messageFunctions["getCounterValue"] = func(name string, _ ...string) int64 {
return store.GetCounterValue(name)
}
func tplGetTagFromMessage(m *irc.Message, params ...string) string {
s, _ := m.GetTag(params[0])
return s
messageFunctions["getTag"] = func(m *irc.Message, params ...string) string {
s, _ := m.GetTag(params[0])
return s
}
messageFunctions["recentGame"] = func(username string, _ ...string) (string, error) {
game, _, err := twitch.getRecentStreamInfo(username)
return game, err
}
}