TurnBasedMinecraftMod/src/main/java/com/seodisparate/TurnBasedMinecraft/common/Battle.java

package com.seodisparate.TurnBasedMinecraft.common;

import java.time.Duration;
import java.time.Instant;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.Map;
import java.util.PriorityQueue;
import java.util.Queue;
import java.util.concurrent.atomic.AtomicInteger;

import com.seodisparate.TurnBasedMinecraft.TurnBasedMinecraftMod;
import com.seodisparate.TurnBasedMinecraft.common.networking.PacketBattleInfo;
import com.seodisparate.TurnBasedMinecraft.common.networking.PacketBattleMessage;
import com.seodisparate.TurnBasedMinecraft.common.networking.PacketHandler;

import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.entity.player.EntityPlayerMP;
import net.minecraft.util.DamageSource;

public class Battle
{
    private final int id;
    private Map<Integer, Combatant> sideA;
    private Map<Integer, Combatant> sideB;
    private Map<Integer, Combatant> players;
    private PriorityQueue<Combatant> turnOrderQueue;
    
    private Instant lastUpdated;
    private State state;
    private AtomicInteger playerCount;
    private AtomicInteger undecidedCount;
    private Duration timer;
    
    private boolean isServer;
    private boolean battleEnded;
    
    public enum State
    {
        DECISION,
        ACTION,
        HEALTH_CHECK
    }
    
    public enum Decision
    {
        UNDECIDED(0),
        ATTACK(1),
        DEFEND(2),
        FLEE(3),
        USE_ITEM(4);
        
        private int value;
        private static Map<Integer, Decision> map = new HashMap<Integer, Decision>();
        
        private Decision(int value)
        {
            this.value = value;
        }
        
        public int getValue()
        {
            return value;
        }
        
        static
        {
            for(Decision decision : Decision.values())
            {
                map.put(decision.value, decision);
            }
        }
        
        public static Decision valueOf(int decisionType)
        {
            return map.get(decisionType);
        }
    }

    public Battle(int id, Collection<Entity> sideA, Collection<Entity> sideB, boolean isServer)
    {
        this.isServer = isServer;
        this.id = id;
        this.sideA = new Hashtable<Integer, Combatant>();
        this.sideB = new Hashtable<Integer, Combatant>();
        players = new Hashtable<Integer, Combatant>();
        turnOrderQueue = new PriorityQueue<Combatant>(new Combatant.CombatantComparator());
        playerCount = new AtomicInteger(0);
        undecidedCount = new AtomicInteger(0);
        if(sideA != null)
        {
            for(Entity e : sideA)
            {
                EntityInfo entityInfo = TurnBasedMinecraftMod.config.getMatchingEntityInfo(e);
                if(entityInfo == null && !(e instanceof EntityPlayer))
                {
                    continue;
                }
                Combatant newCombatant = new Combatant(e, entityInfo);
                newCombatant.isSideA = true;
                this.sideA.put(e.getEntityId(), newCombatant);
                if(e instanceof EntityPlayer)
                {
                    newCombatant.recalcSpeedOnCompare = true;
                    playerCount.incrementAndGet();
                    players.put(e.getEntityId(), newCombatant);
                }
            }
        }
        if(sideB != null)
        {
            for(Entity e : sideB)
            {
                EntityInfo entityInfo = TurnBasedMinecraftMod.config.getMatchingEntityInfo(e);
                if(entityInfo == null && !(e instanceof EntityPlayer))
                {
                    continue;
                }
                Combatant newCombatant = new Combatant(e, entityInfo);
                newCombatant.isSideA = false;
                this.sideB.put(e.getEntityId(), newCombatant);
                if(e instanceof EntityPlayer)
                {
                    newCombatant.recalcSpeedOnCompare = true;
                    playerCount.incrementAndGet();
                    players.put(e.getEntityId(), newCombatant);
                }
            }
        }
        
        for(Combatant c : this.sideA.values())
        {
            sendMessageToAllPlayers(PacketBattleMessage.MessageType.ENTERED, c.entity.getEntityId(), 0, id);
        }
        for(Combatant c : this.sideB.values())
        {
            sendMessageToAllPlayers(PacketBattleMessage.MessageType.ENTERED, c.entity.getEntityId(), 0, id);
        }
        
        lastUpdated = null;
        state = State.DECISION;
        undecidedCount.set(playerCount.get());
        timer = TurnBasedMinecraftMod.BattleDecisionTime;
        battleEnded = false;
        
        notifyPlayersBattleInfo();
    }

