Added SerializedResponseTransformer. It changes the response by the Accept header

Author: JulianFun123

src/main/java/org/javawebstack/framework/WebApplication.java

@@ -14,17 +14,16 @@
 import org.javawebstack.framework.command.crypto.HashCommand;
 import org.javawebstack.framework.command.db.MigrateCommand;
 import org.javawebstack.framework.command.db.SeedCommand;
+import org.javawebstack.framework.command.schedule.WorkCommand;
 import org.javawebstack.framework.config.Config;
 import org.javawebstack.framework.module.Module;
 import org.javawebstack.framework.seed.AllSeeder;
 import org.javawebstack.framework.seed.MergedSeeder;
 import org.javawebstack.framework.seed.Seeder;
 import org.javawebstack.framework.util.*;
 import org.javawebstack.httpserver.HTTPServer;
-import org.javawebstack.httpserver.transformer.response.JsonResponseTransformer;
 import org.javawebstack.injector.Injector;
 import org.javawebstack.injector.SimpleInjector;
-import org.javawebstack.orm.ORM;
 import org.javawebstack.orm.Repo;
 import org.javawebstack.orm.exception.ORMConfigurationException;
 import org.javawebstack.orm.wrapper.MySQL;
@@ -140,8 +139,8 @@ public WebApplication() {
         injector.inject(this);
         server.beforeInterceptor(new CORSPolicy(config.get("http.server.cors", "*")));
         server.beforeInterceptor(new MultipartPolicy(config.get("http.server.tmp", null)));
-        if (config.isEnabled("http.server.json", true))
-            server.responseTransformer(new JsonResponseTransformer().ignoreStrings());
+        if (config.isEnabled("http.server.autoserialization", true))
+            server.responseTransformer(new SerializedResponseTransformer().ignoreStrings());
         if (sql != null)
             server.routeParamTransformer(modelBindParamTransformer);
         modules.forEach(m -> m.beforeSetupServer(this, server));
@@ -169,6 +168,9 @@ public WebApplication() {
         commandSystem.addCommand("generate", new MultiCommand()
                 .add("key", new GenerateKeyCommand())
         );
+        commandSystem.addCommand("schedule", new MultiCommand()
+                .add("work", new WorkCommand())
+        );
     }
 
     public Map<String, Seeder> getSeeders() {

src/main/java/org/javawebstack/framework/util/SerializedResponseTransformer.java

@@ -0,0 +1,52 @@
+package org.javawebstack.framework.util;
+
+import org.javawebstack.abstractdata.AbstractMapper;
+import org.javawebstack.abstractdata.NamingPolicy;
+import org.javawebstack.httpserver.Exchange;
+import org.javawebstack.httpserver.transformer.response.ResponseTransformer;
+
+public class SerializedResponseTransformer implements ResponseTransformer {
+    private final AbstractMapper mapper;
+    private boolean ignoreStrings;
+
+    public SerializedResponseTransformer() {
+        this((new AbstractMapper()).setNamingPolicy(NamingPolicy.SNAKE_CASE));
+    }
+
+    public SerializedResponseTransformer(AbstractMapper mapper) {
+        this.ignoreStrings = false;
+        this.mapper = mapper;
+    }
+
+    public SerializedResponseTransformer ignoreStrings() {
+        this.ignoreStrings = true;
+        return this;
+    }
+
+    public String transform(Exchange exchange, Object object) {
+        if (object instanceof byte[]) {
+            return null;
+        } else {
+            if (this.ignoreStrings && object instanceof String)
+                return null;
+            String accept = exchange.header("Accept");
+
+            if (accept != null) {
+                switch (accept.toLowerCase()) {
+                    case "application/x-yaml":
+                    case "application/yaml":
+                    case "text/yaml":
+                    case "text/x-yaml":
+                        exchange.contentType(accept);
+                        return this.mapper.toAbstract(object).toYaml();
+                    case "application/x-www-form-urlencoded":
+                        exchange.contentType(accept);
+                        return this.mapper.toAbstract(object).toFormDataString();
+                }
+            }
+
+            exchange.contentType("application/json");
+            return this.mapper.toAbstract(object).toJsonString();
+        }
+    }
+}