EntityIconService.php

Go to the documentation of this file.

<?php

namespace Elgg;

use Elgg\Database\EntityTable;
use Elgg\Filesystem\MimeTypeDetector;
use Elgg\Http\Request as HttpRequest;
use ElggEntity;
use ElggFile;
use ElggIcon;
use InvalidParameterException;
use Psr\Log\LoggerInterface;
use Symfony\Component\HttpFoundation\BinaryFileResponse;
use Symfony\Component\HttpFoundation\Response;

class EntityIconService {

    use Loggable;
    use TimeUsing;

    private $config;

    private $hooks;

    private $request;

    private $entities;

    private $uploads;

    private $images;

    public function __construct(
        Config $config,
        PluginHooksService $hooks,
        HttpRequest $request,
        LoggerInterface $logger,
        EntityTable $entities,
        UploadService $uploads,
        ImageService $images
    ) {
        $this->config = $config;
        $this->hooks = $hooks;
        $this->request = $request;
        $this->logger = $logger;
        $this->entities = $entities;
        $this->uploads = $uploads;
        $this->images = $images;
    }

    public function saveIconFromUploadedFile(ElggEntity $entity, $input_name, $type = 'icon', array $coords = []) {
        $input = $this->uploads->getFile($input_name);
        if (empty($input)) {
            return false;
        }
                
        // auto detect cropping coordinates
        if (empty($coords)) {
            $auto_coords = $this->detectCroppingCoordinates();
            if (!empty($auto_coords)) {
                $coords = $auto_coords;
            }
        }

        $tmp = new \ElggTempFile();
        $tmp->setFilename(uniqid() . $input->getClientOriginalName());
        $tmp->open('write');
        $tmp->close();
        
        copy($input->getPathname(), $tmp->getFilenameOnFilestore());

        $tmp->mimetype = (new MimeTypeDetector())->getType($tmp->getFilenameOnFilestore(), $input->getClientMimeType());
        $tmp->simpletype = elgg_get_file_simple_type($tmp->mimetype);

        $result = $this->saveIcon($entity, $tmp, $type, $coords);

        $tmp->delete();

        return $result;
    }

    public function saveIconFromLocalFile(ElggEntity $entity, $filename, $type = 'icon', array $coords = []) {
        if (!file_exists($filename) || !is_readable($filename)) {
            throw new InvalidParameterException(__METHOD__ . " expects a readable local file. $filename is not readable");
        }
                
        $tmp = new \ElggTempFile();
        $tmp->setFilename(uniqid() . basename($filename));
        $tmp->open('write');
        $tmp->close();
        
        copy($filename, $tmp->getFilenameOnFilestore());

        $tmp->mimetype = (new MimeTypeDetector())->getType($tmp->getFilenameOnFilestore());
        $tmp->simpletype = elgg_get_file_simple_type($tmp->mimetype);

        $result = $this->saveIcon($entity, $tmp, $type, $coords);

        $tmp->delete();

        return $result;
    }

    public function saveIconFromElggFile(ElggEntity $entity, ElggFile $file, $type = 'icon', array $coords = []) {
        if (!$file->exists()) {
            throw new InvalidParameterException(__METHOD__ . ' expects an instance of ElggFile with an existing file on filestore');
        }
        
        $tmp = new \ElggTempFile();
        $tmp->setFilename(uniqid() . basename($file->getFilenameOnFilestore()));
        $tmp->open('write');
        $tmp->close();
        
        copy($file->getFilenameOnFilestore(), $tmp->getFilenameOnFilestore());

        $tmp->mimetype = (new MimeTypeDetector())->getType($tmp->getFilenameOnFilestore(), $file->getMimeType());
        $tmp->simpletype = elgg_get_file_simple_type($tmp->mimetype);

        $result = $this->saveIcon($entity, $tmp, $type, $coords);

        $tmp->delete();

        return $result;
    }