    public int getId()
    {
        return id;
    }
    
    public Entity getCombatantEntity(int entityID)
    {
        Combatant c = sideA.get(entityID);
        if(c != null)
        {
            return c.entity;
        }
        c = sideB.get(entityID);
        if(c != null)
        {
            return c.entity;
        }
        return null;
    }
    
    public boolean hasCombatant(int entityID)
    {
        return sideA.containsKey(entityID) || sideB.containsKey(entityID);
    }
    
    public boolean hasCombatantInSideA(int entityID)
    {
        return sideA.containsKey(entityID);
    }
    
    public void addCombatantToSideA(Entity e)
    {
        EntityInfo entityInfo = TurnBasedMinecraftMod.config.getMatchingEntityInfo(e);
        if(entityInfo == null && !(e instanceof EntityPlayer))
        {
            return;
        }
        Combatant newCombatant = new Combatant(e, entityInfo);
        newCombatant.isSideA = true;
        sideA.put(e.getEntityId(), newCombatant);
        if(e instanceof EntityPlayer)
        {
            newCombatant.recalcSpeedOnCompare = true;
            playerCount.incrementAndGet();
            players.put(e.getEntityId(), newCombatant);
            if(state == State.DECISION)
            {
                undecidedCount.incrementAndGet();
            }
        }
        sendMessageToAllPlayers(PacketBattleMessage.MessageType.ENTERED, newCombatant.entity.getEntityId(), 0, id);
        notifyPlayersBattleInfo();
    }
    
    public void addCombatantToSideB(Entity e)
    {
        EntityInfo entityInfo = TurnBasedMinecraftMod.config.getMatchingEntityInfo(e);
        if(entityInfo == null && !(e instanceof EntityPlayer))
        {
            return;
        }
        Combatant newCombatant = new Combatant(e, entityInfo);
        newCombatant.isSideA = false;
        sideB.put(e.getEntityId(), newCombatant);
        if(e instanceof EntityPlayer)
        {
            newCombatant.recalcSpeedOnCompare = true;
            playerCount.incrementAndGet();
            players.put(e.getEntityId(), newCombatant);
            if(state == State.DECISION)
            {
                undecidedCount.incrementAndGet();
            }
        }
        sendMessageToAllPlayers(PacketBattleMessage.MessageType.ENTERED, newCombatant.entity.getEntityId(), 0, id);
        notifyPlayersBattleInfo();
    }
    
    public void clearCombatants()
    {
        sideA.clear();
        sideB.clear();
        players.clear();
        playerCount.set(0);
        undecidedCount.set(0);
    }
    
    public Collection<Combatant> getSideA()
    {
        return sideA.values();
    }
    
    public Collection<Combatant> getSideB()
    {
        return sideB.values();
    }
    
    public Collection<Integer> getSideAIDs()
    {
        Collection<Integer> sideAIDs = new ArrayList<Integer>(sideA.size());
        for(Combatant combatant : sideA.values())
        {
            sideAIDs.add(combatant.entity.getEntityId());
        }
        return sideAIDs;
    }
    
    public Collection<Integer> getSideBIDs()
    {
        Collection<Integer> sideBIDs = new ArrayList<Integer>(sideB.size());
        for(Combatant combatant : sideB.values())
        {
            sideBIDs.add(combatant.entity.getEntityId());
        }
        return sideBIDs;
    }
    
    public Combatant getCombatantByID(int entityID)
    {
        Combatant combatant = sideA.get(entityID);
        if(combatant == null)
        {
            combatant = sideB.get(entityID);
        }
        return combatant;
    }
    
