ProjectDDD/Assets/Plugins/Easy Save 3/Editor/ES3ReferenceMgrEditor.cs

using UnityEditor;
using UnityEngine;
using UnityEngine.SceneManagement;
using System;
using System.Collections;
using System.Collections.Generic;

[CustomEditor(typeof(ES3ReferenceMgr))]
[System.Serializable]
public class ES3ReferenceMgrEditor : UnityEditor.Editor
{
    private bool isDraggingOver = false;
    private bool openReferences = false;

    private ES3ReferenceMgr _mgr = null;
    private ES3ReferenceMgr mgr
    {
        get
        {
            if (_mgr == null)
                _mgr = (ES3ReferenceMgr)serializedObject.targetObject;
            return _mgr;
        }
    }

    public override void OnInspectorGUI()
    {
        EditorGUILayout.HelpBox("This allows Easy Save to maintain references to objects in your scene.\n\nIt is automatically updated when you enter Playmode or build your project.", MessageType.Info);

        if (EditorGUILayout.Foldout(openReferences, "References") != openReferences)
        {
            openReferences = !openReferences;
            if (openReferences == true)
                openReferences = EditorUtility.DisplayDialog("Are you sure?", "Opening this list will display every reference in the manager, which for larger projects can cause the Editor to freeze\n\nIt is strongly recommended that you save your project before continuing.", "Open References", "Cancel");
        }

        // Make foldout drag-and-drop enabled for objects.
        if (GUILayoutUtility.GetLastRect().Contains(Event.current.mousePosition))
        {
            Event evt = Event.current;

            switch (evt.type)
            {
                case EventType.DragUpdated:
                case EventType.DragPerform:
                    isDraggingOver = true;
                    break;
                case EventType.DragExited:
                    isDraggingOver = false;
                    break;
            }

            if (isDraggingOver)
            {
                DragAndDrop.visualMode = DragAndDropVisualMode.Copy;

                if (evt.type == EventType.DragPerform)
                {
                    DragAndDrop.AcceptDrag();
                    Undo.RecordObject(mgr, "Add References to Easy Save 3 Reference List");
                    foreach (UnityEngine.Object obj in DragAndDrop.objectReferences)
                        mgr.Add(obj);
                    // Return now because otherwise we'll change the GUI during an event which doesn't allow it.
                    return;
                }
            }
        }

        if (openReferences)
        {
            EditorGUI.indentLevel++;

            foreach (var kvp in mgr.idRef)
            {
                EditorGUILayout.BeginHorizontal();

                var value = EditorGUILayout.ObjectField(kvp.Value, typeof(UnityEngine.Object), true);
                var key = EditorGUILayout.LongField(kvp.Key);

                EditorGUILayout.EndHorizontal();

                if (value != kvp.Value || key != kvp.Key)
                {
                    Undo.RecordObject(mgr, "Change Easy Save 3 References");
                    // If we're deleting a value, delete it.
                    if (value == null)
                        mgr.Remove(key);
                    // Else, update the ID.
                    else
                        mgr.ChangeId(kvp.Key, key);
                    // Break, as removing or changing Dictionary items will make the foreach out of sync.
                    break;
                }
            }

            EditorGUI.indentLevel--;
        }

        mgr.openPrefabs = EditorGUILayout.Foldout(mgr.openPrefabs, "ES3Prefabs");
        if (mgr.openPrefabs)
        {
            EditorGUI.indentLevel++;

            foreach (var prefab in mgr.prefabs)
            {
                EditorGUILayout.BeginHorizontal();

                EditorGUILayout.ObjectField(prefab, typeof(UnityEngine.Object), true);

                EditorGUILayout.EndHorizontal();
            }

            EditorGUI.indentLevel--;
        }

        var sp = serializedObject.FindProperty("excludeObjects");
        EditorGUILayout.PropertyField(sp);
        serializedObject.ApplyModifiedProperties();

        EditorGUILayout.LabelField("Reference count", mgr.refId.Count.ToString());
        EditorGUILayout.LabelField("Prefab count", mgr.prefabs.Count.ToString());

        if (GUILayout.Button("Refresh"))
        {
            mgr.RefreshDependencies();
        }

        if (GUILayout.Button("Optimize"))
        {
            mgr.Optimize();
        }
    }

