Mise en place de la classe ImageUploader

Model/ImageUploader.php

+<?php
+namespace NicolasBejean\Importer\Model;
+
+use \Magento\Framework\App\Filesystem\DirectoryList;
+use \Magento\Framework\Exception\FileSystemException;
+use \Magento\Framework\Exception\LocalizedException;
+use \Magento\Framework\Filesystem;
+use \Magento\Framework\Filesystem\Directory\WriteInterface;
+use \Magento\Framework\UrlInterface;
+use \Magento\MediaStorage\Helper\File\Storage\Database;
+use \Magento\MediaStorage\Model\File\Uploader;
+use \Magento\MediaStorage\Model\File\UploaderFactory;
+use \Magento\Store\Model\StoreManagerInterface;
+use \Psr\Log\LoggerInterface;
+use \Exception;
+
+/**
+ * Class ImageUploader
+ *
+ * @category PHP
+ * @package  NicolasBejean\Importer\Model
+ * @author   Nicolas Béjean <nicolas@bejean.eu>
+ * @license  https://github.com/nicolasbejean/importer/blob/master/licence.txt BSD Licence
+ * @link     https://www.bejean.eu
+ */
+class ImageUploader
+{
+    /**
+     * Core file storage database
+     *
+     * @var Database
+     */
+    protected $coreFileStorageDatabase;
+
+    /**
+     * Media directory object (writable).
+     *
+     * @var WriteInterface
+     */
+    protected $mediaDirectory;
+
+    /**
+     * Uploader factory
+     *
+     * @var UploaderFactory
+     */
+    private $uploaderFactory;
+
+    /**
+     * Store manager
+     *
+     * @var StoreManagerInterface
+     */
+    protected $storeManager;
+
+    /**
+     * @var LoggerInterface
+     */
+    protected $logger;
+
+    /**
+     * Base tmp path
+     *
+     * @var string
+     */
+    protected $baseTmpPath;
+
+    /**
+     * Base path
+     *
+     * @var string
+     */
+    protected $basePath;
+
+    /**
+     * Allowed extensions
+     *
+     * @var string
+     */
+    protected $allowedExtensions;
+
+    /**
+     * List of allowed image mime types
+     *
+     * @var array
+     */
+    private $allowedMimeTypes = [
+        'image/jpg',
+        'image/jpeg',
+        'image/gif',
+        'image/png',
+    ];
+
+    /**
+     * ImageUploader constructor
+     *
+     * @param Database $coreFileStorageDatabase
+     * @param Filesystem $filesystem
+     * @param UploaderFactory $uploaderFactory
+     * @param StoreManagerInterface $storeManager
+     * @param LoggerInterface $logger
+     * @param string $baseTmpPath
+     * @param string $basePath
+     * @param string[] $allowedExtensions
+     * @param array $allowedMimeTypes
+     * @throws FileSystemException
+     */
+    public function __construct(
+        Database $coreFileStorageDatabase,
+        Filesystem $filesystem,
+        UploaderFactory $uploaderFactory,
+        StoreManagerInterface $storeManager,
+        LoggerInterface $logger,
+        $baseTmpPath,
+        $basePath,
+        $allowedExtensions,
+        $allowedMimeTypes = []
+    ) {
+        $this->coreFileStorageDatabase = $coreFileStorageDatabase;
+        $this->mediaDirectory = $filesystem->getDirectoryWrite(DirectoryList::MEDIA);
+        $this->uploaderFactory = $uploaderFactory;
+        $this->storeManager = $storeManager;
+        $this->logger = $logger;
+        $this->baseTmpPath = $baseTmpPath;
+        $this->basePath = $basePath;
+        $this->allowedExtensions = $allowedExtensions;
+        $this->allowedMimeTypes = $allowedMimeTypes;
+    }
+
+    /**
+     * Set base tmp path
+     *
+     * @param string $baseTmpPath
+     *
+     * @return void
+     */
+    public function setBaseTmpPath($baseTmpPath)
+    {
+        $this->baseTmpPath = $baseTmpPath;
+    }
+
+    /**
+     * Set base path
+     *
+     * @param string $basePath
+     *
+     * @return void
+     */
+    public function setBasePath($basePath)
+    {
+        $this->basePath = $basePath;
+    }
+
+    /**
+     * Set allowed extensions
+     *
+     * @param string[] $allowedExtensions
+     *
+     * @return void
+     */
+    public function setAllowedExtensions($allowedExtensions)
+    {
+        $this->allowedExtensions = $allowedExtensions;
+    }
+
+    /**
+     * Retrieve base tmp path
+     *
+     * @return string
+     */
+    public function getBaseTmpPath()
+    {
+        return $this->baseTmpPath;
+    }
+
+    /**
+     * Retrieve base path
+     *
+     * @return string
+     */
+    public function getBasePath()
+    {
+        return $this->basePath;
+    }
+
+    /**
+     * Retrieve allowed extensions
+     *
+     * @return string[]
+     */
+    public function getAllowedExtensions()
+    {
+        return $this->allowedExtensions;
+    }
+
+    /**
+     * Retrieve path
+     *
+     * @param string $path
+     * @param string $imageName
+     *
+     * @return string
+     */
+    public function getFilePath($path, $imageName)
+    {
+        return rtrim($path, '/') . '/' . ltrim($imageName, '/');
+    }
+
+    /**
+     * Checking file for save and save it to tmp dir
+     *
+     * @param string $fileId
+     *
+     * @return string[]
+     *
+     * @throws LocalizedException
+     * @throws Exception
+     */
+    public function saveFileToTmpDir($fileId)
+    {
+        $baseTmpPath = $this->getBaseTmpPath();
+
+        /** @var Uploader $uploader */
+        $uploader = $this->uploaderFactory->create(['fileId' => $fileId]);
+        $uploader->setAllowedExtensions($this->getAllowedExtensions());
+        $uploader->setAllowRenameFiles(true);
+        if (!$uploader->checkMimeType($this->allowedMimeTypes)) {
+            throw new LocalizedException(__('File validation failed.'));
+        }
+        $result = $uploader->save($this->mediaDirectory->getAbsolutePath($baseTmpPath));
+        // unset($result['path']);
+
+        if (!$result) {
+            throw new LocalizedException(
+                __('File can not be saved to the destination folder.')
+            );
+        }
+
+        /**
+         * Workaround for prototype 1.7 methods "isJSON", "evalJSON" on Windows OS
+         */
+        $result['tmp_name'] = str_replace('\\', '/', $result['tmp_name']);
+        $result['url'] = $this->storeManager
+                ->getStore()
+                ->getBaseUrl(
+                    UrlInterface::URL_TYPE_MEDIA
+                ) . $this->getFilePath($baseTmpPath, $result['file']);
+        $result['name'] = $result['file'];
+
+        if (isset($result['file'])) {
+            try {
+                $relativePath = rtrim($baseTmpPath, '/') . '/' . ltrim($result['file'], '/');
+                $this->coreFileStorageDatabase->saveFile($relativePath);
+            } catch (Exception $e) {
+                $this->logger->critical($e);
+                throw new LocalizedException(
+                    __('Something went wrong while saving the file(s).')
+                );
+            }
+        }
+
+        return $result;
+    }
+}