1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
namespace Magento\Sales\Model;
use Magento\Framework\App\Filesystem\DirectoryList;
use Magento\Framework\Exception\LocalizedException;
/**
* Class Download. Represents download logic for files
*/
class Download
{
/**
* @var \Magento\Framework\Filesystem\Directory\WriteInterface
*/
protected $_rootDir;
/**
* @var \Magento\MediaStorage\Helper\File\Storage\Database
*/
protected $_fileStorageDatabase;
/**
* @var \Magento\MediaStorage\Model\File\Storage\DatabaseFactory
*/
protected $_storageDatabaseFactory;
/**
* @var \Magento\Framework\App\Response\Http\FileFactory
*/
protected $_fileFactory;
/**
* @var string
*/
protected $rootDirBasePath;
/**
* Constructor method
*
* @param \Magento\Framework\Filesystem $filesystem
* @param \Magento\MediaStorage\Helper\File\Storage\Database $fileStorageDatabase
* @param \Magento\MediaStorage\Model\File\Storage\DatabaseFactory $storageDatabaseFactory
* @param \Magento\Framework\App\Response\Http\FileFactory $fileFactory
* @param string $rootDirBasePath
*/
public function __construct(
\Magento\Framework\Filesystem $filesystem,
\Magento\MediaStorage\Helper\File\Storage\Database $fileStorageDatabase,
\Magento\MediaStorage\Model\File\Storage\DatabaseFactory $storageDatabaseFactory,
\Magento\Framework\App\Response\Http\FileFactory $fileFactory,
$rootDirBasePath = DirectoryList::MEDIA
) {
$this->rootDirBasePath = $rootDirBasePath;
$this->_rootDir = $filesystem->getDirectoryWrite($this->rootDirBasePath);
$this->_fileStorageDatabase = $fileStorageDatabase;
$this->_storageDatabaseFactory = $storageDatabaseFactory;
$this->_fileFactory = $fileFactory;
}
/**
* Custom options downloader
*
* @param array $info
* @return void
* @throws \Exception
*/
public function downloadFile($info)
{
$relativePath = $info['order_path'];
if (!$this->_isCanProcessed($relativePath)) {
//try get file from quote
$relativePath = $info['quote_path'];
if (!$this->_isCanProcessed($relativePath)) {
throw new LocalizedException(
__('Path "%1" is not part of allowed directory "%2"', $relativePath, $this->rootDirBasePath)
);
}
}
$this->_fileFactory->create(
$info['title'],
['value' => $this->_rootDir->getRelativePath($relativePath), 'type' => 'filename'],
$this->rootDirBasePath,
$info['type']
);
}
/**
* Method checks, if file can be returned depends on the given filepath
*
* @param string $relativePath
* @return bool
*/
protected function _isCanProcessed($relativePath)
{
$filePath = $this->_rootDir->getAbsolutePath($relativePath);
$pathWithFixedSeparator = str_replace('\\', '/', $this->_rootDir->getDriver()->getRealPath($filePath));
return (strpos($pathWithFixedSeparator, $relativePath) !== false
&& $this->_rootDir->isFile($relativePath) && $this->_rootDir->isReadable($relativePath))
|| $this->_processDatabaseFile($filePath, $relativePath);
}
/**
* Check file in database storage if needed and place it on file system
*
* @param string $filePath
* @param string $relativePath
* @return bool
*/
protected function _processDatabaseFile($filePath, $relativePath)
{
if (!$this->_fileStorageDatabase->checkDbUsage()) {
return false;
}
$file = $this->_storageDatabaseFactory->create()->loadByFilename($relativePath);
if (!$file->getId()) {
return false;
}
$stream = $this->_rootDir->openFile($relativePath, 'w+');
$stream->lock();
$stream->write($filePath, $file->getContent());
$stream->unlock();
$stream->close();
return true;
}
}