Generation of serializers can be deactivated via SmartEnum-/ValueObjectAttribute

Author: PawelGerr

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/KeyedSerializerGeneratorState.cs

@@ -5,6 +5,7 @@ namespace Thinktecture.CodeAnalysis;
    public ITypeInformation Type { get; }
    public IMemberInformation? KeyMember { get; }
    public AttributeInfo AttributeInfo { get; }
+   public SerializationFrameworks SerializationFrameworks { get; }
 
    public string? Namespace => Type.Namespace;
    public IReadOnlyList<ContainingTypeState> ContainingTypes => Type.ContainingTypes;
@@ -14,17 +15,20 @@ namespace Thinktecture.CodeAnalysis;
    public KeyedSerializerGeneratorState(
       ITypeInformation type,
       IMemberInformation? keyMember,
-      AttributeInfo attributeInfo)
+      AttributeInfo attributeInfo,
+      SerializationFrameworks serializationFrameworks)
    {
       Type = type;
       KeyMember = keyMember;
       AttributeInfo = attributeInfo;
+      SerializationFrameworks = serializationFrameworks;
    }
 
    public bool Equals(KeyedSerializerGeneratorState other)
    {
       return TypeInformationComparer.Instance.Equals(Type, other.Type)
              && KeyMember?.TypeFullyQualified == other.KeyMember?.TypeFullyQualified
+             && SerializationFrameworks == other.SerializationFrameworks
              && ContainingTypes.SequenceEqual(other.ContainingTypes)
              && AttributeInfo.Equals(other.AttributeInfo);
    }
@@ -40,6 +44,7 @@ public override int GetHashCode()
       {
          var hashCode = TypeInformationComparer.Instance.GetHashCode(Type);
          hashCode = (hashCode * 397) ^ (KeyMember is null ? 0 : KeyMember.TypeFullyQualified.GetHashCode());
+         hashCode = (hashCode * 397) ^ (int)SerializationFrameworks;
          hashCode = (hashCode * 397) ^ ContainingTypes.ComputeHashCode();
          hashCode = (hashCode * 397) ^ AttributeInfo.GetHashCode();
 

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/SmartEnums/AllEnumSettings.cs

@@ -16,6 +16,7 @@ public sealed class AllEnumSettings : IEquatable<AllEnumSettings>, IKeyMemberSet
    public SwitchMapMethodsGeneration MapMethods { get; }
    public ConversionOperatorsGeneration ConversionFromKeyMemberType { get; }
    public ConversionOperatorsGeneration ConversionToKeyMemberType { get; }
+   public SerializationFrameworks SerializationFrameworks { get; }
 
    public AllEnumSettings(AttributeData attribute)
    {
@@ -33,6 +34,7 @@ public AllEnumSettings(AttributeData attribute)
       MapMethods = attribute.FindMapMethods();
       ConversionToKeyMemberType = attribute.FindConversionToKeyMemberType() ?? ConversionOperatorsGeneration.Implicit;
       ConversionFromKeyMemberType = attribute.FindConversionFromKeyMemberType() ?? ConversionOperatorsGeneration.Explicit;
+      SerializationFrameworks = attribute.FindSerializationFrameworks();
 
       // Comparison operators depend on the equality comparison operators
       if (ComparisonOperators > EqualityComparisonOperators)
@@ -64,7 +66,8 @@ public bool Equals(AllEnumSettings? other)
              && SwitchMethods == other.SwitchMethods
              && MapMethods == other.MapMethods
              && ConversionToKeyMemberType == other.ConversionToKeyMemberType
-             && ConversionFromKeyMemberType == other.ConversionFromKeyMemberType;
+             && ConversionFromKeyMemberType == other.ConversionFromKeyMemberType
+             && SerializationFrameworks == other.SerializationFrameworks;
    }
 
    public override int GetHashCode()
@@ -81,10 +84,11 @@ public override int GetHashCode()
          hashCode = (hashCode * 397) ^ (int)EqualityComparisonOperators;
          hashCode = (hashCode * 397) ^ SkipIFormattable.GetHashCode();
          hashCode = (hashCode * 397) ^ SkipToString.GetHashCode();
-         hashCode = (hashCode * 397) ^ SwitchMethods.GetHashCode();
-         hashCode = (hashCode * 397) ^ MapMethods.GetHashCode();
-         hashCode = (hashCode * 397) ^ ConversionToKeyMemberType.GetHashCode();
-         hashCode = (hashCode * 397) ^ ConversionFromKeyMemberType.GetHashCode();
+         hashCode = (hashCode * 397) ^ (int)SwitchMethods;
+         hashCode = (hashCode * 397) ^ (int)MapMethods;
+         hashCode = (hashCode * 397) ^ (int)ConversionToKeyMemberType;
+         hashCode = (hashCode * 397) ^ (int)ConversionFromKeyMemberType;
+         hashCode = (hashCode * 397) ^ (int)SerializationFrameworks;
 
          return hashCode;
       }

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/SmartEnums/JsonSmartEnumCodeGeneratorFactory.cs

