// Copyright (c) Pixel Crushers. All rights reserved.

using UnityEngine;
using System.Collections.Generic;

namespace PixelCrushers
{

    [AddComponentMenu("")] // Use wrapper instead.
    public class LocalizeUI : MonoBehaviour
    {

        [Tooltip("Overrides the global text table.")]
        [SerializeField]
        private TextTable m_textTable;

        [Tooltip("Overrides the UILocalizationManager's Localized Fonts.")]
        [SerializeField]
        private LocalizedFonts m_localizedFonts;

        [Tooltip("(Optional) If assigned, use this instead of the UI element's text's value as the field lookup value.")]
        [SerializeField]
        private string m_fieldName = string.Empty;

        public TextTable textTable
        {
            get { return m_textTable; }
            set { m_textTable = value; }
        }

        public LocalizedFonts localizedFonts
        {
            get { return m_localizedFonts; }
            set { m_localizedFonts = value; }
        }

        public string fieldName
        {
            get { return string.IsNullOrEmpty(m_fieldName) ? null : m_fieldName; }
            set { m_fieldName = value; }
        }

        private bool m_started = false;
        protected bool started
        {
            get { return m_started; }
            private set { m_started = value; }
        }

        private List<string> m_fieldNames = new List<string>();
        public List<string> fieldNames
        {
            get { return m_fieldNames; }
            set { m_fieldNames = value; }
        }

        private List<string> m_tmpFieldNames = new List<string>();
        public List<string> tmpFieldNames
        {
            get { return m_tmpFieldNames; }
            set { m_tmpFieldNames = value; }
        }
        private UnityEngine.UI.Text m_text = null;
        public UnityEngine.UI.Text text
        {
            get { return m_text; }
            set { m_text = value; }
        }

        private UnityEngine.UI.Dropdown m_dropdown = null;
        public UnityEngine.UI.Dropdown dropdown
        {
            get { return m_dropdown; }
            set { m_dropdown = value; }
        }

#if TMP_PRESENT
        private TMPro.TextMeshPro m_textMeshPro;
        public TMPro.TextMeshPro textMeshPro
        {
            get { return m_textMeshPro; }
            set { m_textMeshPro = value; }
        }
        private TMPro.TextMeshProUGUI m_textMeshProUGUI;
        public TMPro.TextMeshProUGUI textMeshProUGUI
        {
            get { return m_textMeshProUGUI; }
            set { m_textMeshProUGUI = value; }
        }
        private TMPro.TMP_Dropdown m_textMeshProDropdown;
        public TMPro.TMP_Dropdown textMeshProDropdown
        {
            get { return m_textMeshProDropdown; }
            set { m_textMeshProDropdown = value; }
        }
        private bool m_lookedForTMP = false;
#endif

#if USE_STM
        private SuperTextMesh m_superTextMesh = null;
        public SuperTextMesh superTextMesh
        {
            get { return m_superTextMesh; }
            set { m_superTextMesh = value; }
        }
        private bool m_lookedForSTM = false;
#endif

        protected virtual void Start()
        {
            started = true;
            UpdateText();
        }

        protected virtual void OnEnable()
        {
            if (started) UpdateText();
        }

        /// <summary>
        /// If fieldName(s), tmpFieldName(s) are blank, set them to the UI element's text.
        /// </summary>
        public virtual void ValidateFieldNames()
        {
            if (!string.IsNullOrEmpty(fieldName)) return;
            if (fieldNames.Count > 0) return;

            if (text == null && dropdown == null)
            {
                text = GetComponent<UnityEngine.UI.Text>();
                dropdown = GetComponent<UnityEngine.UI.Dropdown>();
            }
            var hasLocalizableComponent = text != null || dropdown != null;

#if TMP_PRESENT
            if (!m_lookedForTMP)
            {
                m_lookedForTMP = true;
                textMeshPro = GetComponent<TMPro.TextMeshPro>();
                textMeshProUGUI = GetComponent<TMPro.TextMeshProUGUI>();
                textMeshProDropdown = GetComponent<TMPro.TMP_Dropdown>();
            }
            hasLocalizableComponent = hasLocalizableComponent || textMeshPro != null ||
                textMeshProUGUI != null || textMeshProDropdown != null;
#endif

#if USE_STM
            if (!m_lookedForSTM)
            {
                m_lookedForSTM = true;
                superTextMesh = GetComponent<SuperTextMesh>();
            }
            hasLocalizableComponent = hasLocalizableComponent || superTextMesh != null;
#endif

            if (!hasLocalizableComponent) return;

            // Get the original values to use as field lookups:
            if (string.IsNullOrEmpty(fieldName))
            {
                fieldName = (text != null) ? text.text : string.Empty;
            }
            if ((dropdown != null) && (fieldNames.Count != dropdown.options.Count))
            {
                fieldNames.Clear();
                dropdown.options.ForEach(opt => fieldNames.Add(opt.text));
            }


#if TMP_PRESENT
            if (textMeshPro != null)
            {
                if (string.IsNullOrEmpty(fieldName))
                {
                    fieldName = (textMeshPro != null) ? textMeshPro.text : string.Empty;
                }
            }
            if (textMeshProUGUI != null)
            {
                if (string.IsNullOrEmpty(fieldName))
                {
                    fieldName = (textMeshProUGUI != null) ? textMeshProUGUI.text : string.Empty;
                }
            }
            if (textMeshProDropdown != null)
            {
                if (tmpFieldNames.Count != textMeshProDropdown.options.Count)
                {
                    tmpFieldNames.Clear();
                    textMeshProDropdown.options.ForEach(opt => tmpFieldNames.Add(opt.text));
                }
            }
#endif

#if USE_STM
            if (superTextMesh != null && string.IsNullOrEmpty(fieldName))
            {
                fieldName = superTextMesh.text;
            }
#endif

        }

