Implement on the fly phase creation

Signed-off-by: Austin L Mayes <me@austinlm.me>

Author: AustinLMayes

sets/competitve-objectives/src/main/java/net/avicus/atlas/sets/competitve/objectives/commands/PhaseCommands.java

@@ -5,7 +5,9 @@
 import com.sk89q.minecraft.util.commands.CommandContext;
 import com.sk89q.minecraft.util.commands.CommandException;
 import com.sk89q.minecraft.util.commands.CommandPermissions;
+import com.sk89q.minecraft.util.commands.CommandUsageException;
 import com.sk89q.minecraft.util.commands.NestedCommand;
+import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Map.Entry;
@@ -15,7 +17,10 @@
 import net.avicus.atlas.core.command.exception.CommandMatchException;
 import net.avicus.atlas.core.match.Match;
 import net.avicus.atlas.core.match.registry.MatchRegistry;
+import net.avicus.atlas.core.module.locales.LocalesModule;
+import net.avicus.atlas.core.module.locales.LocalizedXmlString;
 import net.avicus.atlas.core.module.objectives.ObjectivesModule;
+import net.avicus.atlas.core.module.states.StatesModule;
 import net.avicus.atlas.sets.competitve.objectives.bridges.ObjectivesBridge;
 import net.avicus.atlas.sets.competitve.objectives.destroyable.DestroyableObjective;
 import net.avicus.atlas.sets.competitve.objectives.phases.DestroyablePhase;
@@ -25,14 +30,18 @@
 import net.avicus.compendium.commands.exception.TranslatableCommandErrorException;
 import net.avicus.compendium.commands.exception.TranslatableCommandWarningException;
 import net.avicus.compendium.countdown.Countdown;
+import net.avicus.compendium.countdown.CountdownManager;
 import net.avicus.compendium.countdown.CountdownTask;
+import net.avicus.compendium.inventory.MultiMaterialMatcher;
+import net.avicus.compendium.inventory.SingleMaterialMatcher;
 import net.avicus.compendium.locale.text.UnlocalizedFormat;
 import net.avicus.compendium.locale.text.UnlocalizedText;
 import net.avicus.compendium.plugin.CompendiumPlugin;
 import net.avicus.compendium.utils.Strings;
 import net.md_5.bungee.api.chat.BaseComponent;
 import net.md_5.bungee.api.chat.TextComponent;
 import org.bukkit.ChatColor;
+import org.bukkit.Material;
 import org.bukkit.command.CommandSender;
 import org.joda.time.Duration;
 
@@ -41,6 +50,7 @@ public class PhaseCommands {
   private static final String LINE =
       ChatColor.GOLD + ChatColor.STRIKETHROUGH.toString() + "------" + ChatColor.RESET;
   private static final String INDENT = "  ";
+  private static final String ADD_ARGS = "id|name|success message|delay|find|replace";
 
   private static Match getMatch(CommandSender sender) throws CommandException {
     MustBePlayerCommandException.ensurePlayer(sender);
@@ -127,7 +137,7 @@ private static DestroyablePhase getPhase(CommandSender sender, String id)
     return registry.get(DestroyablePhase.class, id, true).get();
   }
 
-  @Command(aliases = "remove", desc = "Remove a phase by ID", max = 1, min = 1)
+  @Command(aliases = "remove", desc = "Remove a phase by ID", max = 1, min = 1, usage = "<id>")
   @CommandPermissions("atlas.phases.manage")
   public static void remove(CommandContext args, CommandSender sender) throws CommandException {
     String id = args.getString(0);
@@ -155,7 +165,7 @@ public static void remove(CommandContext args, CommandSender sender) throws Comm
     sender.sendMessage(ChatColor.GREEN + "Phase removed");
   }
 
-  @Command(aliases = "modtime", desc = "Modify a phase's application time", max = 2, min = 2)
+  @Command(aliases = "modtime", desc = "Modify a phase's application time", max = 2, min = 2, usage = "<id> <time>")
   @CommandPermissions("atlas.phases.manage")
   public static void modtime(CommandContext args, CommandSender sender) throws CommandException {
     String id = args.getString(0);
@@ -182,6 +192,79 @@ public static void modtime(CommandContext args, CommandSender sender) throws Com
     );
   }
 
