using UnityEngine;
using UnityEngine.UI;
using System.Collections;

public class UnitIconControl : MonoBehaviour
{
	private Button thisButton, startBattleButton;
	private UnitSelectionControl unitSelection;

	[SerializeField] private UnitStatBlock unitStats;
	[SerializeField] private bool isAReward;

    public UnitStatBlock UnitStats { get => unitStats; set => unitStats = value; }

    private void Awake()
    {
		if(!isAReward)
        {
			unitSelection = GameObject.Find("Roster Manager").GetComponent<UnitSelectionControl>();
			startBattleButton = GameObject.Find("Start Button").GetComponent<Button>();
		}
    }

    void Start()
	{
		if(!isAReward)
        {
			thisButton = GetComponent<Button>();
			thisButton.onClick.AddListener(TaskOnClick);
		}
	}

	void TaskOnClick()
	{
		bool maxUnitsReached = unitSelection.MoveToSelectedUnitList(gameObject);
		startBattleButton.interactable = true;

		if (!maxUnitsReached)
			unitSelection.CreateStatBlock(unitStats);
	}
}