    [MenuItem("GameObject/Easy Save 3/Add Dependencies to Manager", false, 33)]
    [MenuItem("Assets/Easy Save 3/Add Dependencies to Manager", false, 33)]
    public static void AddDependenciesToManager()
    {
        var mgr = ES3ReferenceMgr.GetManagerFromScene(SceneManager.GetActiveScene());
        if (mgr == null)
        {
            EditorUtility.DisplayDialog("Could not add reference to manager", "This object could not be added to the reference manager because no reference manager exists in this scene. To create one, go to Tools > Easy Save 3 > Add Manager to Scene", "Ok");
            return;
        }

        var selected = new HashSet<UnityEngine.Object>(Selection.GetFiltered<UnityEngine.Object>(SelectionMode.DeepAssets));
        selected.UnionWith(Selection.GetFiltered<UnityEngine.Object>(SelectionMode.TopLevel));

        if (selected == null || selected.Count == 0)
            return;

        Undo.RecordObject(mgr, "Update Easy Save 3 Reference Manager");

        foreach (var obj in selected)
        {
            if (obj == null)
                continue;

            if (obj.GetType() == typeof(GameObject))
            {
                var go = (GameObject)obj;
                if (ES3EditorUtility.IsPrefabInAssets(go) && go.GetComponent<ES3Internal.ES3Prefab>() != null)
                    mgr.AddPrefab(go.GetComponent<ES3Internal.ES3Prefab>());
            }

            ((ES3ReferenceMgr)mgr).AddDependencies(obj);
        }
    }

    [MenuItem("GameObject/Easy Save 3/Add Reference(s) to Manager", false, 33)]
    [MenuItem("Assets/Easy Save 3/Add Reference(s) to Manager", false, 33)]
    public static void AddReferencesToManager()
    {
        var mgr = ES3ReferenceMgr.Current;
        if (mgr == null)
        {
            EditorUtility.DisplayDialog("Could not add reference to manager", "This object could not be added to the reference manager because no reference manager exists in this scene. To create one, go to Tools > Easy Save 3 > Add Manager to Scene", "Ok");
            return;
        }

        var selected = Selection.GetFiltered<UnityEngine.Object>(SelectionMode.TopLevel);

        if (selected == null || selected.Length == 0)
            return;

        Undo.RecordObject(mgr, "Update Easy Save 3 Reference Manager");

        foreach (var obj in selected)
        {
            if (obj == null)
                continue;

            if (obj.GetType() == typeof(GameObject))
            {
                var go = (GameObject)obj;
                if (ES3EditorUtility.IsPrefabInAssets(go) && go.GetComponent<ES3Internal.ES3Prefab>() != null)
                    mgr.AddPrefab(go.GetComponent<ES3Internal.ES3Prefab>());
            }

            ((ES3ReferenceMgr)mgr).Add(obj);
        }
    }

    [MenuItem("GameObject/Easy Save 3/Add Reference(s) to Manager", true, 33)]
    [MenuItem("Assets/Easy Save 3/Add Reference(s) to Manager", true, 33)]
    [MenuItem("GameObject/Easy Save 3/Add Dependencies to Manager", true, 33)]
    [MenuItem("Assets/Easy Save 3/Add Dependencies to Manager", true, 33)]
    private static bool CanAddReferenceToManager()
    {
        var selected = Selection.GetFiltered<UnityEngine.Object>(SelectionMode.Deep);
        return selected != null && selected.Length > 0 && ES3ReferenceMgr.Current != null;
    }

