src/ZKEACMS/Route/AccountRouteConstraint.cs

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

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

using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Routing;
using Microsoft.Extensions.DependencyInjection;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using ZKEACMS.Setting;

namespace ZKEACMS.Route
{
    public class AccountRouteConstraint : IRouteConstraint
    {
        public bool Match(HttpContext httpContext, IRouter route, string routeKey, RouteValueDictionary values, RouteDirection routeDirection)
        {
            if (httpContext == null || routeDirection == RouteDirection.UrlGeneration) return true;

            IApplicationSettingService applicationSettingService = httpContext.RequestServices.GetService<IApplicationSettingService>();
            bool enableCustomerAccount = applicationSettingService.Get(SettingKeys.EnableCustomerAccount, "true") == "true";
            if (!enableCustomerAccount && !IsAdminActions(httpContext, values[routeKey].ToString())) return false;

            if (IsSignUpActions(values[routeKey].ToString()))
            {
                return applicationSettingService.Get(SettingKeys.EnableSignMeUp, "true") == "true";
            }

            return true;
        }
        private bool IsSignUpActions(string action)
        {
            return action.Equals("SignUp", StringComparison.OrdinalIgnoreCase) ||
                action.Equals("SignUpSuccess", StringComparison.OrdinalIgnoreCase);
        }
        private bool IsAdminActions(HttpContext httpContext, string action)
        {
            if (action.Equals("Forgotten", StringComparison.OrdinalIgnoreCase))
            {
                return httpContext.Request.Query["userType"].ToString().Equals("Administrator", StringComparison.OrdinalIgnoreCase);
            }
            return action.Equals("Login", StringComparison.OrdinalIgnoreCase) ||
                action.Equals("Logout", StringComparison.OrdinalIgnoreCase);
        }
    }
}