garden_client/Assets/Tookits/LabelUtility/LabelUtility.cs

using System.Reflection;

namespace labelUtility
{


#if UNITY_EDITOR
    using System;
    using System.Collections;
    using System.Collections.Generic;
    using System.IO;
    using System.Linq;
    using System.Text;
    using System.Text.RegularExpressions;
    using System.Xml;
    using UnityEditor;
    using UnityEngine;
    using UnityEngine.UI;

    [Serializable]
    public class LabelSet //todo 更新
    {
        public string Name;

        public string LabelScriptPath;
        public string LabelScriptName;
        public string LabePrefix = "public static string";
        public TextAsset LabelScript;

        public string ComponentScriptPath;
        public string ComponentScriptName;
        public string ComponentPrefix = "private";
        public string NameExcludeString;
        //public string RegistPrefix;
        //public string RegistSuffix;
        public string RegistString;
        public List<string> RegistExtraLines;
        public TextAsset ComponentScript;

        public List<TextAsset> Languages; //todo 需要更新为LanguageXml
        public List<GameObject> Prefabs;

        public bool FoldOut;
        public float TotalHeight;
        public List<ComponentPurview> ComponentPurviews = new List<ComponentPurview>();
    }

    public enum ComponentPurview //todo 更新
    {
        Text,
        Slider,
        Image,
        Button,
        InputField,
        Transform,
        RectTransform,
        SpriteRenderer,
        VirtualScrollRectPlus,
    }

    public class LabelUtility : MonoBehaviour
    {
        #region Config

        public List<LabelSet> LabelSets;

        public static string StartMark = "//StartMark-Used by LabelUtility-Do not remove"; //todo 更新
        public static string EndMark = "//EndMark-Used by LabelUtility-Do not remove"; //todo 更新
        public static string RegistStartMark = "//RegistStartMark-Used by LabelUtility-Do not remove"; //todo 更新
        public static string RegistEndMark = "//RegistEndMark-Used by LabelUtility-Do not remove"; //todo 更新
        public static string Prefix = "public static string ";

        #endregion

        public static void CreateLabelScript(LabelSet labelSet)
        {
            if (!EditorUtility.DisplayDialog("注意", "新建LabelScript?", "确定", "取消"))
            {
                return;
            }

            labelSet.LabelScript = CreateScript(labelSet.LabelScriptName, labelSet.LabelScriptPath);
        }

        public static void CreateComponentScript(LabelSet labelSet)
        {
            if (!EditorUtility.DisplayDialog("注意", "新建ComponentScript?", "确定", "取消"))
            {
                return;
            }

            labelSet.ComponentScript = CreateScript(labelSet.ComponentScriptName, labelSet.ComponentScriptPath);
        }

        private static TextAsset CreateScript(string scriptName, string scriptPath)
        {
            string directory = scriptPath.TrimEnd('/', '\\') + "/";

            if (directory.Length < 6 || directory.Substring(0, 6).ToLower() != "assets")
            {
                throw new Exception("ScripPath必须位置Assets目录内");
            }

            if (!Directory.Exists(directory))
            {
                throw new Exception("文件夹不存在");
            }

            if (string.IsNullOrEmpty(scriptName) || scriptName.Any(Path.GetInvalidFileNameChars().Contains))
            {
                throw new Exception("ScripName包含无效字符");
            }

            string fullPath = $"{scriptPath}/{scriptName}.cs";
            if (File.Exists(fullPath))
            {
                throw new Exception($"已经存在一个 {fullPath}");
            }

            string scriptContent = $"public class {scriptName}\r\n{{\r\n\t#region Config\r\n\r\n\t{StartMark}\r\n\t{EndMark}\r\n\r\n\t#endregion\r\n}}";
            File.WriteAllText(fullPath, scriptContent);
            AssetDatabase.Refresh();
            return AssetDatabase.LoadAssetAtPath<TextAsset>(fullPath);
        }


        public static void CreateLabelFromPrefab(LabelSet labelSet) //todo 更新
        {
            if (!EditorUtility.DisplayDialog("注意", "重新生成Prefab Label?", "确定", "取消"))
            {
                return;
            }

            List<string> labels = new List<string>();
            List<Transform> transforms = GetAllTransformFromPrefab(labelSet.Prefabs);
            foreach (var transform in transforms)
            {
                labels.Add($"{labelSet.LabePrefix} {transform.name} = \"{transform.name}\";");
            }
            InsertLineToScript(StartMark, EndMark, labelSet.LabelScript, labels);
        }

