Added crypt and migrate commands

Author: JanHolger

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

@@ -3,10 +3,10 @@
 import com.github.javafaker.Faker;
 import org.javawebstack.command.CommandResult;
 import org.javawebstack.command.CommandSystem;
+import org.javawebstack.command.MultiCommand;
 import org.javawebstack.framework.bind.ModelBindParamTransformer;
 import org.javawebstack.framework.bind.ModelBindTransformer;
-import org.javawebstack.framework.command.ShellCommand;
-import org.javawebstack.framework.command.StartCommand;
+import org.javawebstack.framework.command.*;
 import org.javawebstack.framework.config.Config;
 import org.javawebstack.framework.module.Module;
 import org.javawebstack.framework.util.CORSPolicy;
@@ -105,6 +105,12 @@ public WebApplication(){
         modules.forEach(m -> m.setupCommands(this, commandSystem));
         commandSystem.addCommand("start", new StartCommand());
         commandSystem.addCommand("sh", new ShellCommand());
+        commandSystem.addCommand("migrate", new MigrateCommand());
+        commandSystem.addCommand("crypt", new MultiCommand()
+                .add("generate", new CryptGenerateCommand())
+                .add("encrypt", new CryptEncryptCommand())
+                .add("hash", new CryptHashCommand())
+        );
     }
 
     public WebApplication addModule(Module module){

src/main/java/org/javawebstack/framework/command/CryptEncryptCommand.java

@@ -0,0 +1,22 @@
+package org.javawebstack.framework.command;
+
+import org.javawebstack.command.Command;
+import org.javawebstack.command.CommandResult;
+import org.javawebstack.command.CommandSystem;
+import org.javawebstack.framework.util.Crypt;
+import org.javawebstack.injector.Inject;
+
+import java.util.List;
+import java.util.Map;
+
+public class CryptEncryptCommand implements Command {
+
+    @Inject
+    Crypt crypt;
+
+    public CommandResult execute(CommandSystem commandSystem, List<String> args, Map<String, List<String>> params) {
+        System.out.println(crypt.hash(args.size() > 0 ? args.get(0) : ""));
+        return CommandResult.success();
+    }
+
+}

src/main/java/org/javawebstack/framework/command/CryptGenerateCommand.java

@@ -0,0 +1,51 @@
+package org.javawebstack.framework.command;
+
+import org.javawebstack.command.Command;
+import org.javawebstack.command.CommandResult;
+import org.javawebstack.command.CommandSystem;
+import org.javawebstack.framework.config.Config;
+import org.javawebstack.framework.util.Crypt;
+import org.javawebstack.framework.util.IO;
+import org.javawebstack.injector.Inject;
+
+import java.io.IOException;
+import java.util.List;
+import java.util.Map;
+
+public class CryptGenerateCommand implements Command {
+
+    public CommandResult execute(CommandSystem commandSystem, List<String> args, Map<String, List<String>> params) {
+        String[] lines;
+        try {
+            lines = IO.readTextFile(".env").replace("\r", "").split("\\n");
+        } catch (IOException e) {
+            lines = new String[0];
+        }
+        boolean found = false;
+        for(int i=0; i<lines.length; i++){
+            if(lines[i].startsWith("CRYPT_KEY=")){
+                if(lines[i].length() > 10 && !params.containsKey("f"))
+                    return CommandResult.error("You already have a key. You might loose all your encrypted data if you regenerate it! Use -f to do it anyway.");
+                lines[i] = "CRYPT_KEY=" + Crypt.generateKey();
+                found = true;
+                break;
+            }
+        }
+        if(!found){
+            String[] newLines = new String[lines.length + 3];
+            System.arraycopy(lines, 0, newLines, 0, lines.length);
+            newLines[newLines.length-3] = "";
+            newLines[newLines.length-2] = "# Encryption Key";
+            newLines[newLines.length-1] = "CRYPT_KEY=" + Crypt.generateKey();
+            lines = newLines;
+        }
+        try {
+            IO.writeFile(".env", String.join("\n", lines));
+            System.out.println("Generated a new key!");
+            return CommandResult.success();
+        } catch (IOException e) {
+            return CommandResult.error(e);
+        }
+    }
+
+}

src/main/java/org/javawebstack/framework/command/CryptHashCommand.java

@@ -0,0 +1,22 @@
+package org.javawebstack.framework.command;
+
+import org.javawebstack.command.Command;
+import org.javawebstack.command.CommandResult;
+import org.javawebstack.command.CommandSystem;
+import org.javawebstack.framework.util.Crypt;
+import org.javawebstack.injector.Inject;
+
+import java.util.List;
+import java.util.Map;
+
+public class CryptHashCommand implements Command {
+
+    @Inject
+    Crypt crypt;
+
+    public CommandResult execute(CommandSystem commandSystem, List<String> args, Map<String, List<String>> params) {
+        System.out.println(crypt.hash(args.size() > 0 ? args.get(0) : ""));
+        return CommandResult.success();
+    }
+
+}

src/main/java/org/javawebstack/framework/command/MigrateCommand.java

@@ -0,0 +1,25 @@
+package org.javawebstack.framework.command;
+
+import org.javawebstack.command.Command;
+import org.javawebstack.command.CommandResult;
+import org.javawebstack.command.CommandSystem;
+import org.javawebstack.orm.ORM;
+
+import java.util.List;
+import java.util.Map;
+
+public class MigrateCommand implements Command {
+
+    public CommandResult execute(CommandSystem commandSystem, List<String> args, Map<String, List<String>> params) {
+        if(params.containsKey("f")){
+            // DROP TABLES
+        }
+        if(params.containsKey("a")){
+            ORM.autoMigrate();
+            System.out.println("Auto-Migration successful!");
+            return CommandResult.success();
+        }
+        return CommandResult.error("Manual migration is not yet fully implemented");
+    }
+
+}