src/EasyFrameWork/Mvc/Authorize/DefaultAuthorizer.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

/* http://www.zkea.net/ 
 * Copyright (c) ZKEASOFT. All rights reserved. 
 * http://www.zkea.net/licenses */

using System.Collections.Generic;
using System.Linq;
using Easy.Constant;
using Easy.Extend;
using Easy.Models;
using Easy.Modules.Role;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.DependencyInjection;

namespace Easy.Mvc.Authorize
{
    public class DefaultAuthorizer : IAuthorizer
    {
        private readonly IApplicationContext _applicationContext;
        private readonly IRoleService _roleService;
        private readonly IUserRoleRelationService _userRoleRelationService;
        private readonly Dictionary<string, HashSet<string>> _userPermissions;

        public DefaultAuthorizer(IApplicationContext applicationContext,
            IRoleService roleService,
            IUserRoleRelationService userRoleRelationService)
        {
            _applicationContext = applicationContext;
            _roleService = roleService;
            _userRoleRelationService = userRoleRelationService;
            _userPermissions = new Dictionary<string, HashSet<string>>();
        }
        public bool Authorize(string permission)
        {
            return Authorize(permission, _applicationContext.CurrentUser);
        }

        public bool Authorize(string permission, IUser user)
        {
            if (!_applicationContext.IsAuthenticated)
            {
                return false;
            }
            if (permission.IsNullOrWhiteSpace())
            {
                return true;
            }
            if (user == null)
            {
                return false;
            }
            if (_userPermissions.ContainsKey(user.UserID))
            {
                return _userPermissions[user.UserID].Contains(permission);
            }

            List<int> roles;
            if (user.Roles != null)
            {
                roles = user.Roles.ToList(m => m.RoleID);
            }