JShellWrapper moved sysout char limit into the config

Author: Alathreon

JShellWrapper/src/main/java/Config.java

@@ -1,8 +1,17 @@
 
-public record Config(int evalTimeoutSeconds) {
+public record Config(int evalTimeoutSeconds, int sysOutCharLimit) {
+    static int loadIntEnv(String envName) {
+        return Integer.parseInt(System.getenv(envName));
+    }
     public static Config load() {
         return new Config(
-                Integer.parseInt(System.getenv("evalTimeoutSeconds"))
+                loadIntEnv("evalTimeoutSeconds"),
+                loadIntEnv("sysOutCharLimit")
         );
     }
+
+    public Config {
+        if(evalTimeoutSeconds <= 0) throw new IllegalArgumentException("Invalid evalTimeoutSeconds : " + evalTimeoutSeconds);
+        if(sysOutCharLimit <= 0) throw new IllegalArgumentException("Invalid sysOutCharLimit : " + sysOutCharLimit);
+    }
 }

JShellWrapper/src/main/java/JShellWrapper.java

@@ -16,7 +16,7 @@ public class JShellWrapper {
     public void run(Config config, InputStream in, PrintStream processOut) {
         Scanner processIn = new Scanner(in);
         String startup = desanitize(processIn.nextLine());
-        StringOutputStream jshellOut = new StringOutputStream(1024);
+        StringOutputStream jshellOut = new StringOutputStream(config.sysOutCharLimit());
         try (JShell shell = JShell.builder().out(new PrintStream(jshellOut)).build()) {
             verifyStartupEval(eval(shell, startup, new AtomicBoolean()));
             while(true) {

JShellWrapper/src/test/java/JShellWrapperStartupScriptTest.java

@@ -8,7 +8,7 @@
 class JShellWrapperStartupScriptTest {
     @Test
     void testDoubleSnippets() {
-        Config config = new Config(5);
+        Config config = new Config(5, 1024);
         StringInputStream inputStream = new StringInputStream("""
                 import java.util.*; void println(Object o) { System.out.println(o); }
                 eval

JShellWrapper/src/test/java/JShellWrapperTest.java

@@ -13,7 +13,7 @@ class JShellWrapperTest {
     PrintStream out;
     @BeforeAll
     static void setUp() {
-        config = new Config(5);
+        config = new Config(5, 1024);
         jshell = new JShellWrapper();
     }
     void evalTest(String input, String expectedOutput) {