✨ feat(mcp): IncreaseAvailableCreatures write slice | add MCP server tool

Author: MateuszNaKodach

…/builddwelling/ModelContextProtocol.java …rite/builddwelling/BuildDwellingMcp.javasrc/main/java/com/dddheroes/heroesofddd/creaturerecruitment/write/builddwelling/ModelContextProtocol.java renamed to src/main/java/com/dddheroes/heroesofddd/creaturerecruitment/write/builddwelling/BuildDwellingMcp.java src/main/java/com/dddheroes/heroesofddd/creaturerecruitment/write/builddwelling/ModelContextProtocol.java renamed to src/main/java/com/dddheroes/heroesofddd/creaturerecruitment/write/builddwelling/BuildDwellingMcp.java

@@ -2,6 +2,7 @@
 
 import com.dddheroes.heroesofddd.shared.application.GameMetaData;
 import org.axonframework.commandhandling.gateway.CommandGateway;
+import org.jetbrains.annotations.NotNull;
 import org.springframework.ai.chat.model.ToolContext;
 import org.springframework.ai.tool.ToolCallbackProvider;
 import org.springframework.ai.tool.annotation.Tool;
@@ -14,7 +15,7 @@
 import java.util.Map;
 import java.util.concurrent.CompletableFuture;
 
-public class ModelContextProtocol {
+public class BuildDwellingMcp {
 
     @Component
     static class Tools {
@@ -41,20 +42,30 @@ public CompletableFuture<Map<String, Object>> buildDwelling(
             var command = BuildDwelling.command(dwellingId, creatureId, costPerTroop);
 
             return commandGateway.send(command, GameMetaData.with(gameId, playerId))
-                    .thenApply(_ -> Map.of(
-                            "success", true,
-                            "dwellingId", dwellingId,
-                            "creatureId", creatureId,
-                            "costPerTroop", costPerTroop,
-                            "playerId", playerId,
-                            "message", "Dwelling built successfully"
-                    ))
-                    .exceptionally(throwable -> Map.of(
-                            "success", false,
-                            "error", throwable != null ? throwable.getMessage() : "Unknown error",
-                            "dwellingId", dwellingId,
-                            "message", "Failed to build dwelling: " + (throwable != null ? throwable.getMessage() : "Unknown error")
-                    ));
+                    .thenApply(_ -> getSuccess(dwellingId, creatureId, costPerTroop, playerId))
+                    .exceptionally(throwable -> failure(dwellingId, throwable));
+        }
+
+        @NotNull
+        private static Map<String, Object> failure(String dwellingId, Throwable throwable) {
+            return Map.of(
+                    "success", false,
+                    "error", throwable != null ? throwable.getMessage() : "Unknown error",
+                    "dwellingId", dwellingId,
+                    "message", "Failed to build dwelling: " + (throwable != null ? throwable.getMessage() : "Unknown error")
+            );
+        }
+
+        @NotNull
+        private static Map<String, Object> getSuccess(String dwellingId, String creatureId, Map<String, Integer> costPerTroop, String playerId) {
+            return Map.of(
+                    "success", true,
+                    "dwellingId", dwellingId,
+                    "creatureId", creatureId,
+                    "costPerTroop", costPerTroop,
+                    "playerId", playerId,
+                    "message", "Dwelling built successfully"
+            );
         }
 
         private String playerId(ToolContext toolContext) {
@@ -67,7 +78,7 @@ private String playerId(ToolContext toolContext) {
     static class Config {
 
         @Bean
-        ToolCallbackProvider buildDwellingTool(ModelContextProtocol.Tools sliceTools) {
+        ToolCallbackProvider buildDwellingTool(BuildDwellingMcp.Tools sliceTools) {
             return MethodToolCallbackProvider.builder().toolObjects(sliceTools).build();
         }
 

src/main/java/com/dddheroes/heroesofddd/creaturerecruitment/write/changeavailablecreatures/IncreaseAvailableCreaturesMcp.java

@@ -0,0 +1,86 @@
+package com.dddheroes.heroesofddd.creaturerecruitment.write.changeavailablecreatures;
+
+import com.dddheroes.heroesofddd.shared.application.GameMetaData;
+import org.axonframework.commandhandling.gateway.CommandGateway;
+import org.jetbrains.annotations.NotNull;
+import org.springframework.ai.chat.model.ToolContext;
+import org.springframework.ai.tool.ToolCallbackProvider;
+import org.springframework.ai.tool.annotation.Tool;
+import org.springframework.ai.tool.annotation.ToolParam;
+import org.springframework.ai.tool.method.MethodToolCallbackProvider;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.stereotype.Component;
+
+import java.util.Map;
+import java.util.concurrent.CompletableFuture;
+
+public class IncreaseAvailableCreaturesMcp {
+
+    @Component
+    static class Tools {
+
+        private final CommandGateway commandGateway;
+
+        public Tools(CommandGateway commandGateway) {
+            this.commandGateway = commandGateway;
+        }
+
+        @Tool(
+                name = "increase_available_creatures",
+                description = "Increase the number of available creatures in a dwelling. This affects how many creatures can be recruited from this dwelling. The playerId should be provided via MCP client context (similar to REST API headers)."
+        )
+        public CompletableFuture<Map<String, Object>> increaseAvailableCreatures(
+                @ToolParam(description = "Unique identifier for the game instance") String gameId,
+                @ToolParam(description = "Unique identifier for the dwelling to modify") String dwellingId,
+                @ToolParam(description = "Type of creature to increase availability for (e.g., 'ANGEL', 'DRAGON', 'GRIFFIN')") String creatureId,
+                @ToolParam(description = "Number of creatures to increase availability by (positive integer)") Integer increaseBy,
+                ToolContext toolContext
+        ) {
+            var playerId = playerId(toolContext);
+
+            var command = IncreaseAvailableCreatures.command(dwellingId, creatureId, increaseBy);
+
+            return commandGateway.send(command, GameMetaData.with(gameId, playerId))
+                    .thenApply(_ -> success(dwellingId, creatureId, increaseBy, playerId))
+                    .exceptionally(throwable -> failure(dwellingId, throwable));
+        }
+
+        @NotNull
+        private static Map<String, Object> success(String dwellingId, String creatureId, Integer increaseBy, String playerId) {
+            return Map.of(
+                    "success", true,
+                    "dwellingId", dwellingId,
+                    "creatureId", creatureId,
+                    "increaseBy", increaseBy,
+                    "playerId", playerId,
+                    "message", "Available creatures increased successfully"
+            );
+        }
+
+        @NotNull
+        private static Map<String, Object> failure(String dwellingId, Throwable throwable) {
+            return Map.of(
+                    "success", false,
+                    "error", throwable != null ? throwable.getMessage() : "Unknown error",
+                    "dwellingId", dwellingId,
+                    "message", "Failed to increase available creatures: " + (throwable != null ? throwable.getMessage() : "Unknown error")
+            );
+        }
+
+        private String playerId(ToolContext toolContext) {
+            return (String) toolContext.getContext().getOrDefault("playerId", "default-player-id");
+        }
+
+    }
+
+    @Configuration
+    static class Config {
+
+        @Bean
+        ToolCallbackProvider increaseAvailableCreaturesTool(IncreaseAvailableCreaturesMcp.Tools sliceTools) {
+            return MethodToolCallbackProvider.builder().toolObjects(sliceTools).build();
+        }
+
+    }
+}