1 files changed, 97 insertions, 7 deletions
diff --git a/vendor/gomod.garykim.dev/nc-talk/ocs/message.go b/vendor/gomod.garykim.dev/nc-talk/ocs/message.go
index 6093e638..e23078de 100644
--- a/vendor/gomod.garykim.dev/nc-talk/ocs/message.go
+++ b/vendor/gomod.garykim.dev/nc-talk/ocs/message.go
@@ -14,6 +14,11 @@
 
 package ocs
 
+import (
+	"encoding/json"
+	"strings"
+)
+
 // MessageType describes what kind of message a returned Nextcloud Talk message is
 type MessageType string
 
@@ -30,23 +35,108 @@ const (
 
 // TalkRoomMessageData describes the data part of a ocs response for a Talk room message
 type TalkRoomMessageData struct {
-	Message          string      `json:"message"`
-	ID               int         `json:"id"`
-	ActorID          string      `json:"actorId"`
-	ActorDisplayName string      `json:"actorDisplayName"`
-	SystemMessage    string      `json:"systemMessage"`
-	Timestamp        int         `json:"timestamp"`
-	MessageType      MessageType `json:"messageType"`
+	Message           string                      `json:"message"`
+	ID                int                         `json:"id"`
+	ActorID           string                      `json:"actorId"`
+	ActorDisplayName  string                      `json:"actorDisplayName"`
+	SystemMessage     string                      `json:"systemMessage"`
+	Timestamp         int                         `json:"timestamp"`
+	MessageType       MessageType                 `json:"messageType"`
+	MessageParameters map[string]RichObjectString `json:"-"`
+}
+
+// talkRoomMessageParameters is used to unmarshal only MessageParameters
+type talkRoomMessageParameters struct {
+	MessageParameters map[string]RichObjectString `json:"messageParameters"`
+}
+
+// PlainMessage returns the message string with placeholders replaced
+//
+// * User and group placeholders will be replaced with the name of the user or group respectively.
+//
+// * File placeholders will be replaced with the name of the file.
+func (m *TalkRoomMessageData) PlainMessage() string {
+	tr := m.Message
+	for key, value := range m.MessageParameters {
+		tr = strings.ReplaceAll(tr, "{"+key+"}", value.Name)
+	}
+	return tr
 }
 
 // TalkRoomMessage describes an ocs response for a Talk room message
 type TalkRoomMessage struct {
+	OCS talkRoomMessage `json:"ocs"`
+}
+
+type talkRoomMessage struct {
 	ocs
 	TalkRoomMessage []TalkRoomMessageData `json:"data"`
 }
 
+// TalkRoomMessageDataUnmarshal unmarshals given ocs request data and returns a TalkRoomMessageData
+func TalkRoomMessageDataUnmarshal(data *[]byte) (*TalkRoomMessage, error) {
+	message := &TalkRoomMessage{}
+	err := json.Unmarshal(*data, message)
+	if err != nil {
+		return nil, err
+	}
+
+	// Get RCS
+	var rcs struct {
+		OCS struct {
+			ocs
+			TalkRoomMessage []talkRoomMessageParameters `json:"data"`
+		} `json:"ocs"`
+	}
+	err = json.Unmarshal(*data, &rcs)
+	// There is no RCS data
+	if err != nil {
+		for i := range message.OCS.TalkRoomMessage {
+			message.OCS.TalkRoomMessage[i].MessageParameters = map[string]RichObjectString{}
+		}
+		return message, nil
+	}
+
+	// There is RCS data
+	for i := range message.OCS.TalkRoomMessage {
+		message.OCS.TalkRoomMessage[i].MessageParameters = rcs.OCS.TalkRoomMessage[i].MessageParameters
+	}
+	return message, nil
+}
+
 // TalkRoomSentResponse describes an ocs response for what is returned when a message is sent
 type TalkRoomSentResponse struct {
+	OCS talkRoomSentResponse `json:"ocs"`
+}
+
+type talkRoomSentResponse struct {
 	ocs
 	TalkRoomMessage TalkRoomMessageData `json:"data"`
 }
+
+// TalkRoomSentResponseUnmarshal unmarshals given ocs request data and returns a TalkRoomMessageData
+func TalkRoomSentResponseUnmarshal(data *[]byte) (*TalkRoomSentResponse, error) {
+	message := &TalkRoomSentResponse{}
+	err := json.Unmarshal(*data, message)
+	if err != nil {
+		return nil, err
+	}
+
+	// Get RCS
+	var rcs struct {
+		OCS struct {
+			ocs
+			TalkRoomMessage talkRoomMessageParameters `json:"data"`
+		} `json:"ocs"`
+	}
+	err = json.Unmarshal(*data, &rcs)
+	// There is no RCS data
+	if err != nil {
+		message.OCS.TalkRoomMessage.MessageParameters = map[string]RichObjectString{}
+		return message, nil
+	}
+
+	// There is RCS data
+	message.OCS.TalkRoomMessage.MessageParameters = rcs.OCS.TalkRoomMessage.MessageParameters
+	return message, nil
+}