Project面板显示文件大小

 Project面板显示文件大小，自适应皮肤颜色，两列布局模式下，大图标显示文件则不显示文件大小。

/*******************************************************************************
* 版本声明：v1.0.0
* 类 名 称：FileCapacity
* 创建日期：2020-05-27 16:56:03
* 作者名称：末零
* 功能描述：文件大小 Alt+K
* 修改记录：
* 
******************************************************************************/

using System;
using System.Collections.Generic;
using System.IO;
using System.Reflection;
using UnityEditor;
using UnityEngine;

namespace LastZero
{
    /// 

    /// 文件大小
    /// 
    public class FileCapacity
    {
        private const string REMOVE_STR = "Assets";
        private const string FILESIZE = "FileSize";

        private static readonly int mRemoveCount = REMOVE_STR.Length;
        private static readonly Color professionalColor = new Color(56f / 255, 56f / 255, 56f / 255, 1);
        private static readonly Color personaloColor = new Color(194f / 255, 194f / 255, 194f / 255, 1);
        private static Dictionary DirSizeDictionary = new Dictionary();
        private static List DirList = new List();
        private static bool isShowSize = true;

        [MenuItem("EditorTools/FileSize &K")]
        private static void OpenPlaySize()
        {
            isShowSize = !isShowSize;
            EditorPrefs.SetBool(FILESIZE, isShowSize);
            GetPropjectDirs();
            AssetDatabase.Refresh();
        }

        [InitializeOnLoadMethod]
        private static void InitializeOnLoadMethod()
        {
            EditorApplication.projectChanged += GetPropjectDirs;
            EditorApplication.projectWindowItemOnGUI += OnGUI;
        }

        [UnityEditor.Callbacks.DidReloadScripts]
        private static void OnScriptsReloaded()
        {
            GetPropjectDirs();
        }

        private static void GetPropjectDirs()
        {
            Init();
            if (isShowSize == false) return;
            GetAllDirecotries(Application.dataPath);
            foreach (string path in DirList)
            {
                string newPath = path.Replace("\\", "/");
                DirSizeDictionary.Add(newPath, GetDirectoriesSize(path));
            }
        }
        private static void Init()
        {
            isShowSize = EditorPrefs.GetBool(FILESIZE);
            DirSizeDictionary.Clear();
            DirList.Clear();
        }

        private static void OnGUI(string guid, Rect selectionRect)
        {
            if (isShowSize == false || selectionRect.height > 16) return;//>16为防止文件图标缩放时引起排版错乱
            var dataPath = Application.dataPath;
            var startIndex = dataPath.LastIndexOf(REMOVE_STR);
            var dir = dataPath.Remove(startIndex, mRemoveCount);
            var path = dir + AssetDatabase.GUIDToAssetPath(guid);
            string text = null;

            long fileSize = 0;
            if (DirSizeDictionary.ContainsKey(path))
            {
                fileSize = DirSizeDictionary[path];
            }
            else if (File.Exists(path))
            {
                fileSize = new FileInfo(path).Length;
            }
            else
            {
                return;
            }
            text = GetFormatSizeString((int)fileSize);

            var label = EditorStyles.label;
            var content = new GUIContent(text);
            var width = label.CalcSize(content).x + 10;

            var pos = selectionRect;
            pos.x = pos.xMax - width;
            pos.width = width;

            EditorGUI.DrawRect(pos, UseDark() ? professionalColor : personaloColor);
            Color defaultC = GUI.color;

            if (fileSize > 1024 * 1024)
            {
                GUI.color = Color.red;
            }
            GUI.Label(pos, text);
            GUI.color = defaultC;
        }

        /// 

        /// 获取皮肤
        /// 
        private static bool UseDark()
        {
            PropertyInfo propertyInfo = typeof(EditorGUIUtility).GetProperty("skinIndex", BindingFlags.Static | BindingFlags.NonPublic);
            bool useDark = (int)propertyInfo.GetValue(null) == 1;
            return useDark;
        }

        private static string GetFormatSizeString(int size)
        {
            return GetFormatSizeString(size, 1024);
        }

        private static string GetFormatSizeString(int size, int p)
        {
            return CountSize(size);
        }
        static string CountSize(long Size)
        {
            string[] ns = new string[] { "Byte", "KB", "MB", "GB", "TB", "PB" };
            double baseNum = 1024;
            if (Size <= 0)
            {
                return $"{0.ToString("F2")} {ns[0]}";
            }
            int pow = Math.Min((int)Math.Floor(Math.Log(Size, baseNum)), ns.Length - 1);

            return $"{(Size / Math.Pow(baseNum, pow)).ToString("F2")} {ns[pow]}";
        }

        private static void GetAllDirecotries(string dirPath)
        {
            if (Directory.Exists(dirPath) == false)
            {
                return;
            }
            DirList.Add(dirPath);
            DirectoryInfo[] dirArray = new DirectoryInfo(dirPath).GetDirectories();
            foreach (DirectoryInfo item in dirArray)
            {
                GetAllDirecotries(item.FullName);
            }
        }

        private static long GetDirectoriesSize(string dirPath)
        {
            if (Directory.Exists(dirPath) == false)
            {
                return 0;
            }

            long size = 0;
            DirectoryInfo dir = new DirectoryInfo(dirPath);
            foreach (FileInfo info in dir.GetFiles())
            {
                size += info.Length;
            }

            DirectoryInfo[] dirBotton = dir.GetDirectories();
            foreach (DirectoryInfo info in dirBotton)
            {
                size += GetDirectoriesSize(info.FullName);
            }
            return size;
        }
    }
}

效果截图：