        public static void CreateLabelFromLanguageXml(LabelSet labelSet) //todo 更新
        {
            if (!EditorUtility.DisplayDialog("注意", "重新生成LanguageXml Label?", "确定", "取消"))
            {
                return;
            }

            List<string> labels = new List<string>();
            foreach (var language in labelSet.Languages)
            {
                XmlDocument document = new XmlDocument();
                document.LoadXml(language.text);
                XmlNodeList childNodes = document.SelectSingleNode("lan").ChildNodes;
                for (int i = 0; i < childNodes.Count; i++)
                {
                    labels.AddRange(GetAllLabelFromXmlNode(childNodes[i], labelSet));
                }
            }

            InsertLineToScript(StartMark, EndMark, labelSet.LabelScript, labels);
        }

        private static List<string> GetAllLabelFromXmlNode(XmlNode node, LabelSet labelSet) //todo 增加
        {
            List<string> labels = new List<string>();
            for (int i = 0; i < node.ChildNodes.Count; i++)
            {
                XmlNode childNode = node.ChildNodes[i];
                string label;
                if (i == 0)
                {
                    label = $"\t{labelSet.LabePrefix} {node.Name} = \"{node.Name}\";";
                    labels.Add(label);
                }
                label = $"\t{labelSet.LabePrefix} {node.Name}{LanguageLabel.LanguagePageSeperator}{childNode.Name} = \"{node.Name}{LanguageLabel.LanguagePageSeperator}{childNode.Name}\";";
                labels.Add(label);
            }
            return labels;
        }


        public static void CreateComponentsFromPrefab(LabelSet labelSet)  //todo 更新
        {
            if (!EditorUtility.DisplayDialog("注意", "重新生成PrefabComponent?", "确定", "取消"))
            {
                return;
            }

            List <string> typeNames = new List<string>();
            foreach (var purview in labelSet.ComponentPurviews)
            {
                typeNames.Add(purview.ToString());
            }
            List <Transform> transforms = GetAllTransformFromPrefab(labelSet.Prefabs);
            Dictionary<string, List<string>> namesDictionary = new Dictionary<string, List<string>>();
            foreach (var name in typeNames)
            {
                namesDictionary.Add(name, new List<string>());
            }
            foreach (var kv in namesDictionary)
            {
                Assembly assembly = Assembly.Load("UnityEngine.UI");
                Type type = assembly.GetType($"UnityEngine.UI.{kv.Key}");
                //Debug.Log($"UnityEngine.UI.{kv.Key}");
                if (type == null)
                {
                    assembly = Assembly.Load("UnityEngine");
                    type = assembly.GetType($"UnityEngine.{kv.Key}");
                }
                if (type == null)
                {
                    assembly = Assembly.Load("Assembly-CSharp");
                    type = assembly.GetType($"{kv.Key}");
                }
                foreach (var transform in transforms)
                {
                    if (transform.GetComponent(type) == null) continue;
                    namesDictionary[kv.Key].Add(transform.name);
                }
            }
            List<string> names = new List<string>();
            List<string> components = new List<string>();
            List<string> registStrings = new List<string>();
            registStrings.AddRange(labelSet.RegistExtraLines);
            foreach (var kv in namesDictionary)
            {
                foreach (var name in kv.Value)
                {
                    names.Add(name);
                    string newName = string.IsNullOrEmpty(labelSet.NameExcludeString) ? name : name.Replace(labelSet.NameExcludeString, "");
                    components.Add($"{labelSet.ComponentPrefix} {kv.Key} {newName};");
                    registStrings.Add(labelSet.RegistString.Replace("#NAME", name).Replace("#TYPE", kv.Key).Replace("#NEWNAME", newName));
                }
            }
            //for (int i = 0; i < names.Count; i++)
            //{
            //    components.Add(names[i].Substring(names[i].IndexOf(' ') + 1));
            //}
            InsertLineToScript(StartMark, EndMark, labelSet.ComponentScript, components);
            InsertLineToScript(RegistStartMark, RegistEndMark, labelSet.ComponentScript, registStrings);

            //List<string> names = new List<string>();
            //List<string> textNames = new List<string>();
            //List<string> sliderNames = new List<string>();
            //List<string> imageNames = new List<string>();
            //List<string> buttonNames = new List<string>();
            //List<string> transformNames = new List<string>();
            //List<string> rectTransformNames = new List<string>();
            //List<string> spriteRendererNames = new List<string>();
            //List<Transform> transforms = GetAllTransformFromPrefab(labelSet.Prefabs);
            //foreach (var transform in transforms)
            //{
            //    if (labelSet.ComponentPurviews.Contains(ComponentPurview.Text) && transform.GetComponent<Text>() != null)
            //    {
            //        textNames.Add($"Text {transform.name}");
            //    }
            //    if (labelSet.ComponentPurviews.Contains(ComponentPurview.Slider) && transform.GetComponent<Slider>() != null)
            //    {
            //        sliderNames.Add($"Slider {transform.name}");
            //    }
            //    if (labelSet.ComponentPurviews.Contains(ComponentPurview.Image) && transform.GetComponent<Image>() != null)
            //    {
            //        sliderNames.Add($"Image {transform.name}");
            //    }
            //    if (labelSet.ComponentPurviews.Contains(ComponentPurview.Button) && transform.GetComponent<Button>() != null)
            //    {
            //        buttonNames.Add($"Button {transform.name}");
            //    }
            //    if (labelSet.ComponentPurviews.Contains(ComponentPurview.Transform) && transform.GetComponent<Transform>() != null)
            //    {
            //        transformNames.Add($"Transform {transform.name}");
            //    }
            //    if (labelSet.ComponentPurviews.Contains(ComponentPurview.RectTransform) && transform.GetComponent<RectTransform>() != null)
            //    {
            //        rectTransformNames.Add($"RectTransform {transform.name}");
            //    }
            //    if (labelSet.ComponentPurviews.Contains(ComponentPurview.SpriteRenderer) && transform.GetComponent<SpriteRenderer>() != null)
            //    {
            //        spriteRendererNames.Add($"SpriteRenderer {transform.name}");
            //    }
            //}
            //names.AddRange(textNames);
            //names.AddRange(sliderNames);
            //names.AddRange(imageNames);
            //names.AddRange(buttonNames);
            //names.AddRange(transformNames);
            //names.AddRange(rectTransformNames);
            //names.AddRange(spriteRendererNames);
            //List<string> components = new List<string>();
            //for (int i = 0; i < names.Count; i++)
            //{
            //    string name = string.IsNullOrEmpty(labelSet.NameExcludeString) ? names[i] : names[i].Replace(labelSet.NameExcludeString, "");
            //    components.Add($"{labelSet.ComponentPrefix} {name};");
            //}
            //for (int i = 0; i < names.Count; i++)
            //{
            //    components.Add(names[i].Substring(names[i].IndexOf(' ') + 1));
            //}
            //InsertLineToScript(labelSet.ComponentScript, components);
        }

