// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.ComponentModel.DataAnnotations;
using System.Globalization;
using Microsoft.AspNetCore.Mvc.DataAnnotations;
using Microsoft.AspNetCore.Testing;
using Microsoft.Extensions.Localization;
using Moq;

namespace Microsoft.AspNetCore.Mvc.ModelBinding.Validation;

public class FileExtensionsAttributeAdapterTest
{
    [Theory]
    [InlineData("", ".png,.jpg,.jpeg,.gif")]
    [InlineData(null, ".png,.jpg,.jpeg,.gif")]
    [ReplaceCulture]
    public void AddValidation_WithoutLocalizationAndDefaultFileExtensions(string extensions, string expectedExtensions)
    {
        // Arrange
        var provider = TestModelMetadataProvider.CreateDefaultProvider();
        var metadata = provider.GetMetadataForProperty(typeof(Profile), nameof(Profile.PhotoFileName));

        var attribute = new FileExtensionsAttribute() { Extensions = extensions };
        attribute.ErrorMessage = "{0} expects only the following extensions: {1}";

        // FileExtensionsAttribute formats the extension list for the error message
        var formattedExtensions = string.Join(", ", expectedExtensions.Split(','));
        var expectedErrorMessage = string.Format(CultureInfo.CurrentCulture, attribute.ErrorMessage, nameof(Profile.PhotoFileName), formattedExtensions);

        var adapter = new FileExtensionsAttributeAdapter(attribute, stringLocalizer: null);
        var context = new ClientModelValidationContext(new ActionContext(), metadata, provider, new Dictionary<string, string>());

        // Act
        adapter.AddValidation(context);

        // Assert
        Assert.Collection(
            context.Attributes,
            kvp => { Assert.Equal("data-val", kvp.Key); Assert.Equal("true", kvp.Value); },
            kvp => { Assert.Equal("data-val-fileextensions", kvp.Key); Assert.Equal(expectedErrorMessage, kvp.Value); },
            kvp => { Assert.Equal("data-val-fileextensions-extensions", kvp.Key); Assert.Equal(expectedExtensions, kvp.Value); });
    }

    public static TheoryData<string, string> ExtensionsData
    {
        get
        {
            return new TheoryData<string, string>()
                {
                    { "jpg", ".jpg" },
                    { " j p g ", ".jpg" },
                    { ".jpg", ".jpg" },
                    { ".x", ".x" },
                    { "jpg,png", ".jpg,.png" },
                    { "jpg, png", ".jpg,.png" },
                    { "JPG, Png", ".jpg,.png" },
                    { ".jpg,.png", ".jpg,.png" },
                    { "..jpg,..png", ".jpg,.png" },
                    { ".TXT, .png", ".txt,.png" },
                    { ".pdf , .docx", ".pdf,.docx" },
                };
        }
    }

    [Theory]
    [MemberData(nameof(ExtensionsData))]
    [ReplaceCulture]
    public void AddValidation_WithoutLocalizationAndCustomFileExtensions(string extensions, string expectedExtensions)
    {
        // Arrange
        var provider = TestModelMetadataProvider.CreateDefaultProvider();
        var metadata = provider.GetMetadataForProperty(typeof(Profile), nameof(Profile.PhotoFileName));

        var attribute = new FileExtensionsAttribute() { Extensions = extensions };
        attribute.ErrorMessage = "{0} expects only the following extensions: {1}";

        // FileExtensionsAttribute formats the extension list for the error message
        var formattedExtensions = string.Join(", ", expectedExtensions.Split(','));
        var expectedErrorMessage = string.Format(CultureInfo.CurrentCulture, attribute.ErrorMessage, nameof(Profile.PhotoFileName), formattedExtensions);

        var adapter = new FileExtensionsAttributeAdapter(attribute, stringLocalizer: null);
        var context = new ClientModelValidationContext(new ActionContext(), metadata, provider, new Dictionary<string, string>());

        // Act
        adapter.AddValidation(context);

        // Assert
        Assert.Collection(
            context.Attributes,
            kvp => { Assert.Equal("data-val", kvp.Key); Assert.Equal("true", kvp.Value); },
            kvp => { Assert.Equal("data-val-fileextensions", kvp.Key); Assert.Equal(expectedErrorMessage, kvp.Value); },
            kvp => { Assert.Equal("data-val-fileextensions-extensions", kvp.Key); Assert.Equal(expectedExtensions, kvp.Value); });
    }

    [Theory]
    [MemberData(nameof(ExtensionsData))]
    [ReplaceCulture]
    public void AddValidation_WithLocalization(string extensions, string expectedExtensions)
    {
        // Arrange
        var provider = TestModelMetadataProvider.CreateDefaultProvider();
        var metadata = provider.GetMetadataForProperty(typeof(Profile), nameof(Profile.PhotoFileName));

        var attribute = new FileExtensionsAttribute() { Extensions = extensions };
        attribute.ErrorMessage = "{0} expects only the following extensions: {1}";

        var formattedExtensions = string.Join(", ", expectedExtensions.Split(','));
        var expectedProperties = new object[] { "PhotoFileName", formattedExtensions };
        var expectedErrorMessage = $"{nameof(Profile.PhotoFileName)} expects only the following extensions: {formattedExtensions}";

        var stringLocalizer = new Mock<IStringLocalizer>();
        stringLocalizer
            .Setup(s => s[attribute.ErrorMessage, expectedProperties])
            .Returns(new LocalizedString(attribute.ErrorMessage, expectedErrorMessage));

        var adapter = new FileExtensionsAttributeAdapter(attribute, stringLocalizer: stringLocalizer.Object);
        var context = new ClientModelValidationContext(new ActionContext(), metadata, provider, new Dictionary<string, string>());

        // Act
        adapter.AddValidation(context);

        // Assert
        Assert.Collection(
            context.Attributes,
            kvp => { Assert.Equal("data-val", kvp.Key); Assert.Equal("true", kvp.Value); },
            kvp => { Assert.Equal("data-val-fileextensions", kvp.Key); Assert.Equal(expectedErrorMessage, kvp.Value); },
            kvp => { Assert.Equal("data-val-fileextensions-extensions", kvp.Key); Assert.Equal(expectedExtensions, kvp.Value); });
    }

    [Fact]
    [ReplaceCulture]
    public void AddValidation_DoesNotTrounceExistingAttributes()
    {
        // Arrange
        var provider = TestModelMetadataProvider.CreateDefaultProvider();
        var metadata = provider.GetMetadataForProperty(typeof(Profile), "PhotoFileName");

        var attribute = new FileExtensionsAttribute() { Extensions = "jpg" };
        attribute.ErrorMessage = "{0} expects only the following extensions: {1}";

        var adapter = new FileExtensionsAttributeAdapter(attribute, stringLocalizer: null);
        var context = new ClientModelValidationContext(new ActionContext(), metadata, provider, new Dictionary<string, string>());

        context.Attributes.Add("data-val", "original");
        context.Attributes.Add("data-val-fileextensions", "original");
        context.Attributes.Add("data-val-fileextensions-extensions", "original");

        // Act
        adapter.AddValidation(context);

        // Assert
        Assert.Collection(
            context.Attributes,
            kvp => { Assert.Equal("data-val", kvp.Key); Assert.Equal("original", kvp.Value); },
            kvp => { Assert.Equal("data-val-fileextensions", kvp.Key); Assert.Equal("original", kvp.Value); },
            kvp => { Assert.Equal("data-val-fileextensions-extensions", kvp.Key); Assert.Equal("original", kvp.Value); });
    }

    private class Profile
    {
        public string PhotoFileName { get; set; }
    }
}
