﻿#if UNITY_EDITOR
using UnityEditor;
using UnityEngine;
using static FishNet.Component.Prediction.PredictedObject;

namespace FishNet.Component.Prediction
{


    [CustomEditor(typeof(PredictedObject), true)]
    [CanEditMultipleObjects]
    public class PredictedObjectEditor : Editor
    {
#if !PREDICTION_V2

        private SerializedProperty _implementsPredictionMethods;
        private SerializedProperty _graphicalObject;
        private SerializedProperty _ownerSmoothPosition;
        private SerializedProperty _ownerSmoothRotation;
        private SerializedProperty _ownerInterpolation;
        private SerializedProperty _enableTeleport;
        private SerializedProperty _teleportThreshold;
        private SerializedProperty _predictionType;

        private SerializedProperty _rigidbody;
        private SerializedProperty _rigidbody2d;
        private SerializedProperty _spectatorSmoothPosition;
        private SerializedProperty _spectatorSmoothRotation;
        private SerializedProperty _spectatorSmoothingType;
        private SerializedProperty _customSmoothingData;
        private SerializedProperty _preconfiguredSmoothingDataPreview;
        private SerializedProperty _maintainedVelocity;
        private SerializedProperty _resendType;
        private SerializedProperty _resendInterval;

        private SerializedProperty _networkTransform;

        protected virtual void OnEnable()
        {
            _implementsPredictionMethods = serializedObject.FindProperty(nameof(_implementsPredictionMethods));
            _graphicalObject = serializedObject.FindProperty(nameof(_graphicalObject));
            _ownerSmoothPosition = serializedObject.FindProperty(nameof(_ownerSmoothPosition));
            _ownerSmoothRotation = serializedObject.FindProperty(nameof(_ownerSmoothRotation));
            _ownerInterpolation = serializedObject.FindProperty(nameof(_ownerInterpolation));
            _enableTeleport = serializedObject.FindProperty(nameof(_enableTeleport));
            _teleportThreshold = serializedObject.FindProperty(nameof(_teleportThreshold));
            _predictionType = serializedObject.FindProperty(nameof(_predictionType));

            _rigidbody = serializedObject.FindProperty(nameof(_rigidbody));
            _rigidbody2d = serializedObject.FindProperty(nameof(_rigidbody2d));
            _spectatorSmoothPosition = serializedObject.FindProperty(nameof(_spectatorSmoothPosition));
            _spectatorSmoothRotation = serializedObject.FindProperty(nameof(_spectatorSmoothRotation));
            _spectatorSmoothingType = serializedObject.FindProperty(nameof(_spectatorSmoothingType));
            _customSmoothingData = serializedObject.FindProperty(nameof(_customSmoothingData));
            _preconfiguredSmoothingDataPreview = serializedObject.FindProperty(nameof(_preconfiguredSmoothingDataPreview));

            _maintainedVelocity = serializedObject.FindProperty(nameof(_maintainedVelocity));
            _resendType = serializedObject.FindProperty(nameof(_resendType));
            _resendInterval = serializedObject.FindProperty(nameof(_resendInterval));

            _networkTransform = serializedObject.FindProperty(nameof(_networkTransform));

        }

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

            GUI.enabled = false;
            EditorGUILayout.ObjectField("Script:", MonoScript.FromMonoBehaviour((PredictedObject)target), typeof(PredictedObject), false);
            GUI.enabled = true;

            EditorGUILayout.PropertyField(_implementsPredictionMethods);
            EditorGUILayout.PropertyField(_graphicalObject);
            EditorGUILayout.PropertyField(_enableTeleport);
            if (_enableTeleport.boolValue)
            {
                EditorGUI.indentLevel++;
                EditorGUILayout.PropertyField(_teleportThreshold);
                EditorGUI.indentLevel--;
            }

            EditorGUILayout.LabelField("Owner Settings");
            EditorGUI.indentLevel++;
            EditorGUILayout.PropertyField(_ownerSmoothPosition, new GUIContent("Smooth Position"));
            EditorGUILayout.PropertyField(_ownerSmoothRotation, new GUIContent("Smooth Rotation"));
            EditorGUILayout.PropertyField(_ownerInterpolation, new GUIContent("Interpolation"));
            EditorGUI.indentLevel--;

            EditorGUILayout.PropertyField(_predictionType);
            PredictedObject.PredictionType movementType = (PredictedObject.PredictionType)_predictionType.intValue;
            if (movementType != PredictedObject.PredictionType.Other)
            {
                EditorGUI.indentLevel++;
                EditorGUILayout.HelpBox("When using physics prediction do not include a NetworkTransform; this component will synchronize instead.", MessageType.Info);
                if (movementType == PredictedObject.PredictionType.Rigidbody)
                    EditorGUILayout.PropertyField(_rigidbody);
                else
                    EditorGUILayout.PropertyField(_rigidbody2d, new GUIContent("Rigidbody2D", "Rigidbody2D to predict."));

                EditorGUILayout.LabelField("Spectator Settings");
                EditorGUI.indentLevel++;
                EditorGUILayout.LabelField("Smoothing");
                EditorGUI.indentLevel++;
                EditorGUILayout.PropertyField(_spectatorSmoothPosition, new GUIContent("Smooth Position"));
                EditorGUILayout.PropertyField(_spectatorSmoothRotation, new GUIContent("Smooth Rotation"));
                EditorGUILayout.PropertyField(_spectatorSmoothingType, new GUIContent("Smoothing Type"));
                //Custom.
                if ((SpectatorSmoothingType)_spectatorSmoothingType.intValue == SpectatorSmoothingType.Custom)
                {
                    EditorGUI.indentLevel++;
                    EditorGUILayout.PropertyField(_customSmoothingData);
                    EditorGUI.indentLevel--;
                }
                //Preconfigured.
                else
                {
                    EditorGUI.indentLevel++;
                    GUI.enabled = false;
                    EditorGUILayout.PropertyField(_preconfiguredSmoothingDataPreview);
                    GUI.enabled = true;
                    EditorGUI.indentLevel--;
                }
                EditorGUI.indentLevel--;
                EditorGUILayout.PropertyField(_maintainedVelocity);

                EditorGUILayout.PropertyField(_resendType);
                PredictedObject.ResendType resendType = (PredictedObject.ResendType)_resendType.intValue;
                if (resendType == PredictedObject.ResendType.Interval)
                {
                    EditorGUI.indentLevel++;
                    EditorGUILayout.PropertyField(_resendInterval, new GUIContent("Interval"));
                    EditorGUI.indentLevel--;
                }
                EditorGUI.indentLevel--;

                EditorGUI.indentLevel--;
            }
            else
            {
                EditorGUI.indentLevel++;
                EditorGUILayout.HelpBox("When other is selected another component, such as NetworkTransform, must be used to synchronize.", MessageType.Info);
                EditorGUILayout.PropertyField(_networkTransform);
                EditorGUI.indentLevel--;
            }

            EditorGUILayout.Space();
            serializedObject.ApplyModifiedProperties();
        }
#endif
    }

}
#endif

