54 lines
1.9 KiB
C#
54 lines
1.9 KiB
C#
using Bootstrap.DataAccess;
|
|
using Longbow.Security.Principal;
|
|
using Longbow.Web.Mvc;
|
|
using System;
|
|
using System.Linq;
|
|
using System.Web.Mvc;
|
|
|
|
namespace Bootstrap.Admin
|
|
{
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
[AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, Inherited = true, AllowMultiple = true)]
|
|
class BAAuthorizeAttribute : LgbAuthorizeAttribute
|
|
{
|
|
public override void OnAuthorization(AuthorizationContext filterContext)
|
|
{
|
|
if (filterContext.HttpContext.User.Identity.IsAuthenticated)
|
|
{
|
|
string username = filterContext.HttpContext.User.Identity.Name;
|
|
var roles = RoleHelper.RetrieveRolesByUserName(username).Select(r => r.RoleName);
|
|
filterContext.HttpContext.User = new LgbPrincipal(filterContext.HttpContext.User.Identity, roles);
|
|
}
|
|
base.OnAuthorization(filterContext);
|
|
}
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <param name="userName"></param>
|
|
/// <returns></returns>
|
|
protected override bool AuthenticateRole()
|
|
{
|
|
string url = string.Format("~/{0}/{1}", ControllerName, ActionName);
|
|
Roles = string.Join(";", RoleHelper.RetrieveRolesByUrl(url).Select(r => r.RoleName));
|
|
return base.AuthenticateRole();
|
|
}
|
|
/// <summary>
|
|
///
|
|
/// </summary>
|
|
/// <param name="filterContext"></param>
|
|
protected override void HandleUnauthorizedRequest(System.Web.Mvc.AuthorizationContext filterContext)
|
|
{
|
|
if (!filterContext.HttpContext.User.Identity.IsAuthenticated)
|
|
{
|
|
base.HandleUnauthorizedRequest(filterContext);
|
|
return;
|
|
}
|
|
|
|
var view = new ViewResult();
|
|
view.ViewName = "UnAuthorized";
|
|
filterContext.Result = view;
|
|
}
|
|
}
|
|
} |