feat: pending messages (#235)

GetStream · Aug 16, 2022 · 90480e3 · 90480e3
1 parent 5e8e5d1
commit 90480e3
Show file tree

Hide file tree

Showing 2 changed files with 72 additions and 4 deletions.
diff --git a/message.go b/message.go
@@ -113,8 +113,11 @@ func (m *Message) toRequest() messageRequest {
 }
 
 type messageRequest struct {
-	Message  messageRequestMessage `json:"message"`
-	SkipPush bool                  `json:"skip_push,omitempty"`
+	Message                messageRequestMessage `json:"message"`
+	SkipPush               bool                  `json:"skip_push,omitempty"`
+	SkipEnrichURL          bool                  `json:"skip_enrich_url,omitempty"`
+	IsPendingMessage       bool                  `json:"is_pending_message,omitempty"`
+	PendingMessageMetadata map[string]string     `json:"pending_message_metadata,omitempty"`
 }
 
 type messageRequestMessage struct {
@@ -206,9 +209,32 @@ func MessageSkipPush(r *messageRequest) {
 	}
 }
 
-type MessageResponse struct {
-	Message *Message `json:"message"`
+// MessageSkipEnrichURL is a flag that disables enrichment of the URLs in the message
+func MessageSkipEnrichURL(r *messageRequest) {
+	if r != nil {
+		r.SkipEnrichURL = true
+	}
+}
+
+// MessagePending is a flag that makes this a pending message
+func MessagePending(r *messageRequest) {
+	if r != nil {
+		r.IsPendingMessage = true
+	}
+}
 
+// MessagePendingMessageMetadata saves metadata to the pending message
+func MessagePendingMessageMetadata(metadata map[string]string) SendMessageOption {
+	return func(r *messageRequest) {
+		if r != nil {
+			r.PendingMessageMetadata = metadata
+		}
+	}
+}
+
+type MessageResponse struct {
+	Message                *Message          `json:"message"`
+	PendingMessageMetadata map[string]string `json:"pending_message_metadata,omitempty"`
 	Response
 }
 

diff --git a/message_test.go b/message_test.go
@@ -23,6 +23,48 @@ func TestClient_TranslateMessage(t *testing.T) {
 	require.Equal(t, "mensaje de prueba", translated.Message.I18n["es_text"])
 }
 
+func TestClient_SendMessage_Pending(t *testing.T) {
+	c := initClient(t)
+	user := randomUser(t, c)
+
+	ctx := context.Background()
+
+	ch := initChannel(t, c, user.ID)
+	resp1, err := c.CreateChannel(ctx, ch.Type, ch.ID, user.ID, nil)
+	require.NoError(t, err)
+
+	msg := &Message{Text: "test pending message"}
+	metadata := map[string]string{"my": "metadata"}
+	messageResp, err := resp1.Channel.SendMessage(ctx, msg, user.ID, MessagePending, MessagePendingMessageMetadata(metadata))
+	require.NoError(t, err)
+	require.Equal(t, metadata, messageResp.PendingMessageMetadata)
+
+	gotMsg, err := c.GetMessage(ctx, messageResp.Message.ID)
+	require.NoError(t, err)
+	require.Equal(t, metadata, gotMsg.PendingMessageMetadata)
+}
+
+func TestClient_SendMessage_SkipEnrichURL(t *testing.T) {
+	c := initClient(t)
+	user := randomUser(t, c)
+
+	ctx := context.Background()
+
+	ch := initChannel(t, c, user.ID)
+	resp1, err := c.CreateChannel(ctx, ch.Type, ch.ID, user.ID, nil)
+	require.NoError(t, err)
+
+	msg := &Message{Text: "test message with link to https://getstream.io"}
+	messageResp, err := resp1.Channel.SendMessage(ctx, msg, user.ID, MessageSkipEnrichURL)
+	require.NoError(t, err)
+	require.Len(t, messageResp.Message.Attachments, 0)
+
+	time.Sleep(3 * time.Second)
+	gotMsg, err := c.GetMessage(ctx, messageResp.Message.ID)
+	require.NoError(t, err)
+	require.Len(t, gotMsg.Message.Attachments, 0)
+}
+
 func TestClient_PinMessage(t *testing.T) {
 	c := initClient(t)
 	userA := randomUser(t, c)