Files
Umbraco-CMS/src/Umbraco.Web/WebApi/HttpControllerContextExtensions.cs

67 lines
3.1 KiB
C#
Raw Normal View History

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net.Http;
using System.Runtime.Remoting.Contexts;
using System.Threading;
using System.Threading.Tasks;
using System.Web.Http;
using System.Web.Http.Controllers;
using System.Web.Http.Filters;
using Umbraco.Web.WebApi.Filters;
namespace Umbraco.Web.WebApi
{
internal static class HttpControllerContextExtensions
{
/// <summary>
/// This method will go an execute the authorization filters for the controller action, if any fail
/// it will return their response, otherwise we'll return null.
/// </summary>
/// <param name="controllerContext"></param>
internal static async Task<HttpResponseMessage> InvokeAuthorizationFiltersForRequest(this HttpControllerContext controllerContext)
{
var controllerDescriptor = controllerContext.ControllerDescriptor;
var controllerServices = controllerDescriptor.Configuration.Services;
var actionDescriptor = controllerServices.GetActionSelector().SelectAction(controllerContext);
var actionContext = new HttpActionContext(controllerContext, actionDescriptor);
var filters = actionDescriptor.GetFilterPipeline();
var filterGrouping = new FilterGrouping(filters);
var actionFilters = filterGrouping.ActionFilters;
// Because the continuation gets built from the inside out we need to reverse the filter list
// so that least specific filters (Global) get run first and the most specific filters (Action) get run last.
var authorizationFilters = filterGrouping.AuthorizationFilters.Reverse().ToArray();
if (authorizationFilters.Any())
{
2017-07-20 11:21:28 +02:00
var cancelToken = new CancellationToken();
var filterResult = await FilterContinuation(actionContext, cancelToken, authorizationFilters, 0);
if (filterResult != null)
{
//this means that the authorization filter has returned a result - unauthorized so we cannot continue
return filterResult;
2017-07-20 11:21:28 +02:00
}
}
return null;
}
/// <summary>
/// This method is how you execute a chain of filters, it needs to recursively call in to itself as the continuation for the next filter in the chain
/// </summary>
/// <param name="actionContext"></param>
/// <param name="token"></param>
/// <param name="filters"></param>
/// <param name="index"></param>
/// <returns></returns>
private static async Task<HttpResponseMessage> FilterContinuation(HttpActionContext actionContext, CancellationToken token, IList<IAuthorizationFilter> filters, int index)
{
return await filters[index].ExecuteAuthorizationFilterAsync(actionContext, token,
() => (index + 1) == filters.Count
? Task.FromResult<HttpResponseMessage>(null)
: FilterContinuation(actionContext, token, filters, ++index));
}
2017-07-20 11:21:28 +02:00
}
2017-07-20 11:21:28 +02:00
}