JezK

Edit File: API.php

<?php

/**
 * Matomo - free/libre analytics platform
 *
 * @link    https://matomo.org
 * @license https://www.gnu.org/licenses/gpl-3.0.html GPL v3 or later
 */
namespace Piwik\Plugins\AIAgents;

use Piwik\API\Request;
use Piwik\DataTable;
use Piwik\DataTable\DataTableInterface;
use Piwik\Period;
use Piwik\Piwik;
use Piwik\Plugin\API as PluginAPI;
use Piwik\Plugins\API\DataTable\MergeDataTables;
use Piwik\Segment;
use Piwik\Segment\SegmentExpression;
use Piwik\Site;
class API extends PluginAPI
{
    public const AI_AGENT_SEGMENT = 'aiAgentName!=';
    public const AI_AGENT_COLUMN_SUFFIX = '_ai_agent';
    public const HUMAN_SEGMENT = 'aiAgentName==';
    public const HUMAN_COLUMN_SUFFIX = '_human';
    /**
     * @param int|string $idSite
     * @param string $segment
     * @param string|array<string> $columns
     */
    public function get($idSite, string $period, string $date, string $segment = '', $columns = '') : DataTableInterface
    {
        Piwik::checkUserHasViewAccess($idSite);
        $columns = Piwik::getArrayFromApiParameter($columns);
        if ($idSite === 'all' || count(Site::getIdSitesFromIdSitesString($idSite)) > 1) {
            $resultSet = new DataTable\Map();
            $resultSet->setKeyName('idSite');
        } elseif (Period::isMultiplePeriod($date, $period)) {
            $resultSet = new DataTable\Map();
            $resultSet->setKeyName('period');
        } else {
            $resultSet = new DataTable\Simple();
        }
        $clientTypes = [self::AI_AGENT_COLUMN_SUFFIX => self::AI_AGENT_SEGMENT, self::HUMAN_COLUMN_SUFFIX => self::HUMAN_SEGMENT];
        foreach ($clientTypes as $columnSuffix => $clientTypeSegment) {
            $modifiedSegment = Segment::combine($segment, SegmentExpression::AND_DELIMITER, $clientTypeSegment);
            $columnsForClientType = empty($columns) ? [] : $this->filterColumnsBySuffix($columns, $columnSuffix);
            // Only make the API call if either $columns is empty (i.e. no list of columns was passed in, so we
            // should fetch all columns) or if one of the columns that was passed in is for this client type
            if (!empty($columns) && empty($columnsForClientType)) {
                continue;
            }
            $params = ['idSite' => $idSite, 'period' => $period, 'date' => $date, 'segment' => $modifiedSegment, 'columns' => implode(',', $columnsForClientType), 'format' => 'original', 'format_metrics' => 0];
            /** @var DataTable\Map $response */
            $response = Request::processRequest('VisitsSummary.get', $params);
            $this->addSuffixToTableColumns($response, $columnSuffix);
            $merger = new MergeDataTables();
            $merger->mergeDataTables($resultSet, $response);
        }
        return $resultSet;
    }
    /**
     * @param array<string> $requestedColumns
     *
     * @return array<string>
     */
    protected function filterColumnsBySuffix(array $requestedColumns, string $suffix) : array
    {
        $columns = [];
        foreach ($requestedColumns as $column) {
            if (!str_ends_with($column, $suffix)) {
                continue;
            }
            $columns[] = str_replace($suffix, '', $column);
        }
        return $columns;
    }
    protected function addSuffixToTableColumns(DataTableInterface $table, string $suffix) : void
    {
        $renames = [];
        foreach ($table->getColumns() as $oldColumn) {
            $renames[$oldColumn] = $oldColumn . $suffix;
        }
        $table->filter('ReplaceColumnNames', [$renames]);
    }
}