FIX Add actions to injector configuration

MasonD · Jan 25, 2017 · 67a4c28 · 67a4c28
1 parent be46d60
commit 67a4c28
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 4 deletions.
diff --git a/_config/injetor.yml → _config/injector.yml b/_config/injetor.yml → _config/injector.yml
@@ -28,3 +28,18 @@ SilverStripe\Core\Injector\Injector:
     class: UncleCheese\BetterButtons\Buttons\BetterButton_SaveDraft
   BetterButton_Unpublish:
     class: UncleCheese\BetterButtons\Buttons\BetterButton_Unpublish
+
+  BetterButtonAction:
+    class: UncleCheese\BetterButtons\Actions\BetterButton
+  BetterButtonCancelAction:
+    class: UncleCheese\BetterButtons\Actions\BetterButtonCancelAction
+  BetterButtonCustomAction:
+    class: UncleCheese\BetterButtons\Actions\BetterButtonCustomAction
+  BetterButtonFrontendLinksAction:
+    class: UncleCheese\BetterButtons\Actions\BetterButtonFrontendLinksAction
+  BetterButtonLink:
+    class: UncleCheese\BetterButtons\Actions\BetterButtonLink
+  BetterButtonNestedForm:
+    class: UncleCheese\BetterButtons\Actions\BetterButtonNestedForm
+  BetterButtonPrevNextAction:
+    class: UncleCheese\BetterButtons\Actions\BetterButtonPrevNextAction
diff --git a/src/Extensions/BetterButtonDataObject.php b/src/Extensions/BetterButtonDataObject.php
@@ -174,10 +174,10 @@ protected function createFieldList($buttons)
      */
     protected function instantiateButton($className)
     {
-        if (class_exists($className)) {
-            $buttonObj = Injector::inst()->create($className);
-            return $buttonObj;
-        } else {
+        try {
+            return Injector::inst()->create($className);
+        } catch (Exception $ex) {
+            // Customize the default injector exception
             throw new Exception("The button type $className doesn't exist.");
         }
     }