using MessagePack;
using MessagePack.Resolvers;
using Microsoft.Extensions.Options;

namespace Dpz.Core.WebApi.MessagePackFormatters;

/// <summary>
///     格式化选项设置
/// </summary>
public class MessagePackFormatterMvcOptionsSetup(
    IOptions<MessagePackFormatterOptions> messagePackFormatterOptions
) : IConfigureOptions<MvcOptions>
{
    private readonly IOptions<MessagePackFormatterOptions> _messagePackFormatterOptions =
        messagePackFormatterOptions
        ?? throw new ArgumentNullException(nameof(messagePackFormatterOptions));

    /// <inheritdoc />
    /// <exception cref="T:System.InvalidOperationException">不支持的类型</exception>
    /// <exception cref="T:System.ArgumentNullException"><paramref name="options" /> is <see langword="null" /></exception>
    public void Configure(MvcOptions options)
    {
        ArgumentNullException.ThrowIfNull(options);

        var formatterOptions = _messagePackFormatterOptions.Value;
        var supportedMediaTypes = formatterOptions
            .MediaTypes.Where(t => !string.IsNullOrWhiteSpace(t))
            .ToArray();
        if (supportedMediaTypes.Length == 0)
        {
            throw new InvalidOperationException("No supported media types were specified.");
        }

        var msgpackOptions = MessagePackSerializerOptions.Standard;
        msgpackOptions = msgpackOptions.WithResolver(
            formatterOptions.FormatterResolver ?? ContractlessStandardResolver.Instance
        );
        msgpackOptions = msgpackOptions
            .WithAllowAssemblyVersionMismatch(formatterOptions.AllowAssemblyVersionMismatch)
            .WithCompression(formatterOptions.Compression)
            .WithOldSpec(formatterOptions.UseOldSpecification)
            .WithOmitAssemblyVersion(formatterOptions.OmitAssemblyVersion);

        options.InputFormatters.Add(
            new MessagePackInputFormatter(msgpackOptions, supportedMediaTypes)
        );
        options.OutputFormatters.Add(
            new MessagePackOutputFormatter(msgpackOptions, supportedMediaTypes)
        );

        foreach (var fileExtension in formatterOptions.FileExtensions)
        {
            options.FormatterMappings.SetMediaTypeMappingForFormat(
                fileExtension,
                supportedMediaTypes[0]
            );
        }
    }
}