using UnityEditor;
using UnityEngine;

namespace CreatureControl {

    [CustomEditor(typeof(Ant))]
    public class Ant_Editor : Insect_Editor {

        protected Ant ant;

        public override void OnEnable() {
            this.ant = target as Ant;

            base.OnEnable();
        }

        public override void OnInspectorGUI() {
            base.OnInspectorGUI();

            HomePheromonePrefabInspector();
            FoodPheromonePrefabInspector();

            EditorGUILayout.PropertyField(serializedObject.FindProperty(nameof(Ant.targetDirection)));
            EditorGUILayout.PropertyField(serializedObject.FindProperty(nameof(Ant.foodReceptor)));
            EditorGUILayout.PropertyField(serializedObject.FindProperty(nameof(Ant.homeReceptor)));
            EditorGUILayout.PropertyField(serializedObject.FindProperty(nameof(Ant.hasFood)));
        }

        private void HomePheromonePrefabInspector() {
            SerializedProperty homePheromonePrefabProp = serializedObject.FindProperty(nameof(Ant.homePheromonePrefab));
            homePheromonePrefabProp.objectReferenceValue = (GameObject)EditorGUILayout.ObjectField("Home Pheromone Prefab", homePheromonePrefabProp.objectReferenceValue, typeof(GameObject), true);
        }

        private void FoodPheromonePrefabInspector() {
            SerializedProperty foodPheromonePrefabProp = serializedObject.FindProperty(nameof(Ant.foodPheromonePrefab));
            foodPheromonePrefabProp.objectReferenceValue = (GameObject)EditorGUILayout.ObjectField("Food Pheromone Prefab", foodPheromonePrefabProp.objectReferenceValue, typeof(GameObject), true);
        }
    }

}