From 950ed9539da1f9a40c15398668da31677e9b6f45 Mon Sep 17 00:00:00 2001
From: Jan Trejbal <jan@trejbal.land>
Date: Fri, 26 Jul 2024 11:38:59 +0200
Subject: [PATCH] Update MapperGenerator

---
 src/Riok.Mapperly/MapperGenerator.cs | 27 ++++++++++++---------------
 1 file changed, 12 insertions(+), 15 deletions(-)

diff --git a/src/Riok.Mapperly/MapperGenerator.cs b/src/Riok.Mapperly/MapperGenerator.cs
index bc20ddcfc37..812e460d40e 100644
--- a/src/Riok.Mapperly/MapperGenerator.cs
+++ b/src/Riok.Mapperly/MapperGenerator.cs
@@ -1,4 +1,5 @@
-using Microsoft.CodeAnalysis;
+using System.Collections.Immutable;
+using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp;
 using Riok.Mapperly.Abstractions;
 using Riok.Mapperly.Configuration;
@@ -29,25 +30,21 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
         context.ReportDiagnostics(compilationDiagnostics);
 
         var nestedCompilations = context
-            .MetadataReferencesProvider.Select(
-                (metadataReference, _) =>
-                {
-                    if (metadataReference is CompilationReference compilationReference)
-                    {
-                        return compilationReference;
-                    }
-
-                    return null;
-                }
-            )
-            .Where(compilationReference => compilationReference is not null)
-            .Select((compilationReference, _) => compilationReference!.Compilation)
+            .MetadataReferencesProvider.Select((metadataReference, _) => (metadataReference as CompilationReference)?.Compilation)
             .Collect();
 
         // build the compilation context
         var compilationContext = context
             .CompilationProvider.Combine(nestedCompilations)
-            .Select(static (c, _) => new CompilationContext(c.Left, new WellKnownTypes(c.Left), c.Right, new FileNameBuilder()))
+            .Select(
+                static (c, _) =>
+                    new CompilationContext(
+                        c.Left,
+                        new WellKnownTypes(c.Left),
+                        c.Right.OfType<Compilation>().ToImmutableArray(),
+                        new FileNameBuilder()
+                    )
+            )
             .WithTrackingName(MapperGeneratorStepNames.BuildCompilationContext);
 
         // build the assembly default configurations