diff --git a/src/main/java/noppes/npcs/scripted/ScriptedActionManager.java b/src/main/java/noppes/npcs/scripted/ScriptedActionManager.java index 3f0b50b0d..667fbc6e2 100644 --- a/src/main/java/noppes/npcs/scripted/ScriptedActionManager.java +++ b/src/main/java/noppes/npcs/scripted/ScriptedActionManager.java @@ -110,26 +110,15 @@ public class ScriptedActionManager implements IActionManager { return false; } + @Override - public IAction scheduleConditionalAction(String name, - int checkIntervalTicks, - Supplier predicate, - Consumer task) { - return scheduleConditionalAction(name, checkIntervalTicks, predicate, task, -1); + public IConditionalAction scheduleConditionalAction(String name, Supplier predicate, Consumer task) { + return scheduleConditionalAction(new ConditionalAction(name, predicate, task)); } @Override - public IAction scheduleConditionalAction(String name, - int checkIntervalTicks, - Supplier predicate, - Consumer task, - int maxChecks) { - return scheduleConditionalAction(new ConditionalAction(name, checkIntervalTicks, predicate, task, maxChecks)); - } - - @Override - public IAction scheduleConditionalAction(String name, int checkIntervalTicks, Supplier predicate, Supplier terminateWhen, Consumer task, int maxChecks) { - return scheduleConditionalAction(new ConditionalAction(name, checkIntervalTicks, predicate, terminateWhen, task, maxChecks)); + public IConditionalAction scheduleConditionalAction(String name, Supplier predicate, Supplier terminateWhen, Consumer task) { + return scheduleConditionalAction(new ConditionalAction(name, predicate, terminateWhen, task)); } @Override @@ -284,8 +273,9 @@ public class ScriptedActionManager implements IActionManager { } @Override - public void addData(String key, Object v) { + public IAction addData(String key, Object v) { dataStore.put(key, v); + return this; } @Override @@ -294,8 +284,9 @@ public class ScriptedActionManager implements IActionManager { } @Override - public void setUpdateEveryXTick(int x) { + public IAction setUpdateEveryXTick(int x) { updateEveryXTick = x; + return this; } @Override @@ -444,17 +435,6 @@ public class ScriptedActionManager implements IActionManager { this.terminate = terminate; } - public ConditionalAction(String name, int updateEveryXTick, Supplier predicate, Consumer task, int maxChecks) { - this(name, predicate, task); - this.updateEveryXTick = updateEveryXTick; - this.maxChecks = maxChecks; - } - - public ConditionalAction(String name, int updateEveryXTick, Supplier predicate, Supplier terminate, Consumer task, int maxChecks) { - this(name, updateEveryXTick, predicate, task, maxChecks); - this.terminate = terminate; - } - @Override public void tick(int ticksExisted) { if (isDone()) return; @@ -469,6 +449,12 @@ public class ScriptedActionManager implements IActionManager { } } + @Override + public IConditionalAction setMaxChecks(int maxChecks) { + this.maxChecks = maxChecks; + return this; + } + @Override public int getCheckCount() { return checkCount; } @Override public int getMaxChecks() { return maxChecks; }