LowPolyBattleSim/Assets/Plugins/RaycastPro/Core/Casters/GunCaster.cs

using RaycastPro.Detectors;

namespace RaycastPro
{
    using UnityEngine;
    using System;
    using Bullets;
    using Bullets2D;
    using RaySensors;
    using RaySensors2D;
    using Casters;
    using Casters2D;
    using System.Collections;
    using UnityEngine.Events;
    
#if UNITY_EDITOR
    using UnityEditor;
    using Editor;
#endif

    [Serializable]
    public class BulletEvent : UnityEvent<Bullet> { }
    
    [Serializable]
    public class BulletEvent2D : UnityEvent<Bullet2D> { }
    public abstract class GunCaster<B, C, R> : BaseCaster where B : BaseBullet where C : UnityEngine.Object
    {
        public B[] bullets = Array.Empty<B>();
        public B[] cloneBullets = Array.Empty<B>();
        public C[] ignoreColliders = Array.Empty<C>();
        public Ammo ammo;

        public bool autoShoot = true;


        [Tooltip("This option is exclusive to Tracker Bullet, you need to set the target before shooting and it will be automatically inserted into the bullet.")]
        public Transform trackTarget;

        public int bulletIndex;

        [SerializeField] protected int cloneIndex;
        
        [Tooltip("Using Ammo quickly provides a classic weapon system that you will benefit from the amount of bullets, magazine capacity and reload speed, if it is necessary for your weapon to have a custom form, you can write a separate code and use the Cast() method.")]
        public bool usingAmmo = true;

        [SerializeField] private bool arrayCasting;
        [SerializeField] protected int arrayCapacity = 6;
        
        /// <summary>
        /// Performed in Caster only works with Array Casting
        /// </summary>
        public override bool Performed {
            get
            {
                var any = false;
                foreach (var cloneBullet in cloneBullets)
                {
                    if (cloneBullet && cloneBullet.isActiveAndEnabled) any = true;
                }
                return any;
            }
            protected set {} }
        public enum CastType
        {
            /// <summary>
            /// 0
            /// </summary>
            Together,
            /// <summary>
            /// 1
            /// </summary>
            Sequence,
            /// <summary>
            /// 2
            /// </summary>
            Random,
            /// <summary>
            /// 3
            /// </summary>
            PingPong
        }

        private B tempBullet;
        
        public UnityEvent onCast;

        protected override void OnCast()
        {
            if (autoShoot)
            {
                Cast(bulletIndex);
                onCast?.Invoke();
            }
        }

        /// <summary>
        /// Auto setup new Index or Replace Bullets by ID checking
        /// </summary>
        /// <param name="bulletObject"></param>
        private void OnArrayCast(B bulletObject, Vector3 _base, Vector3 _dir)
        {
            if (cloneBullets[cloneIndex])
            {
                // Replacing
                if (bulletObject.bulletID != cloneBullets[cloneIndex].bulletID)
                {
                    Destroy(cloneBullets[cloneIndex].gameObject);
                    tempBullet = Instantiate(bulletObject, poolManager);
                    // Set Clone Index to this bullet
                    cloneBullets[cloneIndex] = tempBullet;
                    BulletSetup(tempBullet);
                }
                else // Refreshing
                {
                    tempBullet = cloneBullets[cloneIndex];
                    Refresh_ArrayCasting_Clone(tempBullet);
                }
            }
            else
            {
                tempBullet = Instantiate(bulletObject, poolManager);
                // Set Clone Index to this bullet
                cloneBullets[cloneIndex] = tempBullet;
                BulletSetup(tempBullet);
            }
            
            tempBullet.transform.position = _base;
            tempBullet.transform.forward = _dir;
            
            // Revive Again
            tempBullet.ended = false;
            // Go To Next Clone
            cloneIndex = ++cloneIndex % cloneBullets.Length;
            // End function must be to "Disable"
            tempBullet.endFunction = BaseBullet.EndType.Disable;
        }

        /// <summary>
        /// Just Instantiate Bullets
        /// </summary>
        /// <param name="bulletObject"></param>
        private void OnNormalCast(B bulletObject, Vector3 _base, Vector3 _dir)
        {
            tempBullet = Instantiate(bulletObject, poolManager);
            tempBullet.endFunction = BaseBullet.EndType.Destroy;
            tempBullet.transform.position = _base;
            tempBullet.transform.forward = _dir;
            BulletSetup(tempBullet);
        }
        
        // ReSharper disable Unity.PerformanceAnalysis
        private void Refresh_ArrayCasting_Clone(B _bullet)
        {
            _bullet.onEndCast?.Invoke(this);
            _bullet.gameObject.SetActive(true);
            _bullet.position = 0;
            _bullet.Life = 0;
            BulletSetup(_bullet);
        }
        private B _bulletObject;

        #region Public Methods
        
        /// <summary>
        /// Safe Reload
        /// </summary>
        public override void Reload()
        {
            StartCoroutine(ammo.IReload());
            onReload?.Invoke();
        }
        
