commit 5303d9ea6e53c050c3e70b995fb928138cdc7757
Author: Dellle <39235395+Dellle@users.noreply.github.com>
Date: Tue Oct 20 21:22:31 2020 +0000
Add username formatting for all events (matrix) (#1233)
diff --git a/bridge/matrix/matrix.go b/bridge/matrix/matrix.go
index c3605b6..d33c3d8 100644
--- a/bridge/matrix/matrix.go
+++ b/bridge/matrix/matrix.go
@@ -1713 +1716 @@ import (
matrix "github.com/matrix-org/gomatrix"
)
+var (
+ htmlTag = regexp.MustCompile("</.*?>")
+ htmlReplacementTag = regexp.MustCompile("<[^>]*>")
+)
+
type Bmatrix struct {
mc *matrix.Client
UserID string
RoomMap map[string]string
sync.RWMutex
- htmlTag *regexp.Regexp
- htmlReplacementTag *regexp.Regexp
*bridge.Config
}
@@ -3310 +3629 @@ type httpError struct {
RetryAfterMs int `json:"retry_after_ms"`
}
+type matrixUsername struct {
+ plain string
+ formatted string
+}
+
+func newMatrixUsername(username string) *matrixUsername {
+ mUsername := new(matrixUsername)
+
+ // check if we have a </tag>. if we have, we don't escape HTML. #696
+ if htmlTag.MatchString(username) {
+ mUsername.formatted = username
+ // remove the HTML formatting for beautiful push messages #1188
+ mUsername.plain = htmlReplacementTag.ReplaceAllString(username, "")
+ } else {
+ mUsername.formatted = html.EscapeString(username)
+ mUsername.plain = username
+ }
+
+ return mUsername
+}
+
func New(cfg *bridge.Config) bridge.Bridger {
b := &Bmatrix{Config: cfg}
- b.htmlTag = regexp.MustCompile("</.*?>")
- b.htmlReplacementTag = regexp.MustCompile("<[^>]*>")
b.RoomMap = make(map[string]string)
return b
}
@@ -9611 +11814 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) {
channel := b.getRoomID(msg.Channel)
b.Log.Debugf("Channel %s maps to channel id %s", msg.Channel, channel)
+ username := newMatrixUsername(msg.Username)
+
// Make a action /me of the message
if msg.Event == config.EventUserAction {
m := matrix.TextMessage{
- MsgType: "m.emote",
- Body: msg.Username + msg.Text,
+ MsgType: "m.emote",
+ Body: username.plain + msg.Text,
+ FormattedBody: username.formatted + msg.Text,
}
resp, err := b.mc.SendMessageEvent(channel, "m.room.message", m)
if err != nil {
@@ -1397 +16412 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) {
// Use notices to send join/leave events
if msg.Event == config.EventJoinLeave {
- resp, err := b.mc.SendNotice(channel, msg.Username+msg.Text)
+ m := matrix.TextMessage{
+ MsgType: "m.notice",
+ Body: username.plain + msg.Text,
+ FormattedBody: username.formatted + msg.Text,
+ }
+ resp, err := b.mc.SendMessageEvent(channel, "m.room.message", m)
if err != nil {
return "", err
}
@@ -14727 +17715 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) {
}
if b.GetBool("HTMLDisable") {
- resp, err := b.mc.SendText(channel, msg.Username+msg.Text)
+ resp, err := b.mc.SendText(channel, username.plain+msg.Text)
if err != nil {
return "", err
}
return resp.EventID, err
}
- var username string
- var plainUsername string
- // check if we have a </tag>. if we have, we don't escape HTML. #696
- if b.htmlTag.MatchString(msg.Username) {
- username = msg.Username
- // remove the HTML formatting for beautiful push messages #1188
- plainUsername = b.htmlReplacementTag.ReplaceAllString(msg.Username, "")
- } else {
- username = html.EscapeString(msg.Username)
- plainUsername = msg.Username
- }
-
// Post normal message with HTML support (eg riot.im)
- resp, err := b.mc.SendFormattedText(channel, plainUsername+msg.Text, username+helper.ParseMarkdown(msg.Text))
+ resp, err := b.mc.SendFormattedText(channel, username.plain+msg.Text, username.formatted+helper.ParseMarkdown(msg.Text))
if err != nil {
return "", err
}
@@ -33021 +34816 @@ func (b *Bmatrix) handleUploadFiles(msg *config.Message, channel string) (string
// handleUploadFile handles native upload of a file.
func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string, fi *config.FileInfo) {
+ username := newMatrixUsername(msg.Username)
content := bytes.NewReader(*fi.Data)
sp := strings.Split(fi.Name, ".")
mtype := mime.TypeByExtension("." + sp[len(sp)-1])
- if fi.Comment != "" {
- _, err := b.mc.SendText(channel, msg.Username+fi.Comment)
- if err != nil {
- b.Log.Errorf("file comment failed: %#v", err)
- }
- } else {
- // image and video uploads send no username, we have to do this ourself here #715
- _, err := b.mc.SendText(channel, msg.Username)
- if err != nil {
- b.Log.Errorf("file comment failed: %#v", err)
- }
+ // image and video uploads send no username, we have to do this ourself here #715
+ _, err := b.mc.SendFormattedText(channel, username.plain+fi.Comment, username.formatted+fi.Comment)
+ if err != nil {
+ b.Log.Errorf("file comment failed: %#v", err)
}
+
b.Log.Debugf("uploading file: %s %s", fi.Name, mtype)
res, err := b.mc.UploadToContentRepo(content, mtype, int64(len(*fi.Data)))
if err != nil {
diff --git a/bridge/matrix/matrix_test.go b/bridge/matrix/matrix_test.go
new file mode 100644
index 0000000..846d9a4
--- /dev/null
+++ b/bridge/matrix/matrix_test.go
@@ -00 +128 @@
+package bmatrix
+
+import (
+ "testing"
+
+ "github.com/stretchr/testify/assert"
+)
+
+func TestPlainUsername(t *testing.T) {
+ uut := newMatrixUsername("MyUser")
+
+ assert.Equal(t, "MyUser", uut.formatted)
+ assert.Equal(t, "MyUser", uut.plain)
+}
+
+func TestHTMLUsername(t *testing.T) {
+ uut := newMatrixUsername("<b>MyUser</b>")
+
+ assert.Equal(t, "<b>MyUser</b>", uut.formatted)
+ assert.Equal(t, "MyUser", uut.plain)
+}
+
+func TestFancyUsername(t *testing.T) {
+ uut := newMatrixUsername("<MyUser>")
+
+ assert.Equal(t, "<MyUser>", uut.formatted)
+ assert.Equal(t, "<MyUser>", uut.plain)
+}
