H3VR-TNH-Quality-of-Life-Improvements/Assets/Plugins/AssetBundles-Browser/Editor/BundleDetailList.cs

using UnityEditor;
using UnityEngine;
using System.Collections.Generic;
using AssetBundleBrowser.AssetBundleModel;
using UnityEditor.IMGUI.Controls;

namespace AssetBundleBrowser
{
    internal class BundleDetailItem : TreeViewItem
    {
        internal BundleDetailItem(int id, int depth, string displayName, MessageType type) : base(id, depth, displayName)
        {
            MessageLevel = type;
        }

        internal MessageType MessageLevel { get; set; }
    }

    internal class TogglePathTreeViewItem : TreeViewItem
    {
        private static bool m_DisplayAlt = false;

        private string m_DisplayNamePrefix;
        private string m_Path;

        public string Path
        {
            get { return m_Path; }
        }

        public string DisplayNamePrefix
        {
            get { return m_DisplayNamePrefix; }
        }

        public TogglePathTreeViewItem(int id, int depth, string displayName, string path)
        {
            base.depth = depth;
            base.id = id;
            base.displayName = displayName;
            m_Path = path;
            m_DisplayNamePrefix = "";
        }

        public TogglePathTreeViewItem(int id, int depth, string displayNamePrefix, string displayName, string path)
        {
            base.depth = depth;
            base.id = id;
            base.displayName = displayName;
            m_Path = path;
            m_DisplayNamePrefix = displayNamePrefix;
        }

        public override string displayName
        {
            get
            {
                // TODO this is a bit unresponsive here in large projects, see if can be better elsewhere
                Event e = Event.current;
                if (e.alt && e.type == EventType.MouseDown)
                    m_DisplayAlt = !m_DisplayAlt;

                return m_DisplayNamePrefix + (m_DisplayAlt ? m_Path : base.displayName);
            }
            set { base.displayName = value; }
        }
    }

    internal class BundleDetailList : TreeView
    {
        private HashSet<BundleDataInfo> m_Selecteditems;
        private Rect m_TotalRect;

        private const float k_DoubleIndent = 32f;
        private const string k_SizeHeader = "Size: ";
        private const string k_DependencyHeader = "Dependent On:";
        private const string k_DependencyEmpty = k_DependencyHeader + " - None";
        private const string k_MessageHeader = "Messages:";
        private const string k_MessageEmpty = k_MessageHeader + " - None";
        private const string k_ReferencedPrefix = "- ";


        internal BundleDetailList(TreeViewState state) : base(state)
        {
            m_Selecteditems = new HashSet<BundleDataInfo>();
            showBorder = true;
        }

        internal void Update()
        {
            bool dirty = false;
            foreach (BundleDataInfo bundle in m_Selecteditems) dirty |= bundle.dirty;
            if (dirty)
            {
                Reload();
                ExpandAll(2);
            }
        }

        protected override TreeViewItem BuildRoot()
        {
            TreeViewItem root = new TreeViewItem(-1, -1);
            root.children = new List<TreeViewItem>();
            if (m_Selecteditems != null)
                foreach (BundleDataInfo bundle in m_Selecteditems)
                    root.AddChild(AppendBundleToTree(bundle));
            return root;
        }

        protected override void RowGUI(RowGUIArgs args)
        {
            if (args.item as BundleDetailItem != null)
            {
                EditorGUI.HelpBox(
                    new Rect(args.rowRect.x + k_DoubleIndent, args.rowRect.y, args.rowRect.width - k_DoubleIndent, args.rowRect.height),
                    args.item.displayName,
                    (args.item as BundleDetailItem).MessageLevel);
            }
            else
            {
                Color old = GUI.color;
                if (args.item.depth == 1 &&
                    (args.item.displayName == k_MessageEmpty || args.item.displayName == k_DependencyEmpty))
                    GUI.color = Model.k_LightGrey;
                base.RowGUI(args);
                GUI.color = old;
            }
        }

        public override void OnGUI(Rect rect)
        {
            m_TotalRect = rect;
            base.OnGUI(rect);
        }

        protected override float GetCustomRowHeight(int row, TreeViewItem item)
        {
            if (item as BundleDetailItem != null)
            {
                float height = DefaultStyles.backgroundEven.CalcHeight(new GUIContent(item.displayName), m_TotalRect.width);
                return height + 3f;
            }

            return base.GetCustomRowHeight(row, item);
        }


        protected override void SelectionChanged(IList<int> selectedIds)
        {
            base.SelectionChanged(selectedIds);
            List<string> pathList = new List<string>();

            for (int i = 0; i < selectedIds.Count; ++i)
            {
                TreeViewItem item = FindItem(selectedIds[i], rootItem);
                if (item != null) AddDependentAssetsRecursive(item, pathList);
            }

            AssetBundleBrowserMain.instance.m_ManageTab.SetAssetListSelection(pathList);
        }

        private void AddDependentAssetsRecursive(TreeViewItem item, List<string> pathList)
        {
            TogglePathTreeViewItem pathItem = item as TogglePathTreeViewItem;
            if (pathItem != null)
                if (string.IsNullOrEmpty(pathItem.DisplayNamePrefix) == false && pathList.Contains(pathItem.Path) == false)
                    pathList.Add(pathItem.Path);

            if (item.hasChildren)
                for (int i = 0; i < item.children.Count; ++i)
                    AddDependentAssetsRecursive(item.children[i], pathList);
        }

