using System;
using System.Collections.Generic;
using UnityEngine;

    /// <summary>
    /// 延迟几秒执行事件
    /// </summary>
    public class WaitCallBack : MonoBehaviour
    {
        internal float waitTime = 2.0f;

        public bool doOnAwake = false;
        /// <summary>
        /// 时间到达回调
        /// </summary>
        public Action<Component> OnCompelete;
        /// <summary>
        /// 每帧回调
        /// </summary>
        public Action<Component> OnWait;

        internal static Dictionary<GameObject, WaitCallBack> waitDic = new Dictionary<GameObject, WaitCallBack>();

        private float m_Time = 0;

        private bool start = false;

        private Component m_Comp;

        private void Awake()
        {
            if (doOnAwake)
            {
                Play();
            }
        }

        private void OnDisable()
        {
            Stop();
        }

        private void Update()
        {
            if (start)
            {
                m_Time += Time.deltaTime;
                if (m_Time >= waitTime)
                {
                    Stop();
                    if (OnCompelete != null)
                        OnCompelete(m_Comp);
                }
                if (OnWait != null && start)
                    OnWait(m_Comp);
            }
        }

        internal void Play()
        {
            start = true;
        }

        internal void Restart()
        {
            m_Time = 0;
            Play();
        }

        internal void Stop()
        {
            m_Time = 0;
            start = false;
        }

        internal void Pause()
        {
            start = false;
        }

        private void OnDestroy()
        {
            if (waitDic.ContainsKey(gameObject))
            {
                waitDic.Remove(gameObject);
            }
        }

        internal void AddWaitDic(Component go)
        {
            m_Comp = go;
            waitDic.Add(go.gameObject, this);
        }
    }

    public static class Helper
    {
        public static void DoWaitStart(this Component go)
        {
            if (GetWait(go) == null)
                return;
            GetWait(go).Play();
        }

        public static void OnWaitCompelete(this Component go, Action<Component> func)
        {
            WaitCallBack wait = GetWait(go);
            if (wait == null)
                return;
            wait.OnCompelete = func;
        }

        public static void OnWait(this Component go, Action<Component> func)
        {
            WaitCallBack wait = GetWait(go);
            if (wait == null)
                return;
            wait.OnWait = func;
        }

        public static void DoWaitRestart(this Component go)
        {
            if (GetWait(go) == null)
                return;
            GetWait(go).Restart();
        }

        public static void DoWaitStop(this Component go)
        {
            if (GetWait(go) == null)
                return;
            GetWait(go).Stop();
        }

        public static void DoWaitPause(this Component go)
        {
            if (GetWait(go) == null)
                return;
            GetWait(go).Pause();
        }

        public static WaitCallBack GetWait(Component go)
        {
            WaitCallBack wait = null;
            if (go == null)
                return null;
            WaitCallBack.waitDic.TryGetValue(go.gameObject, out wait);
            if (wait == null)
            {
                wait = go.GetComponent<WaitCallBack>();
                if (wait == null)
                {
                    wait = go.gameObject.AddComponent<WaitCallBack>();
                }
                wait.AddWaitDic(go);
            }
            return wait;
        }

        public static void SetWait(this Component comp, float time)
        {
            WaitCallBack wait = GetWait(comp);
            if (wait != null)
            {
                wait.waitTime = time;
            }
        }
    }