meilisearch/Classes/IndexQueue/Indexer.php

722 lines
27 KiB
PHP
Raw Normal View History

2021-04-17 00:26:33 +02:00
<?php
namespace WapplerSystems\Meilisearch\IndexQueue;
/***************************************************************
* Copyright notice
*
* (c) 2009-2015 Ingo Renner <ingo@typo3.org>
* All rights reserved
*
* This script is part of the TYPO3 project. The TYPO3 project 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 3 of the License, or
* (at your option) any later version.
*
* The GNU General Public License can be found at
* http://www.gnu.org/copyleft/gpl.html.
*
* This script 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.
*
* This copyright notice MUST APPEAR in all copies of the script!
***************************************************************/
use WapplerSystems\Meilisearch\ConnectionManager;
use WapplerSystems\Meilisearch\Domain\Search\ApacheSolrDocument\Builder;
use WapplerSystems\Meilisearch\FieldProcessor\Service;
use WapplerSystems\Meilisearch\FrontendEnvironment;
use WapplerSystems\Meilisearch\NoSolrConnectionFoundException;
use WapplerSystems\Meilisearch\Domain\Site\Site;
use WapplerSystems\Meilisearch\System\Logging\SolrLogManager;
use WapplerSystems\Meilisearch\System\Records\Pages\PagesRepository;
use WapplerSystems\Meilisearch\System\Solr\Document\Document;
use WapplerSystems\Meilisearch\System\Solr\ResponseAdapter;
use WapplerSystems\Meilisearch\System\Solr\SolrConnection;
use Exception;
use RuntimeException;
use Solarium\Exception\HttpException;
use TYPO3\CMS\Core\Context\LanguageAspectFactory;
use TYPO3\CMS\Core\Error\Http\ServiceUnavailableException;
use TYPO3\CMS\Core\Exception\SiteNotFoundException;
use TYPO3\CMS\Core\Http\ImmediateResponseException;
use TYPO3\CMS\Core\Site\SiteFinder;
use TYPO3\CMS\Core\Utility\GeneralUtility;
use TYPO3\CMS\Core\Utility\RootlineUtility;
use TYPO3\CMS\Frontend\Page\PageRepository;
/**
* A general purpose indexer to be used for indexing of any kind of regular
* records like tt_news, tt_address, and so on.
* Specialized indexers can extend this class to handle advanced stuff like
* category resolution in tt_news or file indexing.
*
* @author Ingo Renner <ingo@typo3.org>
*/
class Indexer extends AbstractIndexer
{
# TODO change to singular $document instead of plural $documents
/**
* A Solr service instance to interact with the Solr server
*
* @var SolrConnection
*/
protected $solr;
/**
* @var ConnectionManager
*/
protected $connectionManager;
/**
* Holds options for a specific indexer
*
* @var array
*/
protected $options = [];
/**
* To log or not to log... #Shakespeare
*
* @var bool
*/
protected $loggingEnabled = false;
/**
* @var SolrLogManager
*/
protected $logger = null;
/**
* @var PagesRepository
*/
protected $pagesRepository;
/**
* @var Builder
*/
protected $documentBuilder;
/**
* @var FrontendEnvironment
*/
protected $frontendEnvironment = null;
/**
* Constructor
*
* @param array $options array of indexer options
* @param PagesRepository|null $pagesRepository
* @param Builder|null $documentBuilder
* @param SolrLogManager|null $logger
* @param ConnectionManager|null $connectionManager
* @param FrontendEnvironment|null $frontendEnvironment
*/
public function __construct(
array $options = [],
PagesRepository $pagesRepository = null,
Builder $documentBuilder = null,
SolrLogManager $logger = null,
ConnectionManager $connectionManager = null,
FrontendEnvironment $frontendEnvironment = null
)
{
$this->options = $options;
$this->pagesRepository = $pagesRepository ?? GeneralUtility::makeInstance(PagesRepository::class);
$this->documentBuilder = $documentBuilder ?? GeneralUtility::makeInstance(Builder::class);
$this->logger = $logger ?? GeneralUtility::makeInstance(SolrLogManager::class, /** @scrutinizer ignore-type */ __CLASS__);
$this->connectionManager = $connectionManager ?? GeneralUtility::makeInstance(ConnectionManager::class);
$this->frontendEnvironment = $frontendEnvironment ?? GeneralUtility::makeInstance(FrontendEnvironment::class);
}
/**
* Indexes an item from the indexing queue.
*
* @param Item $item An index queue item
* @return bool returns true when indexed, false when not
*/
public function index(Item $item)
{
$indexed = true;
$this->type = $item->getType();
$this->setLogging($item);
$solrConnections = $this->getSolrConnectionsByItem($item);
foreach ($solrConnections as $systemLanguageUid => $solrConnection) {
$this->solr = $solrConnection;
if (!$this->indexItem($item, $systemLanguageUid)) {
/*
* A single language voting for "not indexed" should make the whole
* item count as being not indexed, even if all other languages are
* indexed.
* If there is no translation for a single language, this item counts
* as TRUE since it's not an error which that should make the item
* being reindexed during another index run.
*/
$indexed = false;
}
}
return $indexed;
}
/**
* Creates a single Solr Document for an item in a specific language.
*
* @param Item $item An index queue item to index.
* @param int $language The language to use.
* @return bool TRUE if item was indexed successfully, FALSE on failure
*/
protected function indexItem(Item $item, $language = 0)
{
$itemIndexed = false;
$documents = [];
$itemDocument = $this->itemToDocument($item, $language);
if (is_null($itemDocument)) {
/*
* If there is no itemDocument, this means there was no translation
* for this record. This should not stop the current item to count as
* being valid because not-indexing not-translated items is perfectly
* fine.
*/
return true;
}
$documents[] = $itemDocument;
$documents = array_merge($documents, $this->getAdditionalDocuments($item, $language, $itemDocument));
$documents = $this->processDocuments($item, $documents);
$documents = $this->preAddModifyDocuments($item, $language, $documents);
try {
$response = $this->solr->getWriteService()->addDocuments($documents);
if ($response->getHttpStatus() == 200) {
$itemIndexed = true;
}
} catch (HttpException $e) {
$response = new ResponseAdapter($e->getBody(), $httpStatus = 500, $e->getStatusMessage());
}
$this->log($item, $documents, $response);
return $itemIndexed;
}
/**
* Gets the full item record.
*
* This general record indexer simply gets the record from the item. Other
* more specialized indexers may provide more data for their specific item
* types.
*
* @param Item $item The item to be indexed
* @param int $language Language Id (sys_language.uid)
* @return array|NULL The full record with fields of data to be used for indexing or NULL to prevent an item from being indexed
*/
protected function getFullItemRecord(Item $item, $language = 0)
{
$itemRecord = $this->getItemRecordOverlayed($item, $language);
if (!is_null($itemRecord)) {
$itemRecord['__solr_index_language'] = $language;
}
return $itemRecord;
}
/**
* Returns the overlayed item record.
*
* @param Item $item
* @param int $language
* @return array|mixed|null
*/
protected function getItemRecordOverlayed(Item $item, int $language): ?array
{
$itemRecord = $item->getRecord();
if ($language > 0) {
$page = GeneralUtility::makeInstance(PageRepository::class);
$itemRecord = $page->getLanguageOverlay($item->getType(), $itemRecord);
}
if (!$itemRecord) {
$itemRecord = null;
}
return $itemRecord;
}
/**
* Gets the configuration how to process an item's fields for indexing.
*
* @param Item $item An index queue item
* @param int $language Language ID
* @throws RuntimeException
* @return array Configuration array from TypoScript
*/
protected function getItemTypeConfiguration(Item $item, int $language = 0): array
{
$indexConfigurationName = $item->getIndexingConfigurationName();
$fields = $this->getFieldConfigurationFromItemRecordPage($item, $language, $indexConfigurationName);
if (!$this->isRootPageIdPartOfRootLine($item) || count($fields) === 0) {
$fields = $this->getFieldConfigurationFromItemRootPage($item, $language, $indexConfigurationName);
if (count($fields) === 0) {
throw new RuntimeException('The item indexing configuration "' . $item->getIndexingConfigurationName() .
'" on root page uid ' . $item->getRootPageUid() . ' could not be found!', 1455530112);
}
}
return $fields;
}
/**
* The method retrieves the field configuration of the items record page id (pid).
*
* @param Item $item
* @param integer $language
* @param string $indexConfigurationName
* @return array
*/
protected function getFieldConfigurationFromItemRecordPage(Item $item, $language, $indexConfigurationName): array
{
try {
$pageId = $this->getPageIdOfItem($item);
$solrConfiguration = $this->frontendEnvironment->getSolrConfigurationFromPageId($pageId, $language);
return $solrConfiguration->getIndexQueueFieldsConfigurationByConfigurationName($indexConfigurationName, []);
} catch (Exception $e) {
return [];
}
}
/**
* @param Item $item
* @return int
*/
protected function getPageIdOfItem(Item $item): int
{
if ($item->getType() === 'pages') {
return (int)$item->getRecordUid();
}
return (int)$item->getRecordPageId();
}
/**
* The method returns the field configuration of the items root page id (uid of the related root page).
*
* @param Item $item
* @param integer $language
* @param string $indexConfigurationName
* @return array
*/
protected function getFieldConfigurationFromItemRootPage(Item $item, $language, $indexConfigurationName)
{
$solrConfiguration = $this->frontendEnvironment->getSolrConfigurationFromPageId($item->getRootPageUid(), $language);
return $solrConfiguration->getIndexQueueFieldsConfigurationByConfigurationName($indexConfigurationName, []);
}
/**
* In case of additionalStoragePid config recordPageId can be outside of siteroot.
* In that case we should not read TS config of foreign siteroot.
*
* @param Item $item
* @return bool
*/
protected function isRootPageIdPartOfRootLine(Item $item): bool
{
$rootPageId = (int)$item->getRootPageUid();
$buildRootlineWithPid = $this->getPageIdOfItem($item);
$rootlineUtility = GeneralUtility::makeInstance(RootlineUtility::class, $buildRootlineWithPid);
$rootline = $rootlineUtility->get();
$pageInRootline = array_filter($rootline, function($page) use ($rootPageId) {
return (int)$page['uid'] === $rootPageId;
});
return !empty($pageInRootline);
}
/**
* Converts an item array (record) to a Solr document by mapping the
* record's fields onto Solr document fields as configured in TypoScript.
*
* @param Item $item An index queue item
* @param int $language Language Id
* @return Document|null The Solr document converted from the record
* @throws SiteNotFoundException
* @throws ServiceUnavailableException
* @throws ImmediateResponseException
*/
protected function itemToDocument(Item $item, $language = 0): ?Document
{
$document = null;
if ($item->getType() === 'pages') {
$this->frontendEnvironment->initializeTsfe($item->getRecordUid(), $language);
} else {
$this->frontendEnvironment->initializeTsfe($item->getRootPageUid(), $language);
}
$itemRecord = $this->getFullItemRecord($item, $language);
if (!is_null($itemRecord)) {
$itemIndexingConfiguration = $this->getItemTypeConfiguration($item, $language);
$document = $this->getBaseDocument($item, $itemRecord);
$document = $this->addDocumentFieldsFromTyposcript($document, $itemIndexingConfiguration, $itemRecord);
}
return $document;
}
/**
* Creates a Solr document with the basic / core fields set already.
*
* @param Item $item The item to index
* @param array $itemRecord The record to use to build the base document
* @return Document A basic Solr document
*/
protected function getBaseDocument(Item $item, array $itemRecord)
{
$type = $item->getType();
$rootPageUid = $item->getRootPageUid();
$accessRootLine = $this->getAccessRootline($item);
return $this->documentBuilder->fromRecord($itemRecord, $type, $rootPageUid, $accessRootLine);
}
/**
* Generates an Access Rootline for an item.
*
* @param Item $item Index Queue item to index.
* @return string The Access Rootline for the item
*/
protected function getAccessRootline(Item $item)
{
$accessRestriction = '0';
$itemRecord = $item->getRecord();
// TODO support access restrictions set on storage page
if (isset($GLOBALS['TCA'][$item->getType()]['ctrl']['enablecolumns']['fe_group'])) {
$accessRestriction = $itemRecord[$GLOBALS['TCA'][$item->getType()]['ctrl']['enablecolumns']['fe_group']];
if (empty($accessRestriction)) {
// public
$accessRestriction = '0';
}
}
return 'r:' . $accessRestriction;
}
/**
* Sends the documents to the field processing service which takes care of
* manipulating fields as defined in the field's configuration.
*
* @param Item $item An index queue item
* @param array $documents An array of \WapplerSystems\Meilisearch\System\Solr\Document\Document objects to manipulate.
* @return Document[] array Array of manipulated Document objects.
*/
protected function processDocuments(Item $item, array $documents)
{
// needs to respect the TS settings for the page the item is on, conditions may apply
$solrConfiguration = $this->frontendEnvironment->getSolrConfigurationFromPageId($item->getRootPageUid());
$fieldProcessingInstructions = $solrConfiguration->getIndexFieldProcessingInstructionsConfiguration();
// same as in the FE indexer
if (is_array($fieldProcessingInstructions)) {
$service = GeneralUtility::makeInstance(Service::class);
$service->processDocuments($documents, $fieldProcessingInstructions);
}
return $documents;
}
/**
* Allows third party extensions to provide additional documents which
* should be indexed for the current item.
*
* @param Item $item The item currently being indexed.
* @param int $language The language uid currently being indexed.
* @param Document $itemDocument The document representing the item for the given language.
* @return Document[] array An array of additional Document objects to index.
*/
protected function getAdditionalDocuments(Item $item, $language, Document $itemDocument)
{
$documents = [];
if (is_array($GLOBALS['TYPO3_CONF_VARS']['EXTCONF']['solr']['IndexQueueIndexer']['indexItemAddDocuments'])) {
foreach ($GLOBALS['TYPO3_CONF_VARS']['EXTCONF']['solr']['IndexQueueIndexer']['indexItemAddDocuments'] as $classReference) {
if (!class_exists($classReference)) {
throw new \InvalidArgumentException('Class does not exits' . $classReference, 1490363487);
}
$additionalIndexer = GeneralUtility::makeInstance($classReference);
if ($additionalIndexer instanceof AdditionalIndexQueueItemIndexer) {
$additionalDocuments = $additionalIndexer->getAdditionalItemDocuments($item, $language, $itemDocument);
if (is_array($additionalDocuments)) {
$documents = array_merge($documents,
$additionalDocuments);
}
} else {
throw new \UnexpectedValueException(
get_class($additionalIndexer) . ' must implement interface ' . AdditionalIndexQueueItemIndexer::class,
1326284551
);
}
}
}
return $documents;
}
/**
* Provides a hook to manipulate documents right before they get added to
* the Solr index.
*
* @param Item $item The item currently being indexed.
* @param int $language The language uid of the documents
* @param array $documents An array of documents to be indexed
* @return array An array of modified documents
*/
protected function preAddModifyDocuments(Item $item, $language, array $documents)
{
if (is_array($GLOBALS['TYPO3_CONF_VARS']['EXTCONF']['solr']['IndexQueueIndexer']['preAddModifyDocuments'])) {
foreach ($GLOBALS['TYPO3_CONF_VARS']['EXTCONF']['solr']['IndexQueueIndexer']['preAddModifyDocuments'] as $classReference) {
$documentsModifier = GeneralUtility::makeInstance($classReference);
if ($documentsModifier instanceof PageIndexerDocumentsModifier) {
$documents = $documentsModifier->modifyDocuments($item, $language, $documents);
} else {
throw new RuntimeException(
'The class "' . get_class($documentsModifier)
. '" registered as document modifier in hook
preAddModifyDocuments must implement interface
WapplerSystems\Meilisearch\IndexQueue\PageIndexerDocumentsModifier',
1309522677
);
}
}
}
return $documents;
}
// Initialization
/**
* Gets the Solr connections applicable for an item.
*
* The connections include the default connection and connections to be used
* for translations of an item.
*
* @param Item $item An index queue item
* @return array An array of WapplerSystems\Meilisearch\System\Solr\SolrConnection connections, the array's keys are the sys_language_uid of the language of the connection
*/
protected function getSolrConnectionsByItem(Item $item)
{
$solrConnections = [];
$rootPageId = $item->getRootPageUid();
if ($item->getType() === 'pages') {
$pageId = $item->getRecordUid();
} else {
$pageId = $item->getRecordPageId();
}
// Solr configurations possible for this item
$site = $item->getSite();
$solrConfigurationsBySite = $site->getAllSolrConnectionConfigurations();
$siteLanguages = [];
foreach ($solrConfigurationsBySite as $solrConfiguration) {
$siteLanguages[] = $solrConfiguration['language'];
}
$defaultLanguageUid = $this->getDefaultLanguageUid($item, $site->getRootPage(), $siteLanguages);
$translationOverlays = $this->getTranslationOverlaysWithConfiguredSite((int)$pageId, $site, (array)$siteLanguages);
$defaultConnection = $this->connectionManager->getConnectionByPageId($rootPageId, $defaultLanguageUid, $item->getMountPointIdentifier());
$translationConnections = $this->getConnectionsForIndexableLanguages($translationOverlays);
if ($defaultLanguageUid == 0) {
$solrConnections[0] = $defaultConnection;
}
foreach ($translationConnections as $systemLanguageUid => $solrConnection) {
$solrConnections[$systemLanguageUid] = $solrConnection;
}
return $solrConnections;
}
/**
* @param int $pageId
* @param Site $site
* @param array $siteLanguages
* @return array
*/
protected function getTranslationOverlaysWithConfiguredSite(int $pageId, Site $site, array $siteLanguages): array
{
$translationOverlays = $this->pagesRepository->findTranslationOverlaysByPageId($pageId);
$translatedLanguages = [];
foreach ($translationOverlays as $key => $translationOverlay) {
if (!in_array($translationOverlay['sys_language_uid'], $siteLanguages)) {
unset($translationOverlays[$key]);
} else {
$translatedLanguages[] = (int)$translationOverlay['sys_language_uid'];
}
}
if (count($translationOverlays) + 1 !== count($siteLanguages)) {
// not all Languages are translated
// add Language Fallback
foreach ($siteLanguages as $languageId) {
if ($languageId !== 0 && !in_array((int)$languageId, $translatedLanguages, true)) {
$fallbackLanguageIds = $this->getFallbackOrder($site, (int)$languageId, (int)$pageId);
foreach ($fallbackLanguageIds as $fallbackLanguageId) {
if ($fallbackLanguageId === 0 || in_array((int)$fallbackLanguageId, $translatedLanguages, true)) {
$translationOverlay = [
'pid' => $pageId,
'sys_language_uid' => $languageId,
'l10n_parent' => $pageId
];
$translationOverlays[] = $translationOverlay;
continue 2;
}
}
}
}
}
return $translationOverlays;
}
/**
* @param Site $site
* @param int $languageId
* @param int $pageId
* @return array
*/
protected function getFallbackOrder(Site $site, int $languageId, int $pageId): array
{
$fallbackChain = [];
$siteFinder = GeneralUtility::makeInstance(SiteFinder::class);
try {
$site = $siteFinder->getSiteByRootPageId($site->getRootPageId());
$languageAspect = LanguageAspectFactory::createFromSiteLanguage($site->getLanguageById($languageId));
$fallbackChain = $languageAspect->getFallbackChain();
} catch (SiteNotFoundException $e) {
}
return $fallbackChain;
}
/**
* @param Item $item An index queue item
* @param array $rootPage
* @param array $siteLanguages
*
* @return int
* @throws RuntimeException
*/
protected function getDefaultLanguageUid(Item $item, array $rootPage, array $siteLanguages)
{
$defaultLanguageUid = 0;
if (($rootPage['l18n_cfg'] & 1) == 1 && count($siteLanguages) == 1 && $siteLanguages[min(array_keys($siteLanguages))] > 0) {
$defaultLanguageUid = $siteLanguages[min(array_keys($siteLanguages))];
} elseif (($rootPage['l18n_cfg'] & 1) == 1 && count($siteLanguages) > 1) {
unset($siteLanguages[array_search('0', $siteLanguages)]);
$defaultLanguageUid = $siteLanguages[min(array_keys($siteLanguages))];
} elseif (($rootPage['l18n_cfg'] & 1) == 1 && count($siteLanguages) == 1) {
$message = 'Root page ' . (int)$item->getRootPageUid() . ' is set to hide default translation, but no other language is configured!';
throw new RuntimeException($message);
}
return $defaultLanguageUid;
}
/**
* Checks for which languages connections have been configured and returns
* these connections.
*
* @param array $translationOverlays An array of translation overlays to check for configured connections.
* @return array An array of WapplerSystems\Meilisearch\System\Solr\SolrConnection connections.
*/
protected function getConnectionsForIndexableLanguages(array $translationOverlays)
{
$connections = [];
foreach ($translationOverlays as $translationOverlay) {
$pageId = $translationOverlay['l10n_parent'];
$languageId = $translationOverlay['sys_language_uid'];
try {
$connection = $this->connectionManager->getConnectionByPageId($pageId, $languageId);
$connections[$languageId] = $connection;
} catch (NoSolrConnectionFoundException $e) {
// ignore the exception as we seek only those connections
// actually available
}
}
return $connections;
}
// Utility methods
// FIXME extract log() and setLogging() to WapplerSystems\Meilisearch\IndexQueue\AbstractIndexer
// FIXME extract an interface Tx_Solr_IndexQueue_ItemInterface
/**
* Enables logging dependent on the configuration of the item's site
*
* @param Item $item An item being indexed
* @return void
*/
protected function setLogging(Item $item)
{
$solrConfiguration = $this->frontendEnvironment->getSolrConfigurationFromPageId($item->getRootPageUid());
$this->loggingEnabled = $solrConfiguration->getLoggingIndexingQueueOperationsByConfigurationNameWithFallBack(
$item->getIndexingConfigurationName()
);
}
/**
* Logs the item and what document was created from it
*
* @param Item $item The item that is being indexed.
* @param array $itemDocuments An array of Solr documents created from the item's data
* @param ResponseAdapter $response The Solr response for the particular index document
*/
protected function log(Item $item, array $itemDocuments, ResponseAdapter $response)
{
if (!$this->loggingEnabled) {
return;
}
$message = 'Index Queue indexing ' . $item->getType() . ':' . $item->getRecordUid() . ' - ';
// preparing data
$documents = [];
foreach ($itemDocuments as $document) {
$documents[] = (array)$document;
}
$logData = ['item' => (array)$item, 'documents' => $documents, 'response' => (array)$response];
if ($response->getHttpStatus() == 200) {
$severity = SolrLogManager::NOTICE;
$message .= 'Success';
} else {
$severity = SolrLogManager::ERROR;
$message .= 'Failure';
$logData['status'] = $response->getHttpStatus();
$logData['status message'] = $response->getHttpStatusMessage();
}
$this->logger->log($severity, $message, $logData);
}
}