fix(ProvideAttribute): fix NullReferenceException during codegen

jonisavo · Nov 20, 2022 · 2b03eb7 · 2b03eb7
1 parent 43c8fba
commit 2b03eb7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/.../UIComponents.Roslyn.Generation/Generators/DependencyInjection/ProvideAugmentGenerator.cs b/.../UIComponents.Roslyn.Generation/Generators/DependencyInjection/ProvideAugmentGenerator.cs
@@ -29,7 +29,7 @@ private void GetProvideDescriptions(AugmentGenerationContext context, IList<Prov
                 if (!(member is IFieldSymbol fieldSymbol))
                     continue;
 
-                var memberType = RoslynUtilities.GetMemberType(member) as INamedTypeSymbol;
+                var memberType = fieldSymbol.Type;
 
                 if (memberType == null)
                     continue;
@@ -72,9 +72,13 @@ private void GetProvideDescriptions(AugmentGenerationContext context, IList<Prov
         protected override bool ShouldGenerateSource(AugmentGenerationContext context)
         {
             _provideDescriptions.Clear();
+
+            if (_provideAttributeSymbol == null)
+                return false;
+
             GetProvideDescriptions(context, _provideDescriptions);
 
-            return _provideAttributeSymbol != null && _provideDescriptions.Count > 0;
+            return _provideDescriptions.Count > 0;
         }
 
         protected override void BuildUsingStatements(StringBuilder stringBuilder)