Merge pull request #128 from atrocore/attachments-removing

Added removing related assets when remove attachment
atrocore · Dec 20, 2023 · 56b84d5 · 56b84d5
2 parents 37dac0c + 80d1aaa
commit 56b84d5
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/app/Repositories/Attachment.php b/app/Repositories/Attachment.php
@@ -13,6 +13,7 @@
 
 namespace Dam\Repositories;
 
+use Atro\Core\PseudoTransactionManager;
 use Dam\Core\AssetValidator;
 use Espo\Core\Exceptions\BadRequest;
 use Espo\Core\Exceptions\Error;
@@ -89,6 +90,12 @@ protected function init()
         parent::init();
 
         $this->addDependency(AssetValidator::class);
+        $this->addDependency('pseudoTransactionManager');
+    }
+
+    protected function getPseudoTransactionManager(): PseudoTransactionManager
+    {
+        return $this->getInjection('pseudoTransactionManager');
     }
 
     /**
@@ -173,6 +180,32 @@ protected function afterRemove(Entity $entity, array $options = [])
 
             $this->getFileManager()->unlink($dirPath . '/page-1.png');
         }
+
+        if ($this->getMetadata()->get(['entityDefs', 'Asset', 'fields', 'file', 'required'], false)) {
+            $offset = 0;
+            $limit = 20;
+
+            while(true) {
+                $assets = $this
+                    ->getEntityManager()
+                    ->getRepository('Asset')
+                    ->where([
+                        'fileId' => $entity->id
+                    ])
+                    ->limit($offset, $limit)
+                    ->find();
+
+                if (count($assets) == 0) {
+                    break;
+                }
+
+                foreach ($assets as $asset) {
+                    $this->getPseudoTransactionManager()->pushDeleteEntityJob('Asset', $asset->id);
+                }
+
+                $offset += $limit;
+            }
+        }
     }
 
     protected function isPdf(Entity $entity): bool