    public function saveIcon(ElggEntity $entity, ElggFile $file, $type = 'icon', array $coords = []) {

        $type = (string) $type;
        if (!strlen($type)) {
            $this->logger->error('Icon type passed to ' . __METHOD__ . ' can not be empty');
            return false;
        }
        
        $entity_type = $entity->getType();
        
        $file = $this->hooks->trigger("entity:$type:prepare", $entity_type, [
            'entity' => $entity,
            'file' => $file,
        ], $file);
        
        if (!$file instanceof ElggFile || !$file->exists() || $file->getSimpleType() !== 'image') {
            $this->logger->error('Source file passed to ' . __METHOD__ . ' can not be resolved to a valid image');
            return false;
        }
        
        $this->prepareIcon($file->getFilenameOnFilestore());
        
        $x1 = (int) elgg_extract('x1', $coords);
        $y1 = (int) elgg_extract('y1', $coords);
        $x2 = (int) elgg_extract('x2', $coords);
        $y2 = (int) elgg_extract('y2', $coords);
        
        $created = $this->hooks->trigger("entity:$type:save", $entity_type, [
            'entity' => $entity,
            'file' => $file,
            'x1' => $x1,
            'y1' => $y1,
            'x2' => $x2,
            'y2' => $y2,
        ], false);

        // did someone else handle saving the icon?
        if ($created !== true) {
            // remove existing icons
            $this->deleteIcon($entity, $type, true);
            
            // save master image
            $store = $this->generateIcon($entity, $file, $type, $coords, 'master');
            
            if (!$store) {
                $this->deleteIcon($entity, $type);
                return false;
            }
        }

        if ($type == 'icon') {
            $entity->icontime = time();
            if ($x1 || $y1 || $x2 || $y2) {
                $entity->x1 = $x1;
                $entity->y1 = $y1;
                $entity->x2 = $x2;
                $entity->y2 = $y2;
            }
        } else {
            if ($x1 || $y1 || $x2 || $y2) {
                $entity->{"{$type}_coords"} = serialize([
                    'x1' => $x1,
                    'y1' => $y1,
                    'x2' => $x2,
                    'y2' => $y2,
                ]);
            }
        }
        
        $this->hooks->trigger("entity:$type:saved", $entity->getType(), [
            'entity' => $entity,
            'x1' => $x1,
            'y1' => $y1,
            'x2' => $x2,
            'y2' => $y2,
        ]);
        
        return true;
    }
    
    protected function prepareIcon($filename) {
        
        // fix orientation
        $temp_file = new \ElggTempFile();
        $temp_file->setFilename(uniqid() . basename($filename));
        
        copy($filename, $temp_file->getFilenameOnFilestore());
        
        $rotated = $this->images->fixOrientation($temp_file->getFilenameOnFilestore());

        if ($rotated) {
            copy($temp_file->getFilenameOnFilestore(), $filename);
        }
        
        $temp_file->delete();
    }
    
