using System;
using System.Threading.Tasks;
using TaskOperate;

namespace TaskOperateUnitTest
{
    /// <summary>
    /// 用于测试的具体任务实现类
    /// </summary>
    public class TestRunTask : RunTaskBase
    {
        private readonly int _delayTime;
        private readonly bool _shouldSucceed;
        private bool _isCancellationRequested;

        public TestRunTask(int delayTime = 100, bool shouldSucceed = true)
        {
            _delayTime = delayTime;
            _shouldSucceed = shouldSucceed;
            SetTimeout(1000); // 设置默认超时时间为1秒
        }

        /// <summary>
        /// 设置任务超时时间
        /// </summary>
        /// <param name="timeout">超时时间（毫秒）</param>
        public void SetTimeout(int timeout)
        {
            TaskTimeout = timeout;
        }

        /// <summary>
        /// 请求取消任务
        /// </summary>
        public void RequestCancel()
        {
            _isCancellationRequested = true;
        }

        public override async Task<TaskExecutionResult> StartTask(params object[] parameters)
        {
            InitCancelToken();
            try
            {
                for (int i = 0; i < _delayTime / 100; i++)
                {
                    if (_isCancellationRequested)
                    {
                        return TaskExecutionResult.Canceled("测试任务被取消");
                    }

                    await Task.Delay(100);

                    // 检查是否超时
                    if (i * 100 >= TaskTimeout)
                    {
                        return await HandleTimeoutAsync("测试任务", TaskTimeout);
                    }
                }

                if (!_shouldSucceed)
                {
                    throw new Exception("测试任务执行失败");
                }

                return TaskExecutionResult.Success("测试任务执行成功");
            }
            catch (Exception ex)
            {
                return TaskExecutionResult.Failed("测试任务执行失败", ex);
            }
        }

        protected override async Task<TaskExecutionResult> HandleTimeoutAsync(string taskName, int timeout)
        {
            return TaskExecutionResult.Timeout(
                $"测试任务 {taskName} 执行超时 (超时设置: {timeout}ms)",
                TimeSpan.FromMilliseconds(timeout));
        }
    }
}