        protected override void DoubleClickedItem(int id)
        {
            base.DoubleClickedItem(id);
            TreeViewItem item = FindItem(id, rootItem);
            if (item != null)
            {
                TogglePathTreeViewItem pathItem = item as TogglePathTreeViewItem;
                if (pathItem != null)
                {
                    Object o = AssetDatabase.LoadAssetAtPath<Object>(pathItem.Path);
                    if (o != null)
                    {
                        Selection.activeObject = o;
                        EditorGUIUtility.PingObject(o);
                    }
                }
            }
        }

        internal static TreeViewItem AppendBundleToTree(BundleDataInfo bundle)
        {
            string itemName = bundle.m_Name.fullNativeName;
            TreeViewItem bunRoot = new TreeViewItem(itemName.GetHashCode(), 0, itemName);

            string str = itemName + k_SizeHeader;
            TreeViewItem sz = new TreeViewItem(str.GetHashCode(), 1, k_SizeHeader + bundle.TotalSize());

            str = itemName + k_DependencyHeader;
            TreeViewItem dependency = new TreeViewItem(str.GetHashCode(), 1, k_DependencyEmpty);
            List<BundleDependencyInfo> depList = bundle.GetBundleDependencies();
            if (depList.Count > 0)
            {
                dependency.displayName = k_DependencyHeader;
                foreach (BundleDependencyInfo dep in bundle.GetBundleDependencies())
                {
                    str = itemName + dep.m_BundleName;
                    TreeViewItem newItem = new TreeViewItem(str.GetHashCode(), 2, dep.m_BundleName);
                    newItem.icon = Model.GetBundleIcon();
                    dependency.AddChild(newItem);

                    Dictionary<string, TogglePathTreeViewItem> toAssetItems = new Dictionary<string, TogglePathTreeViewItem>();

                    for (int i = 0; i < dep.m_FromAssets.Count; ++i)
                    {
                        TogglePathTreeViewItem item = null;

                        if (!toAssetItems.TryGetValue(dep.m_ToAssets[i].fullAssetName, out item))
                        {
                            str = itemName + dep.m_BundleName + dep.m_ToAssets[i].displayName;
                            item = new TogglePathTreeViewItem(str.GetHashCode(), 3, "/" + dep.m_ToAssets[i].displayName, "/" + dep.m_ToAssets[i].fullAssetName);
                            item.icon = AssetDatabase.GetCachedIcon(dep.m_ToAssets[i].fullAssetName) as Texture2D;
                            newItem.AddChild(item);
                            toAssetItems.Add(dep.m_ToAssets[i].fullAssetName, item);
                        }

                        str = str + dep.m_FromAssets[i].displayName;
                        TreeViewItem refItem = new TogglePathTreeViewItem(str.GetHashCode(), 4, k_ReferencedPrefix,
                            dep.m_FromAssets[i].displayName, dep.m_FromAssets[i].fullAssetName);
                        refItem.icon = AssetDatabase.GetCachedIcon(dep.m_FromAssets[i].fullAssetName) as Texture2D;
                        item.AddChild(refItem);
                    }
                }
            }

            str = itemName + k_MessageHeader;
            TreeViewItem msg = new TreeViewItem(str.GetHashCode(), 1, k_MessageEmpty);
            if (bundle.HasMessages())
            {
                msg.displayName = k_MessageHeader;
                List<MessageSystem.Message> currMessages = bundle.GetMessages();

                foreach (MessageSystem.Message currMsg in currMessages)
                {
                    str = itemName + currMsg.message;
                    msg.AddChild(new BundleDetailItem(str.GetHashCode(), 2, currMsg.message, currMsg.severity));
                }
            }


            bunRoot.AddChild(sz);
            bunRoot.AddChild(dependency);
            bunRoot.AddChild(msg);

            return bunRoot;
        }


        internal void SetItems(IEnumerable<BundleInfo> items)
        {
            m_Selecteditems.Clear();
            foreach (BundleInfo item in items) CollectBundles(item);
            SetSelection(new List<int>());
            Reload();
            ExpandAll(2);
        }

        internal void CollectBundles(BundleInfo bundle)
        {
            BundleDataInfo bunData = bundle as BundleDataInfo;
            if (bunData != null)
            {
                m_Selecteditems.Add(bunData);
            }
            else
            {
                BundleFolderInfo bunFolder = bundle as BundleFolderInfo;
                foreach (BundleInfo bun in bunFolder.GetChildList()) CollectBundles(bun);
            }
        }

        internal void ExpandAll(int maximumDepth)
        {
            List<int> expanded = new List<int>(GetExpanded());
            FindItems(rootItem, maximumDepth, expanded);
            SetExpanded(expanded);
        }

        internal void FindItems(TreeViewItem item, int maximumDepth, List<int> expanded)
        {
            if (item.depth >= maximumDepth || !item.hasChildren)
                return;

            expanded.Add(item.id);
            for (int i = 0; i < item.children.Count; ++i) FindItems(item.children[i], maximumDepth, expanded);
        }
    }
}