Update DefaultWorldManager.java

Author: Dev7ex

multiworld-bukkit/src/main/java/com/dev7ex/multiworld/world/DefaultWorldManager.java

@@ -8,11 +8,12 @@
 import com.dev7ex.multiworld.api.bukkit.event.world.WorldCloneEvent;
 import com.dev7ex.multiworld.api.bukkit.event.world.WorldCreateEvent;
 import com.dev7ex.multiworld.api.bukkit.event.world.WorldDeleteEvent;
-import com.dev7ex.multiworld.api.bukkit.generator.FlatChunkGenerator;
-import com.dev7ex.multiworld.api.bukkit.generator.VoidChunkGenerator;
 import com.dev7ex.multiworld.api.bukkit.world.BukkitWorldHolder;
 import com.dev7ex.multiworld.api.bukkit.world.BukkitWorldManager;
 import com.dev7ex.multiworld.api.bukkit.world.BukkitWorldProvider;
+import com.dev7ex.multiworld.api.bukkit.world.generator.FlatWorldGenerator;
+import com.dev7ex.multiworld.api.bukkit.world.generator.VoidWorldGenerator;
+import com.dev7ex.multiworld.api.bukkit.world.generator.WaterWorldGenerator;
 import com.dev7ex.multiworld.api.world.WorldDefaultProperty;
 import com.dev7ex.multiworld.api.world.WorldType;
 import lombok.AccessLevel;
@@ -136,7 +137,7 @@ public void createWorld(@NotNull final String creatorName, @NotNull final String
 
         switch (type) {
             case FLAT:
-                worldCreator.generator(new FlatChunkGenerator());
+                worldCreator.generator(new FlatWorldGenerator());
                 worldCreator.generateStructures(false);
                 break;
 
@@ -148,8 +149,13 @@ public void createWorld(@NotNull final String creatorName, @NotNull final String
                 worldCreator.environment(World.Environment.THE_END);
                 break;
 
+            case WATER:
+                worldCreator.generator(new WaterWorldGenerator());
+                worldCreator.generateStructures(false);
+                break;
+
             case VOID:
-                worldCreator.generator(new VoidChunkGenerator());
+                worldCreator.generator(new VoidWorldGenerator());
                 worldCreator.generateStructures(false);
                 break;
 
@@ -388,19 +394,24 @@ public void loadWorld(@NotNull final String creatorName, @NotNull final String n
 
         switch (worldHolder.getType()) {
             case VOID:
-                worldCreator.generator(new VoidChunkGenerator());
+                worldCreator.generator(new VoidWorldGenerator());
                 worldCreator.generateStructures(false);
                 break;
 
             case FLAT:
-                worldCreator.generator(new FlatChunkGenerator());
+                worldCreator.generator(new FlatWorldGenerator());
                 worldCreator.generateStructures(false);
                 break;
 
             case NETHER:
                 worldCreator.environment(World.Environment.NETHER);
                 break;
 
+            case WATER:
+                worldCreator.generator(new WaterWorldGenerator());
+                worldCreator.generateStructures(false);
+                break;
+
             case END:
                 worldCreator.environment(World.Environment.THE_END);
                 break;
@@ -461,4 +472,4 @@ public BukkitWorldProvider getProvider() {
         return MultiWorldPlugin.getInstance().getWorldProvider();
     }
 
-}
+}