        /// <summary>
        /// Set UI element's text (and possibly font) according to current language.
        /// </summary>
        public virtual void UpdateText()
        {
            var language = (UILocalizationManager.instance != null) ? UILocalizationManager.instance.currentLanguage : string.Empty;

            // Skip if no text table or language set:
            if (textTable == null && (UILocalizationManager.instance == null || UILocalizationManager.instance.textTable == null))
            {
                Debug.LogWarning("No localized text table is assigned to " + name + " or a UI Localized Manager component.", this);
                return;
            }

            if (!HasLanguage(language))
            {
                Debug.LogWarning("Text table does not have a language '" + language + "'.", textTable);
                //return; //--- Allow to continue and use default language value.
            }

            // Get LocalizedFonts asset:
            var localizedFonts = (m_localizedFonts != null) ? m_localizedFonts : UILocalizationManager.instance.localizedFonts;
            var localizedFont = (localizedFonts != null) ? localizedFonts.GetFont(language) : null;

            // Make sure we have localizable UI components:
            if (text == null && dropdown == null)
            {
                text = GetComponent<UnityEngine.UI.Text>();
                dropdown = GetComponent<UnityEngine.UI.Dropdown>();
            }
            var hasLocalizableComponent = text != null || dropdown != null;

#if TMP_PRESENT
            var localizedTextMeshProFont = (localizedFonts != null) ? localizedFonts.GetTextMeshProFont(language) : null;
            if (!m_lookedForTMP)
            {
                m_lookedForTMP = true;
                textMeshPro = GetComponent<TMPro.TextMeshPro>();
                textMeshProUGUI = GetComponent<TMPro.TextMeshProUGUI>();
                textMeshProDropdown = GetComponent<TMPro.TMP_Dropdown>();
            }
            hasLocalizableComponent = hasLocalizableComponent || textMeshPro != null ||
                textMeshProUGUI != null || textMeshProDropdown != null;
#endif

#if USE_STM
            if (!m_lookedForSTM)
            {
                m_lookedForSTM = true;
                superTextMesh = GetComponent<SuperTextMesh>();
            }
            hasLocalizableComponent = hasLocalizableComponent || superTextMesh != null;
#endif

            if (!hasLocalizableComponent)
            {
                Debug.LogWarning("Localize UI didn't find a localizable UI component on " + name + ".", this);
                return;
            }

            // Get the original values to use as field lookups:
            if (string.IsNullOrEmpty(fieldName))
            {
                fieldName = (text != null) ? text.text : string.Empty;
            }
            if ((dropdown != null) && (fieldNames.Count != dropdown.options.Count))
            {
                fieldNames.Clear();
                dropdown.options.ForEach(opt => fieldNames.Add(opt.text));
            }

            // Localize Text:
            if (text != null)
            {
                if (!HasField(fieldName))
                {
                    Debug.LogWarning("Text table does not have a field '" + fieldName + "'.", textTable);
                }
                else
                {
                    text.text = GetLocalizedText(fieldName);
                    if (localizedFont != null) text.font = localizedFont;
                }
            }

            // Localize Dropdown:
            if (dropdown != null)
            {
                for (int i = 0; i < dropdown.options.Count; i++)
                {
                    if (i < fieldNames.Count)
                    {
                        dropdown.options[i].text = GetLocalizedText(fieldNames[i]);
                    }
                }
                dropdown.captionText.text = GetLocalizedText(fieldNames[dropdown.value]);
                if (localizedFont != null)
                {
                    dropdown.captionText.font = localizedFont;
                    dropdown.itemText.font = localizedFont;
                }
            }

#if TMP_PRESENT
            if (!m_lookedForTMP)
            {
                m_lookedForTMP = true;
                textMeshPro = GetComponent<TMPro.TextMeshPro>();
                textMeshProUGUI = GetComponent<TMPro.TextMeshProUGUI>();
            }
            if (textMeshPro != null)
            {
                if (string.IsNullOrEmpty(fieldName))
                {
                    fieldName = (textMeshPro != null) ? textMeshPro.text : string.Empty;
                }
                if (!HasField(fieldName))
                {
                    Debug.LogWarning("Text table does not have a field '" + fieldName + "'.", textTable);
                }
                else
                {
                    textMeshPro.text = GetLocalizedText(fieldName);
                    if (localizedTextMeshProFont != null) 
                    {
                        textMeshPro.font = localizedTextMeshProFont;
                        textMeshPro.enabled = false;
                        textMeshPro.enabled = true;
                    }
                }
            }
            if (textMeshProUGUI != null)
            {
                if (string.IsNullOrEmpty(fieldName))
                {
                    fieldName = (textMeshProUGUI != null) ? textMeshProUGUI.text : string.Empty;
                }
                if (!HasField(fieldName))
                {
                    Debug.LogWarning("Text table does not have a field '" + fieldName + "'.", textTable);
                }
                else
                {
                    textMeshProUGUI.text = GetLocalizedText(fieldName);
                    if (localizedTextMeshProFont != null) textMeshProUGUI.font = localizedTextMeshProFont;
                    textMeshProUGUI.enabled = false;
                    textMeshProUGUI.enabled = true;
                }
            }
            if (textMeshProDropdown != null)
            {
                if (tmpFieldNames.Count != textMeshProDropdown.options.Count)
                {
                    tmpFieldNames.Clear();
                    textMeshProDropdown.options.ForEach(opt => tmpFieldNames.Add(opt.text));
                }
                for (int i = 0; i < textMeshProDropdown.options.Count; i++)
                {
                    if (i < tmpFieldNames.Count)
                    {
                        textMeshProDropdown.options[i].text = GetLocalizedText(tmpFieldNames[i]);
                    }
                }
                textMeshProDropdown.captionText.text = GetLocalizedText(tmpFieldNames[textMeshProDropdown.value]);
                if (localizedTextMeshProFont != null)
                {
                    textMeshProDropdown.captionText.font = localizedTextMeshProFont;
                    textMeshProDropdown.itemText.font = localizedTextMeshProFont;
                }
            }
#endif

#if USE_STM
            if (superTextMesh != null)
            {
                if (string.IsNullOrEmpty(fieldName)) fieldName = superTextMesh.text;
                superTextMesh.text = GetLocalizedText(fieldName);
            }
#endif

        }

