csharp ASP .NET授权模块，用于从呈现的页面中删除不可访问的控件

Posted 2021-05-12

using System;
using System.Linq;
using System.Web;
using System.Web.UI;
using System.Web.UI.WebControls;

namespace AuthorizationTest.HttpModules
{
    public class AuthorizationModule : IHttpModule
    {
        private HttpApplication application;

        public void Init(HttpApplication context)
        {
            application = context;

            application.PostMapRequestHandler += HttpApplication_PostMapRequestHandler;
        }

        public void Dispose()
        {
            application.PostMapRequestHandler -= HttpApplication_PostMapRequestHandler;
        }

        private void HttpApplication_PostMapRequestHandler(object sender, EventArgs e)
        {
            var page = application.Context.Handler as Page;
            if (page == null)
                return;
            page.PreRender += new EventHandler(Page_PreRender);
        }

        private void Page_PreRender(object sender, EventArgs e)
        {
            var page = sender as Page;
            var sw = System.Diagnostics.Stopwatch.StartNew();
            EnforceAuthorization(page.Controls);
            System.Diagnostics.Trace.WriteLine(sw.ElapsedMilliseconds);
            page.PreRender -= Page_PreRender;
        }
        
        private void EnforceAuthorization(ControlCollection controls)
        {
            foreach (Control control in controls)
            {
                var webControl = control as WebControl;
                if (webControl != null)
                {
                    var feature = webControl.Attributes["feature"];
                    if (feature != null)
                    {
                        webControl.Visible = HasFeature(feature);

                        webControl.Attributes.Remove("feature");
                    }
                }

                EnforceAuthorization(control.Controls);
            }
        }

        private bool HasFeature(string feature)
        {
            var features = new[] { "Foo", "Bar" };
            return features.Contains(feature);
        }
    }
}