i-lateral / silverstripe-GridFieldAddOns

src/GridFieldUserColumns.php

<?php

namespace SilverStripe\GridFieldAddOns;

use SilverStripe\ORM\ArrayList;
use SilverStripe\View\ArrayData;
use SilverStripe\Security\Member;
use SilverStripe\Security\Security;
use SilverStripe\View\Requirements;
use SilverStripe\View\ViewableData;
use SilverStripe\Core\Config\Config;
use SilverStripe\ORM\ValidationException;
use SilverStripe\Forms\GridField\GridField_URLHandler;
use SilverStripe\Forms\GridField\GridFieldDataColumns;
use SilverStripe\Forms\GridField\GridField_HTMLProvider;
use SilverStripe\Forms\GridField\GridField_ColumnProvider;
use SilverStripe\Forms\GridField\GridFieldConfig_RecordEditor;
use SilverStripe\Forms\GridField\GridFieldConfig_RelationEditor;

class GridFieldUserColumns extends ViewableData implements GridField_ColumnProvider, GridField_HTMLProvider, GridField_URLHandler
{

    static $static_field_for_extra_columns = 'extra_summary_fields';

    protected $gridField;
    protected $default_columns;

    function init($gridField)
    {
        $this->gridField = $gridField;
    }

    /**
     * This component is not providing columns but manipulating the columns provided by GridFieldDataColumns.
     *
     * @param GridField $gridField
     * @param array - List reference of all column names.
     */
    public function augmentColumns($gridField, &$columns)
    {

        $this->init($gridField);

        Requirements::javascript('i-lateral/silverstripe-gridfield-addons:javascript/GridFieldUserColumns.js');
        Requirements::css('i-lateral/silverstripe-gridfield-addons:css/GridFieldUserColumns.css');

        $usercolumns = $this->currentColumns();
        $extracolumns = array_diff($columns, $this->availableColumns());
        $displaycolumns = array_values(array_unique(array_merge(array_keys($usercolumns), $extracolumns)));

        $datacolumnscomponent = $gridField->getConfig()->getComponentByType(GridFieldDataColumns::class);
        $datacolumnscomponent->setDisplayFields($usercolumns);

        $columns = $displaycolumns;
    }

    function defaultColumns()
    {

        if (!$this->default_columns) {
            if (!$this->gridField) {
                throw new ValidationException('GridField not yet set. Do not call GridFieldUserColumns::defaultColumns() before GridFieldUserColumns::augmentColumns().');
            }
            $datacolumnscomponent = $this->gridField->getConfig()->getComponentByType(GridFieldDataColumns::class);
            $this->default_columns = $datacolumnscomponent->getDisplayFields($this->gridField);
        }

        return $this->default_columns;
    }

    function userColumns()
    {

        if (!$this->gridField) {
            throw new ValidationException('GridField not yet set. Do not call GridFieldUserColumns::userColumns() before GridFieldUserColumns::augmentColumns().');
        }
        $member = Security::getCurrentUser();
        if ($member->hasField('GridFieldUserColumns') &&
            $member->GridFieldUserColumns &&
            ($usercolumns = $member->getGridFieldUserColumnsFor($this->gridField->getList()->dataClass()))
        ) {
            return $usercolumns;
        }
        return false;
    }

    function currentColumns()
    {
        $user = $this->userColumns();
        return is_array($user) ? $user : $this->defaultColumns();
    }

    function availableColumns()
    {
        if (!$this->gridField) {
            throw new ValidationException('GridField not yet set. Do not call GridFieldUserColumns::userColumns() before GridFieldUserColumns::augmentColumns().');
        }

        $class = $this->gridField->getList()->dataClass();
        $default = $this->defaultColumns();
        $user = $this->userColumns();
        $extra_cols = Config::inst()->get($class, self::$static_field_for_extra_columns);
        $extra = $this->addFieldLabels($extra_cols);

        $default = is_array($extra) ? array_merge($default, $extra) : $default;

        return is_array($user) ? array_merge($user, $default) : $default;
    }

    /**
     * Look for and add any needed field labels (used mostly for extra_summary_fields)
     * and return new array
     *
     * @return array
     */
    protected function addFieldLabels($fields)
    {
        $class = $this->gridField->getList()->dataClass();
        $obj = $class::singleton();

        // Merge associative / numeric keys
        $return = [];
        foreach ($fields as $key => $value) {
            if (is_int($key)) {
                $key = $value;
            }
            $return[$key] = $value;
        }

        if (!$return) {
            $return = array();
        }

        // Localize fields (if possible)
        foreach ($obj->fieldLabels(false) as $name => $label) {
            // only attempt to localize if the label definition is the same as the field name.
            // this will preserve any custom labels set in the summary_fields configuration
            if (isset($return[$name]) && $name === $return[$name]) {
                $return[$name] = $label;
            }
        }

        return $return;
    }

    // since we're not really provide columns we're not returning anything
    public function getColumnsHandled($gridField)
    {
        return array();
    }
    public function getColumnMetadata($gridField, $columnName)
    {
        return array();
    }
    public function getColumnContent($gridField, $record, $columnName)
    {
        return false;
    }
    public function getColumnAttributes($gridField, $record, $columnName)
    {
        return array();
    }

    /**
     * Returns a map where the keys are fragment names and the values are pieces of HTML to add to these fragments.
     *
     * Here are 4 built-in fragments: 'header', 'footer', 'before', and 'after', but components may also specify
     * fragments of their own.
     *
     * To specify a new fragment, specify a new fragment by including the text "$DefineFragment(fragmentname)" in the
     * HTML that you return.  Fragment names should only contain alphanumerics, -, and _.
     *
     * If you attempt to return HTML for a fragment that doesn't exist, an exception will be thrown when the GridField
     * is rendered.
     *
     * @return Array
     */
    public function getHTMLFragments($gridField)
    {

        $gridField->getList()->dataClass();
        $buttonlabel = _t('GridFieldAddOns.ChangeColumns', "Change Columns");

        return array(
            'buttons-before-right' => "<a class=\"action btn h-100 btn-outline-secondary action gridfield-setup\" data-icon=\"settings\">$buttonlabel</a>",
        );
    }

    public function getURLHandlers($gridField)
    {
        return array(
            'usercolumnsform' => 'UserColumnsForm',
            'saveusercolumns' => 'saveUserColumns',
        );
    }

    function UserColumnsForm($gridField, $request)

The parameter $request is not used and could be removed.

    {
        $this->init($gridField);
        return $this->renderWith(
            self::class.'_Form'
        );
    }

    function Columns()
    {
        $available = $this->availableColumns();
        $current = $this->currentColumns();

        $columns = new ArrayList();
        foreach ($available as $key => $val) {
            $selected = array_search($val, $current) !== false;
            $columns->push(new ArrayData(array('Name' => $key, 'Title' => $val, 'Selected' => $selected)));
        }
        return $columns;
    }

    function saveUserColumns($gridField, $request)
    {
        $this->init($gridField);
        $available = $this->availableColumns();
        $postcolumns = $request->postVar('columns');
        $newcolumns = array();
        if (!is_array($postcolumns)) {
            return json_encode('bad');
        }
        foreach ($postcolumns as $col) {
            list($name, $title) = explode(':', $col);
            if (!isset($available[$name]) || $available[$name] != $title) {
                return json_encode(array('bad', $this->gridField->getList()->dataClass()));
            }
            $newcolumns[$name] = $title;
        }
        Security::getCurrentUser()->setGridFieldUserColumnsFor($gridField->getList()->dataClass(), $newcolumns);
        return json_encode('good');
    }
}