Merge pull request #55 from onvardgmbh/allowed-mime-types

Allowed mime types

Author: bjoernhasse

README.md

@@ -205,6 +205,11 @@ Formendpoint::make('application_form', __('ApplicationForm', 'wptheme'), 'bundle
     ->add_fields([
         // [..]
         Input::make('file', 'avatar', __('Avatar', 'wptheme')),
+    ])
+    //Define allowed mime types (this is optional)
+    ->setAllowedMimeTypes([
+        'jpg' => 'image/jpeg',
+        'png' => 'image/png',
     ]);
 ```
 

src/Formendpoint.php

@@ -183,6 +183,10 @@ public function handleformsubmit()
                 continue;
             }
             $uploadedFiles = [];
+            $overrides = ['action' => $this->posttype];
+            if (!empty($input->allowedMimeTypes)) {
+                $overrides['mimes'] = $input->allowedMimeTypes;
+            }
 
             if ('files' === $input->type) {
                 $files = $_FILES[$input->name];
@@ -195,15 +199,15 @@ public function handleformsubmit()
                             'error' => $files['error'][$key],
                             'size' => $files['size'][$key],
                         ];
-                        $file = wp_handle_upload($file, ['action' => $this->posttype]);
+                        $file = wp_handle_upload($file, $overrides);
                         if (!$file || isset($file['error'])) {
                             wp_die('Fileupload failed.', '', ['response' => 400]);
                         }
                         $uploadedFiles[] = $file;
                     }
                 }
             } else {
-                $file = wp_handle_upload($_FILES[$input->name], ['action' => $this->posttype]);
+                $file = wp_handle_upload($_FILES[$input->name], $overrides);
                 if (!$file || isset($file['error'])) {
                     wp_die('Fileupload failed.', '', ['response' => 400]);
                 }

src/Input.php

@@ -13,6 +13,7 @@ class Input
     public $validationFunction;
     public $title;
     public $type;
+    public $allowedMimeTypes;
 
     public static function make($type, $name, $label = null)
     {
@@ -74,6 +75,13 @@ public function setFormat($callback)
         return $this;
     }
 
+    public function setAllowedMimeTypes($array)
+    {
+        $this->allowedMimeTypes = $array;
+
+        return $this;
+    }
+
     public function hide()
     {
         $this->hide = true;