Forest_Client/Forest/Assets/Editor/GameBuild/GameBuildWindow.cs

using System;
using System.IO;
using UnityEditor;
using UnityEngine;
using UnityEngine.UI;
using Framework.Manager;
using Sirenix.Utilities;
using Framework.GameBuild;
using Sirenix.OdinInspector;
using Sirenix.Utilities.Editor;
using System.Collections.Generic;
using UnityEngine.SceneManagement;
using UnityEditor.SceneManagement;
using Sirenix.OdinInspector.Editor;

public class GameBuildWindow : OdinEditorWindow
{
    [Serializable]
    public class NodeEditor
    {
        [LabelText("挂点")] [ReadOnly] public string name;

        [LabelText("选项数量")] [ReadOnly] public int count;

        [LabelText("选项实例")] public List<GameObject> optionObj;

        [LabelText("选项资源路径")] [FolderPath] public string optionPath;

        public NodeEditor(string name, int count)
        {
            this.name = name;
            this.count = count;
            optionObj = new List<GameObject>();
        }
    }

    private const string NodeRootPath = "UI/UIRoot/UIMainBuild/Build/Root";
    private const string NodeTemplatePath = "Assets/Art/GameBuild/Prefab/Node.prefab";
    private const string OptionTemplatePath = "Option";
    private const string NodeName = "Node{0}";
    private const string OptionName = "Option{0}";
    private const string BtnBubble = "Btn";

    /// <summary>
    /// 当前窗口
    /// </summary>
    private static GameBuildWindow _curWindow;

    /// <summary>
    /// 挂点Root
    /// </summary>
    private GameObject _buildRoot;

    /// <summary>
    /// 当前建造编辑器数据
    /// </summary>
    private BuildEditor _buildEditor;

    [LabelText("当前建造数据")] [VerticalGroup("BuildInfo")]
    public BuildData curBuildData;

    /// <summary>
    /// 打开窗口
    /// </summary>
    public static void OpenWindow(BuildEditor buildEditor)
    {
        _curWindow = GetWindow<GameBuildWindow>();
        _curWindow.position = GUIHelper.GetEditorWindowRect().AlignCenter(800, 600);
        _curWindow._buildEditor = buildEditor;
        _curWindow.LoadBuildData(buildEditor.BuildData);
        _curWindow._buildRoot = GameObject.Find(NodeRootPath);
    }

    private void LoadBuildData(BuildData buildData)
    {
        _curWindow.curBuildData = buildData;
        var root = GameObject.Find(NodeRootPath);
        editorNodes = new List<NodeEditor>(buildData.nodeCount);
        foreach (var node in buildData.nodeInfos)
        {
            var nodeObj = root.transform.Find(node.name).gameObject;
            if (nodeObj != null)
            {
                var editorNode = new NodeEditor(node.name, buildData.thematicCount);
                foreach (Transform child in nodeObj.transform)
                {
                    if (!child.gameObject.name.Equals(BtnBubble))
                    {
                        editorNode.optionObj.Add(child.gameObject);
                    }
                }

                editorNodes.Add(editorNode);
            }
        }
    }

    [VerticalGroup("BuildInfo/Create")]
    [Button("创建/更新 节点实例")]
    public void UpdateNode()
    {
        editorNodeResource = false;
        if (editorNodes != null)
            editorNodes.Clear();

        switch (curBuildData.unlockType)
        {
            case UnlockType.ForGroup:
                InitBuildInfo(curBuildData.nodeCount);
                break;
            case UnlockType.ForThematic:
                InitBuildInfo(curBuildData.thematicCount);
                break;
            default:
                DebugUtil.LogError("解锁条件错误");
                break;
        }

        CreateNodeObj();
    }

