using DigitalData.Core.Exceptions;
using EnvelopeGenerator.Application.Common.Extensions;
using EnvelopeGenerator.Application.Common.Query;
using EnvelopeGenerator.Domain.Entities;
using EnvelopeGenerator.Domain.Interfaces;

namespace EnvelopeGenerator.Application.Common.Extensions;

/// <summary>
/// 
/// </summary>
public static class QueryExtensions
{
    /// <summary>
    /// 
    /// </summary>
    /// <typeparam name="TEntity"></typeparam>
    /// <param name="root"></param>
    /// <param name="query"></param>
    /// <param name="notnull"></param>
    /// <returns></returns>
    /// <exception cref="BadRequestException"></exception>
    public static IQueryable<TEntity> Where<TEntity>(this IQueryable<TEntity> root, EnvelopeQueryBase query, bool notnull = true)
        where TEntity : IHasEnvelope
    {
        if (query.Id is not null)
            root = root.Where(e => e.Envelope!.Id == query.Id);
        else if (query.Uuid is not null)
            root = root.Where(e => e.Envelope!.Uuid == query.Uuid);
        else if (notnull)
            throw new BadRequestException(
                "Either Envelope Id or Envelope Uuid must be provided in the query."
            );

        return root;
    }

    /// <summary>
    /// 
    /// </summary>
    /// <param name="root"></param>
    /// <param name="query"></param>
    /// <param name="notnull"></param>
    /// <returns></returns>
    /// <exception cref="BadRequestException"></exception>
    public static IQueryable<Envelope> Where(this IQueryable<Envelope> root, EnvelopeQueryBase query, bool notnull = true)
    {
        if (query.Id is not null)
            root = root.Where(e => e.Id == query.Id);
        else if (query.Uuid is not null)
            root = root.Where(e => e.Uuid == query.Uuid);
        else if (notnull)
            throw new BadRequestException(
                "Either Envelope Id or Envelope Uuid must be provided in the query."
            );

        return root;
    }

    /// <summary>
    /// 
    /// </summary>
    /// <typeparam name="TEntity"></typeparam>
    /// <param name="root"></param>
    /// <param name="query"></param>
    /// <param name="notnull"></param>
    /// <returns></returns>
    /// <exception cref="BadRequestException"></exception>
    public static IQueryable<TEntity> Where<TEntity>(this IQueryable<TEntity> root, ReceiverQueryBase query, bool notnull = true)
        where TEntity : IHasReceiver
    {
        if (query.Id is not null)
            root = root.Where(e => e.Receiver!.Id == query.Id);
        else if (query.EmailAddress is not null)
            root = root.Where(e => e.Receiver!.EmailAddress == query.EmailAddress);
        else if (query.Signature is not null)
            root = root.Where(e => e.Receiver!.Signature == query.Signature);
        else if (notnull)
            throw new BadRequestException(
                "Receiver must have at least one identifier (Id, EmailAddress, or Signature)."
            );

        return root;
    }

    /// <summary>
    /// 
    /// </summary>
    /// <param name="root"></param>
    /// <param name="query"></param>
    /// <param name="notnull"></param>
    /// <returns></returns>
    /// <exception cref="BadRequestException"></exception>
    public static IQueryable<Receiver> Where(this IQueryable<Receiver> root, ReceiverQueryBase query, bool notnull = true)
    {
        if (query.Id is not null)
            root = root.Where(e => e.Id == query.Id);
        else if (query.EmailAddress is not null)
            root = root.Where(e => e.EmailAddress == query.EmailAddress);
        else if (query.Signature is not null)
            root = root.Where(e => e.Signature == query.Signature);
        else if (notnull)
            throw new BadRequestException(
                "Receiver must have at least one identifier (Id, EmailAddress, or Signature)."
            );

        return root;
    }

    /// <summary>
    /// 
    /// </summary>
    /// <typeparam name="TEntity"></typeparam>
    /// <typeparam name="TEnvelopeQuery"></typeparam>
    /// <typeparam name="TReceiverQuery"></typeparam>
    /// <param name="root"></param>
    /// <param name="query"></param>
    /// <param name="notnull"></param>
    /// <returns></returns>
    public static IQueryable<TEntity> Where<TEntity, TEnvelopeQuery, TReceiverQuery>(this IQueryable<TEntity> root, EnvelopeReceiverQueryBase<TEnvelopeQuery, TReceiverQuery> query, bool notnull = true)
        where TEntity : IHasEnvelope, IHasReceiver
        where TEnvelopeQuery : EnvelopeQueryBase, new()
        where TReceiverQuery : ReceiverQueryBase, new() 
        => root.Where(query.Envelope, notnull).Where(query.Receiver, notnull);
}