namespace Dpz.Core.Service.RepositoryServiceImpl;

public class UserExtendedInfoService(IRepository<UserExtendedInfo> repository)
    : IUserExtendedInfoService
{
    public async Task<string> GetEmailAsync(
        string userId,
        CancellationToken cancellationToken = default
    )
    {
        if (string.IsNullOrWhiteSpace(userId))
        {
            return "";
        }

        var filter = Builders<UserExtendedInfo>.Filter.Eq(x => x.Id, userId);
        var entity = await repository
            .Collection.Find(filter)
            .SingleOrDefaultAsync(cancellationToken);
        return entity?.Email ?? "";
    }

    public async Task<Dictionary<string, string>> GetEmailMapAsync(
        IEnumerable<string> userIds,
        CancellationToken cancellationToken = default
    )
    {
        var userIdList = userIds
            .Where(x => !string.IsNullOrWhiteSpace(x))
            .Distinct(StringComparer.OrdinalIgnoreCase)
            .ToList();
        if (userIdList.Count == 0)
        {
            return new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
        }

        var filter = Builders<UserExtendedInfo>.Filter.In(x => x.Id, userIdList);
        var entities = await repository.Collection.Find(filter).ToListAsync(cancellationToken);
        return entities
            .Where(x => !string.IsNullOrWhiteSpace(x.Email))
            .ToDictionary(x => x.Id, x => x.Email ?? "", StringComparer.OrdinalIgnoreCase);
    }

    public async Task CreateIfMissingAsync(
        string userId,
        CancellationToken cancellationToken = default
    )
    {
        if (string.IsNullOrWhiteSpace(userId))
        {
            return;
        }

        var filter = Builders<UserExtendedInfo>.Filter.Eq(x => x.Id, userId);
        if (await repository.Collection.Find(filter).AnyAsync(cancellationToken))
        {
            return;
        }

        var now = DateTime.Now;
        var entity = new UserExtendedInfo
        {
            Id = userId,
            Email = "",
            WebAuthnCredentials = [],
            CreateTime = now,
            LastUpdateTime = now,
        };
        await repository.InsertAsync(entity, cancellationToken);
    }
}