        private static void InsertLineToScript(string startMark, string endMark, TextAsset textAsset, List<string> insertLines)
        {
            int? startMarkLineIndex = null;
            int? endMarkLineIndex = null;
            List<string> strings = GetScriptContentAfterClearMarks(startMark, endMark, textAsset, ref startMarkLineIndex, ref endMarkLineIndex);
            int prefixIndex = strings[startMarkLineIndex.Value].IndexOf("//");
            string prefix = strings[startMarkLineIndex.Value].Substring(0, prefixIndex);
            for (int i = 0; i < insertLines.Count; i++)
            {
                strings.Insert(startMarkLineIndex.Value + 1, prefix + insertLines[i]);
                startMarkLineIndex++;
            }
            StringBuilder stringBuilder = new StringBuilder();
            for (int i = 0; i < strings.Count; i++)
            {
                stringBuilder.AppendLine(strings[i]);
            }
            string content = stringBuilder.ToString();
            content = content.Substring(0, content.Length - 2);
            File.WriteAllText(AssetDatabase.GetAssetPath(textAsset), content);
            AssetDatabase.Refresh();
        }

        private static List<string> GetScriptContentAfterClearMarks(string startMark, string endMark, TextAsset textAsset, ref int? startMarkLineIndex, ref int? endMarkLineIndex)
        {
            List<string> strings = textAsset.text.Split(new[] {"\r\n"}, StringSplitOptions.None).ToList();
            //List<string> currentDefinedNames = new List<string>();
            for (int i = 0; i < strings.Count; i++)
            {
                if (strings[i].Contains(startMark))
                {
                    startMarkLineIndex = i;
                }
                else if (strings[i].Contains(endMark))
                {
                    endMarkLineIndex = i;
                }

                //if (startMarkLineIndex != null)
                //{
                //    if (endMarkLineIndex != null)
                //    {
                //        if (i > startMarkLineIndex.Value && i < endMarkLineIndex.Value)
                //        {
                //            currentDefinedNames.Add(GetDefinedName(strings[i]));
                //        }
                //    }
                //    else
                //    {
                //        if (i > startMarkLineIndex.Value)
                //        {
                //            currentDefinedNames.Add(GetDefinedName(strings[i]));
                //        }
                //    }
                //}
            }
            int definedCount = endMarkLineIndex.Value - startMarkLineIndex.Value - 1;
            for (int i = 0; i < definedCount; i++)
            {
                strings.RemoveAt(startMarkLineIndex.Value + 1);
            }
            return strings;
        }

        private static List<Transform> GetAllTransformFromPrefab(List<GameObject> prefabs)
        {
            List<Transform> transforms = new List<Transform>();
            for (int i = 0; i < prefabs.Count; i++)
            {
                transforms.AddRange(prefabs[i].GetComponentsInChildren<Transform>(true));
            }
            return transforms;
        }
    }
#endif


}