LeKoala\ExcelImportExport\ExcelGridFieldExportButton

Complexity

Total Complexity

82

Size/Duplication

Total Lines	594
Duplicated Lines	0 %

Importance

Changes	8
Bugs	3	Features	0

35 Methods

Rating	Name	Duplication	Size	Complexity
A	getURLHandlers()	0	3	1
A	getActions()	0	3	1
A	getActionName()	0	4	1
A	handleAction()	0	8	2
A	getHTMLFragments()	0	23	2
A	__construct()	0	4	1
A	setListFilters()	0	4	1
A	setIgnoreFilters()	0	4	1
A	setHasHeader()	0	4	1
F	generateExportFileData()	0	103	26
A	getSanitizeXls()	0	3	1
A	getAfterExportCallback()	0	3	1
A	getListFilters()	0	3	1
A	setExportType()	0	7	2
A	setCheckCanView()	0	4	1
A	sanitizeValue()	0	20	5
A	getRealExportColumns()	0	4	2
A	setButtonTitle()	0	4	1
A	setExportName()	0	4	1
A	getIsLimited()	0	3	1
A	handleExport()	0	29	3
A	setSanitizeXls()	0	4	1
A	getExportColumns()	0	3	1
A	getExportType()	0	3	1
A	getHasHeader()	0	3	1
A	setAfterExportCallback()	0	4	1
A	isSanitizeEnabled()	0	5	3
A	getIgnoreFilters()	0	3	1
A	updateExportName()	0	11	3
A	getCheckCanView()	0	3	1
A	getExportName()	0	3	1
A	setIsLimited()	0	4	1
A	setExportColumns()	0	4	1
B	retrieveList()	0	29	9
A	getButtonTitle()	0	3	1

<?php

namespace LeKoala\ExcelImportExport;

use Generator;
use InvalidArgumentException;
use SilverStripe\ORM\DataList;
use SilverStripe\ORM\ArrayList;
use SilverStripe\ORM\SS_List;
use SilverStripe\Control\HTTPRequest;
use LeKoala\SpreadCompat\SpreadCompat;

The type LeKoala\SpreadCompat\SpreadCompat was not found. Maybe you did not declare it correctly or list all dependencies?

use SilverStripe\Assets\FileNameFilter;
use SilverStripe\Forms\GridField\GridField;
use SilverStripe\Forms\GridField\GridFieldPaginator;
use SilverStripe\Forms\GridField\GridField_FormAction;
use SilverStripe\Forms\GridField\GridField_URLHandler;
use SilverStripe\Forms\GridField\GridFieldFilterHeader;
use SilverStripe\Forms\GridField\GridField_HTMLProvider;
use SilverStripe\Forms\GridField\GridFieldSortableHeader;
use SilverStripe\Forms\GridField\GridField_ActionProvider;

/**
 * Adds an "Export list" button to the bottom of a {@link GridField}.
 */
