93 lines
2.2 KiB
PHP
93 lines
2.2 KiB
PHP
<?php
|
|
use Guzzle\Http\Client;
|
|
|
|
/**
|
|
* Text extractor that calls an Apache Solr instance
|
|
* and extracts content via the "ExtractingRequestHandler" endpoint.
|
|
* Does not alter the Solr index itself, but uses it purely
|
|
* for its file parsing abilities.
|
|
*
|
|
* @author ischommer
|
|
* @see http://wiki.apache.org/solr/ExtractingRequestHandler
|
|
*/
|
|
class SolrCellTextExtractor extends FileTextExtractor {
|
|
|
|
/**
|
|
* Base URL to use for solr text extraction.
|
|
* E.g. http://localhost:8983/solr/update/extract
|
|
*
|
|
* @config
|
|
* @var string
|
|
*/
|
|
private static $base_url;
|
|
|
|
private static $priority = 75;
|
|
|
|
protected $httpClient;
|
|
|
|
public function getHttpClient() {
|
|
if(!$this->config()->get('base_url')) {
|
|
throw new InvalidArgumentException('SolrCellTextExtractor.base_url not specified');
|
|
}
|
|
if(!$this->httpClient) $this->httpClient = new Client($this->config()->get('base_url'));
|
|
return $this->httpClient;
|
|
}
|
|
|
|
public function setHttpClient($client) {
|
|
$this->httpClient = $client;
|
|
}
|
|
|
|
public function isAvailable() {
|
|
$url = $this->config()->get('base_url');
|
|
return (boolean) $url;
|
|
}
|
|
|
|
public function supportsExtension($extension) {
|
|
return in_array(
|
|
strtolower($extension),
|
|
array(
|
|
'pdf', 'doc', 'docx', 'xls', 'xlsx',
|
|
'epub', 'rtf', 'odt', 'fodt', 'ods', 'fods',
|
|
'ppt', 'pptx', 'odp', 'fodp', 'csv'
|
|
)
|
|
);
|
|
}
|
|
|
|
public function supportsMime($mime) {
|
|
// Rely on supportsExtension
|
|
return false;
|
|
}
|
|
|
|
public function getContent($path) {
|
|
if (!$path) return ""; // no file
|
|
|
|
$fileName = basename($path);
|
|
$client = $this->getHttpClient();
|
|
try {
|
|
$request = $client
|
|
->post()
|
|
->addPostFields(array('extractOnly' => 'true', 'extractFormat' => 'text'))
|
|
->addPostFiles(array('myfile' => $path));
|
|
$response = $request->send();
|
|
} catch(InvalidArgumentException $e) {
|
|
SS_Log::log(
|
|
sprintf(
|
|
'Error extracting text from "%s" (message: %s)',
|
|
$path,
|
|
$e->getMessage()
|
|
),
|
|
SS_Log::NOTICE
|
|
);
|
|
return null;
|
|
}
|
|
// Use preg match to avoid SimpleXML running out of memory on large text nodes
|
|
preg_match(
|
|
sprintf('/\<str name\="%s"\>(.*?)\<\/str\>/s', preg_quote($fileName)),
|
|
(string)$response->getBody(),
|
|
$matches
|
|
);
|
|
|
|
return $matches ? $matches[1] : null;
|
|
}
|
|
}
|