Add a delay to game closing

Fixes #2

Author: haykam821

src/main/java/io/github/haykam821/sculkprison/game/SculkPrisonConfig.java

@@ -3,25 +3,31 @@
 import com.mojang.serialization.Codec;
 import com.mojang.serialization.codecs.RecordCodecBuilder;
 
+import net.minecraft.SharedConstants;
+import net.minecraft.util.math.intprovider.ConstantIntProvider;
+import net.minecraft.util.math.intprovider.IntProvider;
 import xyz.nucleoid.plasmid.game.common.config.PlayerConfig;
 
 public class SculkPrisonConfig {
 	public static final Codec<SculkPrisonConfig> CODEC = RecordCodecBuilder.create(instance -> {
 		return instance.group(
 			PlayerConfig.CODEC.fieldOf("players").forGetter(SculkPrisonConfig::getPlayerConfig),
 			Codec.INT.optionalFieldOf("lock_time", 20 * 30).forGetter(SculkPrisonConfig::getLockTime),
-			Codec.INT.optionalFieldOf("survive_time", 20 * 60 * 5).forGetter(SculkPrisonConfig::getSurviveTime)
+			Codec.INT.optionalFieldOf("survive_time", 20 * 60 * 5).forGetter(SculkPrisonConfig::getSurviveTime),
+			IntProvider.NON_NEGATIVE_CODEC.optionalFieldOf("ticks_until_close", ConstantIntProvider.create(SharedConstants.TICKS_PER_SECOND * 5)).forGetter(SculkPrisonConfig::getTicksUntilClose)
 		).apply(instance, SculkPrisonConfig::new);
 	});
 
 	private final PlayerConfig playerConfig;
 	private final int lockTime;
 	private final int surviveTime;
+	private final IntProvider ticksUntilClose;
 
-	public SculkPrisonConfig(PlayerConfig playerConfig, int lockTime, int surviveTime) {
+	public SculkPrisonConfig(PlayerConfig playerConfig, int lockTime, int surviveTime, IntProvider ticksUntilClose) {
 		this.playerConfig = playerConfig;
 		this.lockTime = lockTime;
 		this.surviveTime = surviveTime;
+		this.ticksUntilClose = ticksUntilClose;
 	}
 
 	public PlayerConfig getPlayerConfig() {
@@ -35,4 +41,8 @@ public int getLockTime() {
 	public int getSurviveTime() {
 		return this.surviveTime;
 	}
+
+	public IntProvider getTicksUntilClose() {
+		return this.ticksUntilClose;
+	}
 }

src/main/java/io/github/haykam821/sculkprison/game/phase/SculkPrisonActivePhase.java

@@ -50,6 +50,7 @@ public class SculkPrisonActivePhase implements PlayerAttackEntityEvent, GameActi
 
 	private int lockTime;
 	private int surviveTime;
+	private int ticksUntilClose = -1;
 
 	public SculkPrisonActivePhase(GameSpace gameSpace, ServerWorld world, SculkPrisonMap map, SculkPrisonConfig config, List<ServerPlayerEntity> players, GlobalWidgets widgets) {
 		this.world = world;
@@ -104,7 +105,7 @@ public static void open(GameSpace gameSpace, ServerWorld world, SculkPrisonMap m
 	// Listeners
 	@Override
 	public ActionResult onAttackEntity(ServerPlayerEntity attacker, Hand hand, Entity attacked, EntityHitResult hitResult) {
-		if (attacker.equals(this.warden) && attacked instanceof ServerPlayerEntity) {
+		if (!this.isGameEnding() && attacker.equals(this.warden) && attacked instanceof ServerPlayerEntity) {
 			this.eliminate((ServerPlayerEntity) attacked, Text.translatable("text.sculkprison.eliminated.warden", attacked.getDisplayName(), attacker.getDisplayName()), true);
 		}
 		return ActionResult.FAIL;
@@ -125,6 +126,15 @@ public void onEnable() {
 
 	@Override
 	public void onTick() {
+		// Decrease ticks until game end to zero
+		if (this.isGameEnding()) {
+			if (this.ticksUntilClose == 0) {
+				this.gameSpace.close(GameCloseReason.FINISHED);
+			}
+			this.ticksUntilClose -= 1;
+			return;
+		}
+
 		this.lockTime -= 1;
 		if (this.lockTime < 0) {
 			this.surviveTime -= 1;
@@ -165,7 +175,6 @@ public ActionResult onDeath(ServerPlayerEntity player, DamageSource source) {
 	@Override
 	public void onRemovePlayer(ServerPlayerEntity player) {
 		this.eliminate(player, true);
-		this.players.remove(player);
 	}
 
 	// Utilities
@@ -189,6 +198,8 @@ private void setSpectator(ServerPlayerEntity player) {
 	 * @param remove whether to remove the player from {@link SculkPrisonActivePhase#players}
 	 */
 	private void eliminate(ServerPlayerEntity player, Text message, boolean remove) {
+		if (this.isGameEnding()) return;
+
 		this.gameSpace.getPlayers().sendMessage(message);
 
 		if (remove) {
@@ -214,12 +225,16 @@ private void eliminate(ServerPlayerEntity player, boolean remove) {
 	 */
 	private void endWithWinner(WinTeam team) {
 		this.gameSpace.getPlayers().sendMessage(team.getWinMessage());
-		this.gameSpace.close(GameCloseReason.FINISHED);
+		this.endGame();
 	}
 
 	private void endWithNoWinners() {
 		this.gameSpace.getPlayers().sendMessage(Text.translatable("text.sculkprison.no_winners").formatted(Formatting.RED));
-		this.gameSpace.close(GameCloseReason.FINISHED);
+		this.endGame();
+	}
+
+	private void endGame() {
+		this.ticksUntilClose = this.config.getTicksUntilClose().get(this.world.getRandom());
 	}
 
 	/**
@@ -245,6 +260,10 @@ public int getLockTime() {
 		return this.lockTime;
 	}
 
+	private boolean isGameEnding() {
+		return this.ticksUntilClose >= 0;
+	}
+
 	/**
 	 * Spawns a given player within the map.
 	 * @param warden whether to use the {@linkplain SculkPrisonMap#WARDEN_SPAWN warden spawn} instead of the {@linkplain SculkPrisonMap#SPAWN default spawn}