From 8b0193186b32d4d1a7544aff6ffa12021ad1543c Mon Sep 17 00:00:00 2001
From: Vlasov <vlasov.vlasov2015@gmail.com>
Date: Fri, 15 Nov 2024 11:57:32 +0300
Subject: [PATCH] Added Message.Dialog field for all types of messages

---
 v1/client_test.go | 77 +++++++++++++++++++++++++++++++++++++++++++++++
 v1/types.go       |  9 +++++-
 2 files changed, 85 insertions(+), 1 deletion(-)

diff --git a/v1/client_test.go b/v1/client_test.go
index 72eabc2..249e179 100644
--- a/v1/client_test.go
+++ b/v1/client_test.go
@@ -555,6 +555,83 @@ func TestMgClient_Messages(t *testing.T) {
 	}
 }
 
+func TestMgClient_MessagesDialog(t *testing.T) {
+	t.Parallel()
+
+	c := client()
+
+	defer gock.Off()
+
+	gock.New(mgURL).
+		Get("/api/bot/v1/messages").
+		Reply(200).
+		BodyString(`[
+			{
+				"id": 1,
+				"time": "2024-11-14T20:48:21Z",
+				"type": "system",
+				"scope": "private",
+				"chat_id": 1,
+				"is_read": false,
+				"is_edit": false,
+				"status": "received",
+				"dialog": {
+					"id": 2054
+				},
+				"action": "dialog_closed",
+				"channel_id": 71,
+				"created_at": "2024-11-14T20:48:21.907566Z",
+				"updated_at": "2024-11-14T20:48:21.907566Z"
+			},
+			{
+				"id": 2,
+				"time": "2024-11-14T19:58:42Z",
+				"type": "text",
+				"scope": "public",
+				"chat_id": 1,
+				"is_read": false,
+				"is_edit": false,
+				"status": "failed",
+				"from": {
+					"id": 1,
+					"external_id": "1",
+					"type": "user",
+					"avatar": "http://avatars-test.jpeg",
+					"name": "John Smith",
+					"first_name": "John",
+					"last_name": "Smith",
+					"available": true
+				},
+				"dialog": {
+					"id": 2054
+				},
+				"error": {
+					"code": "malformed_response"
+				},
+				"content": "Message from user John Smith",
+				"quote": null,
+				"channel_id": 71,
+				"created_at": "2024-11-14T19:58:42.933025Z",
+				"updated_at": "2024-11-14T19:58:45.01619Z"
+			}
+		]`)
+
+	req := MessagesRequest{}
+
+	data, status, err := c.Messages(req)
+	if err != nil {
+		t.Errorf("%d %v", status, err)
+	}
+
+	assert.NoError(t, err)
+	assert.Len(t, data, 2)
+
+	for _, m := range data {
+		assert.NotNil(t, m.Message.Dialog)
+		assert.Equal(t, uint64(2054), m.Message.Dialog.ID)
+	}
+}
+
 func TestMgClient_MessageSendText(t *testing.T) {
 	c := client()
 
diff --git a/v1/types.go b/v1/types.go
index d93cd5a..6af1b96 100644
--- a/v1/types.go
+++ b/v1/types.go
@@ -466,6 +466,7 @@ type (
 		From    *UserRef        `json:"from"`
 		Product *MessageProduct `json:"product,omitempty"`
 		Order   *MessageOrder   `json:"order,omitempty"`
+		Dialog  *MessageDialog  `json:"dialog,omitempty"`
 		*TextMessage
 		*SystemMessage
 		*AttachmentList
@@ -482,16 +483,22 @@ type (
 	}
 
 	SystemMessage struct {
-		Action      string               `json:"action"`
+		Action string `json:"action"`
+		// Deprecated: Use Message.Dialog.ID instead.
 		Dialog      *SystemMessageDialog `json:"dialog,omitempty"`
 		User        *UserRef             `json:"user,omitempty"`
 		Responsible *UserRef             `json:"responsible,omitempty"`
 	}
 
+	// Deprecated: Use MessageDialog instead.
 	SystemMessageDialog struct {
 		ID uint64 `json:"id"`
 	}
 
+	MessageDialog struct {
+		ID uint64 `json:"id"`
+	}
+
 	QuoteMessage struct {
 		ID      uint64   `json:"id"`
 		Content string   `json:"content"`