Merge pull request #10 from harrycis/list_channels

more flexible params for channels_list

Author: korotovsky

README.md

@@ -21,6 +21,8 @@ Model Context Protocol (MCP) server for Slack Workspaces. This integration suppo
   - Required inputs:
     - `channel_types` (string): Comma-separated channel types. Allowed values: 'mpim', 'im', 'public_channel', 'private_channel'. Example: 'public_channel,private_channel,im'.
     - `sort` (string): Type of sorting. Allowed values: 'popularity' - sort by number of members/participants in each channel.
+    - `limit` (number, default: 100): Limit of channels to fetch.
+    - `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.
   - Returns: List of channels
 
 ## Setup Guide

pkg/handler/channels.go

@@ -2,6 +2,7 @@ package handler
 
 import (
 	"context"
+	"fmt"
 	"log"
 	"sort"
 	"strings"
@@ -21,6 +22,7 @@ type Channel struct {
 	Topic       string `json:"topic"`
 	Purpose     string `json:"purpose"`
 	MemberCount int    `json:"memberCount"`
+	Cursor      string `json:"cursor"`
 }
 
 type ChannelsHandler struct {
@@ -54,46 +56,65 @@ func (ch *ChannelsHandler) ChannelsHandler(ctx context.Context, request mcp.Call
 		}
 	}
 
+	cursor := request.GetString("cursor", "")
+	limit := request.GetInt("limit", 0)
+	if limit == 0 && cursor == "" {
+		return nil, fmt.Errorf("One of limit or cursor needs to be provided")
+	}
+	if limit == 0 {
+		limit = 100
+	}
+
 	api, err := ch.apiProvider.Provide()
 	if err != nil {
 		return nil, err
 	}
 
-	var channels []slack.Channel
+	var channelList []Channel
 
 	params := &slack.GetConversationsParameters{
 		Types:           channelTypes,
-		Limit:           100,
+		Limit:           limit,
 		ExcludeArchived: true,
+		Cursor:          cursor,
 	}
-	var total int
-	for i := 1; ; i++ {
-		var (
-			chans   []slack.Channel
-			nextcur string
-		)
+	var (
+		total   int
+		nextcur string
+	)
+	for {
+		var chans []slack.Channel
 
 		chans, nextcur, err = api.GetConversationsContext(ctx, params)
-
-		if nextcur == "" {
-			log.Printf("channels fetch complete %v", total)
+		if err != nil {
 			break
 		}
 
-		params.Cursor = nextcur
+		if l := len(chans); l > 0 {
+			for _, channel := range chans {
+				channelList = append(channelList, Channel{
+					ID:          channel.ID,
+					Name:        "#" + channel.Name,
+					Topic:       channel.Topic.Value,
+					Purpose:     channel.Purpose.Value,
+					MemberCount: channel.NumMembers,
+				})
+			}
+
+			total += l
+			params.Limit -= l
+		}
 
-		channels = append(channels, chans...)
-	}
+		if total >= limit {
+			log.Printf("channels fetch limit reached %v", total)
+			break
+		}
 
-	var channelList []Channel
-	for _, channel := range channels {
-		channelList = append(channelList, Channel{
-			ID:          channel.ID,
-			Name:        "#" + channel.Name,
-			Topic:       channel.Topic.Value,
-			Purpose:     channel.Purpose.Value,
-			MemberCount: channel.NumMembers,
-		})
+		if nextcur == "" {
+			log.Printf("channels fetch exhausted")
+			break
+		}
+		params.Cursor = nextcur
 	}
 
 	switch sortType {
@@ -105,6 +126,10 @@ func (ch *ChannelsHandler) ChannelsHandler(ctx context.Context, request mcp.Call
 		// pass
 	}
 
+	if len(channelList) > 0 && nextcur != "" {
+		channelList[len(channelList)-1].Cursor = nextcur
+	}
+
 	csvBytes, err := gocsv.MarshalBytes(&channelList)
 	if err != nil {
 		return nil, err

pkg/server/server.go

@@ -49,6 +49,13 @@ func NewMCPServer(provider *provider.ApiProvider) *MCPServer {
 		mcp.WithString("sort",
 			mcp.Description("Type of sorting. Allowed values: 'popularity' - sort by number of members/participants in each channel."),
 		),
+		mcp.WithNumber("limit",
+			mcp.DefaultNumber(100),
+			mcp.Description("The maximum number of items to return. Must be an integer under 1000."),
+		),
+		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."),
+		),
 	), channelsHandler.ChannelsHandler)
 
 	return &MCPServer{