package com.architectwand.ai; import com.architectwand.config.WandConfig; import com.google.gson.Gson; import com.google.gson.JsonArray; import com.google.gson.JsonObject; import com.google.gson.JsonParser; import java.net.URI; import java.net.http.HttpClient; import java.net.http.HttpRequest; import java.net.http.HttpResponse; import java.time.Duration; import java.util.ArrayList; import java.util.List; import java.util.concurrent.CompletableFuture; import java.util.concurrent.Executors; public class AiStructureBuilder { private static final int MAX_BLOCKS = 2048; private static final Gson GSON = new Gson(); private static final HttpClient HTTP = HttpClient.newBuilder() .version(HttpClient.Version.HTTP_1_1) .connectTimeout(Duration.ofSeconds(30)) .executor(Executors.newCachedThreadPool()) .build(); public record BuildResult(String description, List placements) {} public static CompletableFuture buildAsync( String prompt, String style, int width, int height, int depth, String prevDescription, WandConfig config ) { return CompletableFuture.supplyAsync(() -> { try { String systemPrompt = buildSystemPrompt(width, height, depth); String userPrompt = buildUserPrompt(prompt, style, prevDescription); String requestBody = buildRequestBody(systemPrompt, userPrompt, config); HttpRequest request = HttpRequest.newBuilder() .uri(URI.create(config.apiUrl)) .header("Content-Type", "application/json") .header("Authorization", "Bearer " + config.apiKey) .POST(HttpRequest.BodyPublishers.ofString(requestBody)) .timeout(Duration.ofSeconds(120)) .build(); HttpResponse response = HTTP.send(request, HttpResponse.BodyHandlers.ofString()); if (response.statusCode() != 200) { throw new RuntimeException("API returned status " + response.statusCode() + ": " + response.body()); } return parseResponse(response.body(), width, height, depth); } catch (Exception e) { throw new RuntimeException(e.getMessage(), e); } }); } private static String buildSystemPrompt(int width, int height, int depth) { return "You are a Minecraft structure architect AI. Generate block placements for a structure.\n" + "Region: " + width + " wide (X), " + height + " tall (Y), " + depth + " deep (Z).\n" + "Coordinates: (0,0,0) = minimum corner, (" + (width-1) + "," + (height-1) + "," + (depth-1) + ") = maximum corner.\n" + "Return ONLY a JSON object with NO markdown fences:\n" + "{\"description\":\"brief description\",\"blocks\":[{\"x\":0,\"y\":0,\"z\":0,\"block\":\"minecraft:stone\"}]}\n" + "Rules:\n" + "- Use only valid Minecraft 1.21 block IDs with the 'minecraft:' prefix\n" + "- Omit air blocks; only include blocks that should be placed\n" + "- Maximum " + MAX_BLOCKS + " block entries\n" + "- Stay within region dimensions\n" + "- Create detailed, coherent structures that fill the space meaningfully"; } private static String buildUserPrompt(String prompt, String style, String prevDescription) { StringBuilder sb = new StringBuilder(); if (!prevDescription.isEmpty()) { sb.append("This is a re-roll. Significantly improve upon this previous build:\n"); sb.append("Previous: ").append(prevDescription).append("\n\n"); sb.append("Improvements requested: "); } sb.append(prompt); if (!style.equalsIgnoreCase("any") && !style.isBlank()) { sb.append("\nStyle: ").append(style); } return sb.toString(); } private static String buildRequestBody(String systemPrompt, String userPrompt, WandConfig config) { JsonObject body = new JsonObject(); body.addProperty("model", config.model); body.addProperty("max_tokens", config.maxTokens); JsonArray messages = new JsonArray(); JsonObject systemMsg = new JsonObject(); systemMsg.addProperty("role", "system"); systemMsg.addProperty("content", systemPrompt); messages.add(systemMsg); JsonObject userMsg = new JsonObject(); userMsg.addProperty("role", "user"); userMsg.addProperty("content", userPrompt); messages.add(userMsg); body.add("messages", messages); return GSON.toJson(body); } private static BuildResult parseResponse(String responseBody, int width, int height, int depth) { JsonObject root = JsonParser.parseString(responseBody).getAsJsonObject(); String content = root.getAsJsonArray("choices") .get(0).getAsJsonObject() .getAsJsonObject("message") .get("content").getAsString(); // Strip markdown code fences if present content = content.replaceAll("(?s)```json\\s*", "").replaceAll("(?s)```\\s*", "").trim(); JsonObject structure = JsonParser.parseString(content).getAsJsonObject(); String description = structure.has("description") ? structure.get("description").getAsString() : ""; JsonArray blocksJson = structure.getAsJsonArray("blocks"); List placements = new ArrayList<>(); for (int i = 0; i < blocksJson.size() && placements.size() < MAX_BLOCKS; i++) { JsonObject b = blocksJson.get(i).getAsJsonObject(); int x = b.get("x").getAsInt(); int y = b.get("y").getAsInt(); int z = b.get("z").getAsInt(); String blockId = b.get("block").getAsString(); if (x < 0 || x >= width || y < 0 || y >= height || z < 0 || z >= depth) continue; if (!blockId.contains(":")) continue; placements.add(new BlockPlacement(x, y, z, blockId)); } return new BuildResult(description, placements); } }