File.php

Go to the documentation of this file.

<?php
namespace Magento\MediaStorage\Model\File\Storage;

class File
{
    protected $_eventPrefix = 'media_storage_file_storage_file';

    protected $_mediaBaseDirectory = null;

    protected $_storageHelper = null;

    protected $_mediaHelper = null;

    protected $_data = null;

    protected $_errors = [];

    protected $_logger;

    public function __construct(
        \Psr\Log\LoggerInterface $logger,
        \Magento\MediaStorage\Helper\File\Storage\Database $storageHelper,
        \Magento\MediaStorage\Helper\File\Media $mediaHelper,
        \Magento\MediaStorage\Model\ResourceModel\File\Storage\File $fileUtility
    ) {
        $this->_fileUtility = $fileUtility;
        $this->_storageHelper = $storageHelper;
        $this->_logger = $logger;
        $this->_mediaHelper = $mediaHelper;
    }

    public function init()
    {
        return $this;
    }

    public function getStorageName()
    {
        return __('File system');
    }

    public function getStorageData()
    {
        return $this->_fileUtility->getStorageData();
    }

    public function hasErrors()
    {
        return !empty($this->_errors);
    }

    public function clear()
    {
        $this->_fileUtility->clear();
        return $this;
    }

    public function collectData($offset = 0, $count = 100, $type = 'files')
    {
        if (!in_array($type, ['files', 'directories'])) {
            return false;
        }

        $offset = (int)$offset >= 0 ? (int)$offset : 0;
        $count = (int)$count >= 1 ? (int)$count : 1;

        if (empty($this->_data)) {
            $this->_data = $this->getStorageData();
        }

        if (!array_key_exists($type, $this->_data)) {
            return false;
        }
        $slice = array_slice($this->_data[$type], $offset, $count);
        return $slice ?: false;
    }

    public function getConfigConnectionName()
    {
        return null;
    }

    public function getConnectionName()
    {
        return null;
    }

    public function exportDirectories($offset = 0, $count = 100)
    {
        return $this->collectData($offset, $count, 'directories');
    }

    public function exportFiles($offset = 0, $count = 1)
    {
        $slice = $this->collectData($offset, $count, 'files');

        if (!$slice) {
            return false;
        }

        $result = [];
        foreach ($slice as $fileName) {
            try {
                $fileInfo = $this->_mediaHelper->collectFileInfo($this->getMediaBaseDirectory(), $fileName);
            } catch (\Exception $e) {
                $this->_logger->critical($e);
                continue;
            }

            $result[] = $fileInfo;
        }

        return $result;
    }

    public function import($data, $callback)
    {
        if (!is_array($data) || !method_exists($this, $callback)) {
            return $this;
        }

        foreach ($data as $part) {
            try {
                $this->{$callback}($part);
            } catch (\Exception $e) {
                $this->_errors[] = $e->getMessage();
                $this->_logger->critical($e);
            }
        }

        return $this;
    }

    public function importDirectories($dirs)
    {
        return $this->import($dirs, 'saveDir');
    }

    public function importFiles($files)
    {
        return $this->import($files, 'saveFile');
    }

    public function saveDir($dir)
    {
        return $this->_fileUtility->saveDir($dir);
    }

    public function saveFile($file, $overwrite = true)
    {
        if (isset($file['filename'])
            && !empty($file['filename'])
            && isset($file['content'])
            && !empty($file['content'])
        ) {
            try {
                $filename = isset(
                    $file['directory']
                ) && !empty($file['directory']) ? $file['directory'] . '/' . $file['filename'] : $file['filename'];

                return $this->_fileUtility->saveFile($filename, $file['content'], $overwrite);
            } catch (\Exception $e) {
                $this->_logger->critical($e);
                throw new \Magento\Framework\Exception\LocalizedException(
                    __('Unable to save file "%1" at "%2"', $file['filename'], $file['directory'])
                );
            }
        } else {
            throw new \Magento\Framework\Exception\LocalizedException(__('Wrong file info format'));
        }
    }

    public function getMediaBaseDirectory()
    {
        if ($this->_mediaBaseDirectory === null) {
            $this->_mediaBaseDirectory = $this->_storageHelper->getMediaBaseDir();
        }
        return $this->_mediaBaseDirectory;
    }
}