using MessagePack;
using Microsoft.AspNetCore.Mvc.Formatters;

namespace Dpz.Core.WebApi.MessagePackFormatters;

/// <summary>
/// formatter
/// </summary>
public class MessagePackInputFormatter : InputFormatter
{
    private readonly MessagePackSerializerOptions _options;
    private readonly SerializableTypesCache _serializableTypesCache;

    /// <summary>
    /// ctor
    /// </summary>
    /// <param name="options"></param>
    /// <param name="mediaTypes"></param>
    /// <exception cref="ArgumentNullException"></exception>
    /// <exception cref="ArgumentException"></exception>
    public MessagePackInputFormatter(
        MessagePackSerializerOptions options,
        ICollection<string> mediaTypes
    )
    {
        _options = options ?? throw new ArgumentNullException(nameof(options));
        if (mediaTypes == null)
        {
            throw new ArgumentNullException(nameof(mediaTypes));
        }

        if (mediaTypes.Count == 0)
        {
            throw new ArgumentException("Media type must be specified.", nameof(mediaTypes));
        }

        _serializableTypesCache = new SerializableTypesCache(options.Resolver);

        foreach (var mediaType in mediaTypes)
        {
            SupportedMediaTypes.Add(mediaType);
        }
    }

    /// <inheritdoc />
    public override async Task<InputFormatterResult> ReadRequestBodyAsync(
        InputFormatterContext context
    )
    {
        var httpContext = context.HttpContext;
        var result = await MessagePackSerializer.DeserializeAsync(
            context.ModelType,
            httpContext.Request.Body,
            _options,
            httpContext.RequestAborted
        );
        return await InputFormatterResult.SuccessAsync(result);
    }

    /// <inheritdoc />
    /// <exception cref="T:System.ArgumentNullException"><paramref name="type" /> is <c>null</c>.</exception>
    protected override bool CanReadType(Type type)
    {
        return _serializableTypesCache.CanSerialize(type);
    }
}