twitch-bot/msgformatter.go

51 lines
1 KiB
Go
Raw Normal View History

2020-12-21 00:32:39 +00:00
package main
import (
"bytes"
"text/template"
"time"
"github.com/go-irc/irc"
"github.com/pkg/errors"
)
func formatMessage(tplString string, m *irc.Message, r *rule, fields map[string]interface{}) (string, error) {
compiledFields := map[string]interface{}{}
if config != nil {
configLock.RLock()
for k, v := range config.Variables {
compiledFields[k] = v
}
compiledFields["permitTimeout"] = int64(config.PermitTimeout / time.Second)
configLock.RUnlock()
}
for k, v := range fields {
compiledFields[k] = v
}
if m != nil {
compiledFields["msg"] = m
compiledFields["username"] = m.User
if m.Command == "PRIVMSG" && len(m.Params) > 0 {
compiledFields["channel"] = m.Params[0]
}
}
// Parse and execute template
tpl, err := template.
New(tplString).
Funcs(tplFuncs.GetFuncMap(m, r, compiledFields)).
Parse(tplString)
if err != nil {
return "", errors.Wrap(err, "parse template")
}
buf := new(bytes.Buffer)
err = tpl.Execute(buf, compiledFields)
return buf.String(), errors.Wrap(err, "execute template")
2020-12-21 00:32:39 +00:00
}