X-Git-Url: http://www.aleph1.co.uk/gitweb/?p=yaffs-website;a=blobdiff_plain;f=web%2Fmodules%2Fcontrib%2Fentity_browser%2Fsrc%2FControllers%2FEntityBrowserController.php;fp=web%2Fmodules%2Fcontrib%2Fentity_browser%2Fsrc%2FControllers%2FEntityBrowserController.php;h=c3788809defe424a1254a557308caf66d941d5ba;hp=aebb4be8060a5e9c0441aed6b862f3d9bb270239;hb=059867c3f96750652c80f39e44c442a58c2549ee;hpb=f8fc16ae6b862bef59baaad5d051dd37b7ff11b2

diff --git a/web/modules/contrib/entity_browser/src/Controllers/EntityBrowserController.php b/web/modules/contrib/entity_browser/src/Controllers/EntityBrowserController.php
index aebb4be80..c3788809d 100644
--- a/web/modules/contrib/entity_browser/src/Controllers/EntityBrowserController.php
+++ b/web/modules/contrib/entity_browser/src/Controllers/EntityBrowserController.php
@@ -29,8 +29,16 @@ class EntityBrowserController extends ControllerBase {
    *   containing the edit form.
    */
   public function entityBrowserEdit(EntityInterface $entity, Request $request) {
+
+    // Use edit form class if it exists, otherwise use default form class.
+    $operation = 'default';
+    $entity_type = $entity->getEntityType();
+    if ($entity_type->getFormClass('edit')) {
+      $operation = 'edit';
+    }
+
     // Build the entity edit form.
-    $form_object = $this->entityTypeManager()->getFormObject($entity->getEntityTypeId(), 'edit');
+    $form_object = $this->entityTypeManager()->getFormObject($entity->getEntityTypeId(), $operation);
     $form_object->setEntity($entity);
     $form_state = (new FormState())
       ->setFormObject($form_object)