    /// <summary>
    /// 初始化建造数据
    /// </summary>
    private void InitBuildInfo(int count)
    {
        //挂点初始化
        curBuildData.nodeInfos = new List<BuildData.BuildNode>(curBuildData.nodeCount);
        for (int i = 0; i < curBuildData.nodeCount; i++)
        {
            var node = new BuildData.BuildNode(string.Format(NodeName, i + 1));
            curBuildData.nodeInfos.Add(node);
        }

        //解锁条件初始化
        curBuildData.unlockInfos = new List<BuildData.UnlockInfo>(count);
        for (int i = 0; i < count; i++)
        {
            var unlock = new BuildData.UnlockInfo();
            curBuildData.unlockInfos.Add(unlock);
        }
    }

    /// <summary>
    /// 创建实例
    /// </summary>
    private void CreateNodeObj()
    {
        //清除
        DeleteNode();

        editorNodes = new List<NodeEditor>(curBuildData.nodeCount);
        var nodeTemplate = AssetDatabase.LoadAssetAtPath<GameObject>(NodeTemplatePath);
        foreach (var node in curBuildData.nodeInfos)
        {
            //生成节点
            var nodeObj = Instantiate(nodeTemplate, _buildRoot.transform);
            nodeObj.name = node.name;

            //生成选项 
            var optionTemplate = nodeObj.transform.Find(OptionTemplatePath).gameObject;
            var tempEditorNode = new NodeEditor(node.name, curBuildData.thematicCount);
            for (int i = 0; i < curBuildData.thematicCount; i++)
            {
                var optionObj = Instantiate(optionTemplate, nodeObj.transform);
                optionObj.name = string.Format(OptionName, i + 1);
                tempEditorNode.optionObj.Add(optionObj);
            }

            editorNodes.Add(tempEditorNode);
            DestroyImmediate(optionTemplate);

            //置顶泡泡按钮
            var btn = nodeObj.transform.Find(BtnBubble);
            btn.transform.SetSiblingIndex(curBuildData.thematicCount);
        }
    }

    [VerticalGroup("Node")] [LabelText("编辑挂点资源")]
    public bool editorNodeResource;

    [VerticalGroup("Node/Info")] [LabelText("挂点资源")] [ShowIf("@editorNodeResource")]
    public List<NodeEditor> editorNodes;

    [VerticalGroup("Node/NodeCreate")]
    [Button("创建选项资源")]
    [ShowIf("@editorNodeResource")]
    public void CreateNodeResource()
    {
        foreach (var editorNode in editorNodes)
        {
            //加载路径下的资源
            var fileEntries = Directory.GetFiles(editorNode.optionPath, "*.png");
            var spriteList = new List<Sprite>();
            foreach (var filePath in fileEntries)
            {
                var assetPath = filePath.Replace(Application.dataPath, "").Replace('\\', '/');
                var sprite = AssetDatabase.LoadAssetAtPath<Sprite>(assetPath);
                if (sprite != null)
                {
                    spriteList.Add(sprite);
                }
            }

            //应用资源
            for (var i = 0; i < editorNode.optionObj.Count; i++)
            {
                if (i < spriteList.Count)
                {
                    var spriteSize = new Vector2(spriteList[i].texture.width, spriteList[i].texture.height);
                    var option = editorNode.optionObj[i];
                    var normalObj = option.transform.Find("Normal").gameObject.GetComponent<Image>();
                    normalObj.sprite = spriteList[i];
                    var rect = normalObj.GetComponent<RectTransform>();
                    rect.sizeDelta = spriteSize;
                }
            }
        }
    }


    [VerticalGroup("Save")]
    [Button("保存")]
    public void Save()
    {
        _buildEditor.SaveData();
        var activeScene = SceneManager.GetActiveScene();

        // 保存当前活动的场景
        var saveResult = EditorSceneManager.SaveScene(activeScene);
        if (!saveResult)
            DebugUtil.LogError("保存失败！！！");
    }

    private void DeleteNode()
    {
        if (_buildRoot == null) _buildRoot = GameObject.Find(NodeRootPath);
        foreach (Transform child in _buildRoot.transform)
        {
            DebugUtil.LogError(child.name);
            DestroyImmediate(child.gameObject);
        }
    }
}