UnityExplorer/src/MainMenu/Pages/ScenePage.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using MelonLoader;
using UnityEngine;
using UnityEngine.SceneManagement;

namespace Explorer
{
    public class ScenePage : WindowPage
    {
        public static ScenePage Instance;

        public override string Name { get => "Scene Explorer"; set => base.Name = value; }

        // ----- Holders for GUI elements ----- //

        private string m_currentScene = "";

        // gameobject list
        private Transform m_currentTransform;
        private List<GameObjectCache> m_objectList = new List<GameObjectCache>();

        // search bar
        private bool m_searching = false;
        private string m_searchInput = "";
        private List<GameObjectCache> m_searchResults = new List<GameObjectCache>();

        // ------------ Init and Update ------------ //

        public override void Init()
        {
            Instance = this;
        }

        public void OnSceneChange()
        {
            m_currentScene = UnityHelpers.ActiveSceneName;

            m_currentTransform = null;
            CancelSearch();
        }

        public override void Update()
        {
            if (!m_searching)
            {
                m_objectList = new List<GameObjectCache>();
                if (m_currentTransform)
                {
                    var endAppend = new List<GameObjectCache>();
                    for (int i = 0; i < m_currentTransform.childCount; i++)
                    {
                        var child = m_currentTransform.GetChild(i);

                        if (child)
                        {
                            if (child.childCount > 0)
                                m_objectList.Add(new GameObjectCache(child.gameObject));
                            else
                                endAppend.Add(new GameObjectCache(child.gameObject));
                        }
                    }
                    m_objectList.AddRange(endAppend);
                    endAppend = null;
                }
                else
                {
                    var scene = SceneManager.GetSceneByName(m_currentScene);
                    var rootObjects = scene.GetRootGameObjects();

                    // add objects with children first
                    foreach (var obj in rootObjects.Where(x => x.transform.childCount > 0))
                    {
                        m_objectList.Add(new GameObjectCache(obj));
                    }
                    foreach (var obj in rootObjects.Where(x => x.transform.childCount == 0))
                    {
                        m_objectList.Add(new GameObjectCache(obj));
                    }
                }
            }
        }

        // --------- GUI Draw Functions --------- //        

