feat: Add dynamic toolset enable/disable functionality

- Bump version to 0.1.10
- Add `--enabled-toolsets` and `--disabled-toolsets` command-line options
- Add `ENABLED_TOOLSETS` and `DISABLED_TOOLSETS` environment variables
- Update documentation to reflect new features

Author: JJ-H

Makefile

@@ -1,6 +1,6 @@
 # Makefile for cross-platform build
 BINARY_NAME = mcp-gitee
-NPM_VERSION = 0.1.9
+NPM_VERSION = 0.1.10
 GO = go
 OSES = darwin linux windows
 ARCHS = amd64 arm64

README.md

@@ -8,6 +8,7 @@ Gitee MCP Server is a Model Context Protocol (MCP) server implementation for Git
 - Configurable API base URL to support different Gitee instances
 - Command-line flags for easy configuration
 - Supports both personal, organization, and enterprise operations
+- Dynamic toolset enable/disable
 
 <details>
 <summary><b>Practical scenario: Obtain Issue from the repository, implement and create a Pull Request</b></summary>
@@ -99,18 +100,40 @@ config example: [Click to view more application configuration](./docs/install/)
 
 ### Command-line Options
 
-- `-token`: Gitee access token
-- `-api-base`: Gitee API base URL (default: https://gitee.com/api/v5)
-- `-version`: Show version information
-- `-transport`: Transport type (stdio or sse, default: stdio)
-- `-sse-address`: The host and port to start the SSE server on (default: localhost:8000)
+- `--token`: Gitee access token
+- `--api-base`: Gitee API base URL (default: https://gitee.com/api/v5)
+- `--version`: Show version information
+- `--transport`: Transport type (stdio or sse, default: stdio)
+- `--sse-address`: The host and port to start the SSE server on (default: localhost:8000)
+- `--enabled-toolsets`: Comma-separated list of tools to enable (if specified, only these tools will be enabled)
+- `--disabled-toolsets`: Comma-separated list of tools to disable
 
 ### Environment Variables
 
 You can also configure the server using environment variables:
 
 - `GITEE_ACCESS_TOKEN`: Gitee access token
 - `GITEE_API_BASE`: Gitee API base URL
+- `ENABLED_TOOLSETS`: Comma-separated list of tools to enable
+- `DISABLED_TOOLSETS`: Comma-separated list of tools to disable
+
+### Toolset Management
+
+Toolset management supports two modes:
+
+1. Enable specified tools (whitelist mode):
+   - Use `--enabled-toolsets` parameter or `ENABLED_TOOLSETS` environment variable
+   - Specify after, only listed tools will be enabled, others will be disabled
+   - Example: `--enabled-toolsets="list_user_repos,get_file_content"`
+
+2. Disable specified tools (blacklist mode):
+   - Use `--disabled-toolsets` parameter or `DISABLED_TOOLSETS` environment variable
+   - Specify after, listed tools will be disabled, others will be enabled
+   - Example: `--disabled-toolsets="list_user_repos,get_file_content"`
+
+Note:
+- If both `enabled-toolsets` and `disabled-toolsets` are specified, `enabled-toolsets` takes precedence
+- Tool names are case-sensitive
 
 ## License
 

README_CN.md

@@ -8,6 +8,7 @@ Gitee MCP 服务器是一个用于 Gitee 的模型上下文协议（Model Contex
 - 可配置的 API 基础 URL，支持不同的 Gitee 实例
 - 命令行标志，便于配置
 - 支持个人、组织和企业操作
+- 支持动态启用/禁用工具集
 
 <details>
 <summary><b>实战场景：从仓库获取 Issue，实现并创建 Pull Request</b></summary>
@@ -100,18 +101,40 @@ mcp-gitee --version
 
 ### 命令行选项
 
-- `-token`：Gitee 访问令牌
-- `-api-base`：Gitee API 基础 URL（默认：https://gitee.com/api/v5）
-- `-version`：显示版本信息
-- `-transport`：传输类型（stdio 或 sse，默认：stdio）
-- `-sse-address`：启动 SSE 服务器的主机和端口（默认：localhost:8000）
+- `--token`：Gitee 访问令牌
+- `--api-base`：Gitee API 基础 URL（默认：https://gitee.com/api/v5）
+- `--version`：显示版本信息
+- `--transport`：传输类型（stdio 或 sse，默认：stdio）
+- `--sse-address`：启动 SSE 服务器的主机和端口（默认：localhost:8000）
+- `--enabled-toolsets`: 逗号分隔的要启用的工具列表（如果指定，则只启用这些工具）
+- `--disabled-toolsets`: 逗号分隔的要禁用的工具列表
 
 ### 环境变量
 
 您也可以使用环境变量配置服务器：
 
 - `GITEE_ACCESS_TOKEN`：Gitee 访问令牌
 - `GITEE_API_BASE`：Gitee API 基础 URL
+- `ENABLED_TOOLSETS`: 逗号分隔的要启用的工具列表
+- `DISABLED_TOOLSETS`: 逗号分隔的要禁用的工具列表
+
+### 工具集管理
+
+工具集管理支持两种模式：
+
+1. 启用指定工具（白名单模式）：
+   - 使用 `--enabled-toolsets` 参数或 `ENABLED_TOOLSETS` 环境变量
+   - 指定后，只有列出的工具会被启用，其他工具都会被禁用
+   - 例如：`--enabled-toolsets="update_enterprise_issue,list_enterprise_repositories"`
+
+2. 禁用指定工具（黑名单模式）：
+   - 使用 `--disabled-toolsets` 参数或 `DISABLED_TOOLSETS` 环境变量
+   - 指定后，列出的工具会被禁用，其他工具保持启用
+   - 例如：`--disabled-toolsets="update_enterprise_issue,list_enterprise_repositories"`
+
+注意：
+- 如果同时指定了 `enabled-toolsets` 和 `disabled-toolsets`，则 `enabled-toolsets` 优先
+- 工具名称区分大小写
 
 ## 许可证
 

main.go

@@ -4,8 +4,10 @@ import (
 	"context"
 	"flag"
 	"fmt"
+	"github.com/mark3labs/mcp-go/mcp"
 	"log"
 	"os"
+	"strings"
 
 	"gitee.com/oschina/mcp-gitee/operations/issues"
 	"gitee.com/oschina/mcp-gitee/operations/notifications"
@@ -18,7 +20,9 @@ import (
 )
 
 var (
-	Version = utils.Version
+	Version              = utils.Version
+	disabledToolsetsFlag string
+	enabledToolsetsFlag  string
 )
 
 func newMCPServer() *server.MCPServer {
@@ -30,46 +34,111 @@ func newMCPServer() *server.MCPServer {
 	)
 }
 
+func addTool(s *server.MCPServer, tool mcp.Tool, handleFunc server.ToolHandlerFunc) {
+	enabledToolsets := getEnabledToolsets()
+	if len(enabledToolsets) == 0 {
+		s.AddTool(tool, handleFunc)
+		return
+	}
+
+	for i := range enabledToolsets {
+		enabledToolsets[i] = strings.TrimSpace(enabledToolsets[i])
+	}
+
+	for _, keepTool := range enabledToolsets {
+		if tool.Name == keepTool {
+			s.AddTool(tool, handleFunc)
+			return
+		}
+	}
+}
+
+func disableTools(s *server.MCPServer) {
+	if enabledToolsetsFlag != "" {
+		enabledToolsetsFlag = os.Getenv("ENABLED_TOOLSETS")
+	}
+
+	if enabledToolsetsFlag != "" {
+		return
+	}
+
+	if disabledTools := getDisabledToolsets(); len(disabledTools) > 0 {
+		s.DeleteTools(disabledTools...)
+	}
+}
+
 func addTools(s *server.MCPServer) {
 	// Repository Tools
-	s.AddTool(repository.ListUserReposTool, repository.ListUserReposHandler)
-	s.AddTool(repository.GetFileContentTool, repository.GetFileContentHandler)
-	s.AddTool(repository.NewCreateRepoTool(repository.CreateUserRepo), repository.CreateRepoHandleFunc(repository.CreateUserRepo))
-	s.AddTool(repository.NewCreateRepoTool(repository.CreateOrgRepo), repository.CreateRepoHandleFunc(repository.CreateOrgRepo))
-	s.AddTool(repository.NewCreateRepoTool(repository.CreateEnterRepo), repository.CreateRepoHandleFunc(repository.CreateEnterRepo))
-	s.AddTool(repository.CreateReleaseTool, repository.CreateReleaseHandleFunc)
-	s.AddTool(repository.ListReleasesTool, repository.ListReleasesHandleFunc)
-	s.AddTool(repository.SearchReposTool, repository.SearchOpenSourceReposHandler)
-	s.AddTool(repository.ForkRepositoryTool, repository.ForkRepositoryHandler)
+	addTool(s, repository.ListUserReposTool, repository.ListUserReposHandler)
+	addTool(s, repository.GetFileContentTool, repository.GetFileContentHandler)
+	addTool(s, repository.NewCreateRepoTool(repository.CreateUserRepo), repository.CreateRepoHandleFunc(repository.CreateUserRepo))
+	addTool(s, repository.NewCreateRepoTool(repository.CreateOrgRepo), repository.CreateRepoHandleFunc(repository.CreateOrgRepo))
+	addTool(s, repository.NewCreateRepoTool(repository.CreateEnterRepo), repository.CreateRepoHandleFunc(repository.CreateEnterRepo))
+	addTool(s, repository.CreateReleaseTool, repository.CreateReleaseHandleFunc)
+	addTool(s, repository.ListReleasesTool, repository.ListReleasesHandleFunc)
+	addTool(s, repository.SearchReposTool, repository.SearchOpenSourceReposHandler)
+	addTool(s, repository.ForkRepositoryTool, repository.ForkRepositoryHandler)
 
 	// Pulls Tools
-	s.AddTool(pulls.NewListPullsTool(pulls.ListRepoPullsToolName), pulls.ListPullsHandleFunc(pulls.ListRepoPullsToolName))
-	s.AddTool(pulls.CreatePullTool, pulls.CreatePullHandleFunc)
-	s.AddTool(pulls.UpdatePullTool, pulls.UpdatePullHandleFunc)
-	s.AddTool(pulls.GetPullDetailTool, pulls.GetPullDetailHandleFunc)
-	s.AddTool(pulls.CommentPullTool, pulls.CommentPullHandleFunc)
-	s.AddTool(pulls.MergePullTool, pulls.MergePullHandleFunc)
-	s.AddTool(pulls.ListPullCommentsTool, pulls.ListPullCommentsHandleFunc)
+	addTool(s, pulls.NewListPullsTool(pulls.ListRepoPullsToolName), pulls.ListPullsHandleFunc(pulls.ListRepoPullsToolName))
+	addTool(s, pulls.CreatePullTool, pulls.CreatePullHandleFunc)
+	addTool(s, pulls.UpdatePullTool, pulls.UpdatePullHandleFunc)
+	addTool(s, pulls.GetPullDetailTool, pulls.GetPullDetailHandleFunc)
+	addTool(s, pulls.CommentPullTool, pulls.CommentPullHandleFunc)
+	addTool(s, pulls.MergePullTool, pulls.MergePullHandleFunc)
+	addTool(s, pulls.ListPullCommentsTool, pulls.ListPullCommentsHandleFunc)
 
 	// Issues Tools
-	s.AddTool(issues.CreateIssueTool, issues.CreateIssueHandleFunc)
-	s.AddTool(issues.UpdateIssueTool, issues.UpdateIssueHandleFunc)
-	s.AddTool(issues.NewGetIssueDetailTool(issues.GetRepoIssueDetailToolName), issues.GetIssueDetailHandleFunc(issues.GetRepoIssueDetailToolName))
-	s.AddTool(issues.NewListIssuesTool(issues.ListRepoIssuesToolName), issues.ListIssuesHandleFunc(issues.ListRepoIssuesToolName))
-	s.AddTool(issues.CommentIssueTool, issues.CommentIssueHandleFunc)
-	s.AddTool(issues.ListIssueCommentsTool, issues.ListIssueCommentsHandleFunc)
+	addTool(s, issues.CreateIssueTool, issues.CreateIssueHandleFunc)
+	addTool(s, issues.UpdateIssueTool, issues.UpdateIssueHandleFunc)
+	addTool(s, issues.NewGetIssueDetailTool(issues.GetRepoIssueDetailToolName), issues.GetIssueDetailHandleFunc(issues.GetRepoIssueDetailToolName))
+	addTool(s, issues.NewListIssuesTool(issues.ListRepoIssuesToolName), issues.ListIssuesHandleFunc(issues.ListRepoIssuesToolName))
+	addTool(s, issues.CommentIssueTool, issues.CommentIssueHandleFunc)
+	addTool(s, issues.ListIssueCommentsTool, issues.ListIssueCommentsHandleFunc)
 
 	// Notifications Tools
-	s.AddTool(notifications.ListUserNotificationsTool, notifications.ListUserNotificationsHandler)
+	addTool(s, notifications.ListUserNotificationsTool, notifications.ListUserNotificationsHandler)
 
 	// Users Tools
-	s.AddTool(users.GetUserInfoTool, users.GetUserInfoHandleFunc())
-	s.AddTool(users.SearchUsersTool, users.SearchUsersHandler)
+	addTool(s, users.GetUserInfoTool, users.GetUserInfoHandleFunc())
+	addTool(s, users.SearchUsersTool, users.SearchUsersHandler)
+}
+
+func getDisabledToolsets() []string {
+	if disabledToolsetsFlag == "" {
+		disabledToolsetsFlag = os.Getenv("DISABLED_TOOLSETS")
+	}
+
+	if disabledToolsetsFlag == "" {
+		return nil
+	}
+
+	tools := strings.Split(disabledToolsetsFlag, ",")
+	for i := range tools {
+		tools[i] = strings.TrimSpace(tools[i])
+	}
+
+	return tools
+}
+
+func getEnabledToolsets() []string {
+	if enabledToolsetsFlag == "" {
+		enabledToolsetsFlag = os.Getenv("ENABLED_TOOLSETS")
+	}
+	if enabledToolsetsFlag == "" {
+		return nil
+	}
+	tools := strings.Split(enabledToolsetsFlag, ",")
+	for i := range tools {
+		tools[i] = strings.TrimSpace(tools[i])
+	}
+	return tools
 }
 
 func run(transport, addr string) error {
 	s := newMCPServer()
 	addTools(s)
+	disableTools(s)
 
 	switch transport {
 	case "stdio":
@@ -103,6 +172,8 @@ func main() {
 	showVersion := flag.Bool("version", false, "Show version information")
 	transport := flag.String("transport", "stdio", "Transport type (stdio or sse)")
 	addr := flag.String("sse-address", "localhost:8000", "The host and port to start the sse server on")
+	flag.StringVar(&disabledToolsetsFlag, "disabled-toolsets", "", "Comma-separated list of tools to disable")
+	flag.StringVar(&enabledToolsetsFlag, "enabled-toolsets", "", "Comma-separated list of tools to enable (if specified, only these tools will be available)")
 	flag.Parse()
 
 	if *showVersion {

npm/mcp-gitee-darwin-amd64/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gitee/mcp-gitee-darwin-amd64",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Model Context Protocol (MCP) server for Gitee",
   "os": [
     "darwin"

npm/mcp-gitee-darwin-arm64/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gitee/mcp-gitee-darwin-arm64",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Model Context Protocol (MCP) server for Gitee",
   "os": [
     "darwin"

npm/mcp-gitee-linux-amd64/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gitee/mcp-gitee-linux-amd64",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Model Context Protocol (MCP) server for Gitee",
   "os": [
     "linux"

npm/mcp-gitee-linux-arm64/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gitee/mcp-gitee-linux-arm64",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Model Context Protocol (MCP) server for Gitee",
   "os": [
     "linux"

npm/mcp-gitee-windows-amd64/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gitee/mcp-gitee-windows-amd64",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Model Context Protocol (MCP) server for Gitee",
   "os": [
     "win32"

npm/mcp-gitee-windows-arm64/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gitee/mcp-gitee-windows-arm64",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "Model Context Protocol (MCP) server for Gitee",
   "os": [
     "win32"