using System;
using Cysharp.Threading.Tasks;
using GameFramework;
using GameFramework.DataTable;
using SepCore.DataTable;
using UnityGameFramework.Runtime;

namespace SepCore.AsyncTask
{
    /// <summary>
    /// DataTable 异步扩展方法
    /// </summary>
    public static class DataTableAsyncExtension
    {
        /// <summary>
        /// 异步加载数据表
        /// </summary>
        /// <param name="dataTableComponent">数据表组件</param>
        /// <param name="dataTableAssetName">数据表资源名称</param>
        /// <param name="userData">用户自定义数据</param>
        /// <param name="timeout">超时时间（秒），0表示不超时</param>
        /// <returns>加载成功事件</returns>
        public static UniTask<LoadDataTableSuccessEventArgs> LoadDataTableAsync(this DataTableComponent dataTableComponent,
            string dataTableAssetName,
            object userData = null,
            float timeout = 30f)
        {
            UniTask<LoadDataTableSuccessEventArgs> waitTask = AsyncTaskHelper.WaitSuccessOrFailureAsync<LoadDataTableSuccessEventArgs, LoadDataTableFailureEventArgs>(
                LoadDataTableSuccessEventArgs.EventId,
                LoadDataTableFailureEventArgs.EventId,
                successArgs => successArgs.DataTableAssetName == dataTableAssetName && ReferenceEquals(successArgs.UserData, userData),
                failureArgs => failureArgs.DataTableAssetName == dataTableAssetName && ReferenceEquals(failureArgs.UserData, userData),
                timeout
            );

            string dataTableName = System.IO.Path.GetFileNameWithoutExtension(dataTableAssetName);
            dataTableComponent.LoadDataTable(dataTableName, dataTableAssetName, userData);
            return waitTask;
        }

        /// <summary>
        /// 异步加载数据表（泛型版本）
        /// </summary>
        /// <typeparam name="T">数据行类型</typeparam>
        /// <param name="dataTableComponent">数据表组件</param>
        /// <param name="dataTableAssetName">数据表资源名称</param>
        /// <param name="userData">用户自定义数据</param>
        /// <param name="timeout">超时时间（秒），0表示不超时</param>
        /// <returns>加载的数据表</returns>
        public static UniTask<IDataTable<T>> LoadDataTableAsync<T>(this DataTableComponent dataTableComponent,
            string dataTableAssetName,
            object userData = null,
            float timeout = 30f) where T : IDataRow
        {
            return LoadDataTableAsync(dataTableComponent, dataTableAssetName, userData, timeout)
                .ContinueWith(successArgs =>
                {
                    // 获取数据表名称（去掉路径和扩展名）
                    string dataTableName = System.IO.Path.GetFileNameWithoutExtension(dataTableAssetName);
                    return dataTableComponent.GetDataTable<T>(dataTableName);
                });
        }
    }
}