452 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			452 lines
		
	
	
		
			14 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
namespace WapplerSystems\Meilisearch\System\Meilisearch\Service;
 | 
						|
 | 
						|
/***************************************************************
 | 
						|
 *  Copyright notice
 | 
						|
 *
 | 
						|
 *  (c) 2009-2017 Timo Hund <timo.hund@dkd.de>
 | 
						|
 *  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\PingFailedException;
 | 
						|
use WapplerSystems\Meilisearch\System\Configuration\TypoScriptConfiguration;
 | 
						|
use WapplerSystems\Meilisearch\System\Logging\MeilisearchLogManager;
 | 
						|
use WapplerSystems\Meilisearch\System\Meilisearch\ResponseAdapter;
 | 
						|
use WapplerSystems\Meilisearch\Util;
 | 
						|
use Solarium\Client;
 | 
						|
use Solarium\Core\Client\Endpoint;
 | 
						|
use Solarium\Core\Client\Request;
 | 
						|
use Solarium\Core\Query\QueryInterface;
 | 
						|
use Solarium\Exception\HttpException;
 | 
						|
use TYPO3\CMS\Core\Http\Uri;
 | 
						|
use TYPO3\CMS\Core\Utility\GeneralUtility;
 | 
						|
 | 
						|
abstract class AbstractMeilisearchService
 | 
						|
{
 | 
						|
 | 
						|
    /**
 | 
						|
     * @var array
 | 
						|
     */
 | 
						|
    protected static $pingCache = [];
 | 
						|
 | 
						|
    /**
 | 
						|
     * @var TypoScriptConfiguration
 | 
						|
     */
 | 
						|
    protected $configuration;
 | 
						|
 | 
						|
    /**
 | 
						|
     * @var \WapplerSystems\Meilisearch\System\Logging\MeilisearchLogManager
 | 
						|
     */
 | 
						|
    protected $logger = null;
 | 
						|
 | 
						|
    /**
 | 
						|
     * @var Client
 | 
						|
     */
 | 
						|
    protected $client = null;
 | 
						|
 | 
						|
    /**
 | 
						|
     * MeilisearchReadService constructor.
 | 
						|
     */
 | 
						|
    public function __construct(Client $client, $typoScriptConfiguration = null, $logManager = null)
 | 
						|
    {
 | 
						|
        $this->client = $client;
 | 
						|
        $this->configuration = $typoScriptConfiguration ?? Util::getMeilisearchConfiguration();
 | 
						|
        $this->logger = $logManager ?? GeneralUtility::makeInstance(MeilisearchLogManager::class, /** @scrutinizer ignore-type */ __CLASS__);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Returns the path to the core meilisearch path + core path.
 | 
						|
     *
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    public function getCorePath()
 | 
						|
    {
 | 
						|
        $endpoint = $this->getPrimaryEndpoint();
 | 
						|
        return is_null($endpoint) ? '' : $endpoint->getPath() .'/'. $endpoint->getCore();
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Returns the Solarium client
 | 
						|
     *
 | 
						|
     * @return ?Client
 | 
						|
     */
 | 
						|
    public function getClient(): ?Client
 | 
						|
    {
 | 
						|
        return $this->client;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Return a valid http URL given this server's host, port and path and a provided servlet name
 | 
						|
     *
 | 
						|
     * @param string $servlet
 | 
						|
     * @param array $params
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function _constructUrl($servlet, $params = [])
 | 
						|
    {
 | 
						|
        $queryString = count($params) ? '?' . http_build_query($params, null, '&') : '';
 | 
						|
        return $this->__toString() . $servlet . $queryString;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Creates a string representation of the Meilisearch connection. Specifically
 | 
						|
     * will return the Meilisearch URL.
 | 
						|
     *
 | 
						|
     * @return string The Meilisearch URL.
 | 
						|
     * @TODO: Add support for API version 2
 | 
						|
     */
 | 
						|
    public function __toString()
 | 
						|
    {
 | 
						|
        $endpoint = $this->getPrimaryEndpoint();
 | 
						|
        if (!$endpoint instanceof Endpoint) {
 | 
						|
            return '';
 | 
						|
        }
 | 
						|
 | 
						|
        try {
 | 
						|
            return $endpoint->getCoreBaseUri();
 | 
						|
        } catch (\Exception $exception) {
 | 
						|
        }
 | 
						|
        return  $endpoint->getScheme(). '://' . $endpoint->getHost() . ':' . $endpoint->getPort() . $endpoint->getPath() . '/' . $endpoint->getCore() . '/';
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * @return Endpoint|null
 | 
						|
     */
 | 
						|
    public function getPrimaryEndpoint()
 | 
						|
    {
 | 
						|
        return is_array($this->client->getEndpoints()) ? reset($this->client->getEndpoints()) : null;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Central method for making a get operation against this Meilisearch Server
 | 
						|
     *
 | 
						|
     * @param string $url
 | 
						|
     * @return ResponseAdapter
 | 
						|
     */
 | 
						|
    protected function _sendRawGet($url)
 | 
						|
    {
 | 
						|
        return $this->_sendRawRequest($url, Request::METHOD_GET);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Central method for making a HTTP DELETE operation against the Meilisearch server
 | 
						|
     *
 | 
						|
     * @param string $url
 | 
						|
     * @return ResponseAdapter
 | 
						|
     */
 | 
						|
    protected function _sendRawDelete($url)
 | 
						|
    {
 | 
						|
        return $this->_sendRawRequest($url, Request::METHOD_DELETE);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Central method for making a post operation against this Meilisearch Server
 | 
						|
     *
 | 
						|
     * @param string $url
 | 
						|
     * @param string $rawPost
 | 
						|
     * @param string $contentType
 | 
						|
     * @return ResponseAdapter
 | 
						|
     */
 | 
						|
    protected function _sendRawPost($url, $rawPost, $contentType = 'text/xml; charset=UTF-8')
 | 
						|
    {
 | 
						|
        $initializeRequest = function(Request $request) use ($rawPost, $contentType) {
 | 
						|
            $request->setRawData($rawPost);
 | 
						|
            $request->addHeader('Content-Type: ' . $contentType);
 | 
						|
            return $request;
 | 
						|
        };
 | 
						|
 | 
						|
        return $this->_sendRawRequest($url, Request::METHOD_POST, $rawPost, $initializeRequest);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Method that performs an http request with the solarium client.
 | 
						|
     *
 | 
						|
     * @param string $url
 | 
						|
     * @param string $method
 | 
						|
     * @param string $body
 | 
						|
     * @param ?\Closure $initializeRequest
 | 
						|
     * @return ResponseAdapter
 | 
						|
     */
 | 
						|
    protected function _sendRawRequest(
 | 
						|
        string $url,
 | 
						|
        $method = Request::METHOD_GET,
 | 
						|
        $body = '',
 | 
						|
        \Closure $initializeRequest = null
 | 
						|
    ) {
 | 
						|
        $logSeverity = MeilisearchLogManager::INFO;
 | 
						|
        $exception = null;
 | 
						|
        $url = $this->reviseUrl($url);
 | 
						|
        try {
 | 
						|
            $request = $this->buildSolariumRequestFromUrl($url, $method);
 | 
						|
            if($initializeRequest !== null) {
 | 
						|
                $request = $initializeRequest($request);
 | 
						|
            }
 | 
						|
            $response = $this->executeRequest($request);
 | 
						|
        } catch (HttpException $exception) {
 | 
						|
            $logSeverity = MeilisearchLogManager::ERROR;
 | 
						|
            $response = new ResponseAdapter($exception->getBody(), $exception->getCode(), $exception->getMessage());
 | 
						|
        }
 | 
						|
 | 
						|
        if ($this->configuration->getLoggingQueryRawPost() || $response->getHttpStatus() != 200) {
 | 
						|
            $message = 'Querying Meilisearch using '.$method;
 | 
						|
            $this->writeLog($logSeverity, $message, $url, $response, $exception, $body);
 | 
						|
        }
 | 
						|
 | 
						|
        return $response;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Revise url
 | 
						|
     * - Resolve relative paths
 | 
						|
     *
 | 
						|
     * @param string $url
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function reviseUrl(string $url): string
 | 
						|
    {
 | 
						|
        /* @var Uri $uri */
 | 
						|
        $uri = GeneralUtility::makeInstance(Uri::class, $url);
 | 
						|
 | 
						|
        if ((string)$uri->getPath() === '') {
 | 
						|
            return $url;
 | 
						|
        }
 | 
						|
 | 
						|
        $path = trim($uri->getPath(), '/');
 | 
						|
        $pathsCurrent = explode('/', $path);
 | 
						|
        $pathNew = [];
 | 
						|
        foreach ($pathsCurrent as $pathCurrent) {
 | 
						|
            if ($pathCurrent === '..') {
 | 
						|
                array_pop($pathNew);
 | 
						|
                continue;
 | 
						|
            }
 | 
						|
            if ($pathCurrent === '.') {
 | 
						|
                continue;
 | 
						|
            }
 | 
						|
            $pathNew[] = $pathCurrent;
 | 
						|
        }
 | 
						|
 | 
						|
        $uri = $uri->withPath(implode('/', $pathNew));
 | 
						|
        return (string)$uri;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Build the log data and writes the message to the log
 | 
						|
     *
 | 
						|
     * @param integer $logSeverity
 | 
						|
     * @param string $message
 | 
						|
     * @param string $url
 | 
						|
     * @param ResponseAdapter $meilisearchResponse
 | 
						|
     * @param ?\Exception $exception
 | 
						|
     * @param string $contentSend
 | 
						|
     */
 | 
						|
    protected function writeLog($logSeverity, $message, $url, $meilisearchResponse, $exception = null, $contentSend = '')
 | 
						|
    {
 | 
						|
        $logData = $this->buildLogDataFromResponse($meilisearchResponse, $exception, $url, $contentSend);
 | 
						|
        $this->logger->log($logSeverity, $message, $logData);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Parses the meilisearch information to build data for the logger.
 | 
						|
     *
 | 
						|
     * @param ResponseAdapter $meilisearchResponse
 | 
						|
     * @param ?\Exception $e
 | 
						|
     * @param string $url
 | 
						|
     * @param string $contentSend
 | 
						|
     * @return array
 | 
						|
     */
 | 
						|
    protected function buildLogDataFromResponse(ResponseAdapter $meilisearchResponse, \Exception $e = null, $url = '', $contentSend = '')
 | 
						|
    {
 | 
						|
        $logData = ['query url' => $url, 'response' => (array)$meilisearchResponse];
 | 
						|
 | 
						|
        if ($contentSend !== '') {
 | 
						|
            $logData['content'] = $contentSend;
 | 
						|
        }
 | 
						|
 | 
						|
        if (!empty($e)) {
 | 
						|
            $logData['exception'] = $e->__toString();
 | 
						|
            return $logData;
 | 
						|
        } else {
 | 
						|
            // trigger data parsing
 | 
						|
            // @extensionScannerIgnoreLine
 | 
						|
            $meilisearchResponse->response;
 | 
						|
            $logData['response data'] = print_r($meilisearchResponse, true);
 | 
						|
            return $logData;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Call the /admin/ping servlet, can be used to quickly tell if a connection to the
 | 
						|
     * server is available.
 | 
						|
     *
 | 
						|
     * Simply overrides the MeilisearchPhpClient implementation, changing ping from a
 | 
						|
     * HEAD to a GET request, see http://forge.typo3.org/issues/44167
 | 
						|
     *
 | 
						|
     * Also does not report the time, see https://forge.typo3.org/issues/64551
 | 
						|
     *
 | 
						|
     * @param boolean $useCache indicates if the ping result should be cached in the instance or not
 | 
						|
     * @return bool TRUE if Meilisearch can be reached, FALSE if not
 | 
						|
     */
 | 
						|
    public function ping($useCache = true)
 | 
						|
    {
 | 
						|
        try {
 | 
						|
            $httpResponse = $this->performPingRequest($useCache);
 | 
						|
        } catch (HttpException $exception) {
 | 
						|
            return false;
 | 
						|
        }
 | 
						|
 | 
						|
        return ($httpResponse->getHttpStatus() === 200);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Call the /admin/ping servlet, can be used to get the runtime of a ping request.
 | 
						|
     *
 | 
						|
     * @param boolean $useCache indicates if the ping result should be cached in the instance or not
 | 
						|
     * @return double runtime in milliseconds
 | 
						|
     * @throws \WapplerSystems\Meilisearch\PingFailedException
 | 
						|
     */
 | 
						|
    public function getPingRoundTripRuntime($useCache = true)
 | 
						|
    {
 | 
						|
        try {
 | 
						|
            $start = $this->getMilliseconds();
 | 
						|
            $httpResponse = $this->performPingRequest($useCache);
 | 
						|
            $end = $this->getMilliseconds();
 | 
						|
        } catch (HttpException $e) {
 | 
						|
            $message = 'Meilisearch ping failed with unexpected response code: ' . $e->getCode();
 | 
						|
            /** @var $exception \WapplerSystems\Meilisearch\PingFailedException */
 | 
						|
            $exception = GeneralUtility::makeInstance(PingFailedException::class, /** @scrutinizer ignore-type */ $message);
 | 
						|
            throw $exception;
 | 
						|
        }
 | 
						|
 | 
						|
        if ($httpResponse->getHttpStatus() !== 200) {
 | 
						|
            $message = 'Meilisearch ping failed with unexpected response code: ' . $httpResponse->getHttpStatus();
 | 
						|
            /** @var $exception \WapplerSystems\Meilisearch\PingFailedException */
 | 
						|
            $exception = GeneralUtility::makeInstance(PingFailedException::class, /** @scrutinizer ignore-type */ $message);
 | 
						|
            throw $exception;
 | 
						|
        }
 | 
						|
 | 
						|
        return $end - $start;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Performs a ping request and returns the result.
 | 
						|
     *
 | 
						|
     * @param boolean $useCache indicates if the ping result should be cached in the instance or not
 | 
						|
     * @return ResponseAdapter
 | 
						|
     */
 | 
						|
    protected function performPingRequest($useCache = true)
 | 
						|
    {
 | 
						|
        $cacheKey = (string)($this);
 | 
						|
        if ($useCache && isset(static::$pingCache[$cacheKey])) {
 | 
						|
            return static::$pingCache[$cacheKey];
 | 
						|
        }
 | 
						|
 | 
						|
        $pingQuery = $this->client->createPing();
 | 
						|
        $pingResult = $this->createAndExecuteRequest($pingQuery);
 | 
						|
 | 
						|
        if ($useCache) {
 | 
						|
            static::$pingCache[$cacheKey] = $pingResult;
 | 
						|
        }
 | 
						|
 | 
						|
        return $pingResult;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Returns the current time in milliseconds.
 | 
						|
     *
 | 
						|
     * @return double
 | 
						|
     */
 | 
						|
    protected function getMilliseconds()
 | 
						|
    {
 | 
						|
        return GeneralUtility::milliseconds();
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * @param QueryInterface $query
 | 
						|
     * @return ResponseAdapter
 | 
						|
     */
 | 
						|
    protected function createAndExecuteRequest(QueryInterface $query): ResponseAdapter
 | 
						|
    {
 | 
						|
        $request = $this->client->createRequest($query);
 | 
						|
        return $this->executeRequest($request);
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * @param $request
 | 
						|
     * @return ResponseAdapter
 | 
						|
     */
 | 
						|
    protected function executeRequest($request): ResponseAdapter
 | 
						|
    {
 | 
						|
        $result = $this->client->executeRequest($request);
 | 
						|
        return new ResponseAdapter($result->getBody(), $result->getStatusCode(), $result->getStatusMessage());
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Build the request for Solarium.
 | 
						|
     *
 | 
						|
     * Important: The endpoint already contains the API information.
 | 
						|
     * The internal Solarium will append the information including the core if set.
 | 
						|
     *
 | 
						|
     * @param string $url
 | 
						|
     * @param string $httpMethod
 | 
						|
     * @return Request
 | 
						|
     */
 | 
						|
    protected function buildSolariumRequestFromUrl(string $url, $httpMethod = Request::METHOD_GET): Request
 | 
						|
    {
 | 
						|
        $params = [];
 | 
						|
        parse_str(parse_url($url, PHP_URL_QUERY), $params);
 | 
						|
        $request = new Request();
 | 
						|
        $path = parse_url($url, PHP_URL_PATH);
 | 
						|
        $endpoint = $this->getPrimaryEndpoint();
 | 
						|
        $api = $request->getApi() === Request::API_V1 ? 'meilisearch' : 'api';
 | 
						|
        $coreBasePath = $endpoint->getPath() . '/' . $api . '/' . $endpoint->getCore() . '/';
 | 
						|
 | 
						|
        $handler = $this->buildRelativePath($coreBasePath, $path);
 | 
						|
        $request->setMethod($httpMethod);
 | 
						|
        $request->setParams($params);
 | 
						|
        $request->setHandler($handler);
 | 
						|
        return $request;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Build a relative path from base path to target path.
 | 
						|
     * Required since Solarium contains the core information
 | 
						|
     *
 | 
						|
     * @param string $basePath
 | 
						|
     * @param string $targetPath
 | 
						|
     * @return string
 | 
						|
     */
 | 
						|
    protected function buildRelativePath(string $basePath, string $targetPath): string
 | 
						|
    {
 | 
						|
        $basePath = trim($basePath, '/');
 | 
						|
        $targetPath = trim($targetPath, '/');
 | 
						|
        $baseElements = explode('/', $basePath);
 | 
						|
        $targetElements = explode('/', $targetPath);
 | 
						|
        $targetSegment = array_pop($targetElements);
 | 
						|
        foreach ($baseElements as $i => $segment) {
 | 
						|
            if (isset($targetElements[$i]) && $segment === $targetElements[$i]) {
 | 
						|
                unset($baseElements[$i], $targetElements[$i]);
 | 
						|
            } else {
 | 
						|
                break;
 | 
						|
            }
 | 
						|
        }
 | 
						|
        $targetElements[] = $targetSegment;
 | 
						|
        return str_repeat('../', count($baseElements)) . implode('/', $targetElements);
 | 
						|
    }
 | 
						|
}
 |