using AyCode.Core.Helpers;
using AyCode.Database.DataLayers;
using AyCode.Database.DbContexts;
using AyCode.Interfaces.Entities;

namespace AyCode.Database.Extensions;

public static class AcDalExtension
{
    public static Task<TResultType> SessionAsync<TDbContext, TResultType>(this IAcDalBase<TDbContext> acDal, Func<TDbContext, TResultType> callback) where TDbContext : AcDbContextBase
        => TaskHelper.ToThreadPoolTask(() => acDal.Session(callback));

    public static Task<IEnumerable<TEntity>> SessionAsync<TDbContext, TEntity>(this IAcDalBase<TDbContext> acDal, Func<TDbContext, IQueryable<TEntity>> callback) where TEntity : IEntity where TDbContext : AcDbContextBase
        => TaskHelper.ToThreadPoolTask(() => acDal.Session(callback));

    public static TResultType Session<TDbContext, TResultType>(this IAcDalBase<TDbContext> acDal, Func<TDbContext, TResultType> callback) where TDbContext : AcDbContextBase
    {
        using var ctx = acDal.CreateDbContext();
        
        return ctx.Session(callback);
    }

    public static IEnumerable<TEntity> Session<TDbContext, TEntity>(this IAcDalBase<TDbContext> acDal, Func<TDbContext, IQueryable<TEntity>> callback) where TEntity : IEntity where TDbContext : AcDbContextBase
    {
        using var ctx = acDal.CreateDbContext();
        
        return ctx.Session(callback);
    }

    public static Task<bool> TransactionAsync<TDbContext>(this IAcDalBase<TDbContext> acDal, Func<TDbContext, bool> callbackTransactionBody, bool throwException = false) where TDbContext : AcDbContextBase
        => TaskHelper.ToThreadPoolTask(() => acDal.Transaction(callbackTransactionBody, throwException));

    public static bool Transaction<TDbContext>(this IAcDalBase<TDbContext> acDal, Func<TDbContext, bool> callbackTransactionBody, bool throwException = false) where TDbContext : AcDbContextBase
    {
        using var ctx = acDal.CreateDbContext();
        
        return ctx.Transaction(callbackTransactionBody, throwException);
    }
}