Handle case where arity is 0.

GGG-KILLER · Jan 22, 2024 · 5d46aec · 5d46aec
1 parent 86eadf8
commit 5d46aec
Showing 1 changed file with 17 additions and 4 deletions.
diff --git a/Tsu.TreeSourceGen/src/Generator.cs b/Tsu.TreeSourceGen/src/Generator.cs
@@ -171,7 +171,8 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                                     writer.Write(arg);
                                     writer.WriteLine(" is not null)");
                                     writer.Indent++;
-                                    writer.Write("return ");
+                                    if (visitor.Arity > 0)
+                                        writer.Write("return ");
                                     writer.Write(arg);
                                     writer.Write(".Accept(");
                                     writer.Write("this");
@@ -183,8 +184,11 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                                     }
                                     writer.WriteLine(");");
 
-                                    writer.WriteLine();
-                                    writer.WriteLine("return default;");
+                                    if (visitor.Arity > 0)
+                                    {
+                                        writer.WriteLine();
+                                        writer.WriteLine("return default;");
+                                    }
                                     writer.Indent--;
                                     writer.Write('}');
                                 }
@@ -211,7 +215,16 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
                                         writer.Write(" arg");
                                         writer.Write(idx);
                                     }
-                                    writer.WriteLine(") => default;");
+                                    if (visitor.Arity > 0)
+                                    {
+                                        writer.WriteLine(") => default;");
+                                    }
+                                    else
+                                    {
+                                        writer.WriteLine(')');
+                                        writer.WriteLine('{');
+                                        writer.WriteLine('}');
+                                    }
                                 }
 
                                 foreach (var node in tree.Nodes)