using Dpz.Core.Public.Entity.Auth;
using Dpz.Core.Service.Mediator.Features.Auth.Contracts;

namespace Dpz.Core.Service.Mediator.Features.Auth.Queries;

public class SearchAuthTokensQueryHandler(
    IRepository<DpzToken> tokenRepository,
    IRepository<DpzApplication> applicationRepository
) : IRequestHandler<SearchAuthTokensQuery, PagedListWarp<AuthTokenPageItem>>
{
    public async ValueTask<PagedListWarp<AuthTokenPageItem>> Handle(
        SearchAuthTokensQuery request,
        CancellationToken cancellationToken
    )
    {
        var query = tokenRepository.SearchFor(x => true);

        if (!string.IsNullOrWhiteSpace(request.UserId))
        {
            query = query.Where(x => x.Subject == request.UserId);
        }

        if (
            !string.IsNullOrWhiteSpace(request.ClientId)
            && ObjectId.TryParse(request.ClientId, out var appId)
        )
        {
            query = query.Where(x => x.ApplicationId == appId);
        }

        var pagedList = await query
            .OrderByDescending(x => x.Id)
            .ToPagedListAsync(request.Page, request.Limit, cancellationToken);

        var applicationIds = pagedList.Select(x => x.ApplicationId).ToHashSet();
        var applications = await applicationRepository
            .SearchFor(Builders<DpzApplication>.Filter.In(x => x.Id, applicationIds))
            .ToListAsync(cancellationToken);

        var items = pagedList
            .Select(x =>
            {
                var app = applications.FirstOrDefault(y => y.Id == x.ApplicationId);
                return new AuthTokenPageItem
                {
                    Id = x.Id.ToString(),
                    ApplicationId = x.ApplicationId.ToString(),
                    ApplicationName = app?.DisplayName,
                    AuthorizationId = x.AuthorizationId.ToString(),
                    ConcurrencyToken = x.ConcurrencyToken,
                    CreationDate = x.CreationDate,
                    ExpirationDate = x.ExpirationDate,
                    Payload = x.Payload,
                    Properties = x.Properties,
                    RedemptionDate = x.RedemptionDate,
                    ReferenceId = x.ReferenceId,
                    Status = x.Status,
                    Subject = x.Subject,
                    Type = x.Type,
                };
            })
            .ToList();

        return new PagedListWarp<AuthTokenPageItem>(
            pagedList.CurrentPageIndex,
            pagedList.PageSize,
            pagedList.TotalItemCount,
            items
        );
    }
}