using System.Collections.Generic;
using UI.Xml;
using UnityEngine.UI;

public class TargetGraphicAttribute : CustomXmlAttribute
{
    public override bool UsesApplyMethod { get { return true; } }
    public override List<string> PermittedElements { get { return new List<string>() { "Button", "Dropdown", "Toggle", "InputField", "TextMeshProDropdown", "TextMeshProInputField" }; } }

    public override void Apply(XmlElement xmlElement, string value, AttributeDictionary elementAttributes)
    {
        var selectable = xmlElement.GetComponent<Selectable>();

        if (selectable != null)
        {
            XmlLayoutTimer.AtEndOfFrame(() =>
            {

                var targetElement = xmlElement.GetElementByInternalId<Image>(value);

                if (targetElement == null) targetElement = xmlElement.xmlLayoutInstance.GetElementById<Image>(value);

                if (targetElement != null)
                {
                    selectable.targetGraphic = targetElement;
                }
            }, xmlElement, true);
        }
    }
}