MissileWars/src/de/steamwar/misslewars/countdowns/ItemCountdown.java

/* 
    This file is a part of the SteamWar software.
    
    Copyright (C) 2020  SteamWar.de-Serverteam

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

package de.steamwar.misslewars.countdowns;

import de.steamwar.misslewars.Config;
import de.steamwar.misslewars.FightState;
import de.steamwar.misslewars.MissileWars;
import de.steamwar.misslewars.StateDependent;
import de.steamwar.misslewars.items.SpecialItem;
import org.bukkit.Bukkit;
import org.bukkit.inventory.ItemStack;
import org.bukkit.scheduler.BukkitTask;

import java.util.EnumSet;

public class ItemCountdown extends StateDependent {

	private BukkitTask task;

	public ItemCountdown() {
		super(EnumSet.of(FightState.FIGHTING));
	}

	private void run() {
		int itemCount = Math.max(MissileWars.getBlueTeam().size(), MissileWars.getRedTeam().size());

		for (int i = 0; i < itemCount; i++) {
			ItemStack itemStack = SpecialItem.getRandomItem();

			MissileWars.getBlueTeam().givePlayerItem(itemStack);
			MissileWars.getRedTeam().givePlayerItem(itemStack);
		}
	}

	@Override
	public void enable() {
		task = Bukkit.getScheduler().runTaskTimer(MissileWars.getPlugin(), this::run, 0, Config.ItemTime);
	}

	@Override
	public void disable() {
		if(task != null && !task.isCancelled())
			task.cancel();
	}
}