using UnityEngine;
using System.Collections.Generic;


public class ArmyManager : MonoBehaviour
{
    private List<GameObject> enemyArmy = new List<GameObject>();
    private List<GameObject> playerArmy = new List<GameObject>();
    private GameObject enemyCrownDuck = null;
    private GameObject playerCrownDuck = null;
    [SerializeField] private GameManager gameManagerScript;
    
    // Start is called once before the first execution of Update after the MonoBehaviour is created
    void Start()
    {
        
    }

    // Update is called once per frame
    void Update()
    {
        //Debug.Log("Enemy Count : " + enemyArmy.Count);
        //Debug.Log("Player Count : " + playerArmy.Count);
        //Debug.Log(playerCrownDuck);
    }

    public List<GameObject> getArmy(bool isEnemy){
        return(isEnemy ? enemyArmy : playerArmy);
    }

    public void addTroopToArmy(bool isEnemy, GameObject Duck){
        (isEnemy ? enemyArmy : playerArmy).Add(Duck);
    }

    public void removeTroopFromArmy(bool isEnemy, GameObject Duck)
    {
        (isEnemy ? enemyArmy : playerArmy).Remove(Duck);
    }

    public void setCrownDuck(bool isEnemy, GameObject CrownDuck)
    {
        if (isEnemy) {
            enemyCrownDuck = CrownDuck;
        } else {
            playerCrownDuck = CrownDuck;
        }
    }

    public void removeCrownDuck(bool isEnemy)
    {
        if (isEnemy)
        {
            enemyCrownDuck = null;
        }
        else
        {
            playerCrownDuck = null;
        }
    }

    public GameObject getCrownDuck(bool isEnemy)
    {
        return(isEnemy ? enemyCrownDuck : playerCrownDuck);
    }

    public void kill(bool isEnemy, GameObject Duck, bool hasCrown)
    {
        removeTroopFromArmy(isEnemy, Duck);
        if (hasCrown)
        {
            gameManagerScript.endOfLevel(isEnemy);
        }
        removeTroopFromArmy(isEnemy, gameObject);
    }

    public void giveCrownDuckTo(bool isEnemy, GameObject duckToCrown)
    {
        if ((isEnemy ? enemyCrownDuck : playerCrownDuck) != null)
        {
            (isEnemy ? enemyCrownDuck : playerCrownDuck).GetComponent<BaseDuckScript>().loseMyCrown();
            if (isEnemy)
            {
                enemyCrownDuck = null;
            }
            else
            {
                playerCrownDuck = null;
            }
        }

        //becomeCrownDuck will call army manager later and set it locally
        duckToCrown.GetComponent<BaseDuckScript>().becomeCrownDuck();
    }

    public void removeCrownDuckFrom(bool isEnemy, GameObject duckToRemoveCrown)
    {
        (isEnemy ? enemyCrownDuck : playerCrownDuck).GetComponent<BaseDuckScript>().loseMyCrown();
        if (isEnemy)
        {
            enemyCrownDuck = null;
        }
        else
        {
            playerCrownDuck = null;
        }
    }
}