Merge pull request #1252 from michalkleiner/pulls/1203-attach-file-pe…

…r-recipient-hook Add extension hook to control file attachments per recipient and field
silverstripe · Nov 12, 2023 · 4b85a33 · 4b85a33
2 parents 79e1fb8 + 26a4816
commit 4b85a33
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/code/Control/UserDefinedFormController.php b/code/Control/UserDefinedFormController.php
@@ -355,7 +355,8 @@ public function process($data, $form)
 
                         // attach a file to recipient email only if lower than configured size
                         if ($file->getAbsoluteSize() <= $this->getMaximumAllowedEmailAttachmentSize()) {
-                            $attachments[] = $file;
+                            // using the field name as array index is fine as file upload field only allows one file
+                            $attachments[$field->Name] = $file;
                         }
                     }
                 }
@@ -393,17 +394,22 @@ public function process($data, $form)
                 $mergeFields = $this->getMergeFieldsMap($emailData['Fields']);
 
                 if ($attachments && (bool) $recipient->HideFormData === false) {
-                    foreach ($attachments as $file) {
+                    foreach ($attachments as $uploadFieldName => $file) {
                         /** @var File $file */
                         if ((int) $file->ID === 0) {
                             continue;
                         }
 
-                        $email->addAttachmentFromData(
-                            $file->getString(),
-                            $file->getFilename(),
-                            $file->getMimeType()
-                        );
+                        $canAttachFileForRecipient = true;
+                        $this->extend('updateCanAttachFileForRecipient', $canAttachFileForRecipient, $recipient, $uploadFieldName, $file);
+
+                        if ($canAttachFileForRecipient) {
+                            $email->addAttachmentFromData(
+                                $file->getString(),
+                                $file->getFilename(),
+                                $file->getMimeType()
+                            );
+                        }
                     }
                 }