@@ -15,6 +15,7 @@ private JsonSmartEnumCodeGeneratorFactory()
    public bool MustGenerateCode(KeyedSerializerGeneratorState state)
    {
       return !state.AttributeInfo.HasJsonConverterAttribute
+             && state.SerializationFrameworks.HasFlag(SerializationFrameworks.SystemTextJson)
              && (state.KeyMember is not null || state.AttributeInfo.DesiredFactories.Any(f => f.UseForSerialization.Has(SerializationFrameworks.SystemTextJson)));
    }
 

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/SmartEnums/MessagePackSmartEnumCodeGeneratorFactory.cs

@@ -15,6 +15,7 @@ private MessagePackSmartEnumCodeGeneratorFactory()
    public bool MustGenerateCode(KeyedSerializerGeneratorState state)
    {
       return !state.AttributeInfo.HasMessagePackFormatterAttribute
+             && state.SerializationFrameworks.HasFlag(SerializationFrameworks.MessagePack)
              && (state.KeyMember is not null || state.AttributeInfo.DesiredFactories.Any(f => f.UseForSerialization.Has(SerializationFrameworks.MessagePack)));
    }
 

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/SmartEnums/NewtonsoftJsonSmartEnumCodeGeneratorFactory.cs

@@ -15,6 +15,7 @@ private NewtonsoftJsonSmartEnumCodeGeneratorFactory()
    public bool MustGenerateCode(KeyedSerializerGeneratorState state)
    {
       return !state.AttributeInfo.HasNewtonsoftJsonConverterAttribute
+             && state.SerializationFrameworks.HasFlag(SerializationFrameworks.NewtonsoftJson)
              && (state.KeyMember is not null || state.AttributeInfo.DesiredFactories.Any(f => f.UseForSerialization.Has(SerializationFrameworks.NewtonsoftJson)));
    }
 

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/SmartEnums/SmartEnumSourceGenerator.cs

@@ -191,7 +191,11 @@ private void InitializeSerializerGenerators(IncrementalGeneratorInitializationCo
                                                                                  : states.Distinct())
                                                 .WithComparer(new SetComparer<IKeyedSerializerCodeGeneratorFactory>());
 
-      var serializerGeneratorStates = validStates.Select((state, _) => new KeyedSerializerGeneratorState(state.State, state.KeyMember, state.AttributeInfo))
+      var serializerGeneratorStates = validStates.Select((state, _) => new KeyedSerializerGeneratorState(
+                                                            state.State,
+                                                            state.KeyMember,
+                                                            state.AttributeInfo,
+                                                            state.Settings.SerializationFrameworks))
                                                  .Combine(serializerGeneratorFactories)
                                                  .SelectMany((tuple, _) => ImmutableArray.CreateRange(tuple.Right, (factory, state) => (State: state, Factory: factory), tuple.Left))
                                                  .Where(tuple =>

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/ValueObjects/AllValueObjectSettings.cs

@@ -28,6 +28,7 @@ public sealed class AllValueObjectSettings : IEquatable<AllValueObjectSettings>,
    public ConversionOperatorsGeneration UnsafeConversionToKeyMemberType { get; }
    public ConversionOperatorsGeneration ConversionToKeyMemberType { get; }
    public StringComparison DefaultStringComparison { get; }
+   public SerializationFrameworks SerializationFrameworks { get; }
 
    public AllValueObjectSettings(AttributeData valueObjectAttribute)
    {
@@ -57,6 +58,7 @@ public AllValueObjectSettings(AttributeData valueObjectAttribute)
       UnsafeConversionToKeyMemberType = valueObjectAttribute.FindUnsafeConversionToKeyMemberType() ?? ConversionOperatorsGeneration.Explicit;
       ConversionFromKeyMemberType = valueObjectAttribute.FindConversionFromKeyMemberType() ?? ConversionOperatorsGeneration.Explicit;
       DefaultStringComparison = valueObjectAttribute.FindDefaultStringComparison();
+      SerializationFrameworks = valueObjectAttribute.FindSerializationFrameworks();
 
       // Comparison operators depend on the equality comparison operators
       if (ComparisonOperators > EqualityComparisonOperators)
@@ -100,7 +102,8 @@ public bool Equals(AllValueObjectSettings? other)
              && ConversionToKeyMemberType == other.ConversionToKeyMemberType
              && UnsafeConversionToKeyMemberType == other.UnsafeConversionToKeyMemberType
              && ConversionFromKeyMemberType == other.ConversionFromKeyMemberType
-             && DefaultStringComparison == other.DefaultStringComparison;
+             && DefaultStringComparison == other.DefaultStringComparison
+             && SerializationFrameworks == other.SerializationFrameworks;
    }
 
    public override int GetHashCode()
@@ -133,6 +136,7 @@ public override int GetHashCode()
          hashCode = (hashCode * 397) ^ (int)UnsafeConversionToKeyMemberType;
          hashCode = (hashCode * 397) ^ (int)ConversionFromKeyMemberType;
          hashCode = (hashCode * 397) ^ (int)DefaultStringComparison;
+         hashCode = (hashCode * 397) ^ (int)SerializationFrameworks;
 
          return hashCode;
       }

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/ValueObjects/ComplexSerializerGeneratorState.cs

