138 lines
3.1 KiB
PHP
138 lines
3.1 KiB
PHP
<?php
|
|
|
|
namespace SilverStripe\TextExtraction\Extractor;
|
|
|
|
use SilverStripe\Assets\File;
|
|
use SilverStripe\Core\Environment;
|
|
use SilverStripe\Core\Injector\Injector;
|
|
use SilverStripe\TextExtraction\Rest\TikaRestClient;
|
|
|
|
/**
|
|
* Enables text extraction of file content via the Tika Rest Server
|
|
*
|
|
* {@link http://tika.apache.org/1.7/gettingstarted.html}
|
|
*/
|
|
class TikaServerTextExtractor extends FileTextExtractor
|
|
{
|
|
/**
|
|
* Tika server is pretty efficient so use it immediately if available
|
|
*
|
|
* @var integer
|
|
* @config
|
|
*/
|
|
private static $priority = 80;
|
|
|
|
/**
|
|
* Server endpoint
|
|
*
|
|
* @var string
|
|
* @config
|
|
*/
|
|
private static $server_endpoint;
|
|
|
|
/**
|
|
* @var TikaRestClient
|
|
*/
|
|
protected $client = null;
|
|
|
|
/**
|
|
* Cache of supported mime types
|
|
*
|
|
* @var array
|
|
*/
|
|
protected $supportedMimes = [];
|
|
|
|
/**
|
|
* @return TikaRestClient
|
|
*/
|
|
public function getClient()
|
|
{
|
|
if (!$this->client) {
|
|
$this->client = Injector::inst()->createWithArgs(
|
|
TikaRestClient::class,
|
|
[$this->getServerEndpoint()]
|
|
);
|
|
}
|
|
return $this->client;
|
|
}
|
|
|
|
/**
|
|
* @return string
|
|
*/
|
|
public function getServerEndpoint()
|
|
{
|
|
if ($endpoint = Environment::getEnv('SS_TIKA_ENDPOINT')) {
|
|
return $endpoint;
|
|
}
|
|
|
|
// Default to configured endpoint
|
|
return $this->config()->get('server_endpoint');
|
|
}
|
|
|
|
/**
|
|
* Get the version of Tika installed, or 0 if not installed
|
|
*
|
|
* @return float version of Tika
|
|
*/
|
|
public function getVersion()
|
|
{
|
|
return $this->getClient()->getVersion();
|
|
}
|
|
|
|
/**
|
|
* @return boolean
|
|
*/
|
|
public function isAvailable()
|
|
{
|
|
return $this->getServerEndpoint()
|
|
&& $this->getClient()->isAvailable()
|
|
&& version_compare($this->getVersion() ?? '', '1.7') >= 0;
|
|
}
|
|
|
|
/**
|
|
* @param string $extension
|
|
* @return boolean
|
|
*/
|
|
public function supportsExtension($extension)
|
|
{
|
|
// Determine support via mime type only
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* @param string $mime
|
|
* @return boolean
|
|
*/
|
|
public function supportsMime($mime)
|
|
{
|
|
if (!$this->supportedMimes) {
|
|
$this->supportedMimes = (array) $this->getClient()->getSupportedMimes();
|
|
}
|
|
|
|
// Check if supported (most common / quickest lookup)
|
|
if (isset($this->supportedMimes[$mime])) {
|
|
return true;
|
|
}
|
|
|
|
// Check aliases
|
|
foreach ($this->supportedMimes as $info) {
|
|
if (isset($info['alias']) && in_array($mime, $info['alias'] ?? [])) {
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public function getContent($file)
|
|
{
|
|
$tempFile = $file instanceof File ? $this->getPathFromFile($file) : $file;
|
|
$content = $this->getClient()->tika($tempFile);
|
|
//Cleanup temp file
|
|
if ($file instanceof File) {
|
|
unlink($tempFile ?? '');
|
|
}
|
|
return $content;
|
|
}
|
|
}
|