amigan/stillbox
1
0
Fork 0
Code Issues 16 Pull requests Projects Releases Packages Wiki Activity Actions

Use single template with named subs

Browse source
This commit is contained in:
Daniel 2024-11-08 07:19:13 -05:00
parent 6ea61bf820
commit 2153dc8434
1 changed files with 28 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
pkg/notify/notify.go
View file

@ -50,15 +50,31 @@ var alertFm = template.FuncMap{
"highest": highest, "highest": highest,
} }
const defaultBodyTemplStr = `{{ range . -}} const (
defaultBodyTemplStr = `{{ range . -}}
{{ .TGName }} is active with a score of {{ f .Score.Score 4 }}! ({{ f .Score.RecentCount 0 }}/{{ .Score.Count }} recent calls) {{ .TGName }} is active with a score of {{ f .Score.Score 4 }}! ({{ f .Score.RecentCount 0 }}/{{ .Score.Count }} recent calls)
{{ end -}}` {{ end -}}`
defaultSubjectTemplStr = `Stillbox Alert ({{ highest . }}`
)
var defaultBodyTemplate = template.Must(template.New("body").Funcs(common.FuncMap).Funcs(alertFm).Parse(defaultBodyTemplStr)) var (
defaultTemplate *template.Template
)
var defaultSubjectTemplStr = `Stillbox Alert ({{ highest . }}` func init() {
var defaultSubjectTemplate = template.Must(template.New("subject").Funcs(common.FuncMap).Funcs(alertFm).Parse(defaultSubjectTemplStr)) defaultTemplate = template.New("notification")
defaultTemplate.Funcs(common.FuncMap).Funcs(alertFm)
_, err := defaultTemplate.New("body").Parse(defaultBodyTemplStr)
if err != nil {
panic(err)
}
_, err = defaultTemplate.New("subject").Parse(defaultSubjectTemplStr)
if err != nil {
panic(err)
}
}
// Send renders and sends the Alerts. // Send renders and sends the Alerts.
func (b *backend) Send(ctx context.Context, alerts []alert.Alert) (err error) { func (b *backend) Send(ctx context.Context, alerts []alert.Alert) (err error) {
@ -125,7 +141,10 @@ func (n *notifier) addService(cfg config.NotifyService) (err error) {
switch cfg.SubjectTemplate { switch cfg.SubjectTemplate {
case nil: case nil:
be.subject = defaultSubjectTemplate be.subject = defaultTemplate.Lookup("subject")
if be.subject == nil {
panic("subject template nil")
}
default: default:
be.subject, err = template.New("subject").Funcs(common.FuncMap).Funcs(alertFm).Parse(*cfg.SubjectTemplate) be.subject, err = template.New("subject").Funcs(common.FuncMap).Funcs(alertFm).Parse(*cfg.SubjectTemplate)
if err != nil { if err != nil {
@ -135,7 +154,10 @@ func (n *notifier) addService(cfg config.NotifyService) (err error) {
switch cfg.BodyTemplate { switch cfg.BodyTemplate {
case nil: case nil:
be.body = defaultBodyTemplate be.body = defaultTemplate.Lookup("body")
if be.body == nil {
panic("body template nil")
}
default: default:
be.body, err = template.New("body").Funcs(common.FuncMap).Funcs(alertFm).Parse(*cfg.BodyTemplate) be.body, err = template.New("body").Funcs(common.FuncMap).Funcs(alertFm).Parse(*cfg.BodyTemplate)
if err != nil { if err != nil {