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

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

namespace ES3Editor
{
	public class AddES3Prefab : UnityEditor.Editor 
	{
        [MenuItem("GameObject/Easy Save 3/Enable Easy Save for Prefab(s)", false, 1001)]
        [MenuItem("Assets/Easy Save 3/Enable Easy Save for Prefab(s)", false, 1001)]
        public static void Enable()
        {
            if (Selection.gameObjects == null || Selection.gameObjects.Length == 0)
                return;

            foreach (var obj in Selection.gameObjects)
            {
                // Don't add the Component to a GameObject which already has it.
                if (obj == null  || (obj.GetComponent<ES3Prefab>() != null))
                    continue;

                var go = obj;

                #if UNITY_2018_3_OR_NEWER
                if (PrefabUtility.GetPrefabInstanceStatus(go) != PrefabInstanceStatus.NotAPrefab)
                {
                    go = (GameObject)PrefabUtility.GetCorrespondingObjectFromSource(go);
                    if (go == null)
                        continue;
                }
                #else
			    if(PrefabUtility.GetPrefabType(go) != PrefabType.Prefab)
			    {
				    go = (GameObject)PrefabUtility.GetPrefabParent(go);
				    if(go == null)
					    continue;
			    }
                #endif

                var es3Prefab = Undo.AddComponent<ES3Prefab>(go);
                es3Prefab.GeneratePrefabReferences();

                var mgr = ES3ReferenceMgr.GetManagerFromScene(SceneManager.GetActiveScene());
                if (mgr != null)
                {
                    mgr.AddPrefab(es3Prefab);
                    EditorUtility.SetDirty(mgr);
                }
            }
		}

		[MenuItem("GameObject/Easy Save 3/Enable Easy Save for Prefab(s)", true, 1001)]
		[MenuItem("Assets/Easy Save 3/Enable Easy Save for Prefab(s)", true, 1001)]
		public static bool Validate()
		{
            return Selection.gameObjects != null && Selection.gameObjects.Length > 0;
		}
	}

    public class RemoveES3Prefab : UnityEditor.Editor 
    {
        [MenuItem("GameObject/Easy Save 3/Disable Easy Save for Prefab(s)", false, 1001)]
        [MenuItem("Assets/Easy Save 3/Disable Easy Save for Prefab(s)", false, 1001)]
        public static void Enable()
        {
            if (Selection.gameObjects == null || Selection.gameObjects.Length == 0)
                return;

            foreach (var obj in Selection.gameObjects)
            {
                var es3prefab = obj.GetComponent<ES3Prefab>();
                if (es3prefab != null)
                    Undo.DestroyObjectImmediate(es3prefab);
            }
        }

        [MenuItem("GameObject/Easy Save 3/Disable Easy Save for Prefab(s)", true, 1001)]
        [MenuItem("Assets/Easy Save 3/Disable Easy Save for Prefab(s)", true, 1001)]
        public static bool Validate()
        {
            return Selection.gameObjects != null && Selection.gameObjects.Length > 0;
        }
    }
}