@@ -5,6 +5,7 @@ namespace Thinktecture.CodeAnalysis.ValueObjects;
    public ITypeInformation Type { get; }
    public IReadOnlyList<InstanceMemberInfo> AssignableInstanceFieldsAndProperties { get; }
    public AttributeInfo AttributeInfo { get; }
+   public SerializationFrameworks SerializationFrameworks { get; }
 
    public string? Namespace => Type.Namespace;
    public IReadOnlyList<ContainingTypeState> ContainingTypes => Type.ContainingTypes;
@@ -14,16 +15,19 @@ namespace Thinktecture.CodeAnalysis.ValueObjects;
    public ComplexSerializerGeneratorState(
       ITypeInformation type,
       IReadOnlyList<InstanceMemberInfo> assignableInstanceFieldsAndProperties,
-      AttributeInfo attributeInfo)
+      AttributeInfo attributeInfo,
+      SerializationFrameworks serializationFrameworks)
    {
       Type = type;
       AssignableInstanceFieldsAndProperties = assignableInstanceFieldsAndProperties;
       AttributeInfo = attributeInfo;
+      SerializationFrameworks = serializationFrameworks;
    }
 
    public bool Equals(ComplexSerializerGeneratorState other)
    {
       return TypeInformationComparer.Instance.Equals(Type, other.Type)
+             && SerializationFrameworks == other.SerializationFrameworks
              && AssignableInstanceFieldsAndProperties.SequenceEqual(other.AssignableInstanceFieldsAndProperties)
              && ContainingTypes.SequenceEqual(other.ContainingTypes)
              && AttributeInfo.Equals(other.AttributeInfo);
@@ -39,6 +43,7 @@ public override int GetHashCode()
       unchecked
       {
          var hashCode = TypeInformationComparer.Instance.GetHashCode(Type);
+         hashCode = (hashCode * 397) ^ (int)SerializationFrameworks;
          hashCode = (hashCode * 397) ^ AssignableInstanceFieldsAndProperties.ComputeHashCode();
          hashCode = (hashCode * 397) ^ ContainingTypes.ComputeHashCode();
          hashCode = (hashCode * 397) ^ AttributeInfo.GetHashCode();

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/ValueObjects/JsonValueObjectCodeGeneratorFactory.cs

@@ -15,12 +15,14 @@ private JsonValueObjectCodeGeneratorFactory()
    public bool MustGenerateCode(KeyedSerializerGeneratorState state)
    {
       return !state.AttributeInfo.HasJsonConverterAttribute
+             && state.SerializationFrameworks.HasFlag(SerializationFrameworks.SystemTextJson)
              && (state.KeyMember is not null || state.AttributeInfo.DesiredFactories.Any(f => f.UseForSerialization.Has(SerializationFrameworks.SystemTextJson)));
    }
 
    public bool MustGenerateCode(ComplexSerializerGeneratorState state)
    {
       return !state.AttributeInfo.HasJsonConverterAttribute
+             && state.SerializationFrameworks.HasFlag(SerializationFrameworks.SystemTextJson)
              && !state.AttributeInfo.DesiredFactories.Any(f => f.UseForSerialization.Has(SerializationFrameworks.SystemTextJson));
    }
 

src/Thinktecture.Runtime.Extensions.SourceGenerator/CodeAnalysis/ValueObjects/MessagePackValueObjectCodeGeneratorFactory.cs

@@ -15,12 +15,14 @@ private MessagePackValueObjectCodeGeneratorFactory()
    public bool MustGenerateCode(KeyedSerializerGeneratorState state)
    {
       return !state.AttributeInfo.HasMessagePackFormatterAttribute
+             && state.SerializationFrameworks.HasFlag(SerializationFrameworks.MessagePack)
              && (state.KeyMember is not null || state.AttributeInfo.DesiredFactories.Any(f => f.UseForSerialization.Has(SerializationFrameworks.MessagePack)));
    }
 
    public bool MustGenerateCode(ComplexSerializerGeneratorState state)
    {
       return !state.AttributeInfo.HasMessagePackFormatterAttribute
+             && state.SerializationFrameworks.HasFlag(SerializationFrameworks.MessagePack)
              && !state.AttributeInfo.DesiredFactories.Any(f => f.UseForSerialization.Has(SerializationFrameworks.MessagePack));
    }