refactor: loadResource

Author: ReaJason

packer/src/main/java/com/reajason/javaweb/packer/Packers.java

@@ -113,6 +113,7 @@ public enum Packers {
     Aviator(new AviatorPacker()),
 
     JXPath(new JXPathPacker()),
+    JXPathScriptEngine(new JXPathScriptEnginePacker(), JXPathPacker.class),
     JXPathSpringGzip(new JXPathSpringGzipPacker(), JXPathPacker.class),
     JXPathSpringGzipJDK17(new JXPathSpringGzipJDK17Packer(), JXPathPacker.class),
 

packer/src/main/java/com/reajason/javaweb/packer/Util.java

@@ -0,0 +1,22 @@
+package com.reajason.javaweb.packer;
+
+import lombok.SneakyThrows;
+import org.apache.commons.io.IOUtils;
+
+import java.io.InputStream;
+import java.nio.charset.Charset;
+import java.util.Objects;
+
+/**
+ * @author ReaJason
+ * @since 2025/9/1
+ */
+public class Util {
+
+    @SneakyThrows
+    public static String loadTemplateFromResource(String resourceName) {
+        try (InputStream stream = Util.class.getResourceAsStream(resourceName)) {
+            return IOUtils.toString(Objects.requireNonNull(stream), Charset.defaultCharset());
+        }
+    }
+}

packer/src/main/java/com/reajason/javaweb/packer/XxlJobPacker.java

@@ -16,15 +16,8 @@
  * @since 2025/1/21
  */
 public class XxlJobPacker implements Packer {
-    String template = "";
-
-    public XxlJobPacker() {
-        try {
-            template = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/XXL-Job-DefineClass.java")), Charset.defaultCharset());
-        } catch (IOException ignored) {
-
-        }
-    }
+    private final String template = Util.loadTemplateFromResource("/memshell-party/XXL-Job-DefineClass.java");
+    private final ObjectMapper objectMapper = new ObjectMapper().enable(SerializationFeature.INDENT_OUTPUT);
 
     @Override
     @SneakyThrows
@@ -44,8 +37,6 @@ public String pack(ClassPackerConfig config) {
         map.put("glueUpdatetime", System.currentTimeMillis());
         map.put("broadcastIndex", 0);
         map.put("broadcastTotal", 0);
-        ObjectMapper objectMapper = new ObjectMapper();
-        objectMapper.enable(SerializationFeature.INDENT_OUTPUT); // 美化输出
         return objectMapper.writeValueAsString(map);
     }
 }

packer/src/main/java/com/reajason/javaweb/packer/groovy/GroovyClassDefinerPacker.java

@@ -2,27 +2,15 @@
 
 import com.reajason.javaweb.packer.ClassPackerConfig;
 import com.reajason.javaweb.packer.Packer;
+import com.reajason.javaweb.packer.Util;
 import lombok.SneakyThrows;
-import org.apache.commons.io.IOUtils;
-
-import java.io.IOException;
-import java.nio.charset.Charset;
-import java.util.Objects;
 
 /**
  * @author ReaJason
  * @since 2025/5/11
  */
 public class GroovyClassDefinerPacker implements Packer {
-    String template = null;
-
-    public GroovyClassDefinerPacker() {
-        try {
-            template = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/shell.groovy")), Charset.defaultCharset());
-        } catch (IOException ignored) {
-
-        }
-    }
+    private final String template = Util.loadTemplateFromResource("/memshell-party/shell.groovy");
 
     @Override
     @SneakyThrows

packer/src/main/java/com/reajason/javaweb/packer/jsp/ClassLoaderJspPacker.java

@@ -2,6 +2,7 @@
 
 import com.reajason.javaweb.packer.ClassPackerConfig;
 import com.reajason.javaweb.packer.Packer;
+import com.reajason.javaweb.packer.Util;
 import lombok.SneakyThrows;
 import org.apache.commons.io.IOUtils;
 
@@ -14,15 +15,7 @@
  */
 public class ClassLoaderJspPacker implements Packer {
 
-    String jspTemplate = null;
-
-    public ClassLoaderJspPacker() {
-        try {
-            jspTemplate = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/shell.jsp")), Charset.defaultCharset());
-        } catch (Exception ignored) {
-
-        }
-    }
+    private final String jspTemplate = Util.loadTemplateFromResource("/memshell-party/shell.jsp");
 
     @Override
     @SneakyThrows

packer/src/main/java/com/reajason/javaweb/packer/jsp/DefineClassJspPacker.java

@@ -2,29 +2,17 @@
 
 import com.reajason.javaweb.packer.ClassPackerConfig;
 import com.reajason.javaweb.packer.Packer;
+import com.reajason.javaweb.packer.Util;
 import lombok.SneakyThrows;
-import org.apache.commons.io.IOUtils;
-
-import java.nio.charset.Charset;
-import java.util.Objects;
 
 /**
  * @author ReaJason
  * @since 2024/11/26
  */
 public class DefineClassJspPacker implements Packer {
 
-    String template = null;
-    String bypassTemplate = null;
-
-    public DefineClassJspPacker() {
-        try {
-            template = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/shell1.jsp")), Charset.defaultCharset());
-            bypassTemplate = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/shell2.jsp")), Charset.defaultCharset());
-        } catch (Exception ignored) {
-
-        }
-    }
+    private final String template = Util.loadTemplateFromResource("/memshell-party/shell1.jsp");
+    private final String bypassTemplate = Util.loadTemplateFromResource("/memshell-party/shell2.jsp");
 
     @Override
     @SneakyThrows

packer/src/main/java/com/reajason/javaweb/packer/jsp/JspxPacker.java

@@ -2,6 +2,7 @@
 
 import com.reajason.javaweb.packer.ClassPackerConfig;
 import com.reajason.javaweb.packer.Packer;
+import com.reajason.javaweb.packer.Util;
 import lombok.SneakyThrows;
 import org.apache.commons.io.IOUtils;
 
@@ -14,15 +15,7 @@
  */
 public class JspxPacker implements Packer {
 
-    String jspxTemplate = null;
-
-    public JspxPacker() {
-        try {
-            jspxTemplate = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/shell.jspx")), Charset.defaultCharset());
-        } catch (Exception ignored) {
-
-        }
-    }
+    private final String jspxTemplate = Util.loadTemplateFromResource("/memshell-party/shell.jspx");
 
     @Override
     @SneakyThrows

packer/src/main/java/com/reajason/javaweb/packer/scriptengine/DefaultScriptEnginePacker.java

@@ -2,31 +2,22 @@
 
 import com.reajason.javaweb.packer.ClassPackerConfig;
 import com.reajason.javaweb.packer.Packer;
+import com.reajason.javaweb.packer.Util;
 import lombok.SneakyThrows;
-import org.apache.commons.io.IOUtils;
-
-import java.io.IOException;
-import java.nio.charset.Charset;
-import java.util.Objects;
 
 public class DefaultScriptEnginePacker implements Packer {
-    String jsTemplate = null;
-
-    public DefaultScriptEnginePacker() {
-        try {
-            jsTemplate = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/DefaultScriptEngine.js")), Charset.defaultCharset());
-        } catch (IOException ignored) {
-
-        }
-    }
+    private final String jsTemplate = Util.loadTemplateFromResource("/memshell-party/ScriptEngine.js");
 
     @Override
     @SneakyThrows
     public String pack(ClassPackerConfig config) {
-        return jsTemplate
+        return scriptToSingleLine(jsTemplate
                 .replace("{{className}}", config.getClassName())
-                .replace("{{base64Str}}", config.getClassBytesBase64Str())
-                .replace("\n", "")
+                .replace("{{base64Str}}", config.getClassBytesBase64Str()));
+    }
+
+    public static String scriptToSingleLine(String script) {
+        return script.replace("\n", "")
                 .replaceAll("(?m)^[ \t]+|[ \t]+$", "")
                 .replaceAll("[ \t]{2,}", " ");
     }

packer/src/main/java/com/reajason/javaweb/packer/scriptengine/ScriptEngineBigIntegerPacker.java

@@ -3,32 +3,19 @@
 import com.reajason.javaweb.packer.ClassPackerConfig;
 import com.reajason.javaweb.packer.Packer;
 import com.reajason.javaweb.packer.Packers;
+import com.reajason.javaweb.packer.Util;
 import lombok.SneakyThrows;
-import org.apache.commons.io.IOUtils;
 
-import java.io.IOException;
-import java.nio.charset.Charset;
-import java.util.Objects;
+import static com.reajason.javaweb.packer.scriptengine.DefaultScriptEnginePacker.scriptToSingleLine;
 
 public class ScriptEngineBigIntegerPacker implements Packer {
-    String jsTemplate = null;
-
-    public ScriptEngineBigIntegerPacker() {
-        try {
-            jsTemplate = IOUtils.toString(Objects.requireNonNull(this.getClass().getResourceAsStream("/ScriptEngineBigInteger.js")), Charset.defaultCharset());
-        } catch (IOException ignored) {
-
-        }
-    }
+    private final String jsTemplate = Util.loadTemplateFromResource("/memshell-party/ScriptEngineBigInteger.js");
 
     @Override
     @SneakyThrows
     public String pack(ClassPackerConfig config) {
-        return jsTemplate
+        return scriptToSingleLine(jsTemplate
                 .replace("{{className}}", config.getClassName())
-                .replace("{{bigIntegerStr}}", Packers.BigInteger.getInstance().pack(config))
-                .replace("\n", "")
-                .replaceAll("(?m)^[ \t]+|[ \t]+$", "")
-                .replaceAll("[ \t]{2,}", " ");
+                .replace("{{bigIntegerStr}}", Packers.BigInteger.getInstance().pack(config)));
     }
 }

packer/src/main/java/com/reajason/javaweb/packer/scriptengine/ScriptEngineNoSquareBracketsPacker.java

@@ -0,0 +1,24 @@
+package com.reajason.javaweb.packer.scriptengine;
+
+import com.reajason.javaweb.packer.ClassPackerConfig;
+import com.reajason.javaweb.packer.Packer;
+import com.reajason.javaweb.packer.Util;
+import lombok.SneakyThrows;
+
+public class ScriptEngineNoSquareBracketsPacker implements Packer {
+    private final String jsTemplate = Util.loadTemplateFromResource("/memshell-party/ScriptEngineNoSquareBrackets.js");
+
+    @Override
+    @SneakyThrows
+    public String pack(ClassPackerConfig config) {
+        return scriptToSingleLine(jsTemplate
+                .replace("{{className}}", config.getClassName())
+                .replace("{{base64Str}}", config.getClassBytesBase64Str()));
+    }
+
+    public static String scriptToSingleLine(String script) {
+        return script.replace("\n", "")
+                .replaceAll("(?m)^[ \t]+|[ \t]+$", "")
+                .replaceAll("[ \t]{2,}", " ");
+    }
+}