        public void SetBulletIndex(int _i) => bulletIndex = _i;

        public void RandomBulletIndex() => bulletIndex = UnityEngine.Random.Range(0, bullets.Length);

        public void NextBulletIndex() => bulletIndex = (bulletIndex + 1) % bullets.Length;
        
        public void BackBulletIndex()
        {
            bulletIndex = (bulletIndex - 1);

            if (bulletIndex == 0)
            {
                bulletIndex = bullets.Length - 1;
            }
        }

        public void SetTrackerTarget(Transform target) => trackTarget = target;

        /// <summary>r
        /// Sync TrackTarget to nearest detector Collider
        /// </summary>
        /// <param name="colliderDetector"></param>
        public void SyncTrackTarget(ColliderDetector colliderDetector)
        {
            colliderDetector.GetNearestCollider(out var col);
            trackTarget = col.transform;
        }

        public bool AmmoCheck(int count = 1) => !usingAmmo || ammo.Use(this, count);

        #endregion
        
        private void OnEnable()
        {
            if (ammo == null) return;
            
            if (autoReload && ammo.magazineAmount == 0 && !ammo.IsReloading) Reload();
        }

        private void OnDisable()
        {
            if (ammo == null) return;

            ammo.inRate = false;
            ammo.inReload = false;
        }

        /// <summary>
        /// Switchable Cast Between Array and Normal. 100% be sure to Define. _dir and _base before.
        /// </summary>
        protected void AutoCast(Vector3 _base, Vector3 _dir)
        {
            if (arrayCasting) OnArrayCast(_bulletObject, _base, _dir);
            else OnNormalCast(_bulletObject, _base, _dir);
            _bulletObject.ownerReference = gameObject;
        }
        
        protected bool BulletCast(int _index, R _raySensor = default)
        {
#if UNITY_EDITOR
            alphaCharge = AlphaLifeTime;
#endif
            _bulletObject = bullets[_index];
            
            if (_raySensor is RaySensor r)
            {
                AutoCast(r.Base, r.TipDirection);
            }
            else if (_raySensor is RaySensor2D r2D)
            {
                AutoCast(r2D.Base, r2D.TipDirection);
            }

            
            tempBullet.Cast(this, _raySensor);
            return true;
        }

        protected bool BulletCast(int _index, ColliderDetector _cDetector)
        {
#if UNITY_EDITOR
            alphaCharge = AlphaLifeTime;
#endif
            _bulletObject = bullets[_index];
            
            foreach (var detectedTarget in _cDetector.DetectedColliders)
            {
                if (_bulletObject is Bullet _B)
                {
                    AutoCast(transform.position, (detectedTarget.transform.position - transform.position).normalized);
                    _B.caster = this;
                    _B.DirectCast();
                }

            }
            
            return true;
        }
        
        // ReSharper disable Unity.PerformanceAnalysis
        private void BulletSetup(B _bullet)
        {
            _bullet.enabled = true;
            _bullet.ended = false;
            if (ignoreColliders.Length > 0)
            {
                if (typeof(C) == typeof(Collider))
                {
                    var bulletColliders = _bullet.GetComponentsInChildren<Collider>();
                    IgnoreCollider(bulletColliders, Array.ConvertAll(ignoreColliders, i => i as Collider));
                }
                else if (typeof(C) == typeof(Collider2D))
                {
                    var bulletColliders = _bullet.GetComponentsInChildren<Collider2D>();
                    IgnoreCollider(bulletColliders, Array.ConvertAll(ignoreColliders, i => i as Collider2D));
                }
            }
                
            // Add Track Target 
            if (trackTarget)
            {
                if (_bullet is TrackerBullet _trB)
                {
                    _trB.target = trackTarget;
                }
                else if (_bullet is TrackerBullet2D _trB2D)
                {
                    _trB2D.target = trackTarget;
                }
            }
        }
        
        protected Coroutine multiCast;
        protected IEnumerator IMultiCast(int _index, int count)
        {
            if (ammo.magazineAmount == 0) ammo.IReload();
            for (int i = 0; i < count; i++)
            {
                while (ammo.IsInRate)
                {
                    yield return new WaitForEndOfFrame();
                    if (ammo.IsReloading) // Stops when magazine Off
                    {
                        break;
                    }
                }
                Cast(_index);
            }
        }
        public void MultipleCast(int _index, int count)
        {
            if (multiCast != null) StopCoroutine(multiCast);
            
            multiCast = StartCoroutine(IMultiCast(_index, count));
        }

        public void BulletEnable(Bullet bullet) => bullet.enabled = true;
        public void BulletEnable(Bullet2D bullet) => bullet.enabled = true;
        public void BulletActive(Bullet bullet) => bullet.gameObject.SetActive(true);
        public void BulletActive(Bullet2D bullet) => bullet.gameObject.SetActive(true);
#if UNITY_EDITOR


