Fix

Author: inputfalken

src/DynamoDBGenerator.SourceGenerator/Generations/AttributeExpressionValue.cs

@@ -63,7 +63,7 @@ MarshallerOptions options
 
         var yields = (typeIdentifier switch
             {
-                {IsNullable:true}  => $"if ({param} is null)".CreateScope(
+                { IsNullable: true } => $"if ({param} is null)".CreateScope(
                     $"yield return new ({self}.Value, {AttributeValueUtilityFactory.Null});", "yield break;"),
                 { TypeSymbol.IsReferenceType: true } => $"if ({param} is null)".CreateScope(
                     $"throw {ExceptionHelper.NullExceptionMethod}(\"{structName}\");"),
@@ -129,30 +129,29 @@ internal static IEnumerable<string> CreateClasses(DynamoDBMarshallerArguments[]
     private static CodeFactory CreateStruct(TypeIdentifier typeIdentifier, Func<ITypeSymbol, DynamoDbDataMember[]> fn,
         MarshallerOptions options)
     {
-        var dataMembers =
-            options.IsConvertable(typeIdentifier.TypeSymbol)
-                ? Array
-                    .Empty<(bool IsUnknown, DynamoDbDataMember DDB, string AttributeReference, string
-                        AttributeInterfaceName)>()
-                : fn(typeIdentifier.TypeSymbol)
-                    .Select(x => (
-                        IsUnknown: options.IsUnknown(x.DataMember.TypeIdentifier),
-                        DDB: x,
-                        AttributeReference: TypeName(x.DataMember.TypeIdentifier.TypeSymbol),
-                        AttributeInterfaceName:
-                        $"{Constants.DynamoDBGenerator.Marshaller.AttributeExpressionValueTrackerInterface}<{x.DataMember.TypeIdentifier.AnnotatedString}>"
-                    ))
-                    .ToArray();
+        var dataMembers = options.IsConvertable(typeIdentifier)
+            ? Array.Empty<(bool IsUnknown, DynamoDbDataMember DDB, string AttributeReference, string AttributeInterfaceName)>()
+            : fn(typeIdentifier.TypeSymbol)
+                .Select(x => (
+                    IsUnknown: options.IsUnknown(x.DataMember.TypeIdentifier),
+                    DDB: x,
+                    AttributeReference: TypeName(x.DataMember.TypeIdentifier.TypeSymbol),
+                    AttributeInterfaceName:
+                    $"{Constants.DynamoDBGenerator.Marshaller.AttributeExpressionValueTrackerInterface}<{x.DataMember.TypeIdentifier.AnnotatedString}>"
+                ))
+                .ToArray();
 
         var structName = TypeName(typeIdentifier.TypeSymbol);
         var interfaceName =
             $"{Constants.DynamoDBGenerator.Marshaller.AttributeExpressionValueTrackerInterface}<{typeIdentifier.AnnotatedString}>";
 
         var @struct =
-            $"public readonly struct {structName} : {interfaceName}".CreateScope(TypeContents(typeIdentifier, dataMembers,
+            $"public readonly struct {structName} : {interfaceName}".CreateScope(TypeContents(typeIdentifier,
+                dataMembers,
                 structName, interfaceName, options));
 
-        return new CodeFactory(@struct, dataMembers.Where(x => x.IsUnknown).Select(x => x.DDB.DataMember.TypeIdentifier));
+        return new CodeFactory(@struct,
+            dataMembers.Where(x => x.IsUnknown).Select(x => x.DDB.DataMember.TypeIdentifier));
     }
 
     [MethodImpl(MethodImplOptions.AggressiveInlining)]

src/DynamoDBGenerator.SourceGenerator/Types/MarshallerOptions.cs

@@ -17,9 +17,8 @@ public readonly struct MarshallerOptions
     public string FullName { get; }
     public string FieldDeclaration { get; }
 
-    public bool IsUnknown(TypeIdentifier typeSymbol) =>
-        typeSymbol is UnknownType && IsConvertable(typeSymbol.TypeSymbol) is false;
-
+    public bool IsUnknown(TypeIdentifier typeIdentifier) => typeIdentifier is UnknownType && IsConvertable(typeIdentifier) is false;
+    public bool IsConvertable(TypeIdentifier typeIdentifier) => typeIdentifier.TypeSymbol.TypeKind is TypeKind.Enum || Converters.ContainsKey(typeIdentifier.TypeSymbol);
 
     private MarshallerOptions(
         INamedTypeSymbol originalType,
@@ -90,10 +89,6 @@ or ConversionStrategy.UpperCase
         return null;
     }
 
-    public bool IsConvertable(ITypeSymbol typeSymbol)
-    {
-        return typeSymbol.TypeKind is TypeKind.Enum || Converters.ContainsKey(typeSymbol);
-    }
 
     private Dictionary<ISymbol?, KeyValuePair<string, ITypeSymbol>> Converters { get; }