Merge pull request #13 from korotovsky/conversations-replies-tool

Added conversations_replies tool

Author: korotovsky

README.md

@@ -12,11 +12,20 @@ Model Context Protocol (MCP) server for Slack Workspaces. This integration suppo
   - Get messages from the channel by channelID
   - Required inputs:
     - `channel_id` (string): ID of the channel in format Cxxxxxxxxxx.
-    - `cursor` (string): Cursor for pagination. Use the value of the last row and column in the response as next_cursor field returned from the previous request.
+    - `cursor` (string, default: ""): Cursor for pagination. Use the value of the last row and column in the response as next_cursor field returned from the previous request.
     - `limit` (string, default: 28): Limit of messages to fetch.
   - Returns: List of messages with timestamps, user IDs, and text content
 
-2. `channels_list`
+2. `conversations_replies`
+  - Get a thread of messages posted to a conversation by channelID and thread_ts
+  - Required inputs:
+    - `channel_id` (string): ID of the channel in format Cxxxxxxxxxx.
+    - `thread_ts` (string): Unique identifier of either a thread’s parent message or a message in the thread. ts must be the timestamp in format 1234567890.123456 of an existing message with 0 or more replies.
+    - `cursor` (string, default: ""): Cursor for pagination. Use the value of the last row and column in the response as next_cursor field returned from the previous request.
+    - `limit` (string, default: 28): Limit of messages to fetch.
+  - Returns: List of replies with timestamps, user IDs, and text content
+
+3. `channels_list`
   - Get list of channels
   - Required inputs:
     - `channel_types` (string): Comma-separated channel types. Allowed values: 'mpim', 'im', 'public_channel', 'private_channel'. Example: 'public_channel,private_channel,im'.

pkg/handler/conversations.go

@@ -20,6 +20,7 @@ type Message struct {
 	UserName string `json:"userUser"`
 	RealName string `json:"realName"`
 	Channel  string `json:"channelID"`
+	ThreadTs string `json:"ThreadTs"`
 	Text     string `json:"text"`
 	Time     string `json:"time"`
 	Cursor   string `json:"cursor"`
@@ -87,17 +88,24 @@ func (ch *ConversationsHandler) ConversationsHistoryHandler(ctx context.Context,
 	for _, message := range messages.Messages {
 		textTokenized := text.ProcessText(message.Text)
 		user, ok := usersMap[message.User]
-		if !ok {
-			// TODO: add periodic refetch of users
-			continue
+
+		var (
+			userName = message.User
+			realName = message.User
+		)
+
+		if ok {
+			userName = user.Name
+			realName = user.RealName
 		}
 
 		messageList = append(messageList, Message{
 			UserID:   message.User,
-			UserName: user.Name,
-			RealName: user.RealName,
+			UserName: userName,
+			RealName: realName,
 			Text:     textTokenized,
 			Channel:  channel,
+			ThreadTs: message.ThreadTimestamp,
 			Time:     message.Timestamp,
 		})
 	}
@@ -114,6 +122,97 @@ func (ch *ConversationsHandler) ConversationsHistoryHandler(ctx context.Context,
 	return mcp.NewToolResultText(string(csvBytes)), nil
 }
 
+func (ch *ConversationsHandler) ConversationsRepliesHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	var (
+		err         error
+		paramLimit  int
+		paramOldest string
+		paramLatest string
+	)
+
+	channel := request.GetString("channel_id", "")
+	if channel == "" {
+		return nil, errors.New("channel_id must be a string")
+	}
+
+	threadTs := request.GetString("thread_ts", "")
+	if threadTs == "" {
+		return nil, errors.New("thread_ts must be a string")
+	}
+
+	limit := request.GetString("limit", "")
+	cursor := request.GetString("cursor", "")
+
+	if strings.HasSuffix(limit, "d") {
+		paramLimit, paramOldest, paramLatest, err = limitByDays(limit)
+		if err != nil {
+			return nil, err
+		}
+	} else if cursor == "" {
+		paramLimit, err = limitByNumeric(limit)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	api, err := ch.apiProvider.Provide()
+	if err != nil {
+		return nil, err
+	}
+
+	params := slack.GetConversationRepliesParameters{
+		ChannelID: channel,
+		Timestamp: threadTs,
+		Limit:     paramLimit,
+		Oldest:    paramOldest,
+		Latest:    paramLatest,
+		Cursor:    cursor,
+		Inclusive: false,
+	}
+	messages, hasMore, nextCursor, err := api.GetConversationRepliesContext(ctx, &params)
+	if err != nil {
+		return nil, err
+	}
+
+	usersMap := ch.apiProvider.ProvideUsersMap()
+
+	var messageList []Message
+	for _, message := range messages {
+		textTokenized := text.ProcessText(message.Text)
+		user, ok := usersMap[message.User]
+
+		var (
+			userName = message.User
+			realName = message.User
+		)
+
+		if ok {
+			userName = user.Name
+			realName = user.RealName
+		}
+
+		messageList = append(messageList, Message{
+			UserID:   message.User,
+			UserName: userName,
+			RealName: realName,
+			Text:     textTokenized,
+			Channel:  channel,
+			Time:     message.Timestamp,
+		})
+	}
+
+	if len(messageList) > 0 && hasMore {
+		messageList[len(messageList)-1].Cursor = nextCursor
+	}
+
+	csvBytes, err := gocsv.MarshalBytes(&messageList)
+	if err != nil {
+		return nil, err
+	}
+
+	return mcp.NewToolResultText(string(csvBytes)), nil
+}
+
 func limitByNumeric(limit string) (int, error) {
 	n, err := strconv.Atoi(limit)
 	if err != nil {

pkg/server/server.go

@@ -16,7 +16,7 @@ type MCPServer struct {
 func NewMCPServer(provider *provider.ApiProvider) *MCPServer {
 	s := server.NewMCPServer(
 		"Slack MCP Server",
-		"1.0.0",
+		"1.1.11",
 		server.WithLogging(),
 		server.WithRecovery(),
 	)
@@ -38,6 +38,25 @@ func NewMCPServer(provider *provider.ApiProvider) *MCPServer {
 		),
 	), conversationsHandler.ConversationsHistoryHandler)
 
+	s.AddTool(mcp.NewTool("conversations_replies",
+		mcp.WithDescription("Get a thread of messages posted to a conversation by channelID and thread_ts, the last row/column in the response is used as 'cursor' parameter for pagination if not empty"),
+		mcp.WithString("channel_id",
+			mcp.Required(),
+			mcp.Description("ID of the channel in format Cxxxxxxxxxx"),
+		),
+		mcp.WithString("thread_ts",
+			mcp.Required(),
+			mcp.Description("Unique identifier of either a thread’s parent message or a message in the thread. ts must be the timestamp in format 1234567890.123456 of an existing message with 0 or more replies."),
+		),
+		mcp.WithString("cursor",
+			mcp.Description("Cursor for pagination. Use the value of the last row and column in the response as next_cursor field returned from the previous request."),
+		),
+		mcp.WithString("limit",
+			mcp.DefaultString("1d"),
+			mcp.Description("Limit of messages to fetch in format of maximum ranges of time (e.g. 1d - 1 day, 30d - 30 days, 90d - 90 days which is a default limit for free tier history) or number of messages (e.g. 50). Must be empty when 'cursor' is provided."),
+		),
+	), conversationsHandler.ConversationsRepliesHandler)
+
 	channelsHandler := handler.NewChannelsHandler(provider)
 
 	s.AddTool(mcp.NewTool("channels_list",