using System.Collections.Generic;
using System.Linq;
using System.Text;
using UnityEditor;
using UnityEngine;

namespace TyphoonUI
{
    /// <summary>
    /// 拓展UIObject
    /// </summary>
    public static class UIObjectExtension
    {
        /// <summary>
        /// 获取有效节点
        /// </summary>
        public static HashSet<Object> GetValidateNodesHashList(this UIObject obj)
        {
            var hash = new HashSet<Object>();
            foreach (var element in obj.Nodes)
            {
                if (element != null)
                {
                    hash.Add(element);
                }
            }

            return hash;
        }

        /// <summary>
        /// 元素是否被导出
        /// </summary>
        public static bool Contains(this UIObject obj, Object check)
        {
            return obj.Nodes.Contains(check);
        }

        /// <summary>
        /// 设置导出元素
        /// </summary>
        public static void SetNodes(this UIObject obj, List<Object> nodes)
        {
            obj.Nodes = nodes;
        }

        /// <summary>
        /// 保存
        /// </summary>
        public static void Save(this UIObject obj)
        {
            EditorUtility.SetDirty(obj);
            AssetDatabase.SaveAssets();
        }


        /// <summary>
        /// 添加导出元素
        /// </summary>
        public static void Add(this UIObject obj, Object element, bool apply = true)
        {
            var nodes = GetValidateNodesHashList(obj);
            nodes.Add(element);
            var list = nodes.ToList();
            obj.Nodes = list;
            if (apply)
            {
                obj.Apply();
            }
        }

        /// <summary>
        /// 应用
        /// </summary>
        public static void Apply(this UIObject obj)
        {
            var nodes = GetValidateNodesHashList(obj);
            //按层级排序
            var list = nodes.ToList();
            list.Sort((a, b) =>
            {
                var goA = a is GameObject ? a as GameObject : (a as Component).gameObject;
                var goB = b is GameObject ? b as GameObject : (b as Component).gameObject;
                var code = goA.GetHierarchySiblingIndices().CompareTo(goB.GetHierarchySiblingIndices());
                if (code == 0)
                {
                    var nameA = a.GetType().FullName;
                    var nameB = b.GetType().FullName;
                    return nameA.CompareTo(nameB);
                }

                return code;
            });
            obj.Nodes = list;
            TyphoonUIEditorApplication.InvokeUIObjectChanged(obj);
            EditorUtility.SetDirty(obj);
        }

        /// <summary>
        /// 修改
        /// </summary>
        public static void Modify(this UIObject obj, List<ExportNodesWindow.NodeStatus> status)
        {
            if (status != null)
            {
                bool changed = false;
                var hash = obj.GetValidateNodesHashList();
                foreach (var element in status)
                {
                    if (element.Exported)
                    {
                        changed |= hash.Add(element.Node);
                    }
                    else
                    {
                        changed |= hash.Remove(element.Node);
                    }
                }

                if (changed)
                {
                    obj.Nodes = hash.ToList();
                    obj.Apply();
                }
            }
        }

        public static string ExportCode(this UIObject obj, bool refreshProj = true)
        {
            return CodeTool.ExportCode(obj, refreshProj);
        }

        /// <summary>
        /// 获取相关联的实例ID
        /// </summary>
        public static HashSet<int> GetRelatedInstanceID(this UIObject obj)
        {
            var hash = new HashSet<int>();
            var nodes = obj.Nodes;
            foreach (var node in nodes)
            {
                if (node == null)
                {
                    continue;
                }

                var go = node as GameObject;
                if (go == null)
                {
                    go = (node as Component)?.gameObject;
                }

                hash.Add(go.GetInstanceID());
            }

            return hash;
        }


        /// <summary>
        /// 是否有空节点
        /// </summary>
        public static bool HasNullNode(this UIObject obj)
        {
            foreach (var element in obj.Nodes)
            {
                if (element == null)
                {
                    return true;
                }
            }

            return false;
        }


        public static string GetLinkUIObjectViewFullClass(UIObject uiObject)
        {
            var sb = new StringBuilder();
            var original = PrefabUtility.GetCorrespondingObjectFromOriginalSource(uiObject);
            sb.AppendLine($"#{uiObject}");
            if (original != null)
            {
                //无原始预制物
                sb.AppendLine($"[{original.gameObject.GetHierarchyPath()}]");
            }

            var fullClass = new ViewCodeGenerator(original).GetFullClassName();
            sb.AppendLine($"class name:{fullClass}");
            Debug.Log(sb.ToString());
            return fullClass;
        }
    }
}