mirror of
https://github.com/pushbits/server.git
synced 2025-04-30 02:36:53 +02:00
101 lines
2.7 KiB
Go
101 lines
2.7 KiB
Go
package dispatcher
|
|
|
|
import (
|
|
"fmt"
|
|
"html"
|
|
"log"
|
|
"strings"
|
|
|
|
"github.com/gomarkdown/markdown"
|
|
"github.com/pushbits/server/internal/model"
|
|
)
|
|
|
|
// SendNotification sends a notification to the specified user.
|
|
func (d *Dispatcher) SendNotification(a *model.Application, n *model.Notification) error {
|
|
log.Printf("Sending notification to room %s.", a.MatrixID)
|
|
|
|
plainMessage := strings.TrimSpace(n.Message)
|
|
plainTitle := strings.TrimSpace(n.Title)
|
|
message := d.getFormattedMessage(n)
|
|
title := d.getFormattedTitle(n)
|
|
|
|
text := fmt.Sprintf("%s\n\n%s", plainTitle, plainMessage)
|
|
formattedText := fmt.Sprintf("%s %s", title, message)
|
|
|
|
_, err := d.client.SendFormattedText(a.MatrixID, text, formattedText)
|
|
|
|
return err
|
|
}
|
|
|
|
// HTML-formats the title
|
|
func (d *Dispatcher) getFormattedTitle(n *model.Notification) string {
|
|
trimmedTitle := strings.TrimSpace(n.Title)
|
|
title := html.EscapeString(trimmedTitle)
|
|
|
|
if d.formatting.ColoredTitle {
|
|
title = d.coloredText(d.priorityToColor(n.Priority), title)
|
|
}
|
|
|
|
return "<b>" + title + "</b><br /><br />"
|
|
}
|
|
|
|
// Converts different syntaxes to a HTML-formatted message
|
|
func (d *Dispatcher) getFormattedMessage(n *model.Notification) string {
|
|
trimmedMessage := strings.TrimSpace(n.Message)
|
|
message := strings.Replace(html.EscapeString(trimmedMessage), "\n", "<br />", -1) // default to text/plain
|
|
|
|
if optionsDisplayRaw, ok := n.Extras["client::display"]; ok {
|
|
optionsDisplay, ok := optionsDisplayRaw.(map[string]interface{})
|
|
log.Printf("%s", optionsDisplay)
|
|
|
|
if ok {
|
|
if contentTypeRaw, ok := optionsDisplay["contentType"]; ok {
|
|
contentType := fmt.Sprintf("%v", contentTypeRaw)
|
|
log.Printf("Message content type: %s", contentType)
|
|
|
|
switch contentType {
|
|
case "html", "text/html":
|
|
message = strings.Replace(trimmedMessage, "\n", "<br />", -1)
|
|
case "markdown", "md", "text/md", "text/markdown":
|
|
// allow HTML in Markdown
|
|
message = string(markdown.ToHTML([]byte(trimmedMessage), nil, nil))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return message
|
|
}
|
|
|
|
// Maps priorities to hex colors
|
|
func (d *Dispatcher) priorityToColor(prio int) string {
|
|
switch prio {
|
|
case 0: // emergency - dark red
|
|
return "#cc0000"
|
|
case 1: // alert - red
|
|
return "#ed1f11"
|
|
case 2: // critical - dark orange
|
|
return "#ed6d11"
|
|
case 3: // error - orange
|
|
return "#edab11"
|
|
case 4: // warning - yellow
|
|
return "#edd711"
|
|
case 5: // notice - green
|
|
return "#70ed11"
|
|
case 6: // informational - blue
|
|
return "#118eed"
|
|
case 7: // debug - grey
|
|
return "#828282"
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
// Maps a priority to a color tag
|
|
func (d *Dispatcher) coloredText(color string, text string) string {
|
|
if color == "" {
|
|
return text
|
|
}
|
|
|
|
return "<font data-mx-color='" + color + "'>" + text + "</font>"
|
|
}
|