Isolate creation of TypeIdentifier

Author: inputfalken

src/DynamoDBGenerator.SourceGenerator/Generations/AttributeExpressionName.cs

@@ -25,7 +25,7 @@ internal static IEnumerable<string> CreateClasses(DynamoDBMarshallerArguments[]
 
         return arguments
             .SelectMany(x =>
-                CodeFactory.Create(x.EntityTypeSymbol.TypeIdentifier(), y => CreateStruct(y, getDynamoDbProperties, options), hashSet));
+                CodeFactory.Create(x.EntityTypeSymbol, y => CreateStruct(y, getDynamoDbProperties, options), hashSet));
     }
 
     private static IEnumerable<string> TypeContent(

src/DynamoDBGenerator.SourceGenerator/Generations/AttributeExpressionValue.cs

@@ -123,7 +123,7 @@ internal static IEnumerable<string> CreateClasses(DynamoDBMarshallerArguments[]
 
         return arguments
             .SelectMany(x =>
-                CodeFactory.Create(x.ArgumentType.TypeIdentifier(), y => CreateStruct(y, getDynamoDbProperties, options), hashSet));
+                CodeFactory.Create(x.ArgumentType, y => CreateStruct(y, getDynamoDbProperties, options), hashSet));
     }
 
     private static CodeFactory CreateStruct(TypeIdentifier typeIdentifier, Func<ITypeSymbol, DynamoDbDataMember[]> fn,

src/DynamoDBGenerator.SourceGenerator/Generations/Marshalling/Keys.cs

@@ -78,7 +78,7 @@ internal static IEnumerable<string> CreateKeys(IEnumerable<DynamoDBMarshallerArg
             var hashSet = new HashSet<TypeIdentifier>(TypeIdentifier.Nullable);
 
             return arguments
-                .SelectMany(x => CodeFactory.Create(x.EntityTypeSymbol.TypeIdentifier(),
+                .SelectMany(x => CodeFactory.Create(x.EntityTypeSymbol,
                     y => StaticAttributeValueDictionaryKeys(y, getDynamoDbProperties, options), hashSet));
         }
 

src/DynamoDBGenerator.SourceGenerator/Generations/Marshalling/Marshaller.cs

@@ -64,11 +64,11 @@ private static IEnumerable<string> TypeContent(DynamoDBMarshallerArguments[] arg
 
         return arguments.SelectMany(x => CodeFactory
                 .Create(
-                    x.EntityTypeSymbol.TypeIdentifier(),
+                    x.EntityTypeSymbol,
                     y => CreateMethod(y, getDynamoDbProperties, options),
                     hashset
                 )
-                .Concat(CodeFactory.Create(x.ArgumentType.TypeIdentifier(), y => CreateMethod(y, getDynamoDbProperties, options),
+                .Concat(CodeFactory.Create(x.ArgumentType, y => CreateMethod(y, getDynamoDbProperties, options),
                     hashset))
             )
             .Concat(KeyMarshaller.CreateKeys(arguments, getDynamoDbProperties, options));

src/DynamoDBGenerator.SourceGenerator/Generations/UnMarshaller.cs

@@ -216,7 +216,7 @@ private static IEnumerable<string> CreateTypeContents(IEnumerable<DynamoDBMarsha
         var hashSet = new HashSet<TypeIdentifier>(TypeIdentifier.Nullable);
         return arguments.SelectMany(x =>
             CodeFactory.Create(
-                x.EntityTypeSymbol.TypeIdentifier(),
+                x.EntityTypeSymbol,
                 y => CreateMethod(y, getDynamoDbProperties, options),
                 hashSet
             )

src/DynamoDBGenerator.SourceGenerator/Types/CodeFactory.cs

@@ -1,3 +1,4 @@
+using DynamoDBGenerator.SourceGenerator.Extensions;
 using Microsoft.CodeAnalysis;
 
 namespace DynamoDBGenerator.SourceGenerator.Types;
@@ -27,6 +28,12 @@ public CodeFactory(IEnumerable<string> lines)
     private readonly IEnumerable<TypeIdentifier> _dependantTypes;
 
     public static IEnumerable<string> Create(
+        ITypeSymbol typeSymbol,
+        Func<TypeIdentifier, CodeFactory> codeSelector,
+        ISet<TypeIdentifier> handledTypes
+    ) => Execute(typeSymbol.TypeIdentifier(), codeSelector, handledTypes);
+
+    private static IEnumerable<string> Execute(
         TypeIdentifier typeSymbol,
         Func<TypeIdentifier, CodeFactory> codeSelector,
         ISet<TypeIdentifier> handledTypes
@@ -42,7 +49,7 @@ ISet<TypeIdentifier> handledTypes
             yield return s;
 
         foreach (var nestedTypeSymbol in code._dependantTypes)
-        foreach (var nestedCode in Create(nestedTypeSymbol, codeSelector, handledTypes))
+        foreach (var nestedCode in Execute(nestedTypeSymbol, codeSelector, handledTypes))
             yield return nestedCode;
     }
 }