YanCheng_Metrology/Assets/ThirdPart/UIToolKitFile/Editor/AutoGenerate.cs

using System.Collections;
using System.Collections.Generic;
using System.IO;
using DG.Tweening;
using UnityEditor;
using UnityEditor.SceneManagement;
using UnityEngine;
using UnityEngine.SceneManagement;
using UnityEngine.UIElements;
using System.Reflection;
using System;
using System.Linq;

public class AutoGenerate : EditorWindow
{
    private VisualElement rootElement;
    private Button generateBtn;
    private Label hintText;
    private TextField nameInput;
    private string scenePath;
    private string scriptPath;
    private string prefabPath;
    private string uiSpritePath;

    [MenuItem("Window/Generate Editor Window")]
    public static void ShowExample()
    {
        AutoGenerate wnd = GetWindow<AutoGenerate>();
        wnd.titleContent = new GUIContent("Auto Generate UI Window");
    }

    public void CreateGUI()
    {
        scenePath = "Assets/SandBox/TestUIPanelScenes";
        scriptPath = "Assets/Project/UI/UI_Panel";
        prefabPath = "Assets/Resources/UI/UI_Panel";
        uiSpritePath="Assets/ArtRes/Sprite/UI_Panel";

        // scenePath = "Assets/Test1";
        // scriptPath = "Assets/Test2";
        // prefabPath = "Assets/Test3";

        var visualTree =
            AssetDatabase.LoadAssetAtPath<VisualTreeAsset>(
                "Assets/ThirdPart/UIToolKitFile/UDoc/UXml/AutoGenerateUI.uxml");
        rootElement = visualTree.CloneTree();
        rootVisualElement.Add(rootElement);
        generateBtn = rootElement.Q<Button>("AutoGenerate");
        hintText = rootElement.Q<Label>("HintTitle");
        nameInput = rootElement.Q<TextField>("NameInput");
        generateBtn.clicked += () => { OnGenerateButtonClick(); };
    }

    private void OnGenerateButtonClick()
    {
        DirectoryDontExist(prefabPath);
        DirectoryDontExist(scenePath);
        DirectoryDontExist(scriptPath);
        DirectoryDontExist(uiSpritePath);
        string tempName = nameInput.text;
        if (string.IsNullOrWhiteSpace(tempName))
        {
            ShowHint("输入文件名异常,未添加任何文件");
            return;
        }
        var sceneExist = DoesFileExist(scenePath, tempName);
        var preExist = DoesFileExist(prefabPath, tempName);
        var scriptExist = DoesFileExist(scriptPath, tempName);
        if (sceneExist || preExist || scriptExist)
        {
            ShowHint("三者中至少有一个同名文件已经存在,未添加任何文件");
            return;
        }
        //Selection.activeGameObject;
        GameObject selectedObjectRes = AssetDatabase.LoadAssetAtPath<GameObject>(Path.Combine("Assets", "SandBox",
            "TestUIPanelScenes", "OnlyUITemplate",
            "CustomPanel.prefab"));

        if (selectedObjectRes == null)
        {
            ShowHint("预制体不存在,未添加任何文件");
            return;
        }
       
        GameObject selectedObject = (GameObject)PrefabUtility.InstantiatePrefab(selectedObjectRes);
        PrefabUtility.UnpackPrefabInstance(selectedObject, PrefabUnpackMode.Completely,
            InteractionMode.AutomatedAction);
        selectedObject.transform.localPosition=Vector3.zero;
        selectedObject.transform.localScale=Vector3.one;
        selectedObject.name = tempName;
        var generatePre =
            PrefabUtility.SaveAsPrefabAsset(selectedObject, Path.Combine(prefabPath, tempName + ".prefab"));
        
        string scriptContent = GetScriptTemplate(tempName);
        File.WriteAllText(Path.Combine(scriptPath, tempName + ".cs"), scriptContent);
        AssetDatabase.Refresh();

     


        CreateNewScene(scenePath, tempName);
        // TODO 绑定脚本的功能没实现AttachScriptToGameObject(tempName, generatePre);
        
        //  PrefabUtility.SavePrefabAsset(generatePre);
        
        AssetDatabase.Refresh();
        
        
        // EditorApplication.delayCall = () =>
        // {
        //     AssetDatabase.Refresh();
        //     AttachScriptToGameObject(tempName, generatePre);
        //     PrefabUtility.SavePrefabAsset(generatePre);
        //     AssetDatabase.Refresh();
        //   //  EditorUtility.DisplayDialog("成功", "UI预制体,测试场景,UI脚本已经被创建 ", "OK");
        // };
        //AssetDatabase.Refresh();
        //EditorUtility.DisplayDialog("成功创建", $" " + scenePath, "OK");
    }

