pushbits/internal/dispatcher/notification.go
2021-06-06 19:30:25 +02:00

132 lines
3.4 KiB
Go

package dispatcher
import (
"fmt"
"html"
"log"
"strings"
"github.com/gomarkdown/markdown"
"github.com/pushbits/server/internal/model"
)
type ReplyEvent struct {
Body string `json:"body"`
Msgtype string `json:"msgtype"`
RelatesTo RelatesTo `json:"m.relates_to,omitempty"`
}
type RelatesTo struct {
InReplyTo map[string]string `json:"m.in_reply_to"`
}
// SendNotification sends a notification to the specified user.
func (d *Dispatcher) SendNotification(a *model.Application, n *model.Notification) (id string, err 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)
respSendEvent, err := d.client.SendFormattedText(a.MatrixID, text, formattedText)
return respSendEvent.EventID, err
}
// DeleteNotification sends a notification to the specified user that another notificaion is deleted
func (d *Dispatcher) DeleteNotification(a *model.Application, n *model.DeleteNotification) error {
log.Printf("Sending delete notification to room %s", a.MatrixID)
event := ReplyEvent{
Body: "<i>This message got deleted.</i>",
Msgtype: "m.text",
}
irt := make(map[string]string)
irt["event_id"] = n.ID
rt := RelatesTo{
InReplyTo: irt,
}
event.RelatesTo = rt
_, err := d.client.SendMessageEvent(a.MatrixID, "m.room.message", event)
return err
/*
messages, _ := d.client.Messages(a.MatrixID, "", "", 'b', 10)
js, _ := json.Marshal(messages)
log.Println(string(js))
*/
}
// 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{})
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 {
case prio < 0:
return "#828282"
case prio <= 3: // info - default color
return ""
case prio <= 10: // low - yellow
return "#edd711"
case prio <= 20: // mid - orange
return "#ed6d11"
case prio > 20: // high - red
return "#ed1f11"
}
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>"
}