Fix preparation of file uploads for new objects in ObjectUploadSubscriber

Author: niels-nijens

EventListener/ObjectUploadSubscriber.php

@@ -59,6 +59,7 @@ public function getSubscribedEvents()
     {
         return array(
             Events::preFlush,
+            Events::prePersist,
             Events::postPersist,
             Events::postUpdate,
             Events::postFlush,
@@ -85,6 +86,19 @@ public function preFlush(PreFlushEventArgs $args)
         }
     }
 
+    /**
+     * Prepares upload file references for a new object implementing the UploadObjectInterface.
+     *
+     * @param LifecycleEventArgs $args
+     */
+    public function prePersist(LifecycleEventArgs $args)
+    {
+        $object = $args->getObject();
+        if ($object instanceof UploadObjectInterface) {
+            $this->prepareUploadFileReferences($object);
+        }
+    }
+
     /**
      * Stores the file uploads.
      *