silverstripe / silverstripe-dms

code/cms/DMSGridFieldEditButton.php

<?php

class DMSGridFieldEditButton extends GridFieldEditButton

PSR1 recommends that each class must be in a namespace of at least one level to avoid collisions.

{
    /**
     * Overriding the parent method to change the template that the DMS add button will be rendered with
     *
     * @param GridField $gridField
     * @param DataObject $record
     * @param string $columnName
     * @return HTMLText
     */
    public function getColumnContent($gridField, $record, $columnName)
    {
        $data = new ArrayData(array(
            'Link' => Controller::join_links($gridField->Link('item'), $record->ID, 'edit')
        ));
        $template = $record->canEdit() ? 'GridFieldEditButton' : 'GridFieldViewButton';
        return $data->renderWith($template);
    }
}