using DigitalData.Core.Abstraction.Application.Repository;
using Microsoft.EntityFrameworkCore;
using System.Linq.Expressions;

namespace DigitalData.Core.Infrastructure;

public sealed record ReadQuery<TEntity> : IReadQuery<TEntity>
{
    private IQueryable<TEntity> _query;

    internal ReadQuery(IQueryable<TEntity> queryable)
    {
        _query = queryable;
    }

    public TEntity First() => _query.First();

    public Task<TEntity> FirstAsync(CancellationToken cancellation = default) => _query.FirstAsync(cancellation);

    public TEntity? FirstOrDefault() => _query.FirstOrDefault();


    public Task<TEntity?> FirstOrDefaultAsync(CancellationToken cancellation = default) => _query.FirstOrDefaultAsync(cancellation);


    public TEntity Single() => _query.Single();


    public Task<TEntity> SingleAsync(CancellationToken cancellation = default) => _query.SingleAsync(cancellation);


    public TEntity? SingleOrDefault() => _query.SingleOrDefault();


    public Task<TEntity?> SingleOrDefaultAsync(CancellationToken cancellation = default) => _query.SingleOrDefaultAsync(cancellation);


    public IEnumerable<TEntity> ToList() => _query.ToList();


    public async Task<IEnumerable<TEntity>> ToListAsync(CancellationToken cancellation = default) => await _query.ToListAsync(cancellation);

    public IReadQuery<TEntity> Where(Expression<Func<TEntity, bool>> expression)
    {
        _query = _query.Where(expression);
        return this;
    }
}