        public override void DrawWindow()
        {
            try
            {
                GUILayout.BeginHorizontal(null);
                // Current Scene label
                GUILayout.Label("Current Scene:", new GUILayoutOption[] { GUILayout.Width(120) });
                try
                {
                    // Need to do 'ToList()' so the object isn't cleaned up by Il2Cpp GC.
                    var scenes = SceneManager.GetAllScenes().ToList();

                    if (scenes.Count > 1)
                    {
                        int changeWanted = 0;
                        if (GUILayout.Button("<", new GUILayoutOption[] { GUILayout.Width(30) }))
                        {
                            changeWanted = -1;
                        }
                        if (GUILayout.Button(">", new GUILayoutOption[] { GUILayout.Width(30) }))
                        {
                            changeWanted = 1;
                        }
                        if (changeWanted != 0)
                        {
                            int index = scenes.IndexOf(SceneManager.GetSceneByName(m_currentScene));
                            index += changeWanted;
                            if (index > scenes.Count - 1)
                            {
                                index = 0;
                            }
                            else if (index < 0)
                            {
                                index = scenes.Count - 1;
                            }
                            m_currentScene = scenes[index].name;
                        }
                    }
                }
                catch { }
                GUILayout.Label("<color=cyan>" + m_currentScene + "</color>", null); //new GUILayoutOption[] { GUILayout.Width(250) });

                GUILayout.EndHorizontal();

                // ----- GameObject Search -----
                GUILayout.BeginHorizontal(GUI.skin.box, null);
                GUILayout.Label("<b>Search Scene:</b>", new GUILayoutOption[] { GUILayout.Width(100) });
                m_searchInput = GUILayout.TextField(m_searchInput, null);
                if (GUILayout.Button("Search", new GUILayoutOption[] { GUILayout.Width(80) }))
                {
                    Search();
                }
                GUILayout.EndHorizontal();

                GUILayout.Space(15);

                // ************** GameObject list ***************

                // ----- main explorer ------
                if (!m_searching)
                {
                    if (m_currentTransform != null)
                    {
                        GUILayout.BeginHorizontal(null);
                        if (GUILayout.Button("<-", new GUILayoutOption[] { GUILayout.Width(35) }))
                        {
                            TraverseUp();
                        }
                        else
                        {
                            GUILayout.Label(m_currentTransform.GetGameObjectPath(), null);
                        }
                        GUILayout.EndHorizontal();
                    }
                    else
                    {
                        GUILayout.Label("Scene Root GameObjects:", null);
                    }

                    if (m_objectList.Count > 0)
                    {
                        foreach (var obj in m_objectList)
                        {
                            //UIStyles.GameobjButton(obj, SetTransformTarget, true, MainMenu.MainRect.width - 170);
                            UIHelpers.FastGameobjButton(obj.RefGameObject, obj.EnabledColor, obj.Label, obj.RefGameObject.activeSelf, SetTransformTarget, true, MainMenu.MainRect.width - 170);
                        }
                    }
                }
                else // ------ Scene Search results ------
                {
                    if (GUILayout.Button("<-", new GUILayoutOption[] { GUILayout.Width(35) }))
                    {
                        CancelSearch();
                    }

                    GUILayout.Label("Search Results:", null);

                    if (m_searchResults.Count > 0)
                    {
                        foreach (var obj in m_searchResults)
                        {
                            //UIStyles.GameobjButton(obj, SetTransformTarget, true, MainMenu.MainRect.width - 170);
                            UIHelpers.FastGameobjButton(obj.RefGameObject, obj.EnabledColor, obj.Label, obj.RefGameObject.activeSelf, SetTransformTarget, true, MainMenu.MainRect.width - 170);
                        }
                    }
                    else
                    {
                        GUILayout.Label("<color=red><i>No results found!</i></color>", null);
                    }
                }
            }
            catch
            {
                m_currentTransform = null;
            }
        }



        // -------- Actual Methods (not drawing GUI) ---------- //

        public void SetTransformTarget(GameObject obj)
        {
            m_currentTransform = obj.transform;
            CancelSearch();
        }

        public void TraverseUp()
        {
            if (m_currentTransform.parent != null)
            {
                m_currentTransform = m_currentTransform.parent;
            }
            else
            {
                m_currentTransform = null;
            }
        }

        public void Search()
        {
            m_searchResults = SearchSceneObjects(m_searchInput);
            m_searching = true;
        }

        public void CancelSearch()
        {
            m_searching = false;
        }

        public List<GameObjectCache> SearchSceneObjects(string _search)
        {
            var matches = new List<GameObjectCache>();

            foreach (var obj in Resources.FindObjectsOfTypeAll<GameObject>())
            {
                if (obj.name.ToLower().Contains(_search.ToLower()) && obj.scene.name == m_currentScene)
                {
                    matches.Add(new GameObjectCache(obj));
                }
            }

            return matches;
        }
    
        public class GameObjectCache
        {
            public GameObject RefGameObject;
            public string Label;
            public Color EnabledColor;
            public int ChildCount;

            public GameObjectCache(GameObject obj)
            {
                RefGameObject = obj;
                ChildCount = obj.transform.childCount;

                Label = (ChildCount > 0) ? "[" + obj.transform.childCount + " children] " : "";
                Label += obj.name;

                bool enabled = obj.activeSelf;
                int childCount = obj.transform.childCount;
                if (enabled)
                {
                    if (childCount > 0)
                    {
                        EnabledColor = Color.green;
                    }
                    else
                    {
                        EnabledColor = UIStyles.LightGreen;
                    }
                }
                else
                {
                    EnabledColor = Color.red;
                }
            }
        }
    }
}