    public void setDecision(int entityID, Decision decision, int targetIDOrItemID)
    {
        if(state != State.DECISION)
        {
            return;
        }
        Combatant combatant = players.get(entityID);
        if(combatant == null || combatant.decision != Decision.UNDECIDED)
        {
            return;
        }
        combatant.decision = decision;
        if(decision == Decision.ATTACK)
        {
            combatant.targetEntityID = targetIDOrItemID;
        }
        else if(decision == Decision.USE_ITEM)
        {
            combatant.itemToUse = targetIDOrItemID;
        }
        undecidedCount.decrementAndGet();
    }
    
    public State getState()
    {
        return state;
    }
    
    protected void notifyPlayersBattleInfo()
    {
        if(!isServer)
        {
            return;
        }
        PacketBattleInfo infoPacket = new PacketBattleInfo(getSideAIDs(), getSideBIDs());
        for(Combatant p : players.values())
        {
            PacketHandler.INSTANCE.sendTo(infoPacket, (EntityPlayerMP)p.entity);
        }
    }
    
    /**
     * @return True if battle has ended
     */
    public boolean update()
    {
        if(battleEnded)
        {
            return true;
        }
        if(lastUpdated == null)
        {
            lastUpdated = Instant.now();
            return update(Duration.ZERO);
        }
        else
        {
            Instant now = Instant.now();
            Duration dt = Duration.between(lastUpdated, now);
            lastUpdated = now;
            return update(dt);
        }
    }
    
    private void sendMessageToAllPlayers(PacketBattleMessage.MessageType type, int from, int to, int amount)
    {
        if(!isServer)
        {
            return;
        }
        for(Combatant p : players.values())
        {
            if(p.entity.isEntityAlive())
            {
                PacketHandler.INSTANCE.sendTo(new PacketBattleMessage(type, from, to, amount), (EntityPlayerMP)p.entity);
            }
        }
    }
    
