Added AI :>

Author: RealMangorage

build.gradle

@@ -79,7 +79,7 @@ dependencies {
     installer('org.mangorage:installer:3.0.27')
     bot('org.mangorage:mangobot:11.0.18')
 
-    plugin('org.mangorage:mangobotplugin:11.1.6')
+    plugin('org.mangorage:mangobotplugin:11.1.7')
     plugin('org.mangorage:mangobotwebsite:1.0.10')
 
     compileOnly('org.mangorage:MangoBotGradle:4.0.5')

src/main/java/org/mangorage/mangobotgithub/AICommand.java

@@ -0,0 +1,45 @@
+package org.mangorage.mangobotgithub;
+
+import net.dv8tion.jda.api.entities.Message;
+import org.jetbrains.annotations.NotNull;
+import org.mangorage.mangobotapi.core.commands.Arguments;
+import org.mangorage.mangobotapi.core.commands.CommandResult;
+import org.mangorage.mangobotapi.core.commands.CommandType;
+import org.mangorage.mangobotapi.core.commands.IBasicCommand;
+
+import java.io.IOException;
+
+public class AICommand implements IBasicCommand {
+    @Override
+    public @NotNull CommandResult execute(Message message, Arguments arguments) {
+        var prompt = arguments.getFrom(0);
+        if (prompt.isBlank()) {
+
+        } else {
+            try {
+                var response = ChatGPTBot.askChatGPT(prompt);
+                if (response != null) {
+                    var choices = response.getChoices();
+                    if (!choices.isEmpty()) {
+                        var msg = choices.getFirst().getMessage();
+                        if (msg != null) {
+                            message.reply(msg.getContent()).setSuppressEmbeds(true).mentionRepliedUser(false).queue();
+                        }
+                    }
+                }
+            } catch (IOException ignored) {
+            }
+        }
+        return CommandResult.PASS;
+    }
+
+    @Override
+    public String commandId() {
+        return "askAI";
+    }
+
+    @Override
+    public CommandType commandType() {
+        return CommandType.BOTH;
+    }
+}

src/main/java/org/mangorage/mangobotgithub/ChatGPTBot.java

@@ -0,0 +1,32 @@
+package org.mangorage.mangobotgithub;
+import com.google.gson.Gson;
+import okhttp3.*;
+
+import java.io.IOException;
+
+public class ChatGPTBot {
+    private static final String API_URL = "https://api.openai.com/v1/chat/completions";
+    private static final Gson gson = new Gson();
+
+    public static ChatGPTResponse askChatGPT(String prompt) throws IOException {
+        OkHttpClient client = new OkHttpClient();
+
+        String json = "{ \"model\": \"gpt-4o-mini\", \"messages\": [{\"role\": \"user\", \"content\": \"" + prompt + "\"}] }";
+
+        RequestBody body = RequestBody.create(MediaType.get("application/json"), json);
+        Request request = new Request.Builder()
+                .url(API_URL)
+                .post(body)
+                .addHeader("Authorization", "Bearer " + MangoBotGithub.CHAT_AI_TOKEN.get())
+                .build();
+
+        try (Response response = client.newCall(request).execute()) {
+            if (response.isSuccessful()) {
+                String responseBody = response.body().string();
+                System.out.println(responseBody);
+                return gson.fromJson(responseBody, ChatGPTResponse.class);
+            }
+        }
+        return null;
+    }
+}

src/main/java/org/mangorage/mangobotgithub/ChatGPTResponse.java

@@ -0,0 +1,103 @@
+package org.mangorage.mangobotgithub;
+
+import com.google.gson.annotations.SerializedName;
+
+import java.util.List;
+
+public class ChatGPTResponse {
+
+    private String id;
+    private String object;
+    private int created;
+    private String model;
+
+    @SerializedName("choices")
+    private List<Choice> choices;
+
+    // Define inner class for choices
+    public static class Choice {
+        private Message message;
+        private int index;
+
+        // Getters and setters
+        public Message getMessage() {
+            return message;
+        }
+
+        public void setMessage(Message message) {
+            this.message = message;
+        }
+
+        public int getIndex() {
+            return index;
+        }
+
+        public void setIndex(int index) {
+            this.index = index;
+        }
+    }
+
+    // Define inner class for message
+    public static class Message {
+        private String role;
+        private String content;
+
+        // Getters and setters
+        public String getRole() {
+            return role;
+        }
+
+        public void setRole(String role) {
+            this.role = role;
+        }
+
+        public String getContent() {
+            return content;
+        }
+
+        public void setContent(String content) {
+            this.content = content;
+        }
+    }
+
+    // Getters and setters for top-level fields
+    public String getId() {
+        return id;
+    }
+
+    public void setId(String id) {
+        this.id = id;
+    }
+
+    public String getObject() {
+        return object;
+    }
+
+    public void setObject(String object) {
+        this.object = object;
+    }
+
+    public int getCreated() {
+        return created;
+    }
+
+    public void setCreated(int created) {
+        this.created = created;
+    }
+
+    public String getModel() {
+        return model;
+    }
+
+    public void setModel(String model) {
+        this.model = model;
+    }
+
+    public List<Choice> getChoices() {
+        return choices;
+    }
+
+    public void setChoices(List<Choice> choices) {
+        this.choices = choices;
+    }
+}

src/main/java/org/mangorage/mangobotgithub/MangoBotGithub.java

@@ -27,6 +27,7 @@ public class MangoBotGithub extends AbstractPlugin {
 
     public static final ISetting<String> GITHUB_TOKEN = ConfigSetting.create(CONFIG, "PASTE_TOKEN", "empty");
     public static final ISetting<String> GITHUB_USERNAME = ConfigSetting.create(CONFIG, "GITHUB_USERNAME", "RealMangoRage");
+    public static final ISetting<String> CHAT_AI_TOKEN = ConfigSetting.create(CONFIG, "AI_TOKEN", "empty");
 
 
     private final MangoBotPlugin parent;
@@ -49,9 +50,11 @@ public void onRegistration(StartupEvent event) {
         if (event.phase() == StartupEvent.Phase.REGISTRATION) {
             GuildConfig.loadServerConfigs();
 
+
             var cmdRegistry = parent.getCommandRegistry();
             cmdRegistry.addBasicCommand(new PRScanCommand(parent));
             cmdRegistry.addBasicCommand(new IssueScanCommand(parent));
+            cmdRegistry.addBasicCommand(new AICommand());
 
             new GHPRStatus(parent);
             new GHIssueStatus(parent);