    private void DirectoryDontExist(string directoryPath)
    {
        if (!Directory.Exists(directoryPath))
        {
            Directory.CreateDirectory(directoryPath);
        }
    }

    private string GetScriptTemplate(string scriptName)
    {
        return $@"
using UnityEngine;

public class {scriptName} : BasePanel
{{
public void Init()
   {{
   }}
    void Start()
    {{
       
    }}
 public override void ShowMe()
    {{
      
    }}

    public override void HideMe()
    {{
     
    }}

 protected override void OnClick(string btnName)
    {{
      
        switch (btnName)
        {{
            case """":
                break;
            default:
                break;
        }}
    }}
}}";
    }

    /// <summary>
    /// 提示文本
    /// </summary>
    /// <param name="mes"></param>
    private void ShowHint(string mes)
    {
        hintText.text = mes;
        hintText.style.opacity = 1f; // 确保提示文本完全可见
        DOVirtual.DelayedCall(2, () => { hintText.text = string.Empty; });
    }

    /// <summary>
    /// 检查文件是否存在
    /// </summary>
    /// <param name="path"></param>
    /// <param name="fileName"></param>
    /// <returns></returns>
    public bool DoesFileExist(string path, string fileName)
    {
        string fullPath = path;
        string[] files = Directory.GetFiles(fullPath, "*.*", SearchOption.TopDirectoryOnly); //不递归
        foreach (var file in files)
        {
            if (Path.GetFileNameWithoutExtension(file) == fileName)
            {
                return true;
            }
        }

        return false;
    }

    /// <summary>
    /// 创建场景
    /// </summary>
    /// <param name="scenePath"></param>
    /// <param name="sceneName"></param>
    private void CreateNewScene(string scenePath, string sceneName)
    {
        //TODO 将预制体添加到场景未实现
        var newScene = EditorSceneManager.NewScene(NewSceneSetup.DefaultGameObjects, NewSceneMode.Single);
        // GameObject lightGameObject = new GameObject("Directional Light");
        // Light lightComp = lightGameObject.AddComponent<Light>();
        // lightComp.type = LightType.Directional;
        // lightComp.transform.rotation = Quaternion.Euler(50, -30, 0);
        //
        // GameObject ground = GameObject.CreatePrimitive(PrimitiveType.Plane);
        // ground.transform.position = Vector3.zero;

        // 加载预制体资源///
        GameObject prefab =
            AssetDatabase.LoadAssetAtPath<GameObject>(Path.Combine("Assets", "Resources", "UI", "Base",
                "Canvas.prefab"));
        //GameObject pre = AssetDatabase.LoadAssetAtPath<GameObject>(prefabPath + "/sceneName.prefab");
        if (prefab != null)
        {
            // 实例化预制体并添加到场景中
            GameObject prefabInstance = (GameObject)PrefabUtility.InstantiatePrefab(prefab);
           // GameObject preInstance = (GameObject)PrefabUtility.InstantiatePrefab(pre);
            SceneManager.MoveGameObjectToScene(prefabInstance, newScene);
           // SceneManager.MoveGameObjectToScene(preInstance, newScene);

            // preInstance.transform.parent = prefabInstance.transform.Find("Mid");
            // ((RectTransform)preInstance.transform).localPosition = Vector3.zero;
            // ((RectTransform)preInstance.transform).transform.localScale = Vector3.one;
        }
        else
        {
            Debug.LogError("预制体未找到，请检查路径是否正确！");
        }

        EditorSceneManager.SaveScene(newScene, scenePath + "/" + sceneName + ".unity", true);
    }

    private void AttachScriptToGameObject(string className, GameObject pre)
    {
        // 获取当前的Assembly
        var assembly = Assembly.Load("Assembly-CSharp");

        // 使用反射获取新脚本的类型
        Type scriptType = assembly.GetTypes().FirstOrDefault(t => t.Name == className);

        if (scriptType != null)
        {
            // 查找目标游戏对象（假设场景中有一个名为 "TargetObject" 的对象）

            if (pre != null)
            {
                // 添加脚本到对象上
                pre.AddComponent(scriptType);
                ShowHint($"{className} script attached to {pre.name}");
            }
            else
            {
                Debug.LogError("Target object not found in the scene.");
            }
        }
        else
        {
            Debug.LogError($"Script type {className} not found.");
        }
    }
}