From 47570052518eba809edaa63129dd44042538f971 Mon Sep 17 00:00:00 2001
From: John Doe <miromaxxs@gmail.com>
Date: Wed, 2 Apr 2025 01:47:59 +0300
Subject: [PATCH] dialogs

---
 internal/tg/dialogs.go | 44 ++++++++++++++++++++++++++++++++----------
 serve.go               |  2 +-
 2 files changed, 35 insertions(+), 11 deletions(-)

diff --git a/internal/tg/dialogs.go b/internal/tg/dialogs.go
index f64e747..be6bd5c 100644
--- a/internal/tg/dialogs.go
+++ b/internal/tg/dialogs.go
@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"sort"
+	"time"
 
 	"github.com/gotd/td/tg"
 	mcp "github.com/metoro-io/mcp-golang"
@@ -28,20 +29,27 @@ const (
 	DefaultDialogsLimit = 100
 )
 
-// DialogsArguments contains parameters for getting dialogs
 type DialogsArguments struct {
-	Type  DialogType `json:"type,omitempty" jsonschema:"description=Filter dialogs by type (user, chat, channel or empty for all),enum=,enum=user,enum=chat,enum=channel"`
-	Limit int        `json:"limit,omitempty" jsonschema:"description=Maximum number of dialogs to return (max: 100),default=100"`
+	Type             DialogType `json:"type,omitempty" jsonschema:"description=Filter dialogs by type (user, chat, channel or empty for all),enum=,enum=user,enum=chat,enum=channel"`
+	Limit            int        `json:"limit,omitempty" jsonschema:"description=Maximum number of dialogs to return (max: 100),default=100"`
+	WithLastMessages bool       `json:"with_last_messages,omitempty" jsonschema:"description=Include last messages in response"`
+}
+
+type MessageInfo struct {
+	Who      string `json:"who"`
+	When     string `json:"when"`
+	Text     string `json:"text"`
+	IsUnread bool   `json:"is_unread,omitempty"`
 }
 
-// DialogInfo represents a simplified dialog structure
 type DialogInfo struct {
-	ID            int64  `json:"id"`
-	Type          string `json:"type"`
-	Title         string `json:"title"`
-	UnreadCount   int    `json:"unread_count"`
-	LastMessageID int    `json:"last_message_id"`
-	IsVerified    bool   `json:"is_verified,omitempty"`
+	ID            int64         `json:"id"`
+	Type          string        `json:"type"`
+	Title         string        `json:"title"`
+	UnreadCount   int           `json:"unread_count"`
+	LastMessageID int           `json:"last_message_id"`
+	IsVerified    bool          `json:"is_verified,omitempty"`
+	LastMessages  []MessageInfo `json:"last_messages,omitempty"`
 }
 
 // GetDialogs returns a list of dialogs (chats, channels, groups)
@@ -93,6 +101,22 @@ func (c *Client) GetDialogs(args DialogsArguments) (*mcp.ToolResponse, error) {
 			info.UnreadCount = dialogItem.UnreadCount
 			info.LastMessageID = dialogItem.TopMessage
 
+			if args.WithLastMessages {
+				for _, msg := range dialogs.Messages {
+					message, ok := msg.(*tg.Message)
+					if !ok {
+						continue
+					}
+
+					info.LastMessages = append(info.LastMessages, MessageInfo{
+						Who:      message.FromID.String(),
+						When:     time.Unix(int64(message.Date), 0).Format(time.DateTime),
+						Text:     message.Message,
+						IsUnread: !message.Out,
+					})
+				}
+			}
+
 			switch peer := dialogItem.Peer.(type) {
 			case *tg.PeerUser:
 				if args.Type != DialogTypeAll && args.Type != DialogTypeUser {
diff --git a/serve.go b/serve.go
index 0adc8c6..485c923 100644
--- a/serve.go
+++ b/serve.go
@@ -41,7 +41,7 @@ func serve(ctx context.Context, cmd *cli.Command) error {
 
 		log.Info().RawJSON("answer", data).Msg("Check GetMe: OK")
 
-		answer, err = client.GetDialogs(tg.DialogsArguments{})
+		answer, err = client.GetDialogs(tg.DialogsArguments{WithLastMessages: true})
 		if err != nil {
 			return fmt.Errorf("get dialogs: %w", err)
 		}