events/Classes/Domain/Repository/DateRepository.php

<?php

namespace Wrm\Events\Domain\Repository;

use TYPO3\CMS\Core\Context\Context;
use TYPO3\CMS\Core\Database\ConnectionPool;
use TYPO3\CMS\Core\Utility\GeneralUtility;
use TYPO3\CMS\Extbase\Persistence\Generic\Qom\ConstraintInterface;
use TYPO3\CMS\Extbase\Persistence\Generic\QueryResult;
use TYPO3\CMS\Extbase\Persistence\QueryInterface;
use TYPO3\CMS\Extbase\Persistence\QueryResultInterface;
use TYPO3\CMS\Extbase\Persistence\Repository;
use Wrm\Events\Domain\Model\Dto\DateDemand;
use Wrm\Events\Service\CategoryService;

class DateRepository extends Repository
{
    /**
     * @var Context
     */
    protected $context;

    public function injectContext(Context $context): void
    {
        $this->context = $context;
    }

    public function findByUids(string $uids): QueryResult
    {
        $uids = explode(',', $uids);
        $query = $this->createQuery();
        $query->matching(
            $query->in('uid', $uids)
        );
        return $query->execute();
    }

    public function findByDemand(DateDemand $demand): QueryResult
    {
        $query = $this->createDemandQuery($demand);
        return $query->execute();
    }

    protected function createDemandQuery(DateDemand $demand): QueryInterface
    {
        $query = $this->createQuery();
        $constraints = [
            $this->createEventConstraint($query),
        ];

        $categoriesConstraint = $this->createCategoryConstraint($query, $demand);
        if ($categoriesConstraint instanceof ConstraintInterface) {
            $constraints['categories'] = $categoriesConstraint;
        }

        if ($demand->getFeatures() !== []) {
            $constraints['features'] = $this->createFeaturesConstraint($query, $demand);
        }

        if ($demand->getLocations() !== []) {
            $constraints['locations'] = $query->in('event.location', $demand->getLocations());
        }

        if ($demand->getRegion() !== '') {
            $constraints['region'] = $query->equals('event.region', $demand->getRegion());
        }

        if ($demand->getHighlight() !== false) {
            $constraints['highlight'] = $query->equals('event.highlight', $demand->getHighlight());
        }

        if ($demand->getSearchword() !== '') {
            $constraints['searchword'] = $this->getSearchwordConstraint($query, $demand);
        }

        if ($demand->getUserCategories() !== []) {
            $constraints['userCategories'] = $query->in('event.categories.uid', $demand->getUserCategories());
        }

        $timingConstraint = $this->createTimingConstraint($query, $demand);
        if ($timingConstraint instanceof ConstraintInterface) {
            $constraints['timing'] = $timingConstraint;
        }

        if ($demand->shouldShowFromNow() || $demand->shouldShowFromMidnight()) {
            $now = $this->context->getPropertyFromAspect(
                'date',
                'full',
                new \DateTimeImmutable()
            );
            if (!$now instanceof \DateTimeImmutable) {
                throw new \UnexpectedValueException(
                    'Could not retrieve now as DateTimeImmutable, got "' . gettype($now) . '".',
                    1639382648
                );
            }

            $now = $now->setTimezone(new \DateTimeZone(date_default_timezone_get()));

            if ($demand->shouldShowFromMidnight()) {
                $now = $now->modify('midnight');
            }

            $constraints['nowAndFuture'] = $query->logicalOr([
                $query->greaterThanOrEqual('start', $now),
                $query->greaterThanOrEqual('end', $now)
            ]);
        }

        if ($demand->getLimit() !== '') {
            $query->setLimit((int) $demand->getLimit());
        }

        $query->matching($query->logicalAnd($constraints));

        if ($demand->getSortBy() && $demand->getSortOrder()) {
            $query->setOrderings([$demand->getSortBy() => $demand->getSortOrder()]);
        }

        $callback = $demand->getQueryCalback();
        if ($callback !== '') {
            $params = ['query' => &$query];
            GeneralUtility::callUserFunction($callback, $params, $this);
        }

        return $query;
    }

