框架上传

Assets/00.StaryEvo/Editor/OdinTool/OneKeySearchDuplicateFiles.cs

@@ -0,0 +1,186 @@
+/**************************************************** 
+ 文件：OneKeySearchDuplicateFiles.cs
+ 作者：张铮
+ 邮箱：834207172@qq.com
+ 日期：2022/3/7 18:28:50
+ 功能：
+*****************************************************/
+
+using System;
+using System.Collections;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using UnityEngine;
+using UnityEngine.UI;
+using Sirenix.OdinInspector;
+using Sirenix.OdinInspector.Editor;
+using UnityEditor;
+using System.Threading;
+
+namespace Stary.Evo.Editor
+{
+
+
+    public class OneKeySearchDuplicateFiles : SerializedScriptableObject
+    {
+        private bool IsToggled;
+        private int maxCount;
+        private IEnumerator<FileInfo> fileInfoIEnumerator;
+
+        [PropertySpace(10)]
+        [Title("需要搜索的文件夹", "默认为Asset全目录", titleAlignment: TitleAlignments.Split)]
+        [FolderPath(ParentFolder = "Assets", RequireExistingPath = true, AbsolutePath = true)]
+        [LabelText("选择你要搜索的文件夹")]
+        public string targetSearchFolder;
+
+
+        [ShowInInspector] [DictionaryDrawerSettings(KeyLabel = "MD5值", ValueLabel = "文件名称列表")]
+        private Dictionary<string, List<string>> sameMD5Group = new Dictionary<string, List<string>>();
+
+        [ShowInInspector] [DictionaryDrawerSettings(KeyLabel = "文件名称", ValueLabel = "绝对路径列表")]
+        private Dictionary<string, List<string>> sameNameGroup = new Dictionary<string, List<string>>();
+
+        [ShowInInspector]
+        [TitleGroup("重复文件列表")]
+        [HorizontalGroup("重复文件列表/重复文件")]
+        [BoxGroup("重复文件列表/重复文件/MD5值相同", CenterLabel = true)]
+        [PropertyOrder(1000)]
+        [InfoBox("发现相同MD5值文件.", InfoMessageType.Error, "CheckSameMD5ResultGroup")]
+        [ShowIf("$CheckSameMD5ResultGroup")]
+        [DictionaryDrawerSettings(KeyLabel = "MD5值", ValueLabel = "相同MD5值文件名称")]
+        private Dictionary<string, List<string>> sameMD5Result5Group = new Dictionary<string, List<string>>();
+
+
+        [BoxGroup("重复文件列表/重复文件/名称值相同", CenterLabel = true)]
+        [ShowInInspector]
+        [PropertyOrder(1000)]
+        [InfoBox("发现相同名称文件.", InfoMessageType.Error, "CheckSameNameResultGroup")]
+        [ShowIf("$CheckSameNameResultGroup")]
+        [DictionaryDrawerSettings(KeyLabel = "相同文件名称", ValueLabel = "对应绝对路径列表")]
+        private Dictionary<string, List<string>> sameNameResultGroup = new Dictionary<string, List<string>>();
+
+        public bool CheckSameMD5ResultGroup()
+        {
+            return sameMD5Result5Group.Count > 0;
+        }
+
+        private bool CheckSameNameResultGroup()
+        {
+            return sameNameResultGroup.Count > 0;
+        }
+
+        [PropertySpace(10, 20)]
+        [ShowIf("@ IsToggled== false")]
+        [Button("开始搜索", ButtonSizes.Large)]
+        public void StartSearch()
+        {
+            if (string.IsNullOrEmpty(targetSearchFolder))
+            {
+                targetSearchFolder = Application.dataPath;
+            }
+
+            ResetData();
+            DirectoryInfo directoryInfo = new DirectoryInfo(targetSearchFolder);
+            var filesGroup = directoryInfo.EnumerateFiles("*", SearchOption.AllDirectories)
+                .Where(x => x.Extension != ".meta");
+
+            maxCount = filesGroup.Count();
+            fileInfoIEnumerator = filesGroup.GetEnumerator();
+            IsToggled = true;
+            EditorApplication.update += Updte;
+        }
+
+        private void ResetData()
+        {
+            maxCount = 0;
+            MaxCount = 0;
+            sameMD5Group.Clear();
+            sameNameGroup.Clear();
+            sameMD5Result5Group.Clear();
+            sameNameResultGroup.Clear();
+            fileInfoIEnumerator = null;
+        }
+
+        /// <summary>
+        /// 过滤掉没有重复文件的数据
+        /// </summary>
+        private void FilterDictionary()
+        {
+            sameMD5Result5Group = sameMD5Group.Where(x => x.Value.Count > 1).ToDictionary(p => p.Key, p => p.Value);
+            sameNameResultGroup = sameNameGroup.Where(x => x.Value.Count > 1).ToDictionary(p => p.Key, p => p.Value);
+        }
+
+        [ReadOnly]
+        [ProgressBar(0, "maxCount", DrawValueLabel = true, ValueLabelAlignment = TextAlignment.Left,
+            ColorGetter = "GetHealthBarColor", Height = 30)]
+        [ShowInInspector]
+        [HideLabel]
+        [ShowIf("@ IsToggled== true")]
+        public int MaxCount { get; set; } //绘制进度条
+
+        private Color GetHealthBarColor(int value)
+        {
+            maxCount = maxCount == 0 ? 1 : maxCount;
+            return Color.Lerp(Color.red, Color.green, Mathf.Pow((float)value / maxCount, 2));
+        }
+
+        public void Updte()
+        {
+            if (IsToggled)
+            {
+                if (fileInfoIEnumerator.MoveNext())
+                {
+                    //获取对应Hash值
+                    string hashValue = GetMD5HashFromFile(fileInfoIEnumerator.Current.FullName);
+                    if (!sameMD5Group.ContainsKey(hashValue))
+                    {
+                        sameMD5Group[hashValue] = new List<string>();
+                    }
+
+                    sameMD5Group[hashValue].Add("名称为：" + fileInfoIEnumerator.Current.Name);
+
+                    //获取名称
+                    string fileName = fileInfoIEnumerator.Current.Name;
+
+                    if (!sameNameGroup.ContainsKey(fileName))
+                    {
+                        sameNameGroup[fileName] = new List<string>();
+                    }
+
+                    sameNameGroup[fileName].Add("路径为：" + fileInfoIEnumerator.Current.FullName);
+
+                    ++MaxCount;
+                }
+                else
+                {
+                    EditorApplication.update -= Updte;
+                    IsToggled = false;
+                    FilterDictionary();
+                    Debug.Log("<color=green>注销</color>");
+                }
+            }
+        }
+
+        /// <summary>
+        /// 计算文件MD5值
+        /// </summary>
+        /// <param name="fileFullName"></param>
+        /// <returns></returns>
+        public string GetMD5HashFromFile(string fileFullName)
+        {
+            try
+            {
+                FileStream file = new FileStream(fileFullName, FileMode.Open);
+                System.Security.Cryptography.MD5 md5 = new System.Security.Cryptography.MD5CryptoServiceProvider();
+                byte[] retVal = md5.ComputeHash(file);
+                file.Close();
+                return BitConverter.ToString(retVal).ToLower().Replace("-", "");
+            }
+            catch
+            {
+                throw;
+            }
+        }
+    }
+}