Store original source object, SelectField will convert to an array

Author: wilr

javascript/ImageSelect.jquery.js

@@ -32,7 +32,6 @@
 
             // Attach Ready event before continue with chose
             this.each(function(input_field) {
-
                 $this = $(this);
 
                 $this.on("chosen:ready", function change(e, chosen){
@@ -52,7 +51,6 @@
 
                     var options = form_field.options;
                     var spans = $(chosen.container).find('.chosen-choices span');
-
                     if(options && options.length){
 
                         for(var i = 0 ; i < options.length; i++){

src/DropdownImageField.php

@@ -6,11 +6,18 @@
 use SilverStripe\View\Requirements;
 use SilverStripe\View\ArrayData;
 use SilverStripe\ORM\ArrayList;
+use SilverStripe\ORM\SS_List;
 use SilverStripe\Forms\FormField;
 
 class DropdownImageField extends DropdownField {
 
-    protected $keyField, $labelField, $imageField;
+    protected $keyField = 'ID';
+
+    protected $labelField = 'Title';
+
+    protected $imageField = 'Image';
+
+    protected $sourceObject;
 
     public function __construct($name, $title, $sourceObject, $keyField = 'ID', $labelField = 'Title', $imageField = 'Image', $value = '', $form = null) {
 
@@ -21,13 +28,29 @@ public function __construct($name, $title, $sourceObject, $keyField = 'ID', $lab
         parent::__construct($name, ($title === null) ? $name : $title, $sourceObject, $value, $form);
 
         $this->addExtraClass('dropdown');
+        $this->sourceObject = $sourceObject;
+    }
+
+    public function setSourceObject(SS_List $source)
+    {
+        $this->sourceObject = $source;
+
+        return $this;
+    }
+
+    public function setImageField($field)
+    {
+        $this->imageField = $field;
+
+        return $this;
     }
 
     public function Field($properties = array()) {
 
         $dirName = basename(dirname(dirname(__FILE__)));
-        $source = $this->getSource();
+        $source = $this->sourceObject;
         $options = array();
+
         if ($source) {
             if (is_object($source) && $this->hasEmptyDefault) {
                 $options[] = new ArrayData([
@@ -37,16 +60,22 @@ public function Field($properties = array()) {
                 ]);
             }
 
-            foreach ($source as $item) {
-                $value = $item->{$this->keyField};
-                if (empty($this->labelField)) {
-                    $title = '--nbsp';
+            foreach ($source as $k => $item) {
+                if (is_object($item)) {
+                    $value = $item->{$this->keyField};
+                    if (empty($this->labelField)) {
+                        $title = '--nbsp';
+                    } else {
+                        $title = $item->{$this->labelField};
+                    }
+
+                    $image = $item->{$this->imageField}();
                 } else {
-                    $title = $item->{$this->labelField};
+                    $value = $k;
+                    $image = null;
+                    $title = $item;
                 }
 
-                $image = $item->{$this->imageField}();
-
                 $selected = false;
                 if ($value === '' && ($this->value === '' || $this->value === null)) {
                     $selected = true;