Firesphere/silverstripe-solr-search

<?php
/**
 * class PostConfigStore|Firesphere\SolrSearch\Stores\PostConfigStore Post storage configuration for a core
 *
 * @package Firesphere\Solr\Search
 * @author Simon `Firesphere` Erkelens; Marco `Sheepy` Hermo
 * @copyright Copyright (c) 2018 - now() Firesphere & Sheepy
 */

namespace Firesphere\SolrSearch\Stores;

use Countable;
use Firesphere\SolrSearch\Interfaces\ConfigStore;
use GuzzleHttp\Client;
use LogicException;
use Psr\Http\Message\ResponseInterface;
use Solarium\Exception\RuntimeException;

/**
 * Class PostConfigStore
 *
 * Store configurations for Solr through a POST to a remote system.
 *
 * @package Firesphere\Solr\Search
 */
class PostConfigStore implements ConfigStore
{
    /**
     * @var array file extensions
     */
    protected static $extensions = [
        'xml' => 'text/xml',
        'txt' => 'text/plain',
    ];
    /**
     * @var array Store configuration
     */
    protected $config;

    /**
     * FileConfigStore constructor.
     *
     * @param array $config
     */
    public function __construct($config)
    {
        if (empty($config)) {
            throw new RuntimeException('No config defined', 1);
        }
        if (!isset($config['uri'])) {
            throw new LogicException('No URI endpoint defined');
        }
        if (!isset($config['path'])) {
            $config['path'] = '/';
        }

        if (substr($config['path'], -1) !== '/') {
            $config['path'] .= '/';
        }

        $this->config = $config;
    }

    /**
     * Upload a file to Solr for index $index
     *
     * @param $index string - The name of an index (which is also used as the name of the Solr core for the index)
     * @param $file string - A path to a file to upload. The base name of the file will be used on the remote side
     * @param null|Countable $handler A handler used for testing, not to be used in Live environments
     * @return ResponseInterface
     */
    public function uploadFile($index, $file, $handler = null)
    {
        return $this->uploadString($index, $file, file_get_contents($file), $handler);
    }

    /**
     * Upload a file to Solr from a string for index $index
     *
     * @param string $index - The name of an index (which is also used as the name of the Solr core for the index)
     * @param string $filename - The base name of the file to use on the remote side
     * @param string $string - The content to upload
     * @param null|Countable $handler A handler used for testing, not to be used in Live environments
     * @return ResponseInterface
     */
    public function uploadString($index, $filename, $string, $handler = null)
    {
        $info = pathinfo($filename);
        $clientConfig = [
            'base_uri' => $this->config['uri'],
            'headers'  => [
                'Content-Type' => static::$extensions[$info['extension']] ?? 'text/plain',
            ],
        ];
        // Add auth to the post if needed
        if (isset($this->config['auth'])) {
            $clientConfig['auth'] = [
                $this->config['auth']['username'],
                $this->config['auth']['password'],
            ];
        }
        // For testing purposes, set the handler. Usually not used
        if ($handler) {
            $clientConfig['handler'] = $handler;
        }

        $client = new Client($clientConfig);

        $path = sprintf('%s%s', $this->getPath(), $index);

        return $client->post($path, ['body' => $string]);
    }

    /**
     * Get the path to the config
     *
     * @return string
     */
    public function getPath()
    {
        return $this->config['path'];
    }

    /**
     * Get the instanceDir to tell Solr to use for index $index
     *
     * @param string|null $index string - The name of an index
     *          (which is also used as the name of the Solr core for the index)
     * @return null|string
     */
    public function instanceDir($index)
    {
        return $index;
    }
}