        protected const string CTogether = "Together";
        protected const string CSequence = "Sequence"; 
        protected const string CRandom = "Random"; 
        protected const string CPingPong = "PingPong";
        
        protected const string TTogether = "Shooting the index bullet in all RaySensors Together.";
        protected const string TSequence = "Shooting in sequence of RaySensors index in wait of ammo reload Time and In Between."; 
        protected const string TRandom = "Random Shooting in any RaySensor."; 
        protected const string TPingPong = "Shooting bullets back and forth on the RaySensor index.";
        /// <summary>
        /// This Script worked but No using for now
        /// </summary>
        /// <param name="property"></param>
        protected void BulletField(SerializedProperty property)
        {
            var rect = EditorGUILayout.GetControlRect();
            EditorGUI.BeginProperty(rect, property.displayName.ToContent(), property);
            EditorGUI.BeginChangeCheck();
            var newBullet = EditorGUI.ObjectField(rect, property.objectReferenceValue, typeof(Bullet), false);
            if (EditorGUI.EndChangeCheck())
            {
                property.objectReferenceValue = newBullet;
            }
            EditorGUI.EndProperty();
        }

        protected void GunField(SerializedObject _so)
        {
            EditorGUILayout.PropertyField(_so.FindProperty(nameof(autoShoot)));
            #region ArrayCastingField
            BeginHorizontal();

            GUI.enabled = !Application.isPlaying;
            var arrCasting = _so.FindProperty(nameof(arrayCasting));
            var arrCapacity = _so.FindProperty(nameof(arrayCapacity));
            EditorGUI.BeginChangeCheck();
            EditorGUILayout.PropertyField(arrCasting,
                (arrCasting.boolValue ? "Array Capacity" : CArrayCasting).ToContent(TArrayCasting));
            GUI.enabled &= arrCasting.boolValue;
            PropertyMaxIntField(arrCapacity, GUIContent.none, 1);
            if (EditorGUI.EndChangeCheck())
            {
                if (arrCasting.boolValue)
                {
                    arrCapacity.intValue = Mathf.Max(1, arrCapacity.intValue);
                    _so.FindProperty(nameof(cloneBullets)).arraySize = arrCapacity.intValue;
                }
                else cloneBullets = null;
            }
            
            GUI.enabled = true;
            EndHorizontal();
            #endregion
            BeginVerticalBox();
            
            var bulletsProp = _so.FindProperty(nameof(bullets));
            RCProEditor.PropertyArrayField(bulletsProp,
                "Bullets".ToContent(), i => $"Bullet {i+1}".ToContent($"Index {i}"));
            PropertySliderField(_so.FindProperty(nameof(bulletIndex)), 0,  bullets != null ? Mathf.Max(bulletsProp.arraySize-1,0) : 0 , "Index".ToContent(),
                I => { });
            EndVertical();

            EditorGUILayout.PropertyField(_so.FindProperty(nameof(trackTarget)));
            
            BeginVerticalBox();
            EditorGUILayout.PropertyField(_so.FindProperty(nameof(usingAmmo)));
            if (usingAmmo)
            {
                if (ammo == null) ammo = new Ammo();
                var ammoProp = _so.FindProperty(nameof(ammo));
                ammo?.EditorPanel(ammoProp);
                
                BeginHorizontal();
                if (GUILayout.Button("Cast"))
                {
                    Cast(bulletIndex);
                }
                if (GUILayout.Button("Reload"))
                {
                    Reload();
                }
                ProgressField(ammo.currentReloadTime/ammo.reloadTime, $"{ammo.currentReloadTime:F1}/{ammo.reloadTime:F1} Sec");

                EditorUtility.SetDirty(this);

                
                EndHorizontal();
                
                EditorGUILayout.PropertyField(_so.FindProperty(nameof(autoReload)));
            }
            else ammo = null;
            EndVertical();
            EditorGUILayout.PropertyField(_so.FindProperty(nameof(poolManager)));
        }
        protected void GeneralField(SerializedObject _so)
        {
            BaseField(_so, hasInteraction: false);
            
            BeginVerticalBox();
            RCProEditor.PropertyArrayField(_so.FindProperty(nameof(ignoreColliders)), "Ignore List".ToContent(),
                (i) => $"Collider {i+1}".ToContent($"Index {i}"));
            EndVertical();
        }
        protected void InformationField()
        {
            if (!IsPlaying || cloneBullets == null) return;
            
            InformationField(() =>
            {
                for (var i = 0; i < cloneBullets.Length; i++)
                {
                    if (!cloneBullets[i]) continue;

                    var cloneBullet = cloneBullets[i];
                    BeginHorizontal();
                    EditorGUILayout.LabelField($"{i} {cloneBullet.GetInstanceID().ToString()}",
                        GUILayout.Width(200));
                    PercentProgressField(cloneBullet.life/cloneBullet.lifeTime, "Life");
                    EndHorizontal();
                }
            });
        }
#endif
    }
}