mirror of
https://github.com/silverstripe/silverstripe-fulltextsearch
synced 2024-10-22 14:05:29 +02:00
116 lines
3.2 KiB
PHP
116 lines
3.2 KiB
PHP
|
<?php
|
||
|
|
||
|
/**
|
||
|
* Class SolrConfigStore
|
||
|
*
|
||
|
* The interface Solr_Configure uses to upload configuration files to Solr
|
||
|
*/
|
||
|
interface SolrConfigStore {
|
||
|
/**
|
||
|
* 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
|
||
|
* @return null
|
||
|
*/
|
||
|
function uploadFile($index, $file);
|
||
|
|
||
|
/**
|
||
|
* Upload a file to Solr from a string 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 $filename string - The base name of the file to use on the remote side
|
||
|
* @param $strong string - The contents of the file
|
||
|
* @return null
|
||
|
*/
|
||
|
function uploadString($index, $filename, $string);
|
||
|
|
||
|
/**
|
||
|
* Get the instanceDir to tell Solr to use 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)
|
||
|
*/
|
||
|
function instanceDir($index);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Class SolrConfigStore_File
|
||
|
*
|
||
|
* A ConfigStore that uploads files to a Solr instance on a locally accessible filesystem
|
||
|
* by just using file copies
|
||
|
*/
|
||
|
class SolrConfigStore_File implements SolrConfigStore {
|
||
|
function __construct($config) {
|
||
|
$this->local = $config['path'];
|
||
|
$this->remote = isset($config['remotepath']) ? $config['remotepath'] : $config['path'];
|
||
|
}
|
||
|
|
||
|
function getTargetDir($index) {
|
||
|
$targetDir = "{$this->local}/{$index}/conf";
|
||
|
|
||
|
if (!is_dir($targetDir)) {
|
||
|
$worked = @mkdir($targetDir, 0770, true);
|
||
|
|
||
|
if(!$worked) {
|
||
|
throw new RuntimeException(
|
||
|
sprintf('Failed creating target directory %s, please check permissions', $targetDir)
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return $targetDir;
|
||
|
}
|
||
|
|
||
|
function uploadFile($index, $file) {
|
||
|
$targetDir = $this->getTargetDir($index);
|
||
|
copy($file, $targetDir.'/'.basename($file));
|
||
|
}
|
||
|
|
||
|
function uploadString($index, $filename, $string) {
|
||
|
$targetDir = $this->getTargetDir($index);
|
||
|
file_put_contents("$targetDir/$filename", $string);
|
||
|
}
|
||
|
|
||
|
function instanceDir($index) {
|
||
|
return $this->remote.'/'.$index;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Class SolrConfigStore_WebDAV
|
||
|
*
|
||
|
* A ConfigStore that uploads files to a Solr instance via a WebDAV server
|
||
|
*/
|
||
|
class SolrConfigStore_WebDAV implements SolrConfigStore {
|
||
|
function __construct($config) {
|
||
|
$this->url = implode('', array(
|
||
|
'http://',
|
||
|
isset($config['auth']) ? $config['auth'].'@' : '',
|
||
|
Solr::$solr_options['host'] . ':' . Solr::$solr_options['port'],
|
||
|
$config['path']
|
||
|
));
|
||
|
$this->remote = $config['remotepath'];
|
||
|
}
|
||
|
|
||
|
function getTargetDir($index) {
|
||
|
$indexdir = "{$this->url}/$index";
|
||
|
if (!WebDAV::exists($indexdir)) WebDAV::mkdir($indexdir);
|
||
|
|
||
|
$targetDir = "{$this->url}/$index/conf";
|
||
|
if (!WebDAV::exists($targetDir)) WebDAV::mkdir($targetDir);
|
||
|
|
||
|
return $targetDir;
|
||
|
}
|
||
|
|
||
|
function uploadFile($index, $file) {
|
||
|
$targetDir = $this->getTargetDir($index);
|
||
|
WebDAV::upload_from_file($file, $targetDir.'/'.basename($file));
|
||
|
}
|
||
|
|
||
|
function uploadString($index, $filename, $string) {
|
||
|
$targetDir = $this->getTargetDir($index);
|
||
|
WebDAV::upload_from_string($string, "$targetDir/$filename");
|
||
|
}
|
||
|
|
||
|
function instanceDir($index) {
|
||
|
return $this->remote ? "{$this->remote}/$index" : $index;
|
||
|
}
|
||
|
}
|