From 7768cff79498ecee9908da131b2af7e1e3a84bd7 Mon Sep 17 00:00:00 2001
From: Rekkonnect <kalfas.alex@gmail.com>
Date: Mon, 24 Apr 2023 14:45:55 +0300
Subject: [PATCH] Fix nullref on visitors with no acceptors

---
 VisitorPatternGenerator/VisitorPatternGenerator.cs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/VisitorPatternGenerator/VisitorPatternGenerator.cs b/VisitorPatternGenerator/VisitorPatternGenerator.cs
index 8e43116..21fb626 100644
--- a/VisitorPatternGenerator/VisitorPatternGenerator.cs
+++ b/VisitorPatternGenerator/VisitorPatternGenerator.cs
@@ -1,4 +1,3 @@
-using System;
 using System.Collections.Generic;
 using System.Collections.Immutable;
 using System.Linq;
@@ -82,6 +81,11 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
             context.RegisterSourceOutput(visitorProvider, static (ctx, e) => {
                 var ((visitor, acceptors), rootNamespace) = e;
                 ctx.CancellationToken.ThrowIfCancellationRequested();
+
+                if (acceptors.IsDefaultOrEmpty) {
+                    return;
+                }
+
                 _AddVisitorSource(ctx, rootNamespace, visitor, acceptors);
             });
         }