+  @Command(aliases = "add", desc = "Add a phase", usage = ADD_ARGS)
+  @CommandPermissions("atlas.phases.manage")
+  public static void add(CommandContext args, CommandSender sender) throws CommandException {
+    Match match = getMatch(sender);
+    ObjectivesBridge bridge = getBridge(sender);
+    String[] indArgs = args.getJoinedStrings(0).split("\\|");
+    if (indArgs.length != 6) {
+      throw new CommandUsageException("Wrong number of arguments", "/phases add " + ADD_ARGS);
+    }
+    try {
+      String id = indArgs[0];
+      String nameRaw = indArgs[1];
+      String successRaw = indArgs[2];
+      Duration delay = Strings.toDuration(indArgs[3]);
+      String countdownRaw = "{phases.transition}";
+      String failRaw = "{phases.monument.fail}";
+      LocalizedXmlString name = match.getRequiredModule(LocalesModule.class).parse(nameRaw);
+      LocalizedXmlString countdown = match.getRequiredModule(LocalesModule.class)
+          .parse(countdownRaw);
+      LocalizedXmlString success = match.getRequiredModule(LocalesModule.class).parse(successRaw);
+      LocalizedXmlString fail = match.getRequiredModule(LocalesModule.class).parse(failRaw);
+      LinkedHashMap<MultiMaterialMatcher, SingleMaterialMatcher> materials = new LinkedHashMap<>();
+      materials.put(parseMultiMatcher(indArgs[4]), parseSingleMatcher(indArgs[5]));
+      DestroyablePhase phase = new DestroyablePhase(match, id, name, countdown, success, fail,
+          materials, delay, Optional.empty(), 0,
+          Optional.empty(), Optional.empty(), Optional.empty());
+      match.getRegistry().add(phase);
+      CountdownManager manager = CompendiumPlugin.getInstance().getCountdownManager();
+      if (manager.isRunning(PhaseApplyCountdown.class)) {
+        manager.getCountdowns().keySet().stream().filter(c -> c instanceof PhaseApplyCountdown)
+            .forEach(c -> {
+              ((PhaseApplyCountdown) c).getPhase().addPhase(phase);
+            });
+        sender.sendMessage(ChatColor.GREEN + "Phase added to end of phase chain");
+      } else {
+        for (DestroyableObjective objective : bridge.getLeakables()) {
+          objective.setPhase(Optional.of(phase));
+        }
+        bridge.populatePhaseCache();
+        if (match.getRequiredModule(StatesModule.class).isPlaying()) {
+          bridge.startPhaseCountdowns(match);
+        }
+        sender.sendMessage(ChatColor.GREEN + "Phase added");
+      }
+    } catch (Exception e) {
+      // e.printStackTrace();
+      throw new TranslatableCommandErrorException(
+          new UnlocalizedFormat("Failed to add phase: " + e.getMessage()));
+    }
+  }
+
+  private static MultiMaterialMatcher parseMultiMatcher(String raw) {
+    List<SingleMaterialMatcher> matchers = Lists.newArrayList();
+    for (String s : raw.split(",")) {
+      matchers.add(parseSingleMatcher(s));
+    }
+    return new MultiMaterialMatcher(matchers);
+  }
+
+  private static SingleMaterialMatcher parseSingleMatcher(String raw) {
+    String[] parts = raw.split(":");
+    Material material = Enum.valueOf(
+        Material.class, parts[0].toUpperCase().replace(" ", "_").replace("-", "_"));
+    Optional<Byte> data = Optional.empty();
+    if (parts.length > 1) {
+      data = Optional.of(Byte.valueOf(parts[1]));
+    }
+    if (!material.isSolid()) {
+      throw new IllegalArgumentException("Materials must be solid");
+    }
+    return new SingleMaterialMatcher(material, data);
+  }
+
   public static class Super {
 
     @Command(aliases = "phase", usage = "<>", desc = ".", min = 1)

sets/competitve-objectives/src/main/java/net/avicus/atlas/sets/competitve/objectives/phases/DestroyablePhase.java

@@ -263,6 +263,14 @@ public void removePhase(DestroyablePhase phase) {
     }
   }
 
+  public void addPhase(DestroyablePhase phase) {
+    if (this.passPhase.isPresent()) {
+      this.passPhase.get().addPhase(phase);
+    } else {
+      this.passPhase = Optional.of(phase);
+    }
+  }
+
   @Override
   public boolean equals(Object o) {
     if (this == o) {