    [MenuItem("GameObject/Easy Save 3/Exclude Reference(s) from Manager", false, 33)]
    [MenuItem("Assets/Easy Save 3/Exclude Reference(s) from Manager", false, 33)]
    public static void ExcludeReferenceFromManager()
    {
        var mgr = (ES3ReferenceMgr)ES3ReferenceMgr.Current;
        if (mgr == null)
        {
            EditorUtility.DisplayDialog("Could not exclude reference from manager", "This object could not be excluded from the reference manager because no reference manager exists in this scene. To create one, go to Tools > Easy Save 3 > Add Manager to Scene", "Ok");
            return;
        }

        var selected = Selection.GetFiltered<UnityEngine.Object>(SelectionMode.TopLevel);

        if (selected == null || selected.Length == 0)
            return;

        Undo.RecordObject(mgr, "Exclude from Easy Save 3 Reference Manager");

        foreach (var obj in selected)
        {
            if (obj == null)
                continue;

            mgr.ExcludeObject(obj);
        }

        mgr.RemoveNullOrInvalidValues();
    }

    [MenuItem("GameObject/Easy Save 3/Exclude Dependencies from Manager", false, 33)]
    [MenuItem("Assets/Easy Save 3/Exclude Dependencies from Manager", false, 33)]
    public static void ExcludeDependenciesFromManager()
    {
        var mgr = (ES3ReferenceMgr)ES3ReferenceMgr.Current;
        if (mgr == null)
        {
            EditorUtility.DisplayDialog("Could not exclude reference from manager", "This object could not be excluded from the reference manager because no reference manager exists in this scene. To create one, go to Tools > Easy Save 3 > Add Manager to Scene", "Ok");
            return;
        }

        var selected = Selection.GetFiltered<UnityEngine.Object>(SelectionMode.TopLevel);

        if (selected == null || selected.Length == 0)
            return;

        Undo.RecordObject(mgr, "Exclude from Easy Save 3 Reference Manager");

        var dependencies = EditorUtility.CollectDependencies(selected);

        foreach (var dependency in dependencies)
            mgr.ExcludeObject(dependency);

        mgr.RemoveNullOrInvalidValues();
    }

    [MenuItem("GameObject/Easy Save 3/Exclude Dependencies from Manager", true, 33)]
    [MenuItem("Assets/Easy Save 3/Exclude Dependencies from Manager", true, 33)]
    [MenuItem("GameObject/Easy Save 3/Exclude Reference(s) from Manager", true, 33)]
    [MenuItem("Assets/Easy Save 3/Exclude Reference(s) from Manager", true, 33)]
    private static bool CanExcludeReferencesFromManager()
    {
        var selected = Selection.GetFiltered<UnityEngine.Object>(SelectionMode.Deep);
        return selected != null && selected.Length > 0 && ES3ReferenceMgr.Current != null;
    }

    [MenuItem("GameObject/Easy Save 3/Add Manager to Scene", false, 33)]
    [MenuItem("Assets/Easy Save 3/Add Manager to Scene", false, 33)]
    [MenuItem("Tools/Easy Save 3/Add Manager to Scene", false, 150)]
    public static void EnableForScene()
    {
        if(!SceneManager.GetActiveScene().isLoaded)
            EditorUtility.DisplayDialog("Could not add manager to scene", "Could not add Easy Save 3 Manager to scene because there is not currently a scene open.", "Ok");
        Selection.activeObject = ES3Postprocessor.AddManagerToScene();
    }

    [MenuItem("GameObject/Easy Save 3/Add Manager to Scene", true, 33)]
    [MenuItem("Assets/Easy Save 3/Add Manager to Scene", true, 33)]
    [MenuItem("Tools/Easy Save 3/Add Manager to Scene", true, 150)]
    private static bool CanEnableForScene()
    {
        return ES3ReferenceMgr.GetManagerFromScene(SceneManager.GetActiveScene(), false) == null;
    }
}