61 lines
2.1 KiB
C#
61 lines
2.1 KiB
C#
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.AspNet.Builder;
|
|
using Microsoft.AspNet.Http;
|
|
using Microsoft.Extensions.Options;
|
|
|
|
namespace Microsoft.AspNet.HttpOverrides
|
|
{
|
|
public class HttpMethodOverrideMiddleware
|
|
{
|
|
private const string xHttpMethodOverride = "X-Http-Method-Override";
|
|
private readonly RequestDelegate _next;
|
|
private readonly HttpMethodOverrideOptions _options;
|
|
|
|
public HttpMethodOverrideMiddleware(RequestDelegate next, IOptions<HttpMethodOverrideOptions> options)
|
|
{
|
|
if (next == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(next));
|
|
}
|
|
if (options == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(options));
|
|
}
|
|
_next = next;
|
|
_options = options.Value;
|
|
}
|
|
|
|
public async Task Invoke(HttpContext context)
|
|
{
|
|
if (string.Equals(context.Request.Method, "POST", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
if (_options.FormFieldName != null)
|
|
{
|
|
if (context.Request.HasFormContentType)
|
|
{
|
|
var form = await context.Request.ReadFormAsync();
|
|
var methodType = form[_options.FormFieldName];
|
|
if (!string.IsNullOrEmpty(methodType))
|
|
{
|
|
context.Request.Method = methodType;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
var xHttpMethodOverrideValue = context.Request.Headers[xHttpMethodOverride];
|
|
if (!string.IsNullOrEmpty(xHttpMethodOverrideValue))
|
|
{
|
|
context.Request.Method = xHttpMethodOverrideValue;
|
|
}
|
|
}
|
|
}
|
|
await _next(context);
|
|
}
|
|
}
|
|
}
|