Only add files to module if module is not null.

tritao · tritao · commit f03a56f0a30b · 2020-11-18T01:26:58.000Z
diff --git a/src/Generator/Driver.cs b/src/Generator/Driver.cs
@@ -319,7 +319,9 @@ public void SaveCode(IEnumerable<GeneratorOutput> outputs)
 
                     var file = Path.Combine(outputPath, fileRelativePath);
                     WriteGeneratedCodeToFile(file, template.Generate());
-                    output.TranslationUnit.Module.CodeFiles.Add(file);
+
+                    if (output.TranslationUnit.Module != null)
+                        output.TranslationUnit.Module.CodeFiles.Add(file);
 
                     Diagnostics.Message("Generated '{0}'", fileRelativePath);
                 }

Original file line number	Diff line number	Diff line change
`@@ -319,7 +319,9 @@ public void SaveCode(IEnumerable<GeneratorOutput> outputs)`
`319`	`319`
`320`	`320`	`var file = Path.Combine(outputPath, fileRelativePath);`
`321`	`321`	`WriteGeneratedCodeToFile(file, template.Generate());`
`322`		`- output.TranslationUnit.Module.CodeFiles.Add(file);`
	`322`	`+`
	`323`	`+ if (output.TranslationUnit.Module != null)`
	`324`	`+ output.TranslationUnit.Module.CodeFiles.Add(file);`
`323`	`325`
`324`	`326`	`Diagnostics.Message("Generated '{0}'", fileRelativePath);`
`325`	`327`	`}`