abp-vnext-pro/aspnet-core/services/src/CompanyName.ProjectName.Domain/IdentityServer/IdenityServerApiResourceManager.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using CompanyName.ProjectName.Extension.System;
using Volo.Abp;
using Volo.Abp.IdentityServer.ApiResources;
using IdentityModel;

namespace CompanyName.ProjectName.IdentityServer
{
    public class IdenityServerApiResourceManager : ProjectNameDomainService
    {
        private readonly IApiResourceRepository _apiResourceRepository;

        public IdenityServerApiResourceManager(IApiResourceRepository apiResourceRepository)
        {
            _apiResourceRepository = apiResourceRepository;
        }

        public Task<List<ApiResource>> GetListAsync(
            int skipCount = 0,
            int maxResultCount = 10,
            string filter = null,
            bool includeDetails = false,
            CancellationToken cancellationToken = default)
        {
            return _apiResourceRepository.GetListAsync("CreationTime desc", skipCount,
                maxResultCount, filter,
                includeDetails,
                cancellationToken);
        }

        public Task<long> GetCountAsync(string filter = null,
            CancellationToken cancellationToken = default)
        {
            return _apiResourceRepository.GetCountAsync(filter,
                cancellationToken);
        }

        /// <summary>
        /// 获取所有api resource
        /// </summary>
        /// <returns></returns>
        public async Task<List<ApiResource>> GetResources(
            bool includeDetails = false,
            CancellationToken cancellationToken = default)
        {
            return await _apiResourceRepository.GetListAsync(includeDetails, cancellationToken);
        }

        public async Task<ApiResource> CreateAsync(
            Guid id,
            string name,
            string displayName,
            string description,
            bool enabled,
            string allowedAccessTokenSigningAlgorithms,
            bool showInDiscoveryDocument,
            string secret,
            CancellationToken cancellationToken = default)
        {
            var apiResource =
                await _apiResourceRepository.FindByNameAsync(name.Trim(), false, cancellationToken);
            if (null != apiResource)
            {
                throw new UserFriendlyException(message: "ApiResource已存在");
            }

            apiResource = new ApiResource(id, name, displayName, description)
            {
                AllowedAccessTokenSigningAlgorithms = allowedAccessTokenSigningAlgorithms,
                ShowInDiscoveryDocument = showInDiscoveryDocument,
                Enabled = enabled
            };

            apiResource.AddSecret(secret.ToSha256());

            // scopes?.Distinct().ToList().ForEach(item => { apiResource.AddScope(item.Scope); });
            //
            // claims?.Distinct().ToList().ForEach(item => { apiResource.AddUserClaim(item.Type); });
            //
            // properties?.Distinct().ToList().ForEach(item => { apiResource.AddProperty(item.Key, item.Value); });

            return await _apiResourceRepository.InsertAsync(apiResource,
                cancellationToken: cancellationToken);
        }

        public async Task DeleteAsync(
            Guid id,
            bool autoSave = false,
            CancellationToken cancellationToken = default)
        {
            await _apiResourceRepository.DeleteAsync(id, autoSave, cancellationToken);
        }

        public async Task<ApiResource> UpdateAsync(
            string name,
            string displayName,
            string description,
            bool enabled,
            string allowedAccessTokenSigningAlgorithms,
            bool showInDiscoveryDocument,
            string secret,
            List<string> scopes,
            CancellationToken cancellationToken = default
        )
        {
            var apiResource =
                await _apiResourceRepository.FindByNameAsync(name.Trim(), true, cancellationToken);
            if (null == apiResource)
            {
                throw new UserFriendlyException(message: "ApiResource不存在");
            }

            apiResource.DisplayName = displayName;
            apiResource.Description = description;
            apiResource.Enabled = enabled;
            apiResource.AllowedAccessTokenSigningAlgorithms = allowedAccessTokenSigningAlgorithms;
            apiResource.ShowInDiscoveryDocument = showInDiscoveryDocument;
            if (secret.IsNotNullOrWhiteSpace())
            {
                // 判读密钥是否一样
                if (apiResource.Secrets.Count > 0)
                {
                    if (apiResource.Secrets.First().Value != secret)
                    {
                        apiResource.Secrets.Clear();
                        apiResource.AddSecret(secret.ToSha256());
                    }
                }
                else
                {
                    apiResource.AddSecret(secret.ToSha256());
                }
            }


            apiResource.Scopes.Clear();

            if (scopes != null)
            {
                foreach (var item in scopes)
                {
                    apiResource.AddScope(item);
                }
            }


            return await _apiResourceRepository.UpdateAsync(apiResource,
                cancellationToken: cancellationToken);
        }
    }
}