        protected virtual bool HasLanguage(string language)
        {
            return (textTable != null && textTable.HasLanguage(language)) ||
                UILocalizationManager.instance.HasLanguage(language);
        }

        protected virtual bool HasField(string fieldName)
        {
            return (textTable != null && textTable.HasField(fieldName)) ||
                UILocalizationManager.instance.HasField(fieldName);
        }

        protected virtual string GetLocalizedText(string fieldName)
        {
            return (textTable != null && textTable.HasField(fieldName))
                ? textTable.GetFieldTextForLanguage(fieldName, GlobalTextTable.currentLanguage)
                : UILocalizationManager.instance.GetLocalizedText(fieldName); //---Was: GlobalTextTable.Lookup(fieldName);
        }

        /// <summary>
        /// Sets the field name, which is the key to use in the text table.
        /// By default, the field name is the initial value of the Text component.
        /// </summary>
        /// <param name="fieldName">Field name.</param>
        public virtual void SetFieldName(string newFieldName = "")
        {
            if (text == null) text = GetComponent<UnityEngine.UI.Text>();
            fieldName = (string.IsNullOrEmpty(newFieldName) && text != null) ? text.text : newFieldName;
        }

        /// <summary>
        /// If changing the Dropdown options, call this afterward to update the localization.
        /// </summary>
        public virtual void UpdateDropdownOptions()
        {
            fieldNames.Clear();
            tmpFieldNames.Clear();
            UpdateText();
        }
    }
}