class ExcelGridFieldExportButton implements
    GridField_HTMLProvider,
    GridField_ActionProvider,
    GridField_URLHandler
{
    /**
     * Map of a property name on the exported objects, with values being the column title in the file.
     * Note that titles are only used when {@link $hasHeader} is set to TRUE.
     * @var array<int|string,mixed>
     */
    protected ?array $exportColumns;

    /**
     * Fragment to write the button to
     */
    protected string $targetFragment;

    protected bool $hasHeader = true;

    protected string $exportType = 'xlsx';

    protected ?string $exportName = null;

    protected ?string $buttonTitle = null;

    protected bool $checkCanView = true;

    protected bool $isLimited = true;

    /**
     * @var array<mixed>
     */
    protected array $listFilters = [];

    /**
     *
     * @var null|callable
     */
    protected $afterExportCallback;

    protected bool $ignoreFilters = false;

    protected bool $sanitizeXls = true;

    /**
     * @param string $targetFragment The HTML fragment to write the button into
     * @param array<string> $exportColumns The columns to include in the export
     */
    public function __construct($targetFragment = "after", $exportColumns = null)
    {
        $this->targetFragment = $targetFragment;
        $this->exportColumns = $exportColumns;
    }

    /**
     * @param GridField $gridField
     * @return string
     */
    public function getActionName($gridField)
    {
        $name = strtolower($gridField->getName());
        return 'excelexport_' . $name . '_' . $this->exportType;
    }

    /**
     * Place the export button in a <p> tag below the field
     * @return array<string|int,mixed>
     */
    public function getHTMLFragments($gridField)
    {
        $defaultTitle = _t(
            'ExcelImportExport.FORMATEXPORT',
            'Export to {format}',
            ['format' => $this->exportType]
        );
        $title = $this->buttonTitle ? $this->buttonTitle : $defaultTitle;

        $name = $this->getActionName($gridField);

        $button = new GridField_FormAction(
            $gridField,
            $name,
            $title,
            $name,
            []
        );
        $button->addExtraClass('btn btn-secondary no-ajax font-icon-down-circled action_export');
        $button->setForm($gridField->getForm());

        return array(
            $this->targetFragment => $button->Field()
        );
    }

    /**
     * export is an action button
     * @param GridField $gridField
     * @return array<string>
     */
    public function getActions($gridField)
    {
        return array($this->getActionName($gridField));
    }

    /**
     * @param GridField $gridField
     * @param string $actionName
     * @param array<mixed> $arguments
     * @param array<mixed> $data
     * @return void
     */
    public function handleAction(
        GridField $gridField,
        $actionName,
        $arguments,
        $data
    ) {
        if (in_array($actionName, $this->getActions($gridField))) {
            $this->handleExport($gridField);
        }
    }

    /**
     * it is also a URL
     * @param GridField $gridField
     * @return array<string,string>
     */
    public function getURLHandlers($gridField)
    {
        return array($this->getActionName($gridField) => 'handleExport');
    }

    /**
     * Handle the export, for both the action button and the URL
     * @param GridField $gridField
     * @param HTTPRequest $request
     * @return void
     */
    public function handleExport($gridField, $request = null)
    {
        $now = date("Ymd_Hi");

        $this->updateExportName($gridField);

        $data = $this->generateExportFileData($gridField);

        $ext = $this->exportType;
        $name = $this->exportName;
        $fileName = "$name-$now.$ext";

        if ($this->afterExportCallback) {
            $func = $this->afterExportCallback;
            $func();
        }

        $opts = [
            'extension' => $ext,
        ];

        if ($ext != 'csv') {
            $end = ExcelImportExport::getLetter(count($this->getRealExportColumns($gridField)));
            $opts['creator'] = ExcelImportExport::config()->default_creator;
            $opts['autofilter'] = "A1:{$end}1";
        }

        SpreadCompat::output($data, $fileName, ...$opts);
        exit();

Using exit here is not recommended.

    }


    /**
     * Make sure export name is a valid file name
     * @param GridField|\LeKoala\Tabulator\TabulatorGrid $gridField

The type LeKoala\Tabulator\TabulatorGrid was not found. Maybe you did not declare it correctly or list all dependencies?

     * @return void
     */
    protected function updateExportName($gridField)
    {
        $filter = new FileNameFilter;
        if ($this->exportName) {
            $this->exportName = $filter->filter($this->exportName);
        } else {
            $class = $gridField->getModelClass();
            $singl = singleton($class);
            $plural = $class ? $singl->i18n_plural_name() : '';

            $this->exportName = $filter->filter('export-' . $plural);
        }
    }

    /**
     * @param GridField|\LeKoala\Tabulator\TabulatorGrid $gridField
     * @return DataList|ArrayList|SS_List|null
     */
    protected function retrieveList($gridField)
    {
        // Remove GridFieldPaginator as we're going to export the entire list.
        $gridField->getConfig()->removeComponentsByType(GridFieldPaginator::class);

        /** @var DataList|ArrayList $items */
        $items = $gridField->getManipulatedList();

        // Keep filters
        if (!$this->ignoreFilters) {
            foreach ($gridField->getConfig()->getComponents() as $component) {
                if ($component instanceof GridFieldFilterHeader || $component instanceof GridFieldSortableHeader) {
                    //@phpstan-ignore-next-line
                    $items = $component->getManipulatedData($gridField, $items);
                }
            }
        }

        $list = $items;
        $limit = ExcelImportExport::getExportLimit();
        if ($list instanceof DataList) {
            if ($this->isLimited && $limit > 0) {
                $list = $list->limit($limit);
            }
            if (!empty($this->listFilters)) {
                $list = $list->filter($this->listFilters);
            }
        }
        return $list;
    }

    /**
     * @param GridField|\LeKoala\Tabulator\TabulatorGrid $gridField
     * @return array<int|string,mixed|null>
     */
    protected function getRealExportColumns($gridField)
    {
        $class = $gridField->getModelClass();
        return ($this->exportColumns) ? $this->exportColumns : ExcelImportExport::exportFieldsForClass($class);
    }

    /**
     * Generate export fields for Excel.
     *
     * @param GridField|\LeKoala\Tabulator\TabulatorGrid $gridField
     */
    public function generateExportFileData($gridField): Generator
    {
        $columns = $this->getRealExportColumns($gridField);

        if ($this->hasHeader) {
            $headers = [];

            // determine the headers. If a field is callable (e.g. anonymous function) then use the
            // source name as the header instead
            foreach ($columns as $columnSource => $columnHeader) {
                //@phpstan-ignore-next-line
                if (is_array($columnHeader) && array_key_exists('title', $columnHeader ?? [])) {
                    $headers[] = $columnHeader['title'];
                } else {
                    $headers[] = (!is_string($columnHeader) && is_callable($columnHeader)) ? $columnSource : $columnHeader;
                }
            }

            yield $headers;
        }

        $list = $this->retrieveList($gridField);

        if (!$list) {
            return;
        }

        // Auto format using DBField methods based on column name
        $export_format = ExcelImportExport::config()->export_format;

        $sanitize = $this->isSanitizeEnabled();

        foreach ($list as $item) {
            // This can be really slow for large exports depending on how canView is implemented
            if ($this->checkCanView) {
                $canView = true;
                if ($item->hasMethod('canView') && !$item->canView()) {
                    $canView = false;
                }
                if (!$canView) {
                    continue;
                }
            }

            $dataRow = [];

            // Loop and transforms records as needed
            foreach ($columns as $columnSource => $columnHeader) {
                if (!is_string($columnHeader) && is_callable($columnHeader)) {
                    if ($item->hasMethod($columnSource)) {
                        $relObj = $item->{$columnSource}();
                    } else {
                        $relObj = $item->relObject($columnSource);
                    }

                    $value = $columnHeader($relObj);
                } else {
                    if (is_string($columnSource)) {
                        // It can be a method
                        if (strpos($columnSource, '(') !== false) {
                            $matches = [];
                            preg_match('/([a-zA-Z]*)\((.*)\)/', $columnSource, $matches);
                            $func = $matches[1];
                            $params = explode(",", $matches[2]);
                            // Support only one param for now
                            $value = $item->$func($params[0]);
                        } else {
                            if (array_key_exists($columnSource, $export_format)) {
                                $format = $export_format[$columnSource];
                                $value = $item->dbObject($columnSource)->$format();
                            } else {
                                $value = $gridField->getDataFieldValue($item, $columnSource);
                            }
                        }
                    } else {
                        // We can also use a simple dot notation
                        $parts = explode(".", $columnHeader);

It seems like $columnHeader can also be of type null; however, parameter $string of explode() does only seem to accept string, maybe add an additional type check?

                        if (count($parts) == 1) {
                            $value = $item->$columnHeader;
                        } else {
                            $value = $item->relObject($parts[0]);
                            if ($value) {
                                $relObjField = $parts[1];
                                $value = $value->$relObjField;
                            }
                        }
                    }
                }

                // @link https://owasp.org/www-community/attacks/CSV_Injection
                // [SS-2017-007] Sanitise XLS executable column values with a leading tab
                if ($sanitize && $value && is_string($value)) {
                    $value = self::sanitizeValue($value);
                }

                $dataRow[] = $value;
            }

            if ($item->hasMethod('destroy')) {
                $item->destroy();
            }

            yield $dataRow;
        }
    }

    /**
     * Sanitization is necessary for csv
     * It can be turned off if needed
     * If we have no chars to sanitize, it's not enabled
     *
     * @return boolean
     */
    public function isSanitizeEnabled(): bool
    {
        $sanitize_xls_chars = ExcelImportExport::config()->sanitize_xls_chars ?? "=";
        $sanitize = $this->sanitizeXls && $sanitize_xls_chars && $this->exportType == "csv";
        return $sanitize;
    }

    /**
     * @link https://owasp.org/www-community/attacks/CSV_Injection
     * [SS-2017-007] Sanitise XLS executable column values with a leading tab
     */
    public static function sanitizeValue(string $value = null): ?string
    {
        if (!$value) {
            return $value;
        }

        $sanitize_xls_chars = ExcelImportExport::config()->sanitize_xls_chars ?? "=";
        $sanitize_xls_chars_len = strlen($sanitize_xls_chars);

        // If we have only one char we can make it simpler
        if ($sanitize_xls_chars_len === 1) {
            if ($value[0] === $sanitize_xls_chars) {
                $value = "\t" . $value;
            }
        } else {
            if (preg_match('/^[' . $sanitize_xls_chars . '].*/', $value)) {
                $value = "\t" . $value;
            }
        }
        return $value;
    }

    /**
     * @return array<int|string,mixed>
     */
    public function getExportColumns()
    {
        return $this->exportColumns;
    }

    /**
     * @param array<int|string,mixed> $cols
     * @return $this
     */
    public function setExportColumns($cols)
    {
        $this->exportColumns = $cols;
        return $this;
    }

    /**
     * @return boolean
     */
    public function getHasHeader()
    {
        return $this->hasHeader;
    }

    /**
     * @param boolean $bool
     * @return $this
     */
    public function setHasHeader($bool)
    {
        $this->hasHeader = $bool;
        return $this;
    }

    /**
     * @return string
     */
    public function getExportType()
    {
        return $this->exportType;
    }

    /**
     * @param string $exportType xlsx (default), xls or csv
     * @return $this
     */
    public function setExportType($exportType)
    {
        if (!in_array($exportType, ['xls', 'xlsx', 'csv'])) {
            throw new InvalidArgumentException("Export type must be one of : xls, xlsx, csv");
        }
        $this->exportType = $exportType;
        return $this;
    }

    /**
     * @return string
     */
    public function getExportName()
    {
        return $this->exportName;
    }

    /**
     * @param string $exportName
     * @return ExcelGridFieldExportButton
     */
    public function setExportName($exportName)
    {
        $this->exportName = $exportName;
        return $this;
    }

    /**
     * @return string
     */
    public function getButtonTitle()
    {
        return $this->buttonTitle;
    }

    /**
     * @param string $buttonTitle
     * @return ExcelGridFieldExportButton
     */
    public function setButtonTitle($buttonTitle)
    {
        $this->buttonTitle = $buttonTitle;
        return $this;
    }

    /**
     *
     * @return bool
     */
    public function getCheckCanView()
    {
        return $this->checkCanView;
    }

    /**
     *
     * @param bool $checkCanView
     * @return ExcelGridFieldExportButton
     */
    public function setCheckCanView($checkCanView)
    {
        $this->checkCanView = $checkCanView;
        return $this;
    }

    /**
     *
     * @return array<mixed>
     */
    public function getListFilters()
    {
        return $this->listFilters;
    }

    /**
     *
     * @param array<mixed> $listFilters
     * @return ExcelGridFieldExportButton
     */
    public function setListFilters($listFilters)
    {
        $this->listFilters = $listFilters;
        return $this;
    }

    /**
     *
     * @return callable
     */
    public function getAfterExportCallback()
    {
        return $this->afterExportCallback;
    }

    /**
     *
     * @param callable $afterExportCallback
     * @return ExcelGridFieldExportButton
     */
    public function setAfterExportCallback(callable $afterExportCallback)
    {
        $this->afterExportCallback = $afterExportCallback;
        return $this;
    }

    /**
     * Get the value of isLimited
     */
    public function getIsLimited(): bool
    {
        return $this->isLimited;
    }

    /**
     * Set the value of isLimited
     *
     * @param bool $isLimited
     * @return self
     */
    public function setIsLimited(bool $isLimited)
    {
        $this->isLimited = $isLimited;
        return $this;
    }

    /**
     * Get the value of ignoreFilters
     */
    public function getIgnoreFilters(): bool
    {
        return $this->ignoreFilters;
    }

    /**
     * Set the value of ignoreFilters
     *
     * @param bool $ignoreFilters
     */
    public function setIgnoreFilters(bool $ignoreFilters): self
    {
        $this->ignoreFilters = $ignoreFilters;
        return $this;
    }

    /**
     * Get the value of sanitizeXls
     */
    public function getSanitizeXls(): bool
    {
        return $this->sanitizeXls;
    }

    /**
     * Set the value of sanitizeXls
     *
     * @param bool $sanitizeXls
     */
    public function setSanitizeXls(bool $sanitizeXls): self
    {
        $this->sanitizeXls = $sanitizeXls;
        return $this;
    }
}