152 lines
3.7 KiB
PHP
152 lines
3.7 KiB
PHP
<?php
|
|
|
|
namespace SilverStripe\TextExtraction\Extractor;
|
|
|
|
use SilverStripe\Assets\File;
|
|
use SilverStripe\TextExtraction\Extractor\FileTextExtractor\Exception;
|
|
use function tempnam;
|
|
|
|
/**
|
|
* Text extractor that calls pdftotext to do the conversion.
|
|
* @author mstephens
|
|
*/
|
|
class PDFTextExtractor extends FileTextExtractor
|
|
{
|
|
/**
|
|
* Set to bin path this extractor can execute
|
|
*
|
|
* @var string
|
|
*/
|
|
private static $binary_location = null;
|
|
|
|
/**
|
|
* Used if binary_location isn't set.
|
|
* List of locations to search for a given binary in
|
|
*
|
|
* @config
|
|
* @var array
|
|
*/
|
|
private static $search_binary_locations = [
|
|
'/usr/bin',
|
|
'/usr/local/bin',
|
|
];
|
|
|
|
public function isAvailable()
|
|
{
|
|
$bin = $this->bin('pdftotext');
|
|
return $bin && file_exists($bin) && is_executable($bin);
|
|
}
|
|
|
|
public function supportsExtension($extension)
|
|
{
|
|
return strtolower($extension) === 'pdf';
|
|
}
|
|
|
|
public function supportsMime($mime)
|
|
{
|
|
return in_array(
|
|
strtolower($mime),
|
|
[
|
|
'application/pdf',
|
|
'application/x-pdf',
|
|
'application/x-bzpdf',
|
|
'application/x-gzpdf'
|
|
]
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Accessor to get the location of the binary
|
|
*
|
|
* @param string $program Name of binary
|
|
* @return string
|
|
*/
|
|
protected function bin($program = '')
|
|
{
|
|
// Get list of allowed search paths
|
|
if ($location = $this->config()->get('binary_location')) {
|
|
$locations = [$location];
|
|
} else {
|
|
$locations = $this->config()->get('search_binary_locations');
|
|
}
|
|
|
|
// Find program in each path
|
|
foreach ($locations as $location) {
|
|
$path = "{$location}/{$program}";
|
|
if (file_exists($path)) {
|
|
return $path;
|
|
}
|
|
if (file_exists($path . '.exe')) {
|
|
return $path . '.exe';
|
|
}
|
|
}
|
|
|
|
// Not found
|
|
return null;
|
|
}
|
|
|
|
public function getContent(File $file)
|
|
{
|
|
if (!$file) {
|
|
// no file
|
|
return '';
|
|
}
|
|
$content = $this->getRawOutput($file);
|
|
return $this->cleanupLigatures($content);
|
|
}
|
|
|
|
/**
|
|
* Invoke pdftotext with the given File object
|
|
*
|
|
* @param File $file
|
|
* @return string Output
|
|
* @throws Exception
|
|
*/
|
|
protected function getRawOutput(File $file)
|
|
{
|
|
if (!$this->isAvailable()) {
|
|
throw new Exception("getRawOutput called on unavailable extractor");
|
|
}
|
|
|
|
$path = $this->getPathFromFile($file);
|
|
exec(sprintf('%s %s - 2>&1', $this->bin('pdftotext'), escapeshellarg($path)), $content, $err);
|
|
if ($err) {
|
|
if (!is_array($err) && $err == 1) {
|
|
// For Windows compatibility
|
|
$err = $content;
|
|
}
|
|
|
|
throw new Exception(sprintf(
|
|
'PDFTextExtractor->getContent() failed for %s: %s',
|
|
$path,
|
|
implode(PHP_EOL, $err)
|
|
));
|
|
}
|
|
|
|
return implode(PHP_EOL, $content);
|
|
}
|
|
|
|
/**
|
|
* Removes utf-8 ligatures.
|
|
*
|
|
* @link http://en.wikipedia.org/wiki/Typographic_ligature#Computer_typesetting
|
|
*
|
|
* @param string $input
|
|
* @return string
|
|
*/
|
|
protected function cleanupLigatures($input)
|
|
{
|
|
$mapping = [
|
|
'ff' => 'ff',
|
|
'fi' => 'fi',
|
|
'fl' => 'fl',
|
|
'ffi' => 'ffi',
|
|
'ffl' => 'ffl',
|
|
'ſt' => 'ft',
|
|
'st' => 'st'
|
|
];
|
|
|
|
return str_replace(array_keys($mapping), array_values($mapping), $input);
|
|
}
|
|
}
|