bugfix: show Assets as field types

basilicom · Jan 18, 2023 · d972fcd · d972fcd
1 parent 8b83e1b
commit d972fcd
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/ERDiagramXMLExportBundle/Command/CreateERDiagramXMLExportCommand.php b/ERDiagramXMLExportBundle/Command/CreateERDiagramXMLExportCommand.php
@@ -74,13 +74,20 @@ private function processFieldDefinitions($fieldDefinitions): array
             $fields = [$fieldDefinition->getName() => $fieldDefinition->getFieldtype()];
 
             if ($fieldDefinition instanceof AbstractRelations) {
-                $allowedTypes = array_map(function ($classConfig) use ($fieldDefinition) {
-                    if ($fieldDefinition instanceof ManyToOneRelation) {
-                        return $classConfig['classes'] ?? '';
-                    } else {
-                        return ($classConfig['classes'] . '[]') ?? '';
-                    }
-                }, $fieldDefinition->getClasses());
+                $allowedTypes = [];
+                if ($fieldDefinition->getObjectsAllowed()) {
+                    $allowedTypes = array_map(function ($classConfig) use ($fieldDefinition) {
+                        if ($fieldDefinition instanceof ManyToOneRelation) {
+                            return $classConfig['classes'] ?? '';
+                        } else {
+                            return ($classConfig['classes'] . '[]') ?? '';
+                        }
+                    }, $fieldDefinition->getClasses());
+                }
+
+                if ($fieldDefinition->getAssetsAllowed() || $fieldDefinition->getDocumentsAllowed()) {
+                    $allowedTypes[] = $fieldDefinition instanceof ManyToOneRelation ? 'Asset' : 'Asset[]';
+                }
 
                 $fields = [$fieldDefinition->getName() => !empty($allowedTypes) ? implode(' | ', $allowedTypes) : ''];
             }