using System;
using System.Collections.Generic;

namespace ET
{
    public class ETCancellationToken
    {
        private HashSet<Action> actions = new HashSet<Action>();

        public void Add(Action callback)
        {
            // 如果action是null，绝对不能添加,要抛异常，说明有协程泄漏
            this.actions.Add(callback);
        }
        
        public void Remove(Action callback)
        {
            this.actions?.Remove(callback);
        }

        public void Cancel()
        {
            if (this.actions == null)
            {
                return;
            }
            
            if (this.actions.Count == 0)
            {
                return;
            }

            this.Invoke();
        }

        private void Invoke()
        {
            HashSet<Action> runActions = this.actions;
            this.actions = null;
            try
            {
                foreach (Action action in runActions)
                {
                    action.Invoke();
                }
            }
            catch (Exception e)
            {
                Log.Error(e);
            }
        }

        public async ETVoid CancelAfter(long afterTimeCancel)
        {
            if (this.actions == null)
            {
                return;
            }

            if (this.actions.Count == 0)
            {
                return;
            }

            await TimerComponent.Instance.WaitAsync(afterTimeCancel);
            
            this.Invoke();
        }
    }
}