events/Classes/Service/DestinationDataImportService/Slugger.php

<?php

declare(strict_types=1);

/*
 * Copyright (C) 2022 Daniel Siepmann <coding@daniel-siepmann.de>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */

namespace Wrm\Events\Service\DestinationDataImportService;

use TYPO3\CMS\Core\Database\ConnectionPool;
use TYPO3\CMS\Core\Database\Query\QueryBuilder;
use TYPO3\CMS\Core\DataHandling\SlugHelper;
use TYPO3\CMS\Core\Utility\GeneralUtility;
use Wrm\Events\Service\DestinationDataImportService\Slugger\Registry;
use Wrm\Events\Service\DestinationDataImportService\Slugger\SluggerType;

class Slugger
{
    /**
     * @var Registry
     */
    private $registry;

    /**
     * @var ConnectionPool
     */
    private $connectionPool;

    public function __construct(
        Registry $registry,
        ConnectionPool $connectionPool
    ) {
        $this->registry = $registry;
        $this->connectionPool = $connectionPool;
    }

    public function update(string $tableName): void
    {
        $sluggerType = $this->registry->get($tableName);
        foreach ($this->getRecords($sluggerType) as $record) {
            $this->updateRecord($sluggerType, $record);
        }
    }

    /**
     * @return \Generator<array>
     */
    private function getRecords(SluggerType $sluggerType): \Generator
    {
        $tableName = $sluggerType->getSupportedTableName();
        $slugColumn = $sluggerType->getSlugColumn();
        $queryBuilder = $this->getQueryBuilder($tableName);

        $statement = $queryBuilder->select('*')
            ->from($tableName)
            ->where(
                $queryBuilder->expr()->orX(
                    $queryBuilder->expr()->eq($slugColumn, $queryBuilder->createNamedParameter('', \PDO::PARAM_STR)),
                    $queryBuilder->expr()->isNull($slugColumn)
                )
            )
            ->execute();

        while ($record = $statement->fetch()) {
            if (is_array($record) === false) {
                continue;
            }

            yield $record;
        }
    }

    private function updateRecord(SluggerType $sluggerType, array $record): void
    {
        $tableName = $sluggerType->getSupportedTableName();
        $record = $sluggerType->prepareRecordForSlugGeneration($record);
        $slug = $this->getSlugHelper($sluggerType)->generate($record, (int)$record['pid']);

        $queryBuilder = $this->getQueryBuilder($tableName);
        $queryBuilder->update($tableName)
            ->where(
                $queryBuilder->expr()->eq(
                    'uid',
                    $queryBuilder->createNamedParameter($record['uid'], \PDO::PARAM_INT)
                )
            )
            ->set($sluggerType->getSlugColumn(), $slug);
        $queryBuilder->execute();
    }

    private function getSlugHelper(
        SluggerType $sluggerType
    ): SlugHelper {
        $tableName = $sluggerType->getSupportedTableName();
        $column = $sluggerType->getSlugColumn();

        return GeneralUtility::makeInstance(
            SlugHelper::class,
            $tableName,
            $column,
            $GLOBALS['TCA'][$tableName]['columns'][$column]['config']
        );
    }

    private function getQueryBuilder(
        string $tableName
    ): QueryBuilder {
        $queryBuilder = $this->connectionPool->getQueryBuilderForTable($tableName);
        $queryBuilder->getRestrictions()->removeAll();
        return $queryBuilder;
    }
}