parent
be0ddf02b3
commit
3cc0dd7e12
@ -0,0 +1,90 @@
|
|||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.ComponentModel;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
using Microsoft.AspNetCore.Mvc.ModelBinding;
|
||||||
|
using Microsoft.Extensions.Logging;
|
||||||
|
|
||||||
|
namespace Jellyfin.Api.ModelBinders
|
||||||
|
{
|
||||||
|
/// <summary>
|
||||||
|
/// Comma delimited array model binder.
|
||||||
|
/// Returns an empty array of specified type if there is no query parameter.
|
||||||
|
/// </summary>
|
||||||
|
public class PipeDelimitedArrayModelBinder : IModelBinder
|
||||||
|
{
|
||||||
|
private readonly ILogger<PipeDelimitedArrayModelBinder> _logger;
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// Initializes a new instance of the <see cref="PipeDelimitedArrayModelBinder"/> class.
|
||||||
|
/// </summary>
|
||||||
|
/// <param name="logger">Instance of the <see cref="ILogger{PipeDelimitedArrayModelBinder}"/> interface.</param>
|
||||||
|
public PipeDelimitedArrayModelBinder(ILogger<PipeDelimitedArrayModelBinder> logger)
|
||||||
|
{
|
||||||
|
_logger = logger;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// <inheritdoc/>
|
||||||
|
public Task BindModelAsync(ModelBindingContext bindingContext)
|
||||||
|
{
|
||||||
|
var valueProviderResult = bindingContext.ValueProvider.GetValue(bindingContext.ModelName);
|
||||||
|
var elementType = bindingContext.ModelType.GetElementType() ?? bindingContext.ModelType.GenericTypeArguments[0];
|
||||||
|
var converter = TypeDescriptor.GetConverter(elementType);
|
||||||
|
|
||||||
|
if (valueProviderResult.Length > 1)
|
||||||
|
{
|
||||||
|
var typedValues = GetParsedResult(valueProviderResult.Values, elementType, converter);
|
||||||
|
bindingContext.Result = ModelBindingResult.Success(typedValues);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
var value = valueProviderResult.FirstValue;
|
||||||
|
|
||||||
|
if (value != null)
|
||||||
|
{
|
||||||
|
var splitValues = value.Split('|', StringSplitOptions.RemoveEmptyEntries);
|
||||||
|
var typedValues = GetParsedResult(splitValues, elementType, converter);
|
||||||
|
bindingContext.Result = ModelBindingResult.Success(typedValues);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
var emptyResult = Array.CreateInstance(elementType, 0);
|
||||||
|
bindingContext.Result = ModelBindingResult.Success(emptyResult);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return Task.CompletedTask;
|
||||||
|
}
|
||||||
|
|
||||||
|
private Array GetParsedResult(IReadOnlyList<string> values, Type elementType, TypeConverter converter)
|
||||||
|
{
|
||||||
|
var parsedValues = new object?[values.Count];
|
||||||
|
var convertedCount = 0;
|
||||||
|
for (var i = 0; i < values.Count; i++)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
parsedValues[i] = converter.ConvertFromString(values[i].Trim());
|
||||||
|
convertedCount++;
|
||||||
|
}
|
||||||
|
catch (FormatException e)
|
||||||
|
{
|
||||||
|
_logger.LogWarning(e, "Error converting value.");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var typedValues = Array.CreateInstance(elementType, convertedCount);
|
||||||
|
var typedValueIndex = 0;
|
||||||
|
for (var i = 0; i < parsedValues.Length; i++)
|
||||||
|
{
|
||||||
|
if (parsedValues[i] != null)
|
||||||
|
{
|
||||||
|
typedValues.SetValue(parsedValues[i], typedValueIndex);
|
||||||
|
typedValueIndex++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return typedValues;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,226 @@
|
|||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Globalization;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
using Jellyfin.Api.ModelBinders;
|
||||||
|
using Microsoft.AspNetCore.Http;
|
||||||
|
using Microsoft.AspNetCore.Mvc.ModelBinding;
|
||||||
|
using Microsoft.Extensions.Logging.Abstractions;
|
||||||
|
using Microsoft.Extensions.Primitives;
|
||||||
|
using Moq;
|
||||||
|
using Xunit;
|
||||||
|
|
||||||
|
namespace Jellyfin.Api.Tests.ModelBinders
|
||||||
|
{
|
||||||
|
public sealed class PipeDelimitedArrayModelBinderTests
|
||||||
|
{
|
||||||
|
[Fact]
|
||||||
|
public async Task BindModelAsync_CorrectlyBindsValidPipeDelimitedStringArrayQuery()
|
||||||
|
{
|
||||||
|
var queryParamName = "test";
|
||||||
|
IReadOnlyList<string> queryParamValues = new[] { "lol", "xd" };
|
||||||
|
var queryParamString = "lol|xd";
|
||||||
|
var queryParamType = typeof(string[]);
|
||||||
|
|
||||||
|
var modelBinder = new PipeDelimitedArrayModelBinder(new NullLogger<PipeDelimitedArrayModelBinder>());
|
||||||
|
var valueProvider = new QueryStringValueProvider(
|
||||||
|
new BindingSource(string.Empty, string.Empty, false, false),
|
||||||
|
new QueryCollection(new Dictionary<string, StringValues> { { queryParamName, new StringValues(queryParamString) } }),
|
||||||
|
CultureInfo.InvariantCulture);
|
||||||
|
var bindingContextMock = new Mock<ModelBindingContext>();
|
||||||
|
bindingContextMock.Setup(b => b.ValueProvider).Returns(valueProvider);
|
||||||
|
bindingContextMock.Setup(b => b.ModelName).Returns(queryParamName);
|
||||||
|
bindingContextMock.Setup(b => b.ModelType).Returns(queryParamType);
|
||||||
|
bindingContextMock.SetupProperty(b => b.Result);
|
||||||
|
|
||||||
|
await modelBinder.BindModelAsync(bindingContextMock.Object);
|
||||||
|
|
||||||
|
Assert.True(bindingContextMock.Object.Result.IsModelSet);
|
||||||
|
Assert.Equal((IReadOnlyList<string>?)bindingContextMock.Object?.Result.Model, queryParamValues);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public async Task BindModelAsync_CorrectlyBindsValidDelimitedIntArrayQuery()
|
||||||
|
{
|
||||||
|
var queryParamName = "test";
|
||||||
|
IReadOnlyList<int> queryParamValues = new[] { 42, 0 };
|
||||||
|
var queryParamString = "42|0";
|
||||||
|
var queryParamType = typeof(int[]);
|
||||||
|
|
||||||
|
var modelBinder = new PipeDelimitedArrayModelBinder(new NullLogger<PipeDelimitedArrayModelBinder>());
|
||||||
|
var valueProvider = new QueryStringValueProvider(
|
||||||
|
new BindingSource(string.Empty, string.Empty, false, false),
|
||||||
|
new QueryCollection(new Dictionary<string, StringValues> { { queryParamName, new StringValues(queryParamString) } }),
|
||||||
|
CultureInfo.InvariantCulture);
|
||||||
|
var bindingContextMock = new Mock<ModelBindingContext>();
|
||||||
|
bindingContextMock.Setup(b => b.ValueProvider).Returns(valueProvider);
|
||||||
|
bindingContextMock.Setup(b => b.ModelName).Returns(queryParamName);
|
||||||
|
bindingContextMock.Setup(b => b.ModelType).Returns(queryParamType);
|
||||||
|
bindingContextMock.SetupProperty(b => b.Result);
|
||||||
|
|
||||||
|
await modelBinder.BindModelAsync(bindingContextMock.Object);
|
||||||
|
|
||||||
|
Assert.True(bindingContextMock.Object.Result.IsModelSet);
|
||||||
|
Assert.Equal((IReadOnlyList<int>?)bindingContextMock.Object.Result.Model, queryParamValues);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public async Task BindModelAsync_CorrectlyBindsValidPipeDelimitedEnumArrayQuery()
|
||||||
|
{
|
||||||
|
var queryParamName = "test";
|
||||||
|
IReadOnlyList<TestType> queryParamValues = new[] { TestType.How, TestType.Much };
|
||||||
|
var queryParamString = "How|Much";
|
||||||
|
var queryParamType = typeof(TestType[]);
|
||||||
|
|
||||||
|
var modelBinder = new PipeDelimitedArrayModelBinder(new NullLogger<PipeDelimitedArrayModelBinder>());
|
||||||
|
var valueProvider = new QueryStringValueProvider(
|
||||||
|
new BindingSource(string.Empty, string.Empty, false, false),
|
||||||
|
new QueryCollection(new Dictionary<string, StringValues> { { queryParamName, new StringValues(queryParamString) } }),
|
||||||
|
CultureInfo.InvariantCulture);
|
||||||
|
var bindingContextMock = new Mock<ModelBindingContext>();
|
||||||
|
bindingContextMock.Setup(b => b.ValueProvider).Returns(valueProvider);
|
||||||
|
bindingContextMock.Setup(b => b.ModelName).Returns(queryParamName);
|
||||||
|
bindingContextMock.Setup(b => b.ModelType).Returns(queryParamType);
|
||||||
|
bindingContextMock.SetupProperty(b => b.Result);
|
||||||
|
|
||||||
|
await modelBinder.BindModelAsync(bindingContextMock.Object);
|
||||||
|
|
||||||
|
Assert.True(bindingContextMock.Object.Result.IsModelSet);
|
||||||
|
Assert.Equal((IReadOnlyList<TestType>?)bindingContextMock.Object.Result.Model, queryParamValues);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public async Task BindModelAsync_CorrectlyBindsValidPipeDelimitedEnumArrayQueryWithDoublePipes()
|
||||||
|
{
|
||||||
|
var queryParamName = "test";
|
||||||
|
IReadOnlyList<TestType> queryParamValues = new[] { TestType.How, TestType.Much };
|
||||||
|
var queryParamString = "How||Much";
|
||||||
|
var queryParamType = typeof(TestType[]);
|
||||||
|
|
||||||
|
var modelBinder = new PipeDelimitedArrayModelBinder(new NullLogger<PipeDelimitedArrayModelBinder>());
|
||||||
|
var valueProvider = new QueryStringValueProvider(
|
||||||
|
new BindingSource(string.Empty, string.Empty, false, false),
|
||||||
|
new QueryCollection(new Dictionary<string, StringValues> { { queryParamName, new StringValues(queryParamString) } }),
|
||||||
|
CultureInfo.InvariantCulture);
|
||||||
|
var bindingContextMock = new Mock<ModelBindingContext>();
|
||||||
|
bindingContextMock.Setup(b => b.ValueProvider).Returns(valueProvider);
|
||||||
|
bindingContextMock.Setup(b => b.ModelName).Returns(queryParamName);
|
||||||
|
bindingContextMock.Setup(b => b.ModelType).Returns(queryParamType);
|
||||||
|
bindingContextMock.SetupProperty(b => b.Result);
|
||||||
|
|
||||||
|
await modelBinder.BindModelAsync(bindingContextMock.Object);
|
||||||
|
|
||||||
|
Assert.True(bindingContextMock.Object.Result.IsModelSet);
|
||||||
|
Assert.Equal((IReadOnlyList<TestType>?)bindingContextMock.Object.Result.Model, queryParamValues);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public async Task BindModelAsync_CorrectlyBindsValidEnumArrayQuery()
|
||||||
|
{
|
||||||
|
var queryParamName = "test";
|
||||||
|
IReadOnlyList<TestType> queryParamValues = new[] { TestType.How, TestType.Much };
|
||||||
|
var queryParamString1 = "How";
|
||||||
|
var queryParamString2 = "Much";
|
||||||
|
var queryParamType = typeof(TestType[]);
|
||||||
|
|
||||||
|
var modelBinder = new PipeDelimitedArrayModelBinder(new NullLogger<PipeDelimitedArrayModelBinder>());
|
||||||
|
|
||||||
|
var valueProvider = new QueryStringValueProvider(
|
||||||
|
new BindingSource(string.Empty, string.Empty, false, false),
|
||||||
|
new QueryCollection(new Dictionary<string, StringValues>
|
||||||
|
{
|
||||||
|
{ queryParamName, new StringValues(new[] { queryParamString1, queryParamString2 }) },
|
||||||
|
}),
|
||||||
|
CultureInfo.InvariantCulture);
|
||||||
|
var bindingContextMock = new Mock<ModelBindingContext>();
|
||||||
|
bindingContextMock.Setup(b => b.ValueProvider).Returns(valueProvider);
|
||||||
|
bindingContextMock.Setup(b => b.ModelName).Returns(queryParamName);
|
||||||
|
bindingContextMock.Setup(b => b.ModelType).Returns(queryParamType);
|
||||||
|
bindingContextMock.SetupProperty(b => b.Result);
|
||||||
|
|
||||||
|
await modelBinder.BindModelAsync(bindingContextMock.Object);
|
||||||
|
|
||||||
|
Assert.True(bindingContextMock.Object.Result.IsModelSet);
|
||||||
|
Assert.Equal((IReadOnlyList<TestType>?)bindingContextMock.Object.Result.Model, queryParamValues);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public async Task BindModelAsync_CorrectlyBindsEmptyEnumArrayQuery()
|
||||||
|
{
|
||||||
|
var queryParamName = "test";
|
||||||
|
IReadOnlyList<TestType> queryParamValues = Array.Empty<TestType>();
|
||||||
|
var queryParamType = typeof(TestType[]);
|
||||||
|
|
||||||
|
var modelBinder = new PipeDelimitedArrayModelBinder(new NullLogger<PipeDelimitedArrayModelBinder>());
|
||||||
|
|
||||||
|
var valueProvider = new QueryStringValueProvider(
|
||||||
|
new BindingSource(string.Empty, string.Empty, false, false),
|
||||||
|
new QueryCollection(new Dictionary<string, StringValues>
|
||||||
|
{
|
||||||
|
{ queryParamName, new StringValues(value: null) },
|
||||||
|
}),
|
||||||
|
CultureInfo.InvariantCulture);
|
||||||
|
var bindingContextMock = new Mock<ModelBindingContext>();
|
||||||
|
bindingContextMock.Setup(b => b.ValueProvider).Returns(valueProvider);
|
||||||
|
bindingContextMock.Setup(b => b.ModelName).Returns(queryParamName);
|
||||||
|
bindingContextMock.Setup(b => b.ModelType).Returns(queryParamType);
|
||||||
|
bindingContextMock.SetupProperty(b => b.Result);
|
||||||
|
|
||||||
|
await modelBinder.BindModelAsync(bindingContextMock.Object);
|
||||||
|
|
||||||
|
Assert.True(bindingContextMock.Object.Result.IsModelSet);
|
||||||
|
Assert.Equal((IReadOnlyList<TestType>?)bindingContextMock.Object.Result.Model, queryParamValues);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public async Task BindModelAsync_EnumArrayQuery_BindValidOnly()
|
||||||
|
{
|
||||||
|
var queryParamName = "test";
|
||||||
|
var queryParamString = "🔥|😢";
|
||||||
|
var queryParamType = typeof(IReadOnlyList<TestType>);
|
||||||
|
|
||||||
|
var modelBinder = new PipeDelimitedArrayModelBinder(new NullLogger<PipeDelimitedArrayModelBinder>());
|
||||||
|
var valueProvider = new QueryStringValueProvider(
|
||||||
|
new BindingSource(string.Empty, string.Empty, false, false),
|
||||||
|
new QueryCollection(new Dictionary<string, StringValues> { { queryParamName, new StringValues(queryParamString) } }),
|
||||||
|
CultureInfo.InvariantCulture);
|
||||||
|
var bindingContextMock = new Mock<ModelBindingContext>();
|
||||||
|
bindingContextMock.Setup(b => b.ValueProvider).Returns(valueProvider);
|
||||||
|
bindingContextMock.Setup(b => b.ModelName).Returns(queryParamName);
|
||||||
|
bindingContextMock.Setup(b => b.ModelType).Returns(queryParamType);
|
||||||
|
bindingContextMock.SetupProperty(b => b.Result);
|
||||||
|
|
||||||
|
await modelBinder.BindModelAsync(bindingContextMock.Object);
|
||||||
|
Assert.True(bindingContextMock.Object.Result.IsModelSet);
|
||||||
|
Assert.Empty((IReadOnlyList<TestType>?)bindingContextMock.Object.Result.Model);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public async Task BindModelAsync_EnumArrayQuery_BindValidOnly_2()
|
||||||
|
{
|
||||||
|
var queryParamName = "test";
|
||||||
|
var queryParamString1 = "How";
|
||||||
|
var queryParamString2 = "😱";
|
||||||
|
var queryParamType = typeof(IReadOnlyList<TestType>);
|
||||||
|
|
||||||
|
var modelBinder = new PipeDelimitedArrayModelBinder(new NullLogger<PipeDelimitedArrayModelBinder>());
|
||||||
|
|
||||||
|
var valueProvider = new QueryStringValueProvider(
|
||||||
|
new BindingSource(string.Empty, string.Empty, false, false),
|
||||||
|
new QueryCollection(new Dictionary<string, StringValues>
|
||||||
|
{
|
||||||
|
{ queryParamName, new StringValues(new[] { queryParamString1, queryParamString2 }) },
|
||||||
|
}),
|
||||||
|
CultureInfo.InvariantCulture);
|
||||||
|
var bindingContextMock = new Mock<ModelBindingContext>();
|
||||||
|
bindingContextMock.Setup(b => b.ValueProvider).Returns(valueProvider);
|
||||||
|
bindingContextMock.Setup(b => b.ModelName).Returns(queryParamName);
|
||||||
|
bindingContextMock.Setup(b => b.ModelType).Returns(queryParamType);
|
||||||
|
bindingContextMock.SetupProperty(b => b.Result);
|
||||||
|
|
||||||
|
await modelBinder.BindModelAsync(bindingContextMock.Object);
|
||||||
|
Assert.True(bindingContextMock.Object.Result.IsModelSet);
|
||||||
|
Assert.Single((IReadOnlyList<TestType>?)bindingContextMock.Object.Result.Model);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in new issue