mirror of https://github.com/Radarr/Radarr
32 lines
854 B
C#
32 lines
854 B
C#
using System.Threading.Tasks;
|
|
using Microsoft.AspNetCore.Http;
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
using Radarr.Http.Extensions;
|
|
|
|
namespace Radarr.Http.Middleware
|
|
{
|
|
public class VersionMiddleware
|
|
{
|
|
private const string VERSIONHEADER = "X-Application-Version";
|
|
|
|
private readonly RequestDelegate _next;
|
|
private readonly string _version;
|
|
|
|
public VersionMiddleware(RequestDelegate next)
|
|
{
|
|
_next = next;
|
|
_version = BuildInfo.Version.ToString();
|
|
}
|
|
|
|
public async Task InvokeAsync(HttpContext context)
|
|
{
|
|
if (context.Request.IsApiRequest() && !context.Response.Headers.ContainsKey(VERSIONHEADER))
|
|
{
|
|
context.Response.Headers.Append(VERSIONHEADER, _version);
|
|
}
|
|
|
|
await _next(context);
|
|
}
|
|
}
|
|
}
|