Feature/jshell from fork (#869)

* JShell feature added

* Added option to get the snippets of another user

* Refactored JShell eval into its own class

* Changed RateLimiter so it's a global limiter and not a per user limiter + merged user and oneOffSession together

* Added context action for running java code

* Fixing sonar wranings for jshell

Author: Alathreon

application/config.json.template

@@ -90,4 +90,9 @@
     "logErrorChannelWebhook": "<put_your_webhook_here>",
     "openaiApiKey": "<check pins in #tjbot_discussion for the key>",
     "sourceCodeBaseUrl": "<https://github.com/<your_account_here>/<your_repo_here>/blob/master/application/src/main/java/>"
+    "jshell": {
+        "baseUrl": "http://localhost:8080/jshell/",
+        "rateLimitWindowSeconds": 10,
+        "rateLimitRequestsInWindow": 3
+    }
 }

application/src/main/java/org/togetherjava/tjbot/config/Config.java

@@ -38,6 +38,7 @@ public final class Config {
     private final String logErrorChannelWebhook;
     private final String openaiApiKey;
     private final String sourceCodeBaseUrl;
+    private final JShellConfig jshell;
 
     @SuppressWarnings("ConstructorWithTooManyParameters")
     @JsonCreator(mode = JsonCreator.Mode.PROPERTIES)
@@ -74,7 +75,8 @@ private Config(@JsonProperty(value = "token", required = true) String token,
             @JsonProperty(value = "logErrorChannelWebhook",
                     required = true) String logErrorChannelWebhook,
             @JsonProperty(value = "openaiApiKey", required = true) String openaiApiKey,
-            @JsonProperty(value = "sourceCodeBaseUrl", required = true) String sourceCodeBaseUrl) {
+            @JsonProperty(value = "sourceCodeBaseUrl", required = true) String sourceCodeBaseUrl,
+            @JsonProperty(value = "jshell", required = true) JShellConfig jshell) {
         this.token = Objects.requireNonNull(token);
         this.gistApiKey = Objects.requireNonNull(gistApiKey);
         this.databasePath = Objects.requireNonNull(databasePath);
@@ -99,6 +101,7 @@ private Config(@JsonProperty(value = "token", required = true) String token,
         this.logErrorChannelWebhook = Objects.requireNonNull(logErrorChannelWebhook);
         this.openaiApiKey = Objects.requireNonNull(openaiApiKey);
         this.sourceCodeBaseUrl = Objects.requireNonNull(sourceCodeBaseUrl);
+        this.jshell = Objects.requireNonNull(jshell);
     }
 
     /**
@@ -330,4 +333,13 @@ public String getOpenaiApiKey() {
     public String getSourceCodeBaseUrl() {
         return sourceCodeBaseUrl;
     }
+
+    /**
+     * The configuration about jshell.
+     * 
+     * @return the jshell configuration
+     */
+    public JShellConfig getJshell() {
+        return jshell;
+    }
 }

application/src/main/java/org/togetherjava/tjbot/config/JShellConfig.java

@@ -0,0 +1,22 @@
+package org.togetherjava.tjbot.config;
+
+import com.linkedin.urls.Url;
+
+import java.net.MalformedURLException;
+
+public record JShellConfig(String baseUrl, int rateLimitWindowSeconds,
+        int rateLimitRequestsInWindow) {
+    public JShellConfig {
+        try {
+            Url.create(baseUrl);
+        } catch (MalformedURLException e) {
+            throw new IllegalArgumentException(e);
+        }
+        if (rateLimitWindowSeconds <= 0)
+            throw new IllegalArgumentException(
+                    "Illegal rateLimitWindowSeconds : " + rateLimitWindowSeconds);
+        if (rateLimitRequestsInWindow <= 0)
+            throw new IllegalArgumentException(
+                    "Illegal rateLimitRequestsInWindow : " + rateLimitRequestsInWindow);
+    }
+}

application/src/main/java/org/togetherjava/tjbot/features/Features.java

@@ -13,6 +13,8 @@
 import org.togetherjava.tjbot.features.code.CodeMessageManualDetection;
 import org.togetherjava.tjbot.features.filesharing.FileSharingMessageListener;
 import org.togetherjava.tjbot.features.help.*;
+import org.togetherjava.tjbot.features.jshell.JShellCommand;
+import org.togetherjava.tjbot.features.jshell.JShellEval;
 import org.togetherjava.tjbot.features.mathcommands.TeXCommand;
 import org.togetherjava.tjbot.features.mathcommands.wolframalpha.WolframAlphaCommand;
 import org.togetherjava.tjbot.features.mediaonly.MediaOnlyChannelListener;
@@ -67,13 +69,15 @@ private Features() {
      * @return a collection of all features
      */
     public static Collection<Feature> createFeatures(JDA jda, Database database, Config config) {
+        JShellEval jshellEval = new JShellEval(config.getJshell());
+
         TagSystem tagSystem = new TagSystem(database);
         BookmarksSystem bookmarksSystem = new BookmarksSystem(config, database);
         ModerationActionsStore actionsStore = new ModerationActionsStore(database);
         ModAuditLogWriter modAuditLogWriter = new ModAuditLogWriter(config);
         ScamHistoryStore scamHistoryStore = new ScamHistoryStore(database);
         HelpSystemHelper helpSystemHelper = new HelpSystemHelper(config, database);
-        CodeMessageHandler codeMessageHandler = new CodeMessageHandler();
+        CodeMessageHandler codeMessageHandler = new CodeMessageHandler(jshellEval);
         ChatGptService chatGptService = new ChatGptService(config);
 
         // NOTE The system can add special system relevant commands also by itself,
@@ -143,6 +147,7 @@ public static Collection<Feature> createFeatures(JDA jda, Database database, Con
         features.add(new ReportCommand(config));
         features.add(new BookmarksCommand(bookmarksSystem));
         features.add(new ChatGptCommand(chatGptService));
+        features.add(new JShellCommand(jshellEval));
         return features;
     }
 }

application/src/main/java/org/togetherjava/tjbot/features/code/CodeMessageHandler.java

@@ -20,6 +20,7 @@
 import org.togetherjava.tjbot.features.UserInteractor;
 import org.togetherjava.tjbot.features.componentids.ComponentIdGenerator;
 import org.togetherjava.tjbot.features.componentids.ComponentIdInteractor;
+import org.togetherjava.tjbot.features.jshell.JShellEval;
 import org.togetherjava.tjbot.features.utils.CodeFence;
 import org.togetherjava.tjbot.features.utils.MessageUtils;
 
@@ -63,10 +64,11 @@ public final class CodeMessageHandler extends MessageReceiverAdapter implements
     /**
      * Creates a new instance.
      */
-    public CodeMessageHandler() {
+    public CodeMessageHandler(JShellEval jshellEval) {
         componentIdInteractor = new ComponentIdInteractor(getInteractionType(), getName());
 
-        List<CodeAction> codeActions = List.of(new FormatCodeCommand());
+        List<CodeAction> codeActions =
+                List.of(new FormatCodeCommand(), new EvalCodeCommand(jshellEval));
 
         labelToCodeAction = codeActions.stream()
             .collect(Collectors.toMap(CodeAction::getLabel, Function.identity(), (x, y) -> y,

application/src/main/java/org/togetherjava/tjbot/features/code/EvalCodeCommand.java

@@ -0,0 +1,42 @@
+package org.togetherjava.tjbot.features.code;
+
+import net.dv8tion.jda.api.EmbedBuilder;
+import net.dv8tion.jda.api.entities.MessageEmbed;
+
+import org.togetherjava.tjbot.features.jshell.JShellEval;
+import org.togetherjava.tjbot.features.jshell.render.Colors;
+import org.togetherjava.tjbot.features.utils.CodeFence;
+import org.togetherjava.tjbot.features.utils.RequestFailedException;
+
+/**
+ * Evaluates the given code.
+ */
+final class EvalCodeCommand implements CodeAction {
+    private final JShellEval jshellEval;
+
+    EvalCodeCommand(JShellEval jshellEval) {
+        this.jshellEval = jshellEval;
+    }
+
+    @Override
+    public String getLabel() {
+        return "Run code";
+    }
+
+    @Override
+    public MessageEmbed apply(CodeFence codeFence) {
+        if (codeFence.code().isEmpty()) {
+            return new EmbedBuilder().setColor(Colors.ERROR_COLOR)
+                .setDescription("There is nothing to evaluate")
+                .build();
+        }
+        try {
+            return jshellEval.evaluateAndRespond(null, codeFence.code(), false);
+        } catch (RequestFailedException e) {
+            return new EmbedBuilder().setColor(Colors.ERROR_COLOR)
+                .setDescription("Request failed: " + e.getMessage())
+                .build();
+        }
+    }
+
+}