mirror of
https://github.com/filegator/filegator.git
synced 2025-10-26 21:41:50 +01:00
deploy: c9cb5afa91
This commit is contained in:
189
backend/Controllers/DownloadController.php
Normal file
189
backend/Controllers/DownloadController.php
Normal file
@@ -0,0 +1,189 @@
|
||||
<?php
|
||||
|
||||
/*
|
||||
* This file is part of the FileGator package.
|
||||
*
|
||||
* (c) Milos Stojanovic <alcalbg@gmail.com>
|
||||
*
|
||||
* For the full copyright and license information, please view the LICENSE file
|
||||
*/
|
||||
|
||||
namespace Filegator\Controllers;
|
||||
|
||||
use Filegator\Config\Config;
|
||||
use Filegator\Kernel\Request;
|
||||
use Filegator\Kernel\Response;
|
||||
use Filegator\Kernel\StreamedResponse;
|
||||
use Filegator\Services\Archiver\ArchiverInterface;
|
||||
use Filegator\Services\Auth\AuthInterface;
|
||||
use Filegator\Services\Session\SessionStorageInterface as Session;
|
||||
use Filegator\Services\Storage\Filesystem;
|
||||
use Filegator\Services\Tmpfs\TmpfsInterface;
|
||||
use Symfony\Component\HttpFoundation\HeaderUtils;
|
||||
use Symfony\Component\Mime\MimeTypes;
|
||||
|
||||
class DownloadController
|
||||
{
|
||||
protected $auth;
|
||||
|
||||
protected $session;
|
||||
|
||||
protected $config;
|
||||
|
||||
protected $storage;
|
||||
|
||||
public function __construct(Config $config, Session $session, AuthInterface $auth, Filesystem $storage)
|
||||
{
|
||||
$this->session = $session;
|
||||
$this->config = $config;
|
||||
$this->auth = $auth;
|
||||
|
||||
$user = $this->auth->user() ?: $this->auth->getGuest();
|
||||
|
||||
$this->storage = $storage;
|
||||
$this->storage->setPathPrefix($user->getHomeDir());
|
||||
}
|
||||
|
||||
public function download(Request $request, Response $response, StreamedResponse $streamedResponse)
|
||||
{
|
||||
try {
|
||||
$file = $this->storage->readStream((string) base64_decode($request->input('path')));
|
||||
} catch (\Exception $e) {
|
||||
return $response->redirect('/');
|
||||
}
|
||||
|
||||
$streamedResponse->setCallback(function () use ($file) {
|
||||
// @codeCoverageIgnoreStart
|
||||
set_time_limit(0);
|
||||
if ($file['stream']) {
|
||||
while (! feof($file['stream'])) {
|
||||
echo fread($file['stream'], 1024 * 8);
|
||||
if (ob_get_level() > 0) {ob_flush();}
|
||||
flush();
|
||||
}
|
||||
fclose($file['stream']);
|
||||
}
|
||||
// @codeCoverageIgnoreEnd
|
||||
});
|
||||
|
||||
$extension = pathinfo($file['filename'], PATHINFO_EXTENSION);
|
||||
$mimes = (new MimeTypes())->getMimeTypes($extension);
|
||||
$contentType = !empty($mimes) ? $mimes[0] : 'application/octet-stream';
|
||||
|
||||
$disposition = HeaderUtils::DISPOSITION_ATTACHMENT;
|
||||
|
||||
$download_inline = (array)$this->config->get('download_inline', ['pdf']);
|
||||
if (in_array($extension, $download_inline) || in_array('*', $download_inline)) {
|
||||
$disposition = HeaderUtils::DISPOSITION_INLINE;
|
||||
}
|
||||
|
||||
$contentDisposition = HeaderUtils::makeDisposition($disposition, $file['filename'], 'file');
|
||||
|
||||
$streamedResponse->headers->set(
|
||||
'Content-Disposition',
|
||||
$contentDisposition
|
||||
);
|
||||
$streamedResponse->headers->set(
|
||||
'Content-Type',
|
||||
$contentType
|
||||
);
|
||||
$streamedResponse->headers->set(
|
||||
'Content-Transfer-Encoding',
|
||||
'binary'
|
||||
);
|
||||
if (isset($file['filesize'])) {
|
||||
$streamedResponse->headers->set(
|
||||
'Content-Length',
|
||||
$file['filesize']
|
||||
);
|
||||
}
|
||||
// @codeCoverageIgnoreStart
|
||||
if (APP_ENV == 'development') {
|
||||
$streamedResponse->headers->set(
|
||||
'Access-Control-Allow-Origin',
|
||||
$request->headers->get('Origin')
|
||||
);
|
||||
$streamedResponse->headers->set(
|
||||
'Access-Control-Allow-Credentials',
|
||||
'true'
|
||||
);
|
||||
}
|
||||
// @codeCoverageIgnoreEnd
|
||||
|
||||
// close session so we can continue streaming, note: dev is single-threaded
|
||||
$this->session->save();
|
||||
|
||||
$streamedResponse->send();
|
||||
}
|
||||
|
||||
public function batchDownloadCreate(Request $request, Response $response, ArchiverInterface $archiver)
|
||||
{
|
||||
$items = $request->input('items', []);
|
||||
|
||||
$uniqid = $archiver->createArchive($this->storage);
|
||||
|
||||
// close session
|
||||
$this->session->save();
|
||||
|
||||
foreach ($items as $item) {
|
||||
if ($item->type == 'dir') {
|
||||
$archiver->addDirectoryFromStorage($item->path);
|
||||
}
|
||||
if ($item->type == 'file') {
|
||||
$archiver->addFileFromStorage($item->path);
|
||||
}
|
||||
}
|
||||
|
||||
$archiver->closeArchive();
|
||||
|
||||
return $response->json(['uniqid' => $uniqid]);
|
||||
}
|
||||
|
||||
public function batchDownloadStart(Request $request, StreamedResponse $streamedResponse, TmpfsInterface $tmpfs)
|
||||
{
|
||||
$uniqid = (string) preg_replace('/[^0-9a-zA-Z_]/', '', (string) $request->input('uniqid'));
|
||||
$file = $tmpfs->readStream($uniqid);
|
||||
|
||||
$streamedResponse->setCallback(function () use ($file, $tmpfs, $uniqid) {
|
||||
// @codeCoverageIgnoreStart
|
||||
set_time_limit(0);
|
||||
if ($file['stream']) {
|
||||
while (! feof($file['stream'])) {
|
||||
echo fread($file['stream'], 1024 * 8);
|
||||
if (ob_get_level() > 0) {ob_flush();}
|
||||
flush();
|
||||
}
|
||||
fclose($file['stream']);
|
||||
}
|
||||
$tmpfs->remove($uniqid);
|
||||
// @codeCoverageIgnoreEnd
|
||||
});
|
||||
|
||||
$streamedResponse->headers->set(
|
||||
'Content-Disposition',
|
||||
HeaderUtils::makeDisposition(
|
||||
HeaderUtils::DISPOSITION_ATTACHMENT,
|
||||
$this->config->get('frontend_config.default_archive_name'),
|
||||
'archive.zip'
|
||||
)
|
||||
);
|
||||
$streamedResponse->headers->set(
|
||||
'Content-Type',
|
||||
'application/octet-stream'
|
||||
);
|
||||
$streamedResponse->headers->set(
|
||||
'Content-Transfer-Encoding',
|
||||
'binary'
|
||||
);
|
||||
if (isset($file['filesize'])) {
|
||||
$streamedResponse->headers->set(
|
||||
'Content-Length',
|
||||
$file['filesize']
|
||||
);
|
||||
}
|
||||
// close session so we can continue streaming, note: dev is single-threaded
|
||||
$this->session->save();
|
||||
|
||||
$streamedResponse->send();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user