Skip to content

Commit 75c608e

Browse files
jakobhellermannK0lb3
authored andcommitted
handle null returns from AssetsTools
1 parent 1f80718 commit 75c608e

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

TypeTreeGeneratorAPI/TypeTreeGenerator/AssetsTools/MonoBehaviourTemplateGeneratorPatch.cs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ namespace TypeTreeGeneratorAPI.TypeTreeGenerator.AssetsTools
99
{
1010
public interface IMonoBehaviourTemplateGeneratorPatch : IMonoBehaviourTemplateGenerator
1111
{
12-
virtual AssetTypeTemplateField GetTemplateFieldPatch(AssetTypeTemplateField templateField, string assemblyName, string nameSpace, string className, UnityVersion unityVersionExtra)
12+
virtual AssetTypeTemplateField? GetTemplateFieldPatch(AssetTypeTemplateField templateField, string assemblyName, string nameSpace, string className, UnityVersion unityVersionExtra)
1313
{
1414
return GetTemplateField(templateField, assemblyName, nameSpace, className, unityVersionExtra);
1515
}
@@ -44,7 +44,7 @@ public MonoCecilTempGeneratorPatch() : base("")
4444
{
4545
}
4646

47-
public AssetTypeTemplateField GetTemplateFieldPatch(AssetTypeTemplateField baseField, string assemblyName, string nameSpace, string className, UnityVersion unityVersion)
47+
public AssetTypeTemplateField? GetTemplateFieldPatch(AssetTypeTemplateField baseField, string assemblyName, string nameSpace, string className, UnityVersion unityVersion)
4848
{
4949
// 1:1 copy of the original method, but without filepath check and using the loadedAssemblies dictionary
5050
if (!assemblyName.EndsWith(".dll"))
@@ -54,6 +54,10 @@ public AssetTypeTemplateField GetTemplateFieldPatch(AssetTypeTemplateField baseF
5454
var asm = loadedAssemblies[assemblyName];
5555

5656
List<AssetTypeTemplateField> newFields = Read(asm, nameSpace, className, unityVersion);
57+
if (newFields == null)
58+
{
59+
return null;
60+
}
5761

5862
AssetTypeTemplateField newBaseField = baseField.Clone();
5963
newBaseField.Children.AddRange(newFields);

0 commit comments

Comments
 (0)