// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Security.Claims;
using Microsoft.AspNetCore.Identity.Test;
 
namespace Microsoft.AspNetCore.Identity.InMemory;
 
public class InMemoryStore<TUser, TRole> :
    InMemoryUserStore<TUser>,
    IUserRoleStore<TUser>,
    IUserPasskeyStore<TUser>,
    IQueryableRoleStore<TRole>,
    IRoleClaimStore<TRole>
    where TRole : PocoRole
    where TUser : PocoUser
{
    // RoleId == roleName for InMemory
    public Task AddToRoleAsync(TUser user, string role, CancellationToken cancellationToken = default(CancellationToken))
    {
        var roleEntity = _roles.Values.SingleOrDefault(r => r.NormalizedName == role);
        if (roleEntity != null)
        {
            user.Roles.Add(new PocoUserRole { RoleId = roleEntity.Id, UserId = user.Id });
        }
        return Task.FromResult(0);
    }
 
    // RoleId == roleName for InMemory
    public Task RemoveFromRoleAsync(TUser user, string role, CancellationToken cancellationToken = default(CancellationToken))
    {
        var roleObject = _roles.Values.SingleOrDefault(r => r.NormalizedName == role);
        var roleEntity = user.Roles.SingleOrDefault(ur => ur.RoleId == roleObject.Id);
        if (roleEntity != null)
        {
            user.Roles.Remove(roleEntity);
        }
        return Task.FromResult(0);
    }
 
    public Task<IList<string>> GetRolesAsync(TUser user, CancellationToken cancellationToken = default(CancellationToken))
    {
        IList<string> roles = new List<string>();
        foreach (var r in user.Roles.Select(ur => ur.RoleId))
        {
            roles.Add(_roles[r].Name);
        }
        return Task.FromResult(roles);
    }
 
    public Task<bool> IsInRoleAsync(TUser user, string role, CancellationToken cancellationToken = default(CancellationToken))
    {
        var roleObject = _roles.Values.SingleOrDefault(r => r.NormalizedName == role);
        bool result = roleObject != null && user.Roles.Any(ur => ur.RoleId == roleObject.Id);
        return Task.FromResult(result);
    }
 
    // RoleId == rolename for inmemory store tests
    public Task<IList<TUser>> GetUsersInRoleAsync(string roleName, CancellationToken cancellationToken = default(CancellationToken))
    {
        ArgumentException.ThrowIfNullOrEmpty(roleName);
 
        var role = _roles.Values.Where(x => x.NormalizedName.Equals(roleName)).SingleOrDefault();
        if (role == null)
        {
            return Task.FromResult<IList<TUser>>(new List<TUser>());
        }
        return Task.FromResult<IList<TUser>>(Users.Where(u => (u.Roles.Where(x => x.RoleId == role.Id).Any())).Select(x => x).ToList());
    }
 
    private readonly Dictionary<string, TRole> _roles = new Dictionary<string, TRole>();
 
    public Task<IdentityResult> CreateAsync(TRole role, CancellationToken cancellationToken = default(CancellationToken))
    {
        _roles[role.Id] = role;
        return Task.FromResult(IdentityResult.Success);
    }
 
    public Task<IdentityResult> DeleteAsync(TRole role, CancellationToken cancellationToken = default(CancellationToken))
    {
        if (role == null || !_roles.ContainsKey(role.Id))
        {
            throw new InvalidOperationException("Unknown role");
        }
        _roles.Remove(role.Id);
        return Task.FromResult(IdentityResult.Success);
    }
 
    public Task<string> GetRoleIdAsync(TRole role, CancellationToken cancellationToken = default(CancellationToken))
    {
        return Task.FromResult(role.Id);
    }
 
    public Task<string> GetRoleNameAsync(TRole role, CancellationToken cancellationToken = default(CancellationToken))
    {
        return Task.FromResult(role.Name);
    }
 
    public Task SetRoleNameAsync(TRole role, string roleName, CancellationToken cancellationToken = default(CancellationToken))
    {
        role.Name = roleName;
        return Task.FromResult(0);
    }
 
    public Task<IdentityResult> UpdateAsync(TRole role, CancellationToken cancellationToken = default(CancellationToken))
    {
        _roles[role.Id] = role;
        return Task.FromResult(IdentityResult.Success);
    }
 
    Task<TRole> IRoleStore<TRole>.FindByIdAsync(string roleId, CancellationToken cancellationToken)
    {
        if (_roles.ContainsKey(roleId))
        {
            return Task.FromResult(_roles[roleId]);
        }
        return Task.FromResult<TRole>(null);
    }
 
    Task<TRole> IRoleStore<TRole>.FindByNameAsync(string roleName, CancellationToken cancellationToken)
    {
        return
            Task.FromResult(
                Roles.SingleOrDefault(r => String.Equals(r.NormalizedName, roleName, StringComparison.OrdinalIgnoreCase)));
    }
 
    public Task<IList<Claim>> GetClaimsAsync(TRole role, CancellationToken cancellationToken = default(CancellationToken))
    {
        var claims = role.Claims.Select(c => new Claim(c.ClaimType, c.ClaimValue)).ToList();
        return Task.FromResult<IList<Claim>>(claims);
    }
 
    public Task AddClaimAsync(TRole role, Claim claim, CancellationToken cancellationToken = default(CancellationToken))
    {
        role.Claims.Add(new PocoRoleClaim<string> { ClaimType = claim.Type, ClaimValue = claim.Value, RoleId = role.Id });
        return Task.FromResult(0);
    }
 
    public Task RemoveClaimAsync(TRole role, Claim claim, CancellationToken cancellationToken = default(CancellationToken))
    {
        var entity =
            role.Claims.FirstOrDefault(
                ur => ur.RoleId == role.Id && ur.ClaimType == claim.Type && ur.ClaimValue == claim.Value);
        if (entity != null)
        {
            role.Claims.Remove(entity);
        }
        return Task.FromResult(0);
    }
 
    public Task<string> GetNormalizedRoleNameAsync(TRole role, CancellationToken cancellationToken = default(CancellationToken))
    {
        return Task.FromResult(role.NormalizedName);
    }
 
    public Task SetNormalizedRoleNameAsync(TRole role, string normalizedName, CancellationToken cancellationToken = default(CancellationToken))
    {
        role.NormalizedName = normalizedName;
        return Task.FromResult(0);
    }
 
    public Task AddOrUpdatePasskeyAsync(TUser user, UserPasskeyInfo passkey, CancellationToken cancellationToken)
    {
        var passkeyEntity = user.Passkeys.FirstOrDefault(p => p.CredentialId.SequenceEqual(passkey.CredentialId));
        if (passkeyEntity is null)
        {
            user.Passkeys.Add(ToPocoUserPasskey(user, passkey));
        }
        else
        {
            passkeyEntity.Name = passkey.Name;
            passkeyEntity.SignCount = passkey.SignCount;
            passkeyEntity.IsBackedUp = passkey.IsBackedUp;
            passkeyEntity.IsUserVerified = passkey.IsUserVerified;
        }
        return Task.CompletedTask;
    }
 
    public Task<IList<UserPasskeyInfo>> GetPasskeysAsync(TUser user, CancellationToken cancellationToken)
    {
        return Task.FromResult<IList<UserPasskeyInfo>>(user.Passkeys.Select(ToUserPasskeyInfo).ToList()!);
    }
 
    public Task<TUser> FindByPasskeyIdAsync(byte[] credentialId, CancellationToken cancellationToken)
    {
        return Task.FromResult(Users.FirstOrDefault(u => u.Passkeys.Any(p => p.CredentialId.SequenceEqual(credentialId))));
    }
 
    public Task<UserPasskeyInfo> FindPasskeyAsync(TUser user, byte[] credentialId, CancellationToken cancellationToken)
    {
        return Task.FromResult(ToUserPasskeyInfo(user.Passkeys.FirstOrDefault(p => p.CredentialId.SequenceEqual(credentialId))));
    }
 
    public Task RemovePasskeyAsync(TUser user, byte[] credentialId, CancellationToken cancellationToken)
    {
        var passkey = user.Passkeys.SingleOrDefault(p => p.CredentialId.SequenceEqual(credentialId));
        if (passkey is not null)
        {
            user.Passkeys.Remove(passkey);
        }
 
        return Task.CompletedTask;
    }
 
    private static UserPasskeyInfo ToUserPasskeyInfo(PocoUserPasskey<string> p)
        => p is null ? null : new(
            p.CredentialId,
            p.PublicKey,
            p.CreatedAt,
            p.SignCount,
            p.Transports,
            p.IsUserVerified,
            p.IsBackupEligible,
            p.IsBackedUp,
            p.AttestationObject,
            p.ClientDataJson)
        {
            Name = p.Name
        };
 
    private static PocoUserPasskey<string> ToPocoUserPasskey(TUser user, UserPasskeyInfo p)
        => new()
        {
            UserId = user.Id,
            CredentialId = p.CredentialId,
            PublicKey = p.PublicKey,
            Name = p.Name,
            CreatedAt = p.CreatedAt,
            Transports = p.Transports,
            SignCount = p.SignCount,
            IsUserVerified = p.IsUserVerified,
            IsBackupEligible = p.IsBackupEligible,
            IsBackedUp = p.IsBackedUp,
            AttestationObject = p.AttestationObject,
            ClientDataJson = p.ClientDataJson,
        };
 
    public IQueryable<TRole> Roles
    {
        get { return _roles.Values.AsQueryable(); }
    }
}