CustomEditor

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*\
|                                                               |
|          Made by Jason Helman © Jason Tech And Games          |
|   I do not condone the use of my work without my permission   |
|                                                               |
\*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/

using UnityEditor;

[CustomEditor(typeof(BulletBehaviourObject), true)]
public class BulletBehaviourObjectCustomEditor : Editor
{
    private static readonly string[] variableNames = {"sprite", "size", "color", "particleTrail"};

    private BulletBehaviourObject previousBulletBehaviourObjectOverride;
    private bool previousUseOverrideState = false;

    private bool useOverrides = false;


    public override void OnInspectorGUI()
    {
        serializedObject.Update();

        BulletBehaviourObject bulletBehaviourObjectTarget = (BulletBehaviourObject) target;
        bulletBehaviourObjectTarget.Store(true);

        EditorGUILayout.PropertyField(serializedObject.FindProperty("useOtherBulletBehaviourObject"));

        if (bulletBehaviourObjectTarget.useOtherBulletBehaviourObject)
        {
            EditorGUILayout.PropertyField(serializedObject.FindProperty("bulletBehaviourObjectOverride"));
            useOverrides = EditorGUILayout.Toggle("Use Overrides", useOverrides);

            if (useOverrides)
            {
                for (int i = 0; i < variableNames.Length; i++)
                {
                    EditorGUILayout.PropertyField(serializedObject.FindProperty(variableNames[i]));
                }
            }
        }

        if (previousUseOverrideState && !bulletBehaviourObjectTarget.useOtherBulletBehaviourObject) // was Overriding -> not Overriding
            bulletBehaviourObjectTarget.Load();
        else if (!previousUseOverrideState && bulletBehaviourObjectTarget.useOtherBulletBehaviourObject) // wasn't Overriding -> Overriding
        {
            bulletBehaviourObjectTarget.Store();

            if (bulletBehaviourObjectTarget.bulletBehaviourObjectOverride != null)
            {
                bulletBehaviourObjectTarget.sprite = bulletBehaviourObjectTarget.bulletBehaviourObjectOverride.sprite;
                bulletBehaviourObjectTarget.size = bulletBehaviourObjectTarget.bulletBehaviourObjectOverride.size;
                bulletBehaviourObjectTarget.color = bulletBehaviourObjectTarget.bulletBehaviourObjectOverride.color;
                bulletBehaviourObjectTarget.particleTrail = bulletBehaviourObjectTarget.bulletBehaviourObjectOverride.particleTrail;
            }
            else
                bulletBehaviourObjectTarget.Load();

            serializedObject.ApplyModifiedProperties();
        }
        else if (bulletBehaviourObjectTarget.useOtherBulletBehaviourObject)
        {
            // issue is that if it is itself it needs to access the stored data
            // ^ fixed btw
            if (bulletBehaviourObjectTarget.bulletBehaviourObjectOverride != null && previousBulletBehaviourObjectOverride != bulletBehaviourObjectTarget.bulletBehaviourObjectOverride)
            {
                if (bulletBehaviourObjectTarget.bulletBehaviourObjectOverride == bulletBehaviourObjectTarget)
                    bulletBehaviourObjectTarget.Load();
                else
                {
                    bulletBehaviourObjectTarget.sprite = bulletBehaviourObjectTarget.bulletBehaviourObjectOverride.sprite;
                    bulletBehaviourObjectTarget.size = bulletBehaviourObjectTarget.bulletBehaviourObjectOverride.size;
                    bulletBehaviourObjectTarget.color = bulletBehaviourObjectTarget.bulletBehaviourObjectOverride.color;
                    bulletBehaviourObjectTarget.particleTrail = bulletBehaviourObjectTarget.bulletBehaviourObjectOverride.particleTrail;
                }

                serializedObject.ApplyModifiedProperties();
            }
            else if (bulletBehaviourObjectTarget.bulletBehaviourObjectOverride == null)
                bulletBehaviourObjectTarget.Load();
        }

        using (new EditorGUI.DisabledScope(bulletBehaviourObjectTarget.useOtherBulletBehaviourObject))
        {
            for (int i = 0; i < variableNames.Length; i++)
                EditorGUILayout.PropertyField(serializedObject.FindProperty(variableNames[i]));
        }

        previousBulletBehaviourObjectOverride = bulletBehaviourObjectTarget.bulletBehaviourObjectOverride;
        previousUseOverrideState = bulletBehaviourObjectTarget.useOtherBulletBehaviourObject;

        serializedObject.ApplyModifiedProperties();
    }
}