    private function getSearchwordConstraint(
        QueryInterface $query,
        DateDemand $demand
    ): ConstraintInterface {
        $fieldsToSearch = [
            'event.title',
            'event.teaser',
            'event.categories.title',
            'event.location.name',
            'event.organizer.name',
        ];

        $wordsToSearch = $demand->getSynonymsForSearchword();
        $wordsToSearch[] = $demand->getSearchword();
        $constraints = [];

        $queryBuilder = $this->objectManager->get(ConnectionPool::class)
            ->getQueryBuilderForTable('tx_events_domain_model_date');

        foreach ($wordsToSearch as $word) {
            foreach ($fieldsToSearch as $field) {
                $constraints[] = $query->like($field, '%' . $queryBuilder->escapeLikeWildcards($word) . '%');
            }
        }

        return $query->logicalOr($constraints);
    }

    protected function createCategoryConstraint(
        QueryInterface $query,
        DateDemand $demand
    ): ?ConstraintInterface {
        $categories = $demand->getCategories();
        if ($categories === '') {
            return null;
        }
        $constraints = [];

        if ($demand->getIncludeSubCategories()) {
            $categories = GeneralUtility::makeInstance(CategoryService::class)
                ->getChildrenCategories($categories);
        }

        $categories = GeneralUtility::intExplode(',', $categories, true);
        foreach ($categories as $category) {
            $constraints[] = $query->contains('event.categories', $category);
        }

        if ($constraints === []) {
            return null;
        }

        if ($demand->getCategoryCombination() === 'or') {
            return $query->logicalOr($constraints);
        }

        return $query->logicalAnd($constraints);
    }

    private function createTimingConstraint(
        QueryInterface $query,
        DateDemand $demand
    ): ?ConstraintInterface {
        // Dates might have end of 0 if only start exists.

        if ($demand->getStartObject() !== null && $demand->getEndObject() === null) {
            return $query->logicalOr([
                $query->greaterThanOrEqual('start', $demand->getStartObject()),
                $query->greaterThanOrEqual('end', $demand->getStartObject()),
            ]);
        }

        if ($demand->getStartObject() === null && $demand->getEndObject() !== null) {
            return $query->logicalOr([
                $query->logicalAnd([
                    $query->lessThanOrEqual('end', $demand->getEndObject()),
                    $query->greaterThan('end', 0),
                ]),
                $query->lessThanOrEqual('start', $demand->getEndObject()),
            ]);
        }

        if ($demand->getStartObject() !== null && $demand->getEndObject() !== null) {
            return $query->logicalOr([
                $query->logicalAnd([
                    $query->logicalOr([
                        $query->greaterThanOrEqual('start', $demand->getStartObject()),
                        $query->greaterThanOrEqual('end', $demand->getStartObject()),
                    ]),
                    $query->logicalOr([
                        $query->lessThanOrEqual('start', $demand->getEndObject()),
                        $query->logicalAnd([
                            $query->lessThanOrEqual('end', $demand->getEndObject()),
                            $query->greaterThan('end', 0),
                        ]),
                    ]),
                ]),
                $query->logicalAnd([
                    $query->lessThanOrEqual('start', $demand->getStartObject()),
                    $query->greaterThanOrEqual('end', $demand->getEndObject()),
                ]),
            ]);
        }

        return null;
    }

    private function createFeaturesConstraint(
        QueryInterface $query,
        DateDemand $demand
    ): ConstraintInterface {
        $constraints = [];

        foreach ($demand->getFeatures() as $feature) {
            $constraints[] = $query->contains('event.features', $feature);
        }

        return $query->logicalAnd($constraints);
    }

    public function findSearchWord(string $search): array
    {
        $connection = GeneralUtility::makeInstance(ConnectionPool::class)
            ->getConnectionForTable('tx_events_domain_model_date');

        $queryBuilder = $connection->createQueryBuilder();

        $statement = $queryBuilder
            ->select('*')
            ->from('tx_events_domain_model_date')
            ->join(
                'tx_events_domain_model_date',
                'tx_events_domain_model_event',
                'event',
                $queryBuilder->expr()->eq(
                    'tx_events_domain_model_date.event',
                    $queryBuilder->quoteIdentifier('event.uid')
                )
            )->where(
                $queryBuilder->expr()->like('event.title', $queryBuilder->createNamedParameter('%' . $search . '%'))
            )->orderBy('tx_events_domain_model_date.start');

        return $statement->execute()->fetchAll();
    }

    private function createEventConstraint(
        QueryInterface $query
    ): ConstraintInterface {
        return $query->logicalAnd(
            // Use sub property to trigger join and pulling in event table constraints (hidden)
            $query->logicalNot($query->equals('event.uid', null))
        );
    }
}