handle UserDefinedPin

add some null checks for subCategoryObject

Author: dave-artificial-agency

src/controls/utils/color-utils.ts

@@ -13,7 +13,7 @@ export enum StructClass {
 export class ColorUtils {
 
     public static getPinColor(pin: PinProperty): string {
-        return this.getPinColorByCategory(pin.category, pin.subCategoryObject.class);
+        return this.getPinColorByCategory(pin.category, pin.subCategoryObject?.class);
     }
 
     public static getPinColorByCategory(category: PinCategory, subCategoryObject?: string): string {

src/parser/node-parsers/generic-node.parser.ts

@@ -84,6 +84,7 @@ export class GenericNodeParser extends NodeParser {
         [key: string]: () => CustomPropertyParser
     } = {
         "Pin": () => new PinPropertyParser(),
+        "UserDefinedPin": () => new PinPropertyParser(),
     }
 
     constructor() {
@@ -162,7 +163,8 @@ export class GenericNodeParser extends NodeParser {
                 data.node.customProperties.push(property);
 
                 if (property instanceof PinProperty) {
-                    if ((property as PinProperty).subCategoryObject.class === StructClass.LatentActionInfo) {
+                    const pinProp = property as PinProperty;
+                    if (pinProp.subCategoryObject && pinProp.subCategoryObject.class === StructClass.LatentActionInfo) {
                         data.node.latent = true;
                     }
                 }

src/parser/pin-property.parser.ts

@@ -25,6 +25,13 @@ export class PinPropertyParser implements CustomPropertyParser {
         "PinName": (p: PinProperty, value: string) => { p.name = prettifyText(BlueprintParserUtils.parseString(value)); },
         "PinFriendlyName": (p: PinProperty, value: string) => { p.friendlyName = prettifyText(PinPropertyParser.parsePinFriendlyName(value)); },
         "PinType.PinCategory": (p: PinProperty, value: string) => { p.category = PinPropertyParser.parsePinCategory(value); },
+        "PinType": (p: PinProperty, value: string) => {
+            // UserDefinedPin format: PinType=(PinCategory="bool")
+            const categoryMatch = value.match(/PinCategory="([^"]+)"/);
+            if (categoryMatch) {
+                p.category = PinPropertyParser.parsePinCategory(`"${categoryMatch[1]}"`);
+            }
+        },
         "Direction": (p: PinProperty, value: string) => { p.direction = PinPropertyParser.parseDirection(value); },
         "DesiredPinDirection": (p: PinProperty, value: string) => { p.direction = PinPropertyParser.parseDirection(value); },
         "PinToolTip": (p: PinProperty, value: string) => { p.toolTip = BlueprintParserUtils.parseString(value); },
@@ -271,9 +278,11 @@ export class PinPropertyParser implements CustomPropertyParser {
             case PinCategory.bool:
                 return { control: CheckBoxControl, data: (BlueprintParserUtils.parseString(value).toLowerCase() === "true") };
             case PinCategory.struct:
-                return this.parseDefaultValueStruct(p.subCategoryObject.class, value);
+                if (p.subCategoryObject && p.subCategoryObject.class) {
+                    return this.parseDefaultValueStruct(p.subCategoryObject.class, value);
+                }
             case PinCategory.byte:
-                if (p.subCategoryObject.type === "Enum") {
+                if (p.subCategoryObject && p.subCategoryObject.type === "Enum") {
                     return { control: TextBoxControl, data: BlueprintParserUtils.parseEnumValue(p.subCategoryObject.class, value) };
                 } else {
                     return { control: TextBoxControl, data: BlueprintParserUtils.parseString(value) };
@@ -366,13 +375,15 @@ export class PinPropertyParser implements CustomPropertyParser {
                 p.defaultValue = "  ";
                 p.defaultValueControlClass = TextBoxControl;
             case PinCategory.struct:
-                switch (p.subCategoryObject.class) {
-                    case StructClass.VECTOR2D:
-                        p.defaultValue = [
-                            { key: 'X', value: '0.0' },
-                            { key: 'Y', value: '0.0' }];
-                        p.defaultValueControlClass = StructBoxControl;
-                        break;
+                if (p.subCategoryObject && p.subCategoryObject.class) {
+                    switch (p.subCategoryObject.class) {
+                        case StructClass.VECTOR2D:
+                            p.defaultValue = [
+                                { key: 'X', value: '0.0' },
+                                { key: 'Y', value: '0.0' }];
+                            p.defaultValueControlClass = StructBoxControl;
+                            break;
+                    }
                 }
                 break;
         }