123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207 |
- <?php
- namespace OC\Preview;
- use OC\StreamImage;
- use OCP\Files\File;
- use OCP\Http\Client\IClientService;
- use OCP\IConfig;
- use OCP\IImage;
- use OCP\Image;
- use Psr\Log\LoggerInterface;
- class Imaginary extends ProviderV2 {
-
- private $config;
-
- private $service;
-
- private $logger;
- public function __construct(array $config) {
- parent::__construct($config);
- $this->config = \OC::$server->get(IConfig::class);
- $this->service = \OC::$server->get(IClientService::class);
- $this->logger = \OC::$server->get(LoggerInterface::class);
- }
-
- public function getMimeType(): string {
- return self::supportedMimeTypes();
- }
- public static function supportedMimeTypes(): string {
- return '/(image\/(bmp|x-bitmap|png|jpeg|gif|heic|heif|svg\+xml|tiff|webp)|application\/(pdf|illustrator))/';
- }
- public function getCroppedThumbnail(File $file, int $maxX, int $maxY, bool $crop): ?IImage {
- $maxSizeForImages = $this->config->getSystemValueInt('preview_max_filesize_image', 50);
- $size = $file->getSize();
- if ($maxSizeForImages !== -1 && $size > ($maxSizeForImages * 1024 * 1024)) {
- return null;
- }
- $imaginaryUrl = $this->config->getSystemValueString('preview_imaginary_url', 'invalid');
- if ($imaginaryUrl === 'invalid') {
- $this->logger->error('Imaginary preview provider is enabled, but no url is configured. Please provide the url of your imaginary server to the \'preview_imaginary_url\' config variable.');
- return null;
- }
- $imaginaryUrl = rtrim($imaginaryUrl, '/');
-
- $stream = $file->fopen('r');
- if (!$stream || !is_resource($stream) || feof($stream)) {
- return null;
- }
- $httpClient = $this->service->newClient();
- $convert = false;
- $autorotate = true;
- switch ($file->getMimeType()) {
- case 'image/heic':
-
- $autorotate = false;
- $mimeType = 'jpeg';
- break;
- case 'image/gif':
- case 'image/png':
- $mimeType = 'png';
- break;
- case 'image/svg+xml':
- case 'application/pdf':
- case 'application/illustrator':
- $convert = true;
-
- $autorotate = false;
- $mimeType = 'png';
- break;
- default:
- $mimeType = 'jpeg';
- }
- $preview_format = $this->config->getSystemValueString('preview_format', 'jpeg');
- switch ($preview_format) {
- case 'webp':
- $mimeType = 'webp';
- break;
- default:
- }
- $operations = [];
- if ($convert) {
- $operations[] = [
- 'operation' => 'convert',
- 'params' => [
- 'type' => $mimeType,
- ]
- ];
- } elseif ($autorotate) {
- $operations[] = [
- 'operation' => 'autorotate',
- ];
- }
- switch ($mimeType) {
- case 'jpeg':
- $quality = $this->config->getAppValue('preview', 'jpeg_quality', '80');
- break;
- case 'webp':
- $quality = $this->config->getAppValue('preview', 'webp_quality', '80');
- break;
- default:
- $quality = $this->config->getAppValue('preview', 'jpeg_quality', '80');
- }
- $operations[] = [
- 'operation' => ($crop ? 'smartcrop' : 'fit'),
- 'params' => [
- 'width' => $maxX,
- 'height' => $maxY,
- 'stripmeta' => 'true',
- 'type' => $mimeType,
- 'norotation' => 'true',
- 'quality' => $quality,
- ]
- ];
- try {
- $imaginaryKey = $this->config->getSystemValueString('preview_imaginary_key', '');
- $response = $httpClient->post(
- $imaginaryUrl . '/pipeline', [
- 'query' => ['operations' => json_encode($operations), 'key' => $imaginaryKey],
- 'stream' => true,
- 'content-type' => $file->getMimeType(),
- 'body' => $stream,
- 'nextcloud' => ['allow_local_address' => true],
- 'timeout' => 120,
- 'connect_timeout' => 3,
- ]);
- } catch (\Throwable $e) {
- $this->logger->info('Imaginary preview generation failed: ' . $e->getMessage(), [
- 'exception' => $e,
- ]);
- return null;
- }
- if ($response->getStatusCode() !== 200) {
- $this->logger->info('Imaginary preview generation failed: ' . json_decode($response->getBody())['message']);
- return null;
- }
-
-
-
- if ($response->getHeader('Image-Width') && $response->getHeader('Image-Height')) {
- $image = new StreamImage(
- $response->getBody(),
- $response->getHeader('Content-Type'),
- (int)$response->getHeader('Image-Width'),
- (int)$response->getHeader('Image-Height'),
- );
- } else {
- $image = new Image();
- $image->loadFromFileHandle($response->getBody());
- }
- return $image->valid() ? $image : null;
- }
-
- public function getThumbnail(File $file, int $maxX, int $maxY): ?IImage {
- return $this->getCroppedThumbnail($file, $maxX, $maxY, false);
- }
- }
|