flack / openpsa

src/midcom/grid/provider.php

<?php
/**
 * @package midcom.grid
 * @author CONTENT CONTROL http://www.contentcontrol-berlin.de/
 * @copyright CONTENT CONTROL http://www.contentcontrol-berlin.de/
 * @license http://www.gnu.org/licenses/gpl.html GNU General Public License
 */

namespace midcom\grid;

use midcom\grid\provider\client;
use midcom_core_query;
use midcom_error;
use midcom;

/**
 * Manager for retrieving grid data by AJAX
 *
 * @package midcom.grid
 */
class provider
{
    /**
     * The class responsible for getting and formatting rows
     */
    private client $_client;

    /**
     * The rows to show
     */
    private ?array $_rows = null;

    /**
     * The total number of rows
     */
    private ?int $_total_rows = null;

    /**
     * How many items should be shown per page
     */
    private int $_results_per_page = 20;

    /**
     * The current offset
     */
    private int $_offset = 0;

    /**
     * The field for sorting
     */
    private ?string $_sort_field = null;

    /**
     * The direction for sorting (ASC or DESC)
     */
    private string $_sort_direction = 'ASC';

    /**
     * The grid we're working with
     */
    private ?grid $_grid = null;

    /**
     * The datatype we're working with
     */
    private string $_datatype;

    /**
     * The midcom query object
     */
    private ?midcom_core_query $_query = null;

    /**
     * Search parameters
     */
    private array $_search = [];

    public function __construct(array|client $source, string $datatype = 'json')
    {
        $this->_datatype = $datatype;
        if ($source instanceof client) {

$source is never a sub-type of midcom\grid\provider\client.

            $this->_client = $source;
        } else {
            $this->set_rows($source);
        }
    }

    /**
     * Adds an initial order to the resultset.
     *
     * This can be overwritten by GET parameters
     */
    public function add_order(string $field, string $direction = 'ASC')
    {
        $this->_sort_field = $field;
        $this->_sort_direction = $direction;
    }

    public function set_grid(grid $grid)
    {
        $this->_grid = $grid;
        $this->_grid->set_provider($this);
        $this->_datatype = $grid->get_option('datatype');
    }

    public function get_grid(?string $identifier = null) : grid
    {
        if (null !== $identifier) {
            $this->_grid = new grid($identifier, $this->_datatype);
            $this->_grid->set_provider($this);
            if (!empty($this->_sort_field)) {
                $this->_grid->set_option('sortname', $this->_sort_field);
                $this->_grid->set_option('sortorder', strtolower($this->_sort_direction));
            }
        }
        return $this->_grid;
    }

    public function set_rows(array $rows)
    {
        $this->_rows = $rows;
        if ($this->_datatype == 'local') {
            $this->_total_rows = count($this->_rows);
        }
    }

    public function get_rows() : array
    {
        if ($this->_rows === null) {
            $this->_get_rows();
        }
        return $this->_rows;
    }

    public function set_query(midcom_core_query $query)
    {
        $this->_rows = null;
        $this->_total_rows = null;
        $this->_query = $query;
    }

    /**
     * returns the query (uncached)
     */
    public function get_query() : midcom_core_query
    {
        if ($this->_datatype == 'json') {
            $this->_parse_query($_GET);
        }
        $field = $this->_sort_field;
        if ($field !== null) {
            $field = str_replace('index_', '', $field);
        }

        return $this->_client->get_qb($field, $this->_sort_direction, $this->_search);
    }

    public function count_rows() : int
    {
        $this->_total_rows ??= $this->_prepare_query()->count();
        return $this->_total_rows;
    }

    public function get_column_total(string $column)
    {
        $ret = 0;
        $rows = $this->get_rows();
        foreach ($rows as $row) {
            if (array_key_exists($column, $row)) {
                $ret += $row[$column];
            }
        }
        return $ret;
    }

    public function setup_grid()
    {
        if ($this->_datatype == 'local') {
            $this->_grid->prepend_js($this->_convert_to_localdata());

The method prepend_js() does not exist on null.

            $this->_grid->set_option('data', $this->_grid->get_identifier() . '_entries', false);
            if (null === $this->_get_grid_option('rowNum')) {
                $this->_grid->set_option('rowNum', $this->count_rows());
            }
        }
    }

    public function render()
    {
        match ($this->_datatype) {
            'json' => $this->_render_json(),
            'local' => $this->get_grid()->render()
        };
    }

    private function _get_grid_option(string $key, $default = null)
    {
        return $this->_grid ? $this->_grid->get_option($key) : $default;
    }

    private function _convert_to_localdata() : string
    {
        return "var " . $this->_grid->get_identifier() . '_entries = ' .  json_encode($this->get_rows()) . ";\n";
    }

    private function _render_json()
    {
        $rows = $this->get_rows();
        $this->_total_rows ??= count($rows);

        $response = [
            'total' => ceil($this->_total_rows / $this->_results_per_page),
            'page' => ($this->_offset / $this->_results_per_page) + 1,
            'records' => $this->_total_rows,
            'rows' => $rows
        ];
        midcom::get()->header('Content-Type: application/json; charset=UTF-8');

        echo json_encode($response);
    }

    private function _parse_query(array $query)
    {
        if (!empty($query['rows'])) {
            $this->_results_per_page = (int) $query['rows'];
            if (!empty($query['page'])) {
                $this->_offset = ($this->_results_per_page * ($query['page'] - 1));
            }
        }
        if (!empty($query['sidx'])) {
            $this->_sort_field = $query['sidx'];
            $this->_sort_direction = strtoupper($query['sord'] ?? 'ASC');
        }
        if (   !empty($query['_search'])
            && $query['_search'] === 'true') {
            foreach ($query as $field => $value) {
                if (in_array($field, ['_search', 'nd', 'page', 'rows', 'sidx', 'sord'])) {
                    continue;
                }
                $this->_search[str_replace('index_', '', $field)] = $value;
            }
        }
    }

    private function _prepare_query() : midcom_core_query
    {
        $this->_query ??= $this->get_query();
        return $this->_query;
    }

    private function _get_rows()
    {
        $query = $this->_prepare_query();

        $this->_total_rows = $query->count();

        if (   $this->_datatype == 'json'
            && !empty($this->_results_per_page)) {
            $query->set_limit($this->_results_per_page);
            if (!empty($this->_offset)) {
                $query->set_offset($this->_offset);
            }
        }
        $this->_rows = [];

        if ($query instanceof \midcom_core_collector) {
            $items = $query->get_objects();
        } else {
            $items = $query->execute();
        }
        foreach ($items as $item) {
            $this->_rows[] = $this->_client->get_row($item);
        }
    }
}