Restricting rules when an attachment file can actually be deleted (correctly handling exercise cloning and shared links).

Author: krulis-martin

app/V1Module/presenters/ExerciseFilesPresenter.php

@@ -17,6 +17,7 @@
 use App\Model\Entity\SupplementaryExerciseFile;
 use App\Model\Entity\UploadedFile;
 use App\Model\Entity\AttachmentFile;
+use App\Model\Repository\Assignments;
 use App\Model\Repository\AttachmentFiles;
 use App\Model\Repository\Exercises;
 use App\Model\Entity\Exercise;
@@ -37,6 +38,12 @@ class ExerciseFilesPresenter extends BasePresenter
      */
     public $exercises;
 
+    /**
+     * @var Assignments
+     * @inject
+     */
+    public $assignments;
+
     /**
      * @var UploadedFiles
      * @inject
@@ -345,7 +352,37 @@ public function actionDeleteAttachmentFile(string $id, string $fileId)
         $exercise->removeAttachmentFile($file);
         $this->exercises->flush();
 
-        $this->fileStorage->deleteAttachmentFile($file);
+        $this->attachmentFiles->refresh($file);
+        if ($file->getExercises()->isEmpty()) {
+            // file has no attachments to exercises, let's check the assignments
+            $isUsed = false;
+            foreach ($file->getAssignments() as $assignment) {
+                $group = $assignment->getGroup();
+                if ($group && !$group->isArchived()) {
+                    $isUsed = true;  // only non-archived assignments are considered relevant
+                    break;
+                }
+            }
+
+            if (!$isUsed) {
+                if ($file->getAssignments()->isEmpty()) {
+                    // only if no attachments exists (except for deleted ones)
+                    // remove all links to deleted entities and remove the file record
+                    foreach ($file->getExercisesAndIReallyMeanAllOkay() as $exercise) {
+                        $exercise->removeAttachmentFile($file);
+                        $this->exercises->persist($exercise, false);
+                    }
+                    foreach ($file->getAssignmentsAndIReallyMeanAllOkay() as $assignment) {
+                        $assignment->removeAttachmentFile($file);
+                        $this->assignments->persist($assignment, false);
+                    }
+
+                    $this->attachmentFiles->remove($file);
+                }
+
+                $this->fileStorage->deleteAttachmentFile($file);
+            }
+        }
 
         $this->sendSuccessResponse("OK");
     }

app/model/entity/AttachmentFile.php

@@ -32,6 +32,14 @@ function (Exercise $exercise) {
         );
     }
 
+    /**
+     * @return Collection
+     */
+    public function getExercisesAndIReallyMeanAllOkay()
+    {
+        return $this->exercises;
+    }
+
     /**
      * @ORM\ManyToMany(targetEntity="Assignment", mappedBy="attachmentFiles")
      */
@@ -49,6 +57,13 @@ function (Assignment $assignment) {
         );
     }
 
+    /**
+     * @return Collection
+     */
+    public function getAssignmentsAndIReallyMeanAllOkay()
+    {
+        return $this->assignments;
+    }
 
     /**
      * AttachmentFile constructor.