Added seeding to migrate

Author: JanHolger

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

@@ -9,6 +9,7 @@
 import org.javawebstack.framework.config.Config;
 import org.javawebstack.framework.job.*;
 import org.javawebstack.framework.module.Module;
+import org.javawebstack.framework.seed.AllSeeder;
 import org.javawebstack.framework.seed.FileSeeder;
 import org.javawebstack.framework.seed.MergedSeeder;
 import org.javawebstack.framework.seed.Seeder;
@@ -63,6 +64,8 @@ public WebApplication(){
         crypt = new Crypt(config.has("crypt.key") ? config.get("crypt.key") : Crypt.generateKey());
         injector.setInstance(Crypt.class, crypt);
 
+        addSeeder("all", new AllSeeder());
+
         modules.forEach(m -> m.setupConfig(this, config));
         if(config.get("database.driver", "none").equalsIgnoreCase("sqlite")){
             sql = new SQLite(config.get("database.file", "db.sqlite"));
@@ -129,6 +132,10 @@ public WebApplication(){
         );
     }
 
+    public Map<String, Seeder> getSeeders(){
+        return seeders;
+    }
+
     public void addTranslation(Locale locale, ClassLoader classLoader, String resource){
         if(!resource.endsWith(".json"))
             resource += ".json";

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

@@ -3,13 +3,18 @@
 import org.javawebstack.command.Command;
 import org.javawebstack.command.CommandResult;
 import org.javawebstack.command.CommandSystem;
+import org.javawebstack.framework.WebApplication;
+import org.javawebstack.injector.Inject;
 import org.javawebstack.orm.ORM;
 
 import java.util.List;
 import java.util.Map;
 
 public class DBMigrateCommand implements Command {
 
+    @Inject
+    private WebApplication app;
+
     public CommandResult execute(CommandSystem commandSystem, List<String> args, Map<String, List<String>> params) {
         if(params.containsKey("a")){
             if(params.containsKey("d")){
@@ -18,6 +23,8 @@ public CommandResult execute(CommandSystem commandSystem, List<String> args, Map
                 return CommandResult.success();
             }
             ORM.autoMigrate(params.containsKey("f"));
+            if(params.containsKey("s"))
+                app.getSeeder("all").seed();
             System.out.println("Auto-Migration successful!");
             return CommandResult.success();
         }

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

@@ -12,7 +12,7 @@
 public class DBSeedCommand implements Command {
 
     @Inject
-    WebApplication application;
+    private WebApplication application;
 
     public CommandResult execute(CommandSystem commandSystem, List<String> args, Map<String, List<String>> params) {
         if(args.size() == 0)

src/main/java/org/javawebstack/framework/seed/AllSeeder.java

@@ -0,0 +1,15 @@
+package org.javawebstack.framework.seed;
+
+import org.javawebstack.framework.WebApplication;
+import org.javawebstack.injector.Inject;
+
+public class AllSeeder implements Seeder {
+
+    @Inject
+    private WebApplication app;
+
+    public void seed() {
+        app.getSeeders().forEach((n,s) -> s.seed());
+    }
+
+}