    protected function generateIcon(ElggEntity $entity, ElggFile $file, $type = 'icon', $coords = [], $icon_size = '') {
        
        if (!$file->exists()) {
            $this->logger->error('Trying to generate an icon from a non-existing file');
            return false;
        }
        
        $x1 = (int) elgg_extract('x1', $coords);
        $y1 = (int) elgg_extract('y1', $coords);
        $x2 = (int) elgg_extract('x2', $coords);
        $y2 = (int) elgg_extract('y2', $coords);
        
        $sizes = $this->getSizes($entity->getType(), $entity->getSubtype(), $type);
        
        if (!empty($icon_size) && !isset($sizes[$icon_size])) {
            $this->logger->warning("The provided icon size '{$icon_size}' doesn't exist for icon type '{$type}'");
            return false;
        }
        
        foreach ($sizes as $size => $opts) {
            if (!empty($icon_size) && ($icon_size !== $size)) {
                // only generate the given icon size
                continue;
            }
            
            // check if the icon config allows cropping
            if (!(bool) elgg_extract('crop', $opts, true)) {
                $coords = [
                    'x1' => 0,
                    'y1' => 0,
                    'x2' => 0,
                    'y2' => 0,
                ];
            }

            $icon = $this->getIcon($entity, $size, $type, false);

            // We need to make sure that file path is readable by
            // Imagine\Image\ImagineInterface::save(), as it fails to
            // build the directory structure on owner's filestore otherwise
            $icon->open('write');
            $icon->close();
            
            // Save the image without resizing or cropping if the
            // image size value is an empty array
            if (is_array($opts) && empty($opts)) {
                copy($file->getFilenameOnFilestore(), $icon->getFilenameOnFilestore());
                continue;
            }

            $source = $file->getFilenameOnFilestore();
            $destination = $icon->getFilenameOnFilestore();

            $resize_params = array_merge($opts, $coords);

            $image_service = _elgg_services()->imageService;
            $image_service->setLogger($this->logger);

            if (!_elgg_services()->imageService->resize($source, $destination, $resize_params)) {
                $this->logger->error("Failed to create {$size} icon from
                    {$file->getFilenameOnFilestore()} with coords [{$x1}, {$y1}],[{$x2}, {$y2}]");
                return false;
            }
        }

        return true;
    }

    public function getIcon(ElggEntity $entity, $size, $type = 'icon', $generate = true) {

        $size = elgg_strtolower($size);

        $params = [
            'entity' => $entity,
            'size' => $size,
            'type' => $type,
        ];

        $entity_type = $entity->getType();

        $default_icon = new ElggIcon();
        $default_icon->owner_guid = $entity->guid;
        $default_icon->setFilename("icons/$type/$size.jpg");

        $icon = $this->hooks->trigger("entity:$type:file", $entity_type, $params, $default_icon);
        if (!$icon instanceof ElggIcon) {
            throw new InvalidParameterException("'entity:$type:file', $entity_type hook must return an instance of ElggIcon");
        }
        
        if ($icon->exists() || !$generate) {
            return $icon;
        }
        
        if ($size === 'master') {
            // don't try to generate for master
            return $icon;
        }
        
        // try to generate icon based on master size
        $master_icon = $this->getIcon($entity, 'master', $type, false);
        if (!$master_icon->exists()) {
            return $icon;
        }
        
        if ($type === 'icon') {
            $coords = [
                'x1' => $entity->x1,
                'y1' => $entity->y1,
                'x2' => $entity->x2,
                'y2' => $entity->y2,
            ];
        } else {
            $coords = $entity->{"{$type}_coords"};
            $coords = empty($coords) ? [] : unserialize($coords);
        }
        
        $this->generateIcon($entity, $master_icon, $type, $coords, $size);
        
        return $icon;
    }

    public function deleteIcon(ElggEntity $entity, $type = 'icon', $retain_master = false) {
        $delete = $this->hooks->trigger("entity:$type:delete", $entity->getType(), [
            'entity' => $entity,
        ], true);

        if ($delete === false) {
            return false;
        }
        
        $result = true;

        $sizes = array_keys($this->getSizes($entity->getType(), $entity->getSubtype(), $type));
        foreach ($sizes as $size) {
            if ($size === 'master' && $retain_master) {
                continue;
            }
            
            $icon = $this->getIcon($entity, $size, $type, false);
            $result &= $icon->delete();
        }

        if ($type == 'icon') {
            unset($entity->icontime);
            unset($entity->x1);
            unset($entity->y1);
            unset($entity->x2);
            unset($entity->y2);
        } else {
            unset($entity->{"{$type}_coords"});
        }
        
        return $result;
    }

    public function getIconURL(ElggEntity $entity, $params = []) {
        if (is_array($params)) {
            $size = elgg_extract('size', $params, 'medium');
        } else {
            $size = is_string($params) ? $params : 'medium';
            $params = [];
        }

        $size = elgg_strtolower($size);

        $params['entity'] = $entity;
        $params['size'] = $size;

        $type = elgg_extract('type', $params) ? : 'icon';
        $entity_type = $entity->getType();

        $url = $this->hooks->trigger("entity:$type:url", $entity_type, $params, null);
        if ($url == null) {
            if ($this->hasIcon($entity, $size, $type)) {
                $icon = $this->getIcon($entity, $size, $type);
                $default_use_cookie = (bool) elgg_get_config('session_bound_entity_icons', false);
                $url = $icon->getInlineURL((bool) elgg_extract('use_cookie', $params, $default_use_cookie));
            } else {
                $url = $this->getFallbackIconUrl($entity, $params);
            }
        }

        if ($url) {
            return elgg_normalize_url($url);
        }
    }

    public function getFallbackIconUrl(ElggEntity $entity, array $params = []) {

        $type = elgg_extract('type', $params) ? : 'icon';
        $size = elgg_extract('size', $params) ? : 'medium';
        
        $entity_type = $entity->getType();
        $entity_subtype = $entity->getSubtype();

        $exts = ['svg', 'gif', 'png', 'jpg'];

        foreach ($exts as $ext) {
            foreach ([$entity_subtype, 'default'] as $subtype) {
                if ($ext == 'svg' && elgg_view_exists("$type/$entity_type/$subtype.svg")) {
                    return elgg_get_simplecache_url("$type/$entity_type/$subtype.svg");
                }
                if (elgg_view_exists("$type/$entity_type/$subtype/$size.$ext")) {
                    return elgg_get_simplecache_url("$type/$entity_type/$subtype/$size.$ext");
                }
            }
        }

        if (elgg_view_exists("$type/default/$size.png")) {
            return elgg_get_simplecache_url("$type/default/$size.png");
        }
    }

    public function getIconLastChange(ElggEntity $entity, $size, $type = 'icon') {
        $icon = $this->getIcon($entity, $size, $type);
        if ($icon->exists()) {
            return $icon->getModifiedTime();
        }
    }

    public function hasIcon(\ElggEntity $entity, $size, $type = 'icon') {
        return $this->getIcon($entity, $size, $type)->exists();
    }

    public function getSizes($entity_type = null, $entity_subtype = null, $type = 'icon') {
        $sizes = [];
        if (!$type) {
            $type = 'icon';
        }
        if ($type == 'icon') {
            $sizes = $this->config->icon_sizes;
        }
        $params = [
            'type' => $type,
            'entity_type' => $entity_type,
            'entity_subtype' => $entity_subtype,
        ];
        if ($entity_type) {
            $sizes = $this->hooks->trigger("entity:$type:sizes", $entity_type, $params, $sizes);
        }

        if (!is_array($sizes)) {
            throw new InvalidParameterException("The icon size configuration for image type '$type' " .
                "must be an associative array of image size names and their properties");
        }

        // lazy generation of icons requires a 'master' size
        // this ensures a default config for 'master' size
        $sizes['master'] = elgg_extract('master', $sizes, [
            'w' => 10240,
            'h' => 10240,
            'square' => false,
            'upscale' => false,
            'crop' => false,
        ]);
        
        if (!isset($sizes['master']['crop'])) {
            $sizes['master']['crop'] = false;
        }
        
        return $sizes;
    }

    public function handleServeIconRequest($allow_removing_headers = true) {

        $response = new Response();
        $response->setExpires($this->getCurrentTime('-1 day'));
        $response->prepare($this->request);

        if ($allow_removing_headers) {
            // clear cache-boosting headers set by PHP session
            header_remove('Cache-Control');
            header_remove('Pragma');
            header_remove('Expires');
        }

        $path = implode('/', $this->request->getUrlSegments());
        if (!preg_match('~serve-icon/(\d+)/(.*+)$~', $path, $m)) {
            return $response->setStatusCode(400)->setContent('Malformatted request URL');
        }

        list(, $guid, $size) = $m;

        $entity = $this->entities->get($guid);
        if (!$entity instanceof \ElggEntity) {
            return $response->setStatusCode(404)->setContent('Item does not exist');
        }

        $thumbnail = $entity->getIcon($size);
        if (!$thumbnail->exists()) {
            return $response->setStatusCode(404)->setContent('Icon does not exist');
        }

        $if_none_match = $this->request->headers->get('if_none_match');
        if (!empty($if_none_match)) {
            // strip mod_deflate suffixes
            $this->request->headers->set('if_none_match', str_replace('-gzip', '', $if_none_match));
        }

        $filenameonfilestore = $thumbnail->getFilenameOnFilestore();
        $last_updated = filemtime($filenameonfilestore);
        $etag = '"' . $last_updated . '"';

        $response->setPrivate()
            ->setEtag($etag)
            ->setExpires($this->getCurrentTime('+1 day'))
            ->setMaxAge(86400);

        if ($response->isNotModified($this->request)) {
            return $response;
        }

        $headers = [
            'Content-Type' => (new MimeTypeDetector())->getType($filenameonfilestore),
        ];
        $response = new BinaryFileResponse($filenameonfilestore, 200, $headers, false, 'inline');
        $response->prepare($this->request);

        $response->setPrivate()
            ->setEtag($etag)
            ->setExpires($this->getCurrentTime('+1 day'))
            ->setMaxAge(86400);

        return $response;
    }
    
    protected function detectCroppingCoordinates() {
        
        $auto_coords = [
            'x1' => get_input('x1'),
            'x2' => get_input('x2'),
            'y1' => get_input('y1'),
            'y2' => get_input('y2'),
        ];
        
        $auto_coords = array_filter($auto_coords, function($value) {
            return !elgg_is_empty($value) && is_numeric($value);
        });
        
        if (count($auto_coords) !== 4) {
            return false;
        }
        
        // make ints
        array_walk($auto_coords, function (&$value) {
            $value = (int) $value;
        });
        
        return $auto_coords;
    }

}