add --exclude-tool option to exclude tools by name (#253)

Closes #251

cc @DanTup

Author: jakemac53

pkgs/dart_mcp_server/CHANGELOG.md

@@ -11,6 +11,7 @@
 * Change the widget tree to the full version instead of the summary. The summary
   tends to hide nested text widgets which makes it difficult to find widgets
   based on their text values.
+* Add an `--exclude-tool` command line flag to exclude tools by name.
 
 # 0.1.0 (Dart SDK 3.9.0)
 

pkgs/dart_mcp_server/lib/src/arg_parser.dart

@@ -47,13 +47,18 @@ ArgParser createArgParser({
           help:
               'Path to a file to log all MPC protocol traffic to. File will be '
               'overwritten if it exists.',
+        )
+        ..addMultiOption(
+          excludeToolOption,
+          help: 'The names of tools to exclude from this run of the server.',
         );
 
   if (includeHelp) parser.addFlag(helpFlag, abbr: 'h', help: 'Show usage text');
   return parser;
 }
 
 const dartSdkOption = 'dart-sdk';
+const excludeToolOption = 'exclude-tool';
 const flutterSdkOption = 'flutter-sdk';
 const forceRootsFallbackFlag = 'force-roots-fallback';
 const helpFlag = 'help';

pkgs/dart_mcp_server/lib/src/server.dart

@@ -48,10 +48,16 @@ final class DartMCPServer extends MCPServer
         ProcessManagerSupport,
         FileSystemSupport,
         SdkSupport {
+  /// A list of tool names to exclude from this version of the server.
+  ///
+  /// Used in [registerTool] to skip registering these tools.
+  final List<String> excludedTools;
+
   DartMCPServer(
     super.channel, {
     required this.sdk,
     this.analytics,
+    this.excludedTools = const [],
     @visibleForTesting this.processManager = const LocalProcessManager(),
     @visibleForTesting this.fileSystem = const LocalFileSystem(),
     this.forceRootsFallback = false,
@@ -96,6 +102,7 @@ final class DartMCPServer extends MCPServer
       () {
         server = DartMCPServer(
           stdioChannel(input: io.stdin, output: io.stdout),
+          excludedTools: parsedArgs.multiOption(excludeToolOption),
           forceRootsFallback: parsedArgs.flag(forceRootsFallbackFlag),
           sdk: Sdk.find(
             dartSdkPath: dartSdkPath,
@@ -167,6 +174,9 @@ final class DartMCPServer extends MCPServer
     FutureOr<CallToolResult> Function(CallToolRequest) impl, {
     bool validateArguments = true,
   }) {
+    // Check manually excluded tools and skip them.
+    if (excludedTools.contains(tool.name)) return;
+
     // For type promotion.
     final analytics = this.analytics;
 

pkgs/dart_mcp_server/test/dart_tooling_mcp_server_test.dart

@@ -6,6 +6,9 @@ import 'dart:async';
 import 'dart:io';
 
 import 'package:dart_mcp/server.dart';
+import 'package:dart_mcp_server/src/arg_parser.dart';
+import 'package:dart_mcp_server/src/mixins/analyzer.dart';
+import 'package:dart_mcp_server/src/mixins/dtd.dart';
 import 'package:dart_mcp_server/src/server.dart';
 import 'package:dart_mcp_server/src/utils/analytics.dart';
 import 'package:test/test.dart';
@@ -256,6 +259,28 @@ void main() {
       await doWithRetries(() => File(logDescriptor.io.path).delete());
     });
   });
+
+  test('Tools can be disabled with --exclude-tool', () async {
+    final testHarness = await TestHarness.start(
+      cliArgs: [
+        '--$excludeToolOption',
+        DartAnalyzerSupport.analyzeFilesTool.name,
+        '--$excludeToolOption',
+        DartToolingDaemonSupport.connectTool.name,
+      ],
+    );
+    final connection = testHarness.serverConnectionPair.serverConnection;
+    final tools = (await connection.listTools()).tools;
+    expect(
+      tools,
+      isNot(contains(equals(DartAnalyzerSupport.analyzeFilesTool))),
+    );
+    expect(
+      tools,
+      isNot(contains(equals(DartToolingDaemonSupport.connectTool))),
+    );
+    expect(tools, contains(equals(DartAnalyzerSupport.hoverTool)));
+  });
 }
 
 /// Performs [action] up to [maxRetries] times, backing off an extra 50ms