feat(server): Integrate McpServerCapabilities and McpServerChangeNotification config into McpServerInfo

Author: codeboyzhou

src/main/java/com/github/codeboyzhou/mcp/declarative/configuration/McpServerCapabilities.java

@@ -5,4 +5,9 @@
 public record McpServerCapabilities(
     @JsonProperty("resource") boolean resource,
     @JsonProperty("prompt") boolean prompt,
-    @JsonProperty("tool") boolean tool) {}
+    @JsonProperty("tool") boolean tool) {
+
+  public McpServerCapabilities() {
+    this(true, true, true);
+  }
+}

src/main/java/com/github/codeboyzhou/mcp/declarative/configuration/McpServerChangeNotification.java

@@ -5,4 +5,9 @@
 public record McpServerChangeNotification(
     @JsonProperty("resource") boolean resource,
     @JsonProperty("prompt") boolean prompt,
-    @JsonProperty("tool") boolean tool) {}
+    @JsonProperty("tool") boolean tool) {
+
+  public McpServerChangeNotification() {
+    this(true, true, true);
+  }
+}

src/main/java/com/github/codeboyzhou/mcp/declarative/server/McpServerInfo.java

@@ -1,5 +1,7 @@
 package com.github.codeboyzhou.mcp.declarative.server;
 
+import com.github.codeboyzhou.mcp.declarative.configuration.McpServerCapabilities;
+import com.github.codeboyzhou.mcp.declarative.configuration.McpServerChangeNotification;
 import com.github.codeboyzhou.mcp.declarative.util.Strings;
 import java.time.Duration;
 
@@ -13,11 +15,17 @@ public class McpServerInfo {
 
   private final Duration requestTimeout;
 
+  private final McpServerCapabilities capabilities;
+
+  private final McpServerChangeNotification changeNotification;
+
   protected McpServerInfo(Builder<?> builder) {
     this.name = builder.name;
     this.version = builder.version;
     this.instructions = builder.instructions;
     this.requestTimeout = builder.requestTimeout;
+    this.capabilities = builder.capabilities;
+    this.changeNotification = builder.changeNotification;
   }
 
   public static Builder<?> builder() {
@@ -40,6 +48,14 @@ public Duration requestTimeout() {
     return requestTimeout;
   }
 
+  public McpServerCapabilities capabilities() {
+    return capabilities;
+  }
+
+  public McpServerChangeNotification changeNotification() {
+    return changeNotification;
+  }
+
   @SuppressWarnings("unchecked")
   public static class Builder<T extends Builder<T>> {
 
@@ -51,6 +67,10 @@ public static class Builder<T extends Builder<T>> {
 
     protected Duration requestTimeout = Duration.ofSeconds(20);
 
+    protected McpServerCapabilities capabilities = new McpServerCapabilities();
+
+    protected McpServerChangeNotification changeNotification = new McpServerChangeNotification();
+
     protected T self() {
       return (T) this;
     }
@@ -78,5 +98,15 @@ public T requestTimeout(Duration requestTimeout) {
       this.requestTimeout = requestTimeout;
       return self();
     }
+
+    public T capabilities(McpServerCapabilities capabilities) {
+      this.capabilities = capabilities;
+      return self();
+    }
+
+    public T changeNotification(McpServerChangeNotification changeNotification) {
+      this.changeNotification = changeNotification;
+      return self();
+    }
   }
 }

src/main/java/com/github/codeboyzhou/mcp/declarative/server/factory/AbstractMcpServerFactory.java

@@ -1,6 +1,8 @@
 package com.github.codeboyzhou.mcp.declarative.server.factory;
 
 import com.github.codeboyzhou.mcp.declarative.common.NamedThreadFactory;
+import com.github.codeboyzhou.mcp.declarative.configuration.McpServerCapabilities;
+import com.github.codeboyzhou.mcp.declarative.configuration.McpServerChangeNotification;
 import com.github.codeboyzhou.mcp.declarative.server.McpServerInfo;
 import com.github.codeboyzhou.mcp.declarative.server.component.McpServerComponentRegister;
 import io.modelcontextprotocol.server.McpSyncServer;
@@ -18,18 +20,26 @@ public void startServer(S serverInfo) {
     McpSyncServer server =
         sync(serverInfo)
             .serverInfo(serverInfo.name(), serverInfo.version())
-            .capabilities(serverCapabilities())
+            .capabilities(serverCapabilities(serverInfo))
             .instructions(serverInfo.instructions())
             .requestTimeout(serverInfo.requestTimeout())
             .build();
     McpServerComponentRegister.of(server).registerComponents();
   }
 
-  private McpSchema.ServerCapabilities serverCapabilities() {
-    return McpSchema.ServerCapabilities.builder()
-        .resources(true, true)
-        .prompts(true)
-        .tools(true)
-        .build();
+  private McpSchema.ServerCapabilities serverCapabilities(S serverInfo) {
+    McpSchema.ServerCapabilities.Builder capabilities = McpSchema.ServerCapabilities.builder();
+    McpServerCapabilities capabilitiesConfig = serverInfo.capabilities();
+    McpServerChangeNotification serverChangeNotification = serverInfo.changeNotification();
+    if (capabilitiesConfig.resource()) {
+      capabilities.resources(true, serverChangeNotification.resource());
+    }
+    if (capabilitiesConfig.prompt()) {
+      capabilities.prompts(serverChangeNotification.prompt());
+    }
+    if (capabilitiesConfig.tool()) {
+      capabilities.tools(serverChangeNotification.tool());
+    }
+    return capabilities.build();
   }
 }