    private boolean update(final Duration dt)
    {
        if(battleEnded)
        {
            return true;
        }
        switch(state)
        {
        case DECISION:
            timer = timer.minus(dt);
            if(timer.isNegative() || timer.isZero() || undecidedCount.get() <= 0)
            {
                for(Combatant c : sideA.values())
                {
                    // picking decision for sideA non-players
                    if(!(c.entity instanceof EntityPlayer) && c.decision == Decision.UNDECIDED && c.entityInfo != null)
                    {
                        int percentage = (int)(Math.random() * 100);
                        if(percentage < c.entityInfo.decisionAttack)
                        {
                            c.decision = Decision.ATTACK;
                        }
                        else if(percentage - c.entityInfo.decisionAttack < c.entityInfo.decisionDefend)
                        {
                            c.decision = Decision.DEFEND;
                        }
                        else if(percentage - c.entityInfo.decisionAttack - c.entityInfo.decisionDefend < c.entityInfo.decisionFlee)
                        {
                            c.decision = Decision.FLEE;
                        }
                    }
                }
                for(Combatant c : sideB.values())
                {
                    if(!(c.entity instanceof EntityPlayer) && c.decision == Decision.UNDECIDED && c.entityInfo != null)
                    {
                        int percentage = (int)(Math.random() * 100);
                        if(percentage < c.entityInfo.decisionAttack)
                        {
                            c.decision = Decision.ATTACK;
                        }
                        else if(percentage - c.entityInfo.decisionAttack < c.entityInfo.decisionDefend)
                        {
                            c.decision = Decision.DEFEND;
                        }
                        else if(percentage - c.entityInfo.decisionAttack - c.entityInfo.decisionDefend < c.entityInfo.decisionFlee)
                        {
                            c.decision = Decision.FLEE;
                        }
                    }
                }
                state = State.ACTION;
                timer = TurnBasedMinecraftMod.BattleDecisionTime;
                turnOrderQueue.clear();
                for(Combatant c : sideA.values())
                {
                    turnOrderQueue.add(c);
                }
                for(Combatant c : sideB.values())
                {
                    turnOrderQueue.add(c);
                }
                update(Duration.ZERO);
            }
            break;
        case ACTION:
        {
            Combatant next = turnOrderQueue.poll();
            while(next != null)
            {
                if(!next.entity.isEntityAlive())
                {
                    next = turnOrderQueue.poll();
                    continue;
                }
                switch(next.decision)
                {
                case UNDECIDED:
                    sendMessageToAllPlayers(PacketBattleMessage.MessageType.DID_NOTHING, next.entity.getEntityId(), 0, 0);
                    next = turnOrderQueue.poll();
                    continue;
                case ATTACK:
                    Combatant target = null;
                    if(next.entity instanceof EntityPlayer)
                    {
                        if(next.isSideA)
                        {
                            target = sideB.get(next.targetEntityID);
                        }
                        else
                        {
                            target = sideA.get(next.targetEntityID);
                        }
                        if(target == null || !target.entity.isEntityAlive())
                        {
                            next = turnOrderQueue.poll();
                            continue;
                        }
                        int hitChance = TurnBasedMinecraftMod.config.getPlayerAttackProbability();
                        if(target.entity instanceof EntityPlayer)
                        {
                            hitChance -= TurnBasedMinecraftMod.config.getPlayerEvasion();
                        }
                        else
                        {
                            hitChance -= target.entityInfo.evasion;
                        }
                        if((int)(Math.random() * 100) < hitChance)
                        {
                            if(target.remainingDefenses <= 0)
                            {
                                // attack
                                // TODO damage via bow and arrow
                                TurnBasedMinecraftMod.attackingEntity = next.entity;
                                ((EntityPlayer)next.entity).attackTargetEntityWithCurrentItem(target.entity);
                                TurnBasedMinecraftMod.attackingEntity = null;
                                sendMessageToAllPlayers(PacketBattleMessage.MessageType.ATTACK, next.entity.getEntityId(), target.entity.getEntityId(), TurnBasedMinecraftMod.attackingDamage);
                                if(!(target.entity instanceof EntityPlayer) && target.entityInfo.defenseDamage > 0)
                                {
                                    if((int)(Math.random() * 100) < target.entityInfo.defenseDamageProbability)
                                    {
                                        // defense damage
                                        DamageSource defenseDamageSource = DamageSource.causeMobDamage((EntityLivingBase)target.entity);
                                        TurnBasedMinecraftMod.attackingEntity = target.entity;
                                        next.entity.attackEntityFrom(defenseDamageSource, target.entityInfo.defenseDamage);
                                        TurnBasedMinecraftMod.attackingEntity = null;
                                        sendMessageToAllPlayers(PacketBattleMessage.MessageType.DEFENSE_DAMAGE, target.entity.getEntityId(), next.entity.getEntityId(), target.entityInfo.defenseDamage);
                                    }
                                }
                            }
                            else
                            {
                                // blocked
                                --target.remainingDefenses;
                                sendMessageToAllPlayers(PacketBattleMessage.MessageType.DEFEND, target.entity.getEntityId(), next.entity.getEntityId(), 0);
                            }
                        }
                        else
                        {
                            // miss
                            sendMessageToAllPlayers(PacketBattleMessage.MessageType.MISS, next.entity.getEntityId(), target.entity.getEntityId(), 0);
                        }
                    }
                    else
                    {
                        if(next.isSideA)
                        {
                            int randomTargetIndex = (int)(Math.random() * sideB.size());
                            for(Combatant c : sideB.values())
                            {
                                if(randomTargetIndex-- == 0)
                                {
                                    target = c;
                                    break;
                                }
                            }
                        }
                        else
                        {
                            int randomTargetIndex = (int)(Math.random() * sideA.size());
                            for(Combatant c : sideA.values())
                            {
                                if(randomTargetIndex-- == 0)
                                {
                                    target = c;
                                    break;
                                }
                            }
                        }
                        if(target == null || !target.entity.isEntityAlive())
                        {
                            next = turnOrderQueue.poll();
                            continue;
                        }
                        int hitChance = next.entityInfo.attackProbability;
                        if(target.entity instanceof EntityPlayer)
                        {
                            hitChance -= TurnBasedMinecraftMod.config.getPlayerEvasion();
                        }
                        else
                        {
                            hitChance -= target.entityInfo.evasion;
                        }
                        if((int)(Math.random() * 100) < hitChance)
                        {
                            if(target.remainingDefenses <= 0)
                            {
                                DamageSource damageSource = DamageSource.causeMobDamage((EntityLivingBase)next.entity);
                                int damageAmount = next.entityInfo.attackPower;
                                if(next.entityInfo.attackVariance > 0)
                                {
                                    damageAmount += (int)(Math.random() * (next.entityInfo.attackVariance * 2 + 1)) - next.entityInfo.attackVariance;
                                }
                                // attack
                                TurnBasedMinecraftMod.attackingEntity = next.entity;
                                target.entity.attackEntityFrom(damageSource, damageAmount);
                                TurnBasedMinecraftMod.attackingEntity = null;
                                sendMessageToAllPlayers(PacketBattleMessage.MessageType.ATTACK, next.entity.getEntityId(), target.entity.getEntityId(), damageAmount);
                                if(!(target.entity instanceof EntityPlayer) && target.entityInfo.defenseDamage > 0)
                                {
                                    if((int)(Math.random() * 100) < target.entityInfo.defenseDamageProbability)
                                    {
                                        // defense damage
                                        DamageSource defenseDamageSource = DamageSource.causeMobDamage((EntityLivingBase)target.entity);
                                        TurnBasedMinecraftMod.attackingEntity = target.entity;
                                        next.entity.attackEntityFrom(defenseDamageSource, target.entityInfo.defenseDamage);
                                        TurnBasedMinecraftMod.attackingEntity = null;
                                        sendMessageToAllPlayers(PacketBattleMessage.MessageType.DEFENSE_DAMAGE, target.entity.getEntityId(), next.entity.getEntityId(), target.entityInfo.defenseDamage);
                                    }
                                }
                            }
                            else
                            {
                                // blocked
                                --target.remainingDefenses;
                                sendMessageToAllPlayers(PacketBattleMessage.MessageType.DEFEND, target.entity.getEntityId(), next.entity.getEntityId(), 0);
                            }
                        }
                        else
                        {
                            // miss
                            sendMessageToAllPlayers(PacketBattleMessage.MessageType.MISS, next.entity.getEntityId(), target.entity.getEntityId(), 0);
                        }
                    }
                    break;
                case DEFEND:
                    next.remainingDefenses = TurnBasedMinecraftMod.config.getDefenseDuration();
                    sendMessageToAllPlayers(PacketBattleMessage.MessageType.DEFENDING, next.entity.getEntityId(), 0, 0);
                    break;
                case FLEE:
                    int fastestEnemySpeed = 0;
                    if(next.isSideA)
                    {
                        for(Combatant c : sideB.values())
                        {
                            if(c.entity instanceof EntityPlayer)
                            {
                                if(TurnBasedMinecraftMod.config.getPlayerSpeed() > fastestEnemySpeed)
                                {
                                    fastestEnemySpeed = TurnBasedMinecraftMod.config.getPlayerSpeed();
                                }
                            }
                            else
                            {
                                if(c.entityInfo.speed > fastestEnemySpeed)
                                {
                                    fastestEnemySpeed = c.entityInfo.speed;
                                }
                            }
                        }
                    }
                    else
                    {
                        for(Combatant c : sideA.values())
                        {
                            if(c.entity instanceof EntityPlayer)
                            {
                                if(TurnBasedMinecraftMod.config.getPlayerSpeed() > fastestEnemySpeed)
                                {
                                    fastestEnemySpeed = TurnBasedMinecraftMod.config.getPlayerSpeed();
                                }
                            }
                            else
                            {
                                if(c.entityInfo.speed > fastestEnemySpeed)
                                {
                                    fastestEnemySpeed = c.entityInfo.speed;
                                }
                            }
                        }
                    }
                    int fleeProbability = 0;
                    if(next.entity instanceof EntityPlayer)
                    {
                        if(fastestEnemySpeed >= TurnBasedMinecraftMod.config.getPlayerSpeed())
                        {
                            fleeProbability = TurnBasedMinecraftMod.config.getFleeBadProbability();
                        }
                        else
                        {
                            fleeProbability = TurnBasedMinecraftMod.config.getFleeGoodProbability();
                        }
                    }
                    else
                    {
                        if(fastestEnemySpeed >= next.entityInfo.speed)
                        {
                            fleeProbability = TurnBasedMinecraftMod.config.getFleeBadProbability();
                        }
                        else
                        {
                            fleeProbability = TurnBasedMinecraftMod.config.getFleeGoodProbability();
                        }
                    }
                    if((int)(Math.random() * 100) < fleeProbability)
                    {
                        // flee success
                        if(next.isSideA)
                        {
                            sideA.remove(next.entity.getEntityId());
                        }
                        else
                        {
                            sideB.remove(next.entity.getEntityId());
                        }
                        sendMessageToAllPlayers(PacketBattleMessage.MessageType.FLEE, next.entity.getEntityId(), 0, 1);
                        if(next.entity instanceof EntityPlayer)
                        {
                            players.remove(next.entity.getEntityId());
                            playerCount.decrementAndGet();
                            PacketHandler.INSTANCE.sendTo(new PacketBattleMessage(PacketBattleMessage.MessageType.ENDED, 0, 0, 0), (EntityPlayerMP)next.entity);
                        }
                    }
                    else
                    {
                        // flee fail
                        sendMessageToAllPlayers(PacketBattleMessage.MessageType.FLEE, next.entity.getEntityId(), 0, 0);
                    }
                    break;
                case USE_ITEM:
                    break;
                }
                next = turnOrderQueue.poll();
            }
            for(Combatant c : sideA.values())
            {
                c.decision = Decision.UNDECIDED;
            }
            for(Combatant c : sideB.values())
            {
                c.decision = Decision.UNDECIDED;
            }
            state = State.HEALTH_CHECK;
            update(Duration.ZERO);
            break;
        }
        case HEALTH_CHECK:
            Queue<Integer> removeQueue = new ArrayDeque<Integer>();
            for(Combatant c : sideA.values())
            {
                if(!c.entity.isEntityAlive())
                {
                    removeQueue.add(c.entity.getEntityId());
                    if(c.entity instanceof EntityPlayer)
                    {
                        PacketHandler.INSTANCE.sendTo(new PacketBattleMessage(PacketBattleMessage.MessageType.ENDED, c.entity.getEntityId(), 0, 0), (EntityPlayerMP)c.entity);
                    }
                    sendMessageToAllPlayers(PacketBattleMessage.MessageType.DIED, c.entity.getEntityId(), 0, 0);
                }
            }
            for(Combatant c : sideB.values())
            {
                if(!c.entity.isEntityAlive())
                {
                    removeQueue.add(c.entity.getEntityId());
                    if(c.entity instanceof EntityPlayer)
                    {
                        PacketHandler.INSTANCE.sendTo(new PacketBattleMessage(PacketBattleMessage.MessageType.ENDED, c.entity.getEntityId(), 0, 0), (EntityPlayerMP)c.entity);
                    }
                    sendMessageToAllPlayers(PacketBattleMessage.MessageType.DIED, c.entity.getEntityId(), 0, 0);
                }
            }
            boolean didRemove = !removeQueue.isEmpty();
            Integer toRemove = removeQueue.poll();
            while(toRemove != null)
            {
                sideA.remove(toRemove);
                sideB.remove(toRemove);
                if(players.remove(toRemove) != null)
                {
                    playerCount.decrementAndGet();
                }
                toRemove = removeQueue.poll();
            }
            if(players.isEmpty() || sideA.isEmpty() || sideB.isEmpty())
            {
                battleEnded = true;
                sendMessageToAllPlayers(PacketBattleMessage.MessageType.ENDED, 0, 0, 0);
            }
            else if(didRemove)
            {
                notifyPlayersBattleInfo();
            }
            state = State.DECISION;
            undecidedCount.set(playerCount.get());
            timer = TurnBasedMinecraftMod.BattleDecisionTime;
            break;
        }
        return battleEnded;
    }
}