filestore.php

Go to the documentation of this file.

<?php
use Symfony\Component\HttpFoundation\File\UploadedFile;

function get_dir_size($dir, $total_size = 0) {
    if (!is_dir($dir)) {
        return $total_size;
    }
    
    $handle = opendir($dir);
    while (($file = readdir($handle)) !== false) {
        if (in_array($file, ['.', '..'])) {
            continue;
        }
        if (is_dir($dir . $file)) {
            $total_size = get_dir_size($dir . $file . "/", $total_size);
        } else {
            $total_size += filesize($dir . $file);
        }
    }
    closedir($handle);

    return($total_size);
}

function elgg_save_resized_image($source, $destination = null, array $params = []) {
    return _elgg_services()->imageService->resize($source, $destination, $params);
}

function delete_directory($directory) {

    if (!file_exists($directory)) {
        return true;
    }

    if (!is_dir($directory)) {
        return false;
    }

    // sanity check: must be a directory
    if (!$handle = opendir($directory)) {
        return false;
    }

    // loop through all files
    while (($file = readdir($handle)) !== false) {
        if (in_array($file, ['.', '..'])) {
            continue;
        }

        $path = "$directory/$file";
        if (is_dir($path)) {
            // recurse down through directory
            if (!delete_directory($path)) {
                return false;
            }
        } else {
            // delete file
            unlink($path);
        }
    }

    // remove empty directory
    closedir($handle);
    return rmdir($directory);
}

function elgg_get_file_simple_type($mime_type) {
    $params = ['mime_type' => $mime_type];
    return elgg_trigger_plugin_hook('simple_type', 'file', $params, 'general');
}

function _elgg_filestore_init() {

    // Fix MIME type detection for Microsoft zipped formats
    elgg_register_plugin_hook_handler('mime_type', 'file', '_elgg_filestore_detect_mimetype');

    // Parse category of file from MIME type
    elgg_register_plugin_hook_handler('simple_type', 'file', '_elgg_filestore_parse_simpletype');

    // Unit testing
    elgg_register_plugin_hook_handler('unit_test', 'system', '_elgg_filestore_test');

    // Touch entity icons if entity access id has changed
    elgg_register_event_handler('update:after', 'object', '_elgg_filestore_touch_icons');
    elgg_register_event_handler('update:after', 'group', '_elgg_filestore_touch_icons');

    // Move entity icons if entity owner has changed
    elgg_register_event_handler('update:after', 'object', '_elgg_filestore_move_icons');
    elgg_register_event_handler('update:after', 'group', '_elgg_filestore_move_icons');
}

function _elgg_filestore_detect_mimetype($hook, $type, $mime_type, $params) {

    $original_filename = elgg_extract('original_filename', $params);
    $ext = pathinfo($original_filename, PATHINFO_EXTENSION);

    return (new \Elgg\Filesystem\MimeTypeDetector())->fixDetectionErrors($mime_type, $ext);
}

function _elgg_filestore_parse_simpletype($hook, $type, $simple_type, $params) {

    $mime_type = elgg_extract('mime_type', $params);

    switch ($mime_type) {
        case "application/msword":
        case "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
        case "application/pdf":
            return "document";

        case "application/ogg":
            return "audio";
    }

    if (preg_match('~^(audio|image|video)/~', $mime_type, $m)) {
        return $m[1];
    }
    if (0 === strpos($mime_type, 'text/') || false !== strpos($mime_type, 'opendocument')) {
        return "document";
    }

    // unrecognized MIME
    return $simple_type;
}

function _elgg_filestore_test($hook, $type, $value) {
    $value[] = ElggCoreFilestoreTest::class;
    return $value;
}

function elgg_get_download_url(\ElggFile $file, $use_cookie = true, $expires = '+2 hours') {
    return $file->getDownloadURL($use_cookie, $expires);
}

function elgg_get_inline_url(\ElggFile $file, $use_cookie = false, $expires = '') {
    return $file->getInlineURL($use_cookie, $expires);
}

function elgg_get_embed_url(\ElggEntity $entity, $size) {
    return elgg_normalize_url(elgg_generate_url('serve-icon', [
        'guid' => $entity->guid,
        'size' => $size,
    ]));
}

function _elgg_filestore_serve_icon_handler() {
    $response = _elgg_services()->iconService->handleServeIconRequest();
    $response->send();
    exit;
}

function _elgg_filestore_touch_icons($event, $type, $entity) {
    $original_attributes = $entity->getOriginalAttributes();
    if (!array_key_exists('access_id', $original_attributes)) {
        return;
    }
    if ($entity instanceof \ElggFile) {
        // we touch the file to invalidate any previously generated download URLs
        $entity->setModifiedTime();
    }
    $sizes = array_keys(elgg_get_icon_sizes($entity->getType(), $entity->getSubtype()));
    foreach ($sizes as $size) {
        $icon = $entity->getIcon($size);
        if ($icon->exists()) {
            $icon->setModifiedTime();
        }
    }
}

function _elgg_filestore_move_icons($event, $type, $entity) {

    $original_attributes = $entity->getOriginalAttributes();
    if (empty($original_attributes['owner_guid'])) {
        return;
    }

    $previous_owner_guid = $original_attributes['owner_guid'];
    $new_owner_guid = $entity->owner_guid;

    $sizes = elgg_get_icon_sizes($entity->getType(), $entity->getSubtype());

    foreach ($sizes as $size => $opts) {
        $new_icon = $entity->getIcon($size);
        if ($new_icon->owner_guid == $entity->guid) {
            // we do not need to update icons that are owned by the entity itself
            continue;
        }

        if ($new_icon->owner_guid != $new_owner_guid) {
            // a plugin implements some custom logic
            continue;
        }

        $old_icon = new \ElggIcon();
        $old_icon->owner_guid = $previous_owner_guid;
        $old_icon->setFilename($new_icon->getFilename());
        if (!$old_icon->exists()) {
            // there is no icon to move
            continue;
        }

        if ($new_icon->exists()) {
            // there is already a new icon
            // just removing the old one
            $old_icon->delete();
            elgg_log("Entity $entity->guid has been transferred to a new owner but an icon was "
                . "left behind under {$old_icon->getFilenameOnFilestore()}. "
                . "Old icon has been deleted", 'NOTICE');
            continue;
        }

        $old_icon->transfer($new_icon->owner_guid, $new_icon->getFilename());
        elgg_log("Entity $entity->guid has been transferred to a new owner. "
        . "Icon was moved from {$old_icon->getFilenameOnFilestore()} to {$new_icon->getFilenameOnFilestore()}.", 'NOTICE');
    }
}

function elgg_get_uploaded_files($input_name) {
    return _elgg_services()->uploads->getFiles($input_name);
}

function elgg_get_uploaded_file($input_name, $check_for_validity = true) {
    return _elgg_services()->uploads->getFile($input_name, $check_for_validity);
}

function elgg_get_temp_file() {
    return new ElggTempFile();
}

return function(\Elgg\EventsService $events, \Elgg\HooksRegistrationService $hooks) {
    $events->registerHandler('init', 'system', '_elgg_filestore_init', 100);
};