MissileWars/src/de/steamwar/misslewars/MWTeam.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;

import de.steamwar.comms.packets.TablistNamePacket;
import de.steamwar.sql.SteamwarUser;
import org.bukkit.*;
import org.bukkit.enchantments.Enchantment;
import org.bukkit.entity.Player;
import org.bukkit.inventory.Inventory;
import org.bukkit.inventory.ItemStack;
import org.bukkit.inventory.meta.ItemMeta;
import org.bukkit.scoreboard.Objective;
import org.bukkit.scoreboard.Team;

import java.util.HashSet;
import java.util.LinkedList;
import java.util.Objects;
import java.util.Set;

public class MWTeam {
    private static final ItemStack bow = new ItemStack(Material.BOW);

    static {
        ItemMeta bowMeta = Objects.requireNonNull(bow.getItemMeta());
        bowMeta.addEnchant(Enchantment.ARROW_FIRE, 1, true);
        bowMeta.addEnchant(Enchantment.ARROW_KNOCKBACK, 1, true);
        bowMeta.addEnchant(Enchantment.KNOCKBACK, 1, true);
        bowMeta.addEnchant(Enchantment.DAMAGE_ALL, 2, true);
        bowMeta.setUnbreakable(true);
        bow.setItemMeta(bowMeta);
    }

    private final ChatColor color;
    private final String teamName;
    private final Team sbteam; //scoreboard-Team
    private final Location spawn;
    private final int portalZ;

    private final LinkedList<Player> players =  new LinkedList<>();
    private final Set<Player> openInvitations = new HashSet<>();

    MWTeam(ChatColor color, Location spawn, String teamName, int portalZ) {
        this.teamName = teamName;
        this.color = color;
        this.spawn = spawn;
        this.portalZ = portalZ;
        if(FightScoreboard.getScoreboard().getTeam(teamName) == null)
            sbteam = FightScoreboard.getScoreboard().registerNewTeam(teamName);
        else
            sbteam = FightScoreboard.getScoreboard().getTeam(teamName);
        assert sbteam != null;
        sbteam.setAllowFriendlyFire(false);
        sbteam.setColor(color);
    }

    public void countdown(int timeDelay, int offset) {
        if (players.isEmpty()) {
            return;
        }

        int i = 0;
        for (Player p : players) {
            int time = timeDelay * (i + 1) - offset;
            i++;

            p.setExp(time / (float) Config.ItemTime);
            p.setLevel(0);
        }
    }

    public void givePlayerItem(ItemStack item) {
    	if (players.isEmpty()) {
    		return;
		}
        Player p = players.removeFirst();
        players.addLast(p);

        Inventory inventory = p.getInventory();
        for (int i = 0; i <= 35; i++) { //35 is the last normal inventory slot
            ItemStack itemStack = inventory.getItem(i);
            if (itemStack != null && itemStack.isSimilar(item) && itemStack.getAmount() != itemStack.getMaxStackSize()) {
                itemStack.setAmount(itemStack.getAmount() + item.getAmount());
                inventory.setItem(i, itemStack);
                p.updateInventory();
                return;
            }
        }

        for (int i = 0; i <= 35; i++) { //35 is the last normal inventory slot
            ItemStack itemStack = inventory.getItem(i);
            if (itemStack == null || itemStack.getType().equals(Material.AIR)) {
                inventory.setItem(i, item);
                p.updateInventory();
                return;
            }
        }
    }

    public void teamScoreboard(Objective objective) {
        players.forEach(p -> objective.getScore(getColorCode() + p.getName()).setScore(1));
    }

    public int size() {
        return players.size();
    }

    public int getPortalZ() {
        return portalZ;
    }

    public Location getSpawn(){
        return spawn;
    }

    public void join(Player p) {
        players.add(p);
        p.teleport(spawn);
        p.setGameMode(GameMode.SURVIVAL);
        p.getInventory().setItem(0, bow);
        sbteam.addPlayer(p);
        p.setDisplayName(color + p.getName());
        if (MissileWars.getFightState() == FightState.WAITING && !enemy().players.isEmpty())
            MissileWars.startRound();

        if (!Config.test())
            Bukkit.getScheduler().runTaskLater(MissileWars.getPlugin(), () -> new TablistNamePacket(SteamwarUser.get(p.getUniqueId()).getId(), color + p.getName()).send(p), 5);
    }

    public void leave(Player p) {
        if (!players.contains(p)) return;

        players.remove(p);
        for (ItemStack stack : p.getInventory().getContents()) {
            if (stack == null) continue;
            if (stack.getType() == Material.AIR) continue;
            if (stack.isSimilar(bow)) continue;
            givePlayerItem(stack);
        }

        p.getInventory().clear();

        p.setDisplayName("§7" + p.getName());
        sbteam.removePlayer(p);
        if (players.isEmpty() && MissileWars.getFightState() == FightState.FIGHTING)
            MissileWars.end(WinReasons.NO_ENEMY, enemy());

        if (!Config.test())
            new TablistNamePacket(SteamwarUser.get(p.getUniqueId()).getId(), "§7§o" + p.getName()).send(p);
    }

    public void invitePlayer(Player p) {
        if (enemy().openInvitations.contains(p)) return;
        openInvitations.add(p);
    }

    public void acceptInvite(Player p) {
        removeInvitations(p);
        join(p);
    }

    private MWTeam enemy() {
        if (this == MissileWars.getRedTeam())
            return MissileWars.getBlueTeam();

        return MissileWars.getRedTeam();
    }

    public String getColorCode(){
        return "§" + color.getChar();
    }

    public boolean hasPlayer(Player p) {
        return players.contains(p);
    }

    public boolean hasInvite(Player p) {
        return openInvitations.contains(p);
    }

    public String getColoredName() {
        return color.toString() + teamName;
    }

    public LinkedList<Player> getPlayers() {
        return players;
    }

    public static void removeInvitations(Player p) {
        MissileWars.getRedTeam().openInvitations.remove(p);
        MissileWars.getBlueTeam().openInvitations.remove(p);
    }

}