diff --git a/code/GridFieldAddNewMultiClass.php b/code/GridFieldAddNewMultiClass.php index e2d4b7b..8f3f5c0 100755 --- a/code/GridFieldAddNewMultiClass.php +++ b/code/GridFieldAddNewMultiClass.php @@ -17,6 +17,11 @@ class GridFieldAddNewMultiClass implements GridField_HTMLProvider, GridField_URL private $classes; + /** + * @var String + */ + protected $itemRequestClass = 'GridFieldAddNewMultiClassHandler'; + /** * @param string $fragment the fragment to render the button in */ @@ -129,8 +134,8 @@ class GridFieldAddNewMultiClass implements GridField_HTMLProvider, GridField_URL throw new SS_HTTPResponse_Exception(400); } - $handler = new GridFieldAddNewMultiClassHandler( - $grid, $component, new $class(), $grid->getForm()->getController(), 'add-multi-class' + $handler = Object::create($this->itemRequestClass, + $grid, $component, new $class(), $grid->getForm()->getController(), 'add-multi-class' ); $handler->setTemplate($component->getTemplate()); @@ -173,4 +178,12 @@ class GridFieldAddNewMultiClass implements GridField_HTMLProvider, GridField_URL ); } + /** + * @param String + */ + public function setItemRequestClass($class) { + $this->itemRequestClass = $class; + return $this; + } + }