DelayOneFrame.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;
using UnityEngine.Events;

namespace KongZEE.Timer.Delay
{
    public class DelayOneFrame
    {
        private MonoBehaviour monoBehaviour;
        Dictionary<OneFrameKey, Coroutine>  coroutines = new ();
        private OneFrameKey delayKeyDisposable;
        
        public DelayOneFrame(MonoBehaviour monoBehaviour)
        {
            this.monoBehaviour = monoBehaviour;
        }

        public void StartDelayDisposable(UnityAction<bool> action = null)
        {
            if (coroutines.Count > 0 && coroutines.ContainsKey(delayKeyDisposable))
            {
                monoBehaviour.StopCoroutine(coroutines[delayKeyDisposable]);
                coroutines.Remove(delayKeyDisposable);
            }            
            
            delayKeyDisposable = StartDelay(action);
        }

        public OneFrameKey StartDelay(UnityAction<bool> action = null)
        {
            OneFrameKey delayKey = new OneFrameKey();
            
            if (action != null)
            {
                action += isNextFrame => { delayKey.nextFrame = isNextFrame; };
            }
            
            Coroutine coroutine = monoBehaviour.StartCoroutine(Timer(action));
            coroutines.Add(delayKey, coroutine);
            delayKey.action = action;

            return delayKey;
        }

        public void StartDelay(OneFrameKey delayKey = null)
        {
            if (coroutines.Count < 0) return;
            
            if (delayKey == null)
            {
                if (coroutines.Count > 0)
                {
                    monoBehaviour.StopCoroutine(coroutines.Values.ElementAt(0));
                    coroutines.Remove(coroutines.Keys.ElementAt(0));
                }
            }else if (coroutines.ContainsKey(delayKey))
            {
                monoBehaviour.StopCoroutine(coroutines[delayKey]);
                coroutines.Remove(delayKey);
            }
            
            OneFrameKey currentDelayKey = delayKey ?? coroutines.Keys.ElementAt(0);
            currentDelayKey.nextFrame = false;
            Coroutine coroutine = monoBehaviour.StartCoroutine(Timer(currentDelayKey.action));
            coroutines.Add(currentDelayKey, coroutine);
        }
        
        private IEnumerator Timer(UnityAction<bool> action)
        {
            Debug.Log($"Old Time: {Time.deltaTime}");
            action?.Invoke(false);
            yield return null;
            action?.Invoke(true);
            Debug.Log($"New Time: {Time.deltaTime}");
            
        }
    }

    [Serializable]
    public class OneFrameKey
    {
        public bool nextFrame;
        public UnityAction<bool> action;
        
        public OneFrameKey()
        {
            
        }
    }
}