ElggFile.php

Go to the documentation of this file.

<?php

use Elgg\Exceptions\Filesystem\IOException;
use Elgg\Exceptions\InvalidArgumentException as ElggInvalidArgumentException;
use Elgg\Exceptions\InvalidParameterException;
use Elgg\Project\Paths;
use Symfony\Component\HttpFoundation\File\Exception\FileException;
use Symfony\Component\HttpFoundation\File\UploadedFile;

class ElggFile extends ElggObject {

    private $handle;

    protected function initializeAttributes() {
        parent::initializeAttributes();

        $this->attributes['subtype'] = 'file';
    }
    
    public function __set($name, $value) {
        switch ($name) {
            case 'filename':
                // ensure sanitization
                $this->setFilename($value);
                return;
        }
        
        parent::__set($name, $value);
    }
    
    public function __get($name) {
        switch ($name) {
            case 'filename':
                // ensure sanitization
                return $this->getFilename();
        }
        
        return parent::__get($name);
    }

    public function setFilename($filename) {
        $filename = ltrim(Paths::sanitize($filename, false), '/');
        
        parent::__set('filename', $filename);
    }

    public function getFilename() {
        $filename = parent::__get('filename');
        if (empty($filename)) {
            return '';
        }
        
        return ltrim(Paths::sanitize($filename, false), '/');
    }

    public function getFilenameOnFilestore() {
        return $this->getFilestore()->getFilenameOnFilestore($this);
    }

    public function getFilestoreSize($prefix = '', $container_guid = 0) {
        if (!$container_guid) {
            $container_guid = $this->container_guid;
        }
        
        return (int) $this->getFilestore()->getSize($prefix, $container_guid);
    }

    public function getMimeType() {
        if ($this->mimetype) {
            return $this->mimetype;
        }
        
        try {
            return _elgg_services()->mimetype->getMimeType($this->getFilenameOnFilestore());
        } catch (ElggInvalidArgumentException $e) {
            // the file has no file on the filesystem
            // can happen in tests etc.
        }
        return false;
    }

    public function setMimeType($mimetype) {
        return $this->mimetype = $mimetype;
    }

    public function getSimpleType() {
        if (isset($this->simpletype)) {
            return $this->simpletype;
        }
        
        return _elgg_services()->mimetype->getSimpleType($this->getMimeType() ?: '');
    }

    public function open($mode) {
        if (!$this->getFilename()) {
            throw new IOException('You must specify a name before opening a file.');
        }

        // See if file has already been saved
        // seek on datastore, parameters and name?
        // Sanity check
        if (
                ($mode != 'read') &&
                ($mode != 'write') &&
                ($mode != 'append')
        ) {
            throw new InvalidParameterException("Unrecognized file mode '{$mode}'");
        }

        // Open the file handle
        $this->handle = $this->getFilestore()->open($this, $mode);

        return $this->handle;
    }

    public function write($data) {
        return $this->getFilestore()->write($this->handle, $data);
    }

    public function read($length, $offset = 0) {
        return $this->getFilestore()->read($this->handle, $length, $offset);
    }

    public function grabFile() {
        return $this->getFilestore()->grabFile($this);
    }

    public function close() {
        if ($this->getFilestore()->close($this->handle)) {
            $this->handle = null;

            return true;
        }

        return false;
    }

    public function delete($follow_symlinks = true) {
        $result = $this->getFilestore()->delete($this, $follow_symlinks);

        if ($this->getGUID() && $result) {
            $result = parent::delete();
        }

        return $result;
    }

    public function seek($position) {
        $this->getFilestore()->seek($this->handle, $position);
    }

    public function tell() {
        return $this->getFilestore()->tell($this->handle);
    }

    public function setModifiedTime() {
        $filestorename = $this->getFilenameOnFilestore();
        
        $modified = touch($filestorename);
        if ($modified) {
            clearstatcache(true, $filestorename);
        } else {
            elgg_log("Unable to update modified time for {$filestorename}", 'ERROR');
        }
        
        return $modified;
    }

    public function getModifiedTime() {
        return filemtime($this->getFilenameOnFilestore());
    }

    public function getSize() {
        return $this->getFilestore()->getFileSize($this);
    }

    public function eof() {
        return $this->getFilestore()->eof($this->handle);
    }

    public function exists() {
        return $this->getFilestore()->exists($this);
    }

    protected function getFilestore() {
        return _elgg_services()->filestore;
    }

    public function transfer($owner_guid, $filename = null) {
        if (!$owner_guid) {
            return false;
        }

        if (!$this->exists()) {
            return false;
        }

        if (!$filename) {
            $filename = $this->getFilename();
        }
        $filestorename = $this->getFilenameOnFilestore();

        $this->owner_guid = $owner_guid;
        $this->setFilename($filename);
        $this->open('write');
        $this->close();

        return rename($filestorename, $this->getFilenameOnFilestore());
    }

    public function acceptUploadedFile(UploadedFile $upload) {
        if (!$upload->isValid()) {
            return false;
        }

        $old_filestorename = '';
        if ($this->exists()) {
            $old_filestorename = $this->getFilenameOnFilestore();
        }

        $originalfilename = $upload->getClientOriginalName();
        $this->originalfilename = $originalfilename;
        if (empty($this->title)) {
            $this->title = htmlspecialchars($this->originalfilename, ENT_QUOTES, 'UTF-8');
        }

        $this->upload_time = time();
        $prefix = $this->filestore_prefix ?: 'file';
        $prefix = trim($prefix, '/');
        $filename = elgg_strtolower("{$prefix}/{$this->upload_time}{$this->originalfilename}");
        $this->setFilename($filename);
        $this->filestore_prefix = $prefix;

        $hook_params = [
            'file' => $this,
            'upload' => $upload,
        ];

        $uploaded = _elgg_services()->hooks->trigger('upload', 'file', $hook_params);
        if ($uploaded !== true && $uploaded !== false) {
            $filestorename = $this->getFilenameOnFilestore();
            try {
                $uploaded = $upload->move(pathinfo($filestorename, PATHINFO_DIRNAME), pathinfo($filestorename, PATHINFO_BASENAME));
            } catch (FileException $ex) {
                _elgg_services()->logger->error($ex->getMessage());
                $uploaded = false;
            }
        }

        if ($uploaded) {
            if ($old_filestorename && $old_filestorename != $this->getFilenameOnFilestore()) {
                // remove old file
                unlink($old_filestorename);
            }
            
            try {
                // try to detect mimetype
                $mime_type = _elgg_services()->mimetype->getMimeType($this->getFilenameOnFilestore());
                $this->setMimeType($mime_type);
                $this->simpletype = _elgg_services()->mimetype->getSimpleType($mime_type);
            } catch (ElggInvalidArgumentException $e) {
                // this can fail if the upload hooks returns true, but the file is not present on the filestore
                // this happens in a unittest
            }
            _elgg_services()->events->triggerAfter('upload', 'file', $this);
            return true;
        }

        return false;
    }

    public function __sleep() {
        return array_diff(array_keys(get_object_vars($this)), [
            // a resource
            'handle',
        ]);
    }

    public function canDownload($user_guid = 0, $default = true) {
        return _elgg_services()->userCapabilities->canDownload($this, $user_guid, $default);
    }

    public function getDownloadURL($use_cookie = true, $expires = '+2 hours') {
        $file_svc = new \Elgg\FileService\File();
        $file_svc->setFile($this);
        if (!empty($expires)) {
            $file_svc->setExpires($expires);
        }
        $file_svc->setDisposition('attachment');
        $file_svc->bindSession($use_cookie);

        $params = [
            'entity' => $this,
        ];
        return _elgg_services()->hooks->trigger('download:url', 'file', $params, $file_svc->getURL());
    }

    public function getInlineURL($use_cookie = false, $expires = '') {
        $file_svc = new \Elgg\FileService\File();
        $file_svc->setFile($this);
        if (!empty($expires)) {
            $file_svc->setExpires($expires);
        }
        $file_svc->setDisposition('inline');
        $file_svc->bindSession($use_cookie);

        $params = [
            'entity' => $this,
        ];
        return _elgg_services()->hooks->trigger('inline:url', 'file', $params, $file_svc->getURL());
    }
}