NEW Scaffold formfields for File and Image using new API

silverstripe · Jun 4, 2024 · 57a9026 · 57a9026
1 parent 159e152
commit 57a9026
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/src/File.php b/src/File.php
@@ -16,6 +16,8 @@
 use SilverStripe\Core\Manifest\ModuleLoader;
 use SilverStripe\Core\Resettable;
 use SilverStripe\Forms\FieldList;
+use SilverStripe\Forms\FileHandleField;
+use SilverStripe\Forms\FormField;
 use SilverStripe\Forms\HTMLReadonlyField;
 use SilverStripe\Forms\TextField;
 use SilverStripe\ORM\ArrayList;
@@ -587,6 +589,19 @@ public function getCMSFields()
         return $fields;
     }
 
+    public function scaffoldFieldForHasOne(
+        string $fieldName,
+        string $fieldTitle,
+        string $relationName,
+        DataObject $ownerRecord
+    ): FormField&FileHandleField {
+        $field = Injector::inst()->create(FileHandleField::class, $relationName, $fieldTitle);
+        if ($field->hasMethod('setAllowedMaxFileNumber')) {
+            $field->setAllowedMaxFileNumber(1);
+        }
+        return $field;
+    }
+
     /**
      * Get title for current file status
      *

diff --git a/src/Image.php b/src/Image.php
@@ -2,6 +2,10 @@
 
 namespace SilverStripe\Assets;
 
+use SilverStripe\Forms\FileHandleField;
+use SilverStripe\Forms\FormField;
+use SilverStripe\ORM\DataObject;
+
 /**
  * Represents an Image
  */
@@ -40,6 +44,17 @@ public function __construct($record = null, $isSingleton = false, $queryParams =
         $this->File->setAllowedCategories('image/supported');
     }
 
+    public function scaffoldFieldForHasOne(
+        string $fieldName,
+        string $fieldTitle,
+        string $relationName,
+        DataObject $ownerRecord
+    ): FormField&FileHandleField {
+        $field = parent::scaffoldFieldForHasOne($fieldName, $fieldTitle, $relationName, $ownerRecord);
+        $field->setAllowedFileCategories('image/supported');
+        return $field;
+    }
+
     public function getIsImage()
     {
         return true;