IndieGame/client/Assets/Ether/Editor/Frames/FrameCodeGenerate.cs

/********************************************************************
	文件: FrameCodeGenerate.cs
	作者: 梦语
	邮箱: 1982614048@qq.com
	创建时间: 2024/03/29 17:28:19
	最后修改: 梦语
	最后修改时间: 2024/04/04 16:34:30
	功能: 生成界面基类
*********************************************************************/

using System;
using System.IO;
using System.Collections;
using System.Collections.Generic;
using System.Text.RegularExpressions;
using UnityEditor;
using UnityEngine;
using System.Linq;
using UnityEditor.PackageManager.UI;

namespace Ether
{
    public class FrameCodeGenerate
    {
        private static string templeteBaseFileName = Application.dataPath + $"/Ether/Editor/Frames/FrameBaseTemplete.txt";

        public static string GenerateFrameBaseCode(string prefabPath, string baseDirectoryPath, Dictionary<string, string> componentTypeDic)
        {
            // 加载预制体
            GameObject prefab = AssetDatabase.LoadAssetAtPath<GameObject>(prefabPath);

            if (prefab && PrefabUtility.IsPartOfAnyPrefab(prefab))
            {
                Transform[] childs = prefab.GetComponentsInChildren<Transform>(true);

                //     子物体路径  子物体名称   组件列表 
                Dictionary<string, (string, List<string>)> componentDic = new Dictionary<string, (string, List<string>)>();

                foreach (Transform child in childs)
                {
                    if (child.name == prefab.name)
                    {
                        continue;
                    }
                    (string, List<string>) childComponent = CheckClildName(child, componentTypeDic);
                    string childPath = GetChildPath(child);
                    if (!componentDic.ContainsKey(childPath))
                    {
                        componentDic.Add(childPath, childComponent);
                    }

                }
                GenerateFrameBase(prefabPath, baseDirectoryPath, componentTypeDic, componentDic);
            }

            return null;
        }
        /// <summary>
        /// 检查子物体的名字，获取对应的组件
        /// </summary>
        /// <param name="child"></param>
        /// <returns></returns>
        private static (string, List<string>) CheckClildName(Transform child, Dictionary<string, string> componentTypeDic)
        {
            string input = child.name;
            string pattern = @"<(.*?)>";

            List<string> childTypeList = new List<string>();
            string childName = input;

            MatchCollection matches = Regex.Matches(input, pattern);
            foreach (Match match in matches)
            {
                childName = childName.Replace(match.Groups[0].Value, "");
                string content = match.Groups[1].Value;

                if (componentTypeDic.ContainsKey(content))
                {
                    childTypeList.Add(content);
                }
            }

            return (childName, childTypeList);
        }


        /// <summary>
        /// 获取子物体的路径
        /// </summary>
        /// <param name="child"></param>
        /// <returns></returns>
        private static string GetChildPath(Transform child)
        {
            string path = child.name;
            Transform parent = child.parent;
            while (parent != null)
            {
                path = parent.name + "/" + path;
                parent = parent.parent;
            }

            //去除本身的路径
            int slashIndex = path.IndexOf("/");
            if (slashIndex >= 0)
            {
                path = path.Substring(slashIndex + 1);
            }

            return path;
        }

        private static void GenerateFrameBase(string prefabPath, string baseDirectoryPath, Dictionary<string, string> componentTypeDic, Dictionary<string, (string, List<string>)> componentDic)
        {
            string relativePrefabPath = PathTools.GetRelativePathToResources(prefabPath);

            string define = "";
            string code = "";

            string name = "";
            foreach (var component in componentDic)
            {
                name = component.Value.Item1;

                foreach (var type in component.Value.Item2)
                {
                    string tempType;
                    if (componentTypeDic.ContainsKey(type))
                    {
                        tempType = componentTypeDic[type];
                    }
                    else
                    {
                        tempType = type.ToString();
                    }
                    define += $"protected {tempType} _{type}{name};\n\t\t";
                    code += GetCodeByComponentType(type, tempType, name, component.Key);
                }
            }
            string tempFileStr = FileTools.ReadFile(templeteBaseFileName);

            if (!string.IsNullOrEmpty(tempFileStr))
            {
                tempFileStr = tempFileStr.Replace("$FrameBaseName$", $"{Path.GetFileNameWithoutExtension(prefabPath)}Base");
                tempFileStr = tempFileStr.Replace("$PrefabPath$", relativePrefabPath);
                tempFileStr = tempFileStr.Replace("$PropertyList$", define);
                tempFileStr = tempFileStr.Replace("$PropertyInitList$", code);

                string basePath = $"{Application.dataPath}/{baseDirectoryPath}/{Path.GetFileNameWithoutExtension(prefabPath)}Base.cs";

                if (File.Exists(basePath)) File.Delete(basePath);

                FileTools.WriteFile(basePath, tempFileStr);

            }
        }

        private static string GetCodeByComponentType(string type, string componentName, string name, string path)
        {
            string code = "";
            if (type == "Go")
            {
                code = $"_{type}{name} = GetChild(\"{path}\").gameObject;\n\t\t\t";
            }
            else
            {
                code = $"_{type}{name} = GetComponent<{componentName}>(\"{path}\");\n\t\t\t";
            }
            return code;
        }


        //=================================== 以下是老代码 =========================================
        //public enum ComponentType
        //{
        //    Image,
        //    Go,
        //    Btn,
        //    BtnEx,
        //    Text,
        //    SRContent,
        //    ScrollRect,
        //    Animator,
        //    Trans,
        //}

        //private static Dictionary<ComponentType, string> relationDic = new Dictionary<ComponentType, string>()
        //{
        //    { ComponentType.Image, "Image" },
        //    { ComponentType.Go, "GameObject" },
        //    { ComponentType.Trans, "Transform" },
        //    { ComponentType.Btn, "Button" },
        //    { ComponentType.BtnEx, "ButtonEx" },
        //    { ComponentType.Text, "TextMeshProUGUI" },
        //    { ComponentType.SRContent, "ScrollContent" },
        //    { ComponentType.ScrollRect, "ScrollRect" },
        //    { ComponentType.Animator, "Animator" },
        //};

        //private static string templeteFileName = Application.dataPath + $"/Ether/Editor/Frames/FrameBaseTemplete.txt";

        ///// <summary>
        ///// 生成FrameBase
        ///// </summary>
        ///// <param name="componentDic"></param>
        //private static void GenerateFrameBase(GameObject selectedPrefab, Dictionary<string, (string, List<ComponentType>)> componentDic)
        //{
        //    string selectPath = AssetDatabase.GetAssetPath(selectedPrefab);
        //    string framePart = "Prefabs";

        //    int selectIndex = selectPath.IndexOf(framePart);
        //    string framePrefabPath = selectPath.Substring(selectIndex).Replace(".prefab", "");

        //    Debug.Log($"selectPath:{selectPath}");

        //    string frameTemplete = FileTools.ReadFile(templeteFileName);

        //    Debug.Log(frameTemplete);

        //    (string, string) generateComponentCode = GenerateComponentCodeStr(componentDic);

        //    string pattern = @"\$(.*?)\$";

        //    MatchCollection matches = Regex.Matches(frameTemplete, pattern);

        //    foreach (Match match in matches)
        //    {
        //        string property = match.Groups[0].Value; //带$的原字符串
        //        string content = match.Groups[1].Value; //仅两个$中的内容

        //        string replaceStr = "";

        //        switch (content)
        //        {
        //            case "FrameBaseName":
        //                replaceStr = $"{selectedPrefab.name}Base";
        //                break;
        //            case "PrefabPath":
        //                replaceStr = framePrefabPath;
        //                break;
        //            case "PropertyList":
        //                replaceStr = generateComponentCode.Item1;
        //                break;
        //            case "PropertyInitList":
        //                replaceStr = generateComponentCode.Item2;
        //                break;
        //            default:
        //                break;
        //        }

        //        //Debug.Log(content);

        //        frameTemplete = frameTemplete.Replace(property, replaceStr);
        //    }

        //    string basePath = Application.dataPath + $"/Scripts/AutoGenerated/FrameBase/{selectedPrefab.name}Base.cs";

        //    if (File.Exists(basePath)) File.Delete(basePath);

        //    FileTools.WriteFile(basePath, frameTemplete);

        //    AssetDatabase.Refresh();
        //    return;

        //}

        //private static (string, string) GenerateComponentCodeStr(Dictionary<string, (string, List<ComponentType>)> componentDic)
        //{
        //    string define = "";
        //    string code = "";

        //    string name = "";
        //    foreach (var component in componentDic)
        //    {
        //        name = component.Value.Item1;

        //        foreach (var type in component.Value.Item2)
        //        {
        //            string tempType;
        //            if (relationDic.ContainsKey(type))
        //            {
        //                tempType = relationDic[type];
        //            }
        //            else
        //            {
        //                tempType = type.ToString();
        //            }
        //            define += $"protected {tempType} _{type}{name};\n\t\t";
        //            code += GetCodeByComponentType(type, tempType, name, component.Key);
        //        }
        //    }

        //    return (define, code);
        //}

        //private static string GetCodeByComponentType(ComponentType type, string componentName, string name, string path)
        //{
        //    string code = "";
        //    switch (type)
        //    {
        //        case ComponentType.Go:
        //            code = $"_{type}{name} = GetChild(\"{path}\").gameObject;\n\t\t\t";
        //            break;
        //        case ComponentType.Image:
        //        case ComponentType.Btn:
        //        case ComponentType.BtnEx:
        //        case ComponentType.Text:
        //        default:
        //            code = $"_{type}{name} = GetComponent<{componentName}>(\"{path}\");\n\t\t\t";
        //            break;
        //    }
        //    return code;
        //}
    }
}