squidex/backend/src/Squidex.Domain.Users/DefaultUserResolver.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Identity;
using Microsoft.Extensions.DependencyInjection;
using Squidex.Infrastructure;
using Squidex.Shared.Users;

namespace Squidex.Domain.Users
{
    public sealed class DefaultUserResolver : IUserResolver
    {
        private readonly IServiceProvider serviceProvider;

        public DefaultUserResolver(IServiceProvider serviceProvider)
        {
            Guard.NotNull(serviceProvider);

            this.serviceProvider = serviceProvider;
        }

        public async Task<bool> CreateUserIfNotExistsAsync(string email, bool invited)
        {
            Guard.NotNullOrEmpty(email);

            using (var scope = serviceProvider.CreateScope())
            {
                var userFactory = scope.ServiceProvider.GetRequiredService<IUserFactory>();
                var userManager = scope.ServiceProvider.GetRequiredService<UserManager<IdentityUser>>();

                var user = userFactory.Create(email);

                try
                {
                    var result = await userManager.CreateAsync(user);

                    if (result.Succeeded)
                    {
                        var values = new UserValues { DisplayName = email, Invited = invited };

                        await userManager.UpdateAsync(user, values);
                    }

                    return result.Succeeded;
                }
                catch
                {
                    return false;
                }
            }
        }

        public async Task<IUser?> FindByIdOrEmailAsync(string idOrEmail)
        {
            Guard.NotNullOrEmpty(idOrEmail);

            using (var scope = serviceProvider.CreateScope())
            {
                var userFactory = scope.ServiceProvider.GetRequiredService<IUserFactory>();
                var userManager = scope.ServiceProvider.GetRequiredService<UserManager<IdentityUser>>();

                if (userFactory.IsId(idOrEmail))
                {
                    return await userManager.FindByIdWithClaimsAsync(idOrEmail);
                }
                else
                {
                    return await userManager.FindByEmailWithClaimsAsync(idOrEmail);
                }
            }
        }

        public async Task<List<IUser>> QueryByEmailAsync(string email)
        {
            Guard.NotNullOrEmpty(email);

            using (var scope = serviceProvider.CreateScope())
            {
                var userManager = scope.ServiceProvider.GetRequiredService<UserManager<IdentityUser>>();

                var result = await userManager.QueryByEmailAsync(email);

                return result.OfType<IUser>().ToList();
            }
        }

        public async Task<Dictionary<string, IUser>> QueryManyAsync(string[] ids)
        {
            Guard.NotNull(ids);

            using (var scope = serviceProvider.CreateScope())
            {
                var userManager = scope.ServiceProvider.GetRequiredService<UserManager<IdentityUser>>();
                var userFactory = scope.ServiceProvider.GetRequiredService<IUserFactory>();

                ids = ids.Where(x => userFactory.IsId(x)).ToArray();

                var result = await userManager.QueryByIdsAync(ids);

                return result.OfType<IUser>().ToDictionary(x => x.Id);
            }
        }
    }
}