package com.amethystgear.ai; import com.amethystgear.AmethystGear; import com.google.gson.JsonArray; import com.google.gson.JsonObject; import com.google.gson.JsonParser; import lombok.Getter; import org.bukkit.Bukkit; import org.bukkit.entity.Player; import java.io.*; import java.net.HttpURLConnection; import java.net.URL; import java.nio.charset.StandardCharsets; import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; import java.util.*; import java.util.concurrent.CompletableFuture; import java.util.concurrent.ConcurrentHashMap; import java.util.stream.Collectors; @Getter public class AIPluginGenerator { private final AmethystGear plugin; private final Map activeRequests; private final Map> conversationHistory; private final String apiKey; private final String apiEndpoint; private final String model; private final int maxTokens; private final double temperature; public AIPluginGenerator(AmethystGear plugin) { this.plugin = plugin; this.activeRequests = new ConcurrentHashMap<>(); this.conversationHistory = new ConcurrentHashMap<>(); this.apiKey = plugin.getConfig().getString("ai.api-key", ""); this.apiEndpoint = plugin.getConfig().getString("ai.endpoint", "https://api.openai.com/v1/chat/completions"); this.model = plugin.getConfig().getString("ai.model", "gpt-4"); this.maxTokens = plugin.getConfig().getInt("ai.max-tokens", 8000); this.temperature = plugin.getConfig().getDouble("ai.temperature", 0.7); } public CompletableFuture generatePlugin(Player player, String description) { UUID playerId = player.getUniqueId(); if (activeRequests.containsKey(playerId)) { CompletableFuture future = new CompletableFuture<>(); future.complete(new GenerationResult(false, "You already have an active plugin generation request!", null)); return future; } GenerationRequest request = new GenerationRequest(player, description, System.currentTimeMillis()); activeRequests.put(playerId, request); player.sendMessage("§d§l[AI Plugin Generator]"); player.sendMessage("§7Starting plugin generation..."); player.sendMessage("§7Description: §f" + description); player.sendMessage("§7This may take 30-60 seconds..."); return CompletableFuture.supplyAsync(() -> { try { return processGeneration(player, description); } catch (Exception e) { plugin.getLogger().severe("Error generating plugin: " + e.getMessage()); e.printStackTrace(); return new GenerationResult(false, "Error: " + e.getMessage(), null); } finally { activeRequests.remove(playerId); } }); } private GenerationResult processGeneration(Player player, String description) { List history = conversationHistory.computeIfAbsent(player.getUniqueId(), k -> new ArrayList<>()); String systemPrompt = buildSystemPrompt(); String userPrompt = buildUserPrompt(description, history); try { String aiResponse = callAIAPI(systemPrompt, userPrompt); history.add("User: " + description); history.add("Assistant: " + aiResponse); if (history.size() > 20) { history.subList(0, history.size() - 20).clear(); } PluginCode code = extractPluginCode(aiResponse); if (code == null || code.getFiles().isEmpty()) { return new GenerationResult(false, "AI did not generate valid plugin code. Try being more specific.", null); } String pluginName = code.getPluginName(); Path pluginDir = Paths.get(plugin.getDataFolder().getParent(), pluginName); boolean compiled = compileAndPackage(code, pluginDir, pluginName); if (compiled) { return new GenerationResult(true, "Plugin '" + pluginName + "' generated successfully!", code); } else { return new GenerationResult(false, "Plugin code generated but compilation failed. Check console for errors.", code); } } catch (Exception e) { return new GenerationResult(false, "API Error: " + e.getMessage(), null); } } private String buildSystemPrompt() { return "You are an expert Minecraft plugin developer. Generate complete, working Spigot/Paper plugins based on user descriptions. " + "Always include:\n" + "1. Complete Java source files with proper package structure\n" + "2. plugin.yml with correct metadata\n" + "3. config.yml if needed\n" + "4. All necessary imports and dependencies\n" + "5. Proper event listeners and command handlers\n" + "6. Error handling and null checks\n\n" + "Format your response with clear file markers:\n" + "FILE: src/main/java/com/example/Plugin.java\n" + "```java\n[code here]\n```\n\n" + "FILE: src/main/resources/plugin.yml\n" + "```yaml\n[yaml here]\n```\n\n" + "Use Spigot API 1.20.1, Java 17, and follow best practices. Make plugins production-ready."; } private String buildUserPrompt(String description, List history) { StringBuilder prompt = new StringBuilder(); if (!history.isEmpty()) { prompt.append("Previous conversation:\n"); for (String msg : history) { prompt.append(msg).append("\n"); } prompt.append("\n"); } prompt.append("Create a Minecraft plugin with the following requirements:\n"); prompt.append(description); prompt.append("\n\nGenerate complete, working code for all necessary files."); return prompt.toString(); } private String callAIAPI(String systemPrompt, String userPrompt) throws IOException { if (apiKey.isEmpty() || apiKey.equals("your-api-key-here")) { throw new IOException("API key not configured! Set ai.api-key in config.yml"); } URL url = new URL(apiEndpoint); HttpURLConnection conn = (HttpURLConnection) url.openConnection(); conn.setRequestMethod("POST"); conn.setRequestProperty("Content-Type", "application/json"); conn.setRequestProperty("Authorization", "Bearer " + apiKey); conn.setDoOutput(true); JsonObject requestBody = new JsonObject(); requestBody.addProperty("model", model); requestBody.addProperty("max_tokens", maxTokens); requestBody.addProperty("temperature", temperature); JsonArray messages = new JsonArray(); JsonObject systemMessage = new JsonObject(); systemMessage.addProperty("role", "system"); systemMessage.addProperty("content", systemPrompt); messages.add(systemMessage); JsonObject userMessage = new JsonObject(); userMessage.addProperty("role", "user"); userMessage.addProperty("content", userPrompt); messages.add(userMessage); requestBody.add("messages", messages); try (OutputStream os = conn.getOutputStream()) { byte[] input = requestBody.toString().getBytes(StandardCharsets.UTF_8); os.write(input, 0, input.length); } int responseCode = conn.getResponseCode(); if (responseCode != 200) { try (BufferedReader br = new BufferedReader(new InputStreamReader(conn.getErrorStream(), StandardCharsets.UTF_8))) { String error = br.lines().collect(Collectors.joining("\n")); throw new IOException("API returned " + responseCode + ": " + error); } } try (BufferedReader br = new BufferedReader(new InputStreamReader(conn.getInputStream(), StandardCharsets.UTF_8))) { String response = br.lines().collect(Collectors.joining("\n")); JsonObject jsonResponse = JsonParser.parseString(response).getAsJsonObject(); return jsonResponse.getAsJsonArray("choices") .get(0).getAsJsonObject() .getAsJsonObject("message") .get("content").getAsString(); } } private PluginCode extractPluginCode(String aiResponse) { Map files = new HashMap<>(); String pluginName = "GeneratedPlugin"; String[] lines = aiResponse.split("\n"); String currentFile = null; StringBuilder currentContent = new StringBuilder(); boolean inCodeBlock = false; for (String line : lines) { if (line.startsWith("FILE:") || line.startsWith("File:")) { if (currentFile != null && currentContent.length() > 0) { files.put(currentFile, currentContent.toString().trim()); } currentFile = line.substring(5).trim(); currentContent = new StringBuilder(); inCodeBlock = false; } else if (line.trim().startsWith("```")) { inCodeBlock = !inCodeBlock; } else if (inCodeBlock && currentFile != null) { currentContent.append(line).append("\n"); } } if (currentFile != null && currentContent.length() > 0) { files.put(currentFile, currentContent.toString().trim()); } for (Map.Entry entry : files.entrySet()) { if (entry.getKey().contains("plugin.yml")) { String content = entry.getValue(); String[] yamlLines = content.split("\n"); for (String yamlLine : yamlLines) { if (yamlLine.trim().startsWith("name:")) { pluginName = yamlLine.split(":", 2)[1].trim(); break; } } break; } } if (files.isEmpty()) { return null; } return new PluginCode(pluginName, files); } private boolean compileAndPackage(PluginCode code, Path pluginDir, String pluginName) { try { Files.createDirectories(pluginDir); for (Map.Entry entry : code.getFiles().entrySet()) { Path filePath = pluginDir.resolve(entry.getKey()); Files.createDirectories(filePath.getParent()); Files.write(filePath, entry.getValue().getBytes(StandardCharsets.UTF_8)); } Path buildGradle = pluginDir.resolve("build.gradle"); if (!Files.exists(buildGradle)) { String gradleContent = generateBuildGradle(pluginName); Files.write(buildGradle, gradleContent.getBytes(StandardCharsets.UTF_8)); } plugin.getLogger().info("Plugin files written to: " + pluginDir); plugin.getLogger().info("To compile, run: cd " + pluginDir + " && gradle build"); return true; } catch (IOException e) { plugin.getLogger().severe("Error writing plugin files: " + e.getMessage()); e.printStackTrace(); return false; } } private String generateBuildGradle(String pluginName) { return "plugins {\n" + " id 'java'\n" + "}\n\n" + "group = 'com.generated'\n" + "version = '1.0.0'\n\n" + "repositories {\n" + " mavenCentral()\n" + " maven { url = 'https://hub.spigotmc.org/nexus/content/repositories/snapshots/' }\n" + "}\n\n" + "dependencies {\n" + " compileOnly 'org.spigotmc:spigot-api:1.20.1-R0.1-SNAPSHOT'\n" + "}\n\n" + "java {\n" + " toolchain {\n" + " languageVersion = JavaLanguageVersion.of(17)\n" + " }\n" + "}\n"; } public void clearHistory(UUID playerId) { conversationHistory.remove(playerId); } public boolean hasActiveRequest(UUID playerId) { return activeRequests.containsKey(playerId); } @Getter public static class GenerationRequest { private final Player player; private final String description; private final long startTime; public GenerationRequest(Player player, String description, long startTime) { this.player = player; this.description = description; this.startTime = startTime; } } @Getter public static class GenerationResult { private final boolean success; private final String message; private final PluginCode code; public GenerationResult(boolean success, String message, PluginCode code) { this.success = success; this.message = message; this.code = code; } } @Getter public static class PluginCode { private final String pluginName; private final Map files; public PluginCode(String pluginName, Map files) { this.pluginName = pluginName; this.files = files; } } }