JsonCreateDefaultRegistryEnricher.cs

using System;
using Microsoft.CodeAnalysis.CSharp;
using Microsoft.CodeAnalysis.CSharp.Syntax;
using Yardarm.Enrichment.Registration;
using Yardarm.SystemTextJson.Internal;
using static Microsoft.CodeAnalysis.CSharp.SyntaxFactory;

namespace Yardarm.SystemTextJson
{
    public class JsonCreateDefaultRegistryEnricher : ReturnValueRegistrationEnricher, ICreateDefaultRegistryEnricher
    {
        private readonly IJsonSerializationNamespace _jsonSerializationNamespace;

        public JsonCreateDefaultRegistryEnricher(IJsonSerializationNamespace jsonSerializationNamespace)
        {
            ArgumentNullException.ThrowIfNull(jsonSerializationNamespace);

            _jsonSerializationNamespace = jsonSerializationNamespace;
        }

        protected override ExpressionSyntax EnrichReturnValue(ExpressionSyntax target) =>
            // Don't use the Add<T> overload here because it will cause trimming to retain
            // all constructors. This will then cause IL2026 warnings if trimming is enabled.
            // Instead create a new instance directly using the default constructor and add it.
            InvocationExpression(
                MemberAccessExpression(SyntaxKind.SimpleMemberAccessExpression,
                    target,
                    IdentifierName("Add")),
                ArgumentList(SeparatedList(
                [
                    Argument(MemberAccessExpression(SyntaxKind.SimpleMemberAccessExpression,
                        _jsonSerializationNamespace.JsonTypeSerializer,
                        IdentifierName("SupportedMediaTypes"))),
                    Argument(ObjectCreationExpression(_jsonSerializationNamespace.JsonTypeSerializer,
                        ArgumentList(SingletonSeparatedList(
                            Argument(MemberAccessExpression(SyntaxKind.SimpleMemberAccessExpression,
                                QualifiedName(_jsonSerializationNamespace.Name, IdentifierName(JsonSerializerContextGenerator.TypeName)),
                                IdentifierName("Default"))))),
                            null))
                ])));
    }
}