Your IP : 18.191.73.161
<?php
namespace Bitrix\Disk;
use Bitrix\Disk\Integration\TransformerManager;
use Bitrix\Disk\Internals\Error\Error;
use Bitrix\Disk\Security\ParameterSigner;
use Bitrix\Main\EventResult;
use Bitrix\Main\Loader;
use Bitrix\Main\Localization\Loc;
Loc::loadMessages(__FILE__);
class DownloadController extends Internals\Controller
{
const ERROR_COULD_NOT_FIND_VERSION = 'DISK_DC_22003';
const ERROR_COULD_NOT_FIND_FILE = 'DISK_DC_22004';
const ERROR_BAD_RIGHTS = 'DISK_DC_22005';
const ERROR_COULD_NOT_FIND_REAL_FILE = 'DISK_DC_22006';
protected $fileId;
protected $versionId;
/** @var File */
protected $file;
/** @var Version */
protected $version;
protected function listActions()
{
return array(
'showFile' => array(
'method' => array('GET'),
'redirect_on_auth' => true,
'close_session' => true,
),
'showPreview' => array(
'method' => array('GET'),
'close_session' => true,
),
'showView' => array(
'method' => array('GET'),
'close_session' => true,
),
'showViewHtml' => array(
'method' => array('GET'),
'close_session' => true,
),
'showVersionView' => array(
'method' => array('GET'),
'close_session' => true,
),
'transformOnOpen' => array(
'method' => array('GET'),
'close_session' => true,
),
'showTransformationInfo' => array(
'method' => array('GET'),
'close_session' => true,
),
'downloadFile',
'downloadVersion',
'downloadArchive' => array(
'method' => array('GET'),
'close_session' => true,
),
'copyToMe' => array(
'method' => array('POST', 'GET'),
'check_csrf_token' => true,
),
'downloadByExternalLink' => array(
'method' => array('GET'),
'close_session' => true,
),
);
}
protected function onBeforeActionShowFile()
{
return $this->onWorkWithOneFile();
}
protected function onBeforeActionShowPreview()
{
return $this->onWorkWithOneFile();
}
protected function onBeforeActionShowView()
{
return $this->onWorkWithOneFile();
}
protected function onBeforeActionShowViewHtml()
{
return $this->onWorkWithOneFile();
}
protected function onBeforeActionShowVersionView()
{
return $this->onWorkWithOneFile();
}
protected function onBeforeActionShowTransformationInfo()
{
return $this->onWorkWithOneFile();
}
protected function onBeforeActionDownloadFile()
{
return $this->onWorkWithOneFile();
}
protected function onBeforeActionDownloadVersion()
{
return $this->onWorkWithOneFile();
}
protected function onBeforeActionCopyToMe()
{
return $this->onWorkWithOneFile();
}
protected function onBeforeActionTransformOnOpen()
{
return $this->onWorkWithOneFile();
}
private function onWorkWithOneFile()
{
if(!$this->checkRequiredGetParams(array('fileId')))
{
return new EventResult(EventResult::ERROR);
}
$this->fileId = (int)$this->request->getQuery('fileId');
if($this->request->getQuery('versionId'))
{
$this->versionId = (int)$this->request->getQuery('versionId');
}
$this->file = File::loadById($this->fileId, array('STORAGE'));
if(!$this->file)
{
$this->errorCollection->addOne(new Error(Loc::getMessage('DISK_DOWNLOAD_CONTROLLER_ERROR_COULD_NOT_FIND_FILE'), self::ERROR_COULD_NOT_FIND_FILE));
return new EventResult(EventResult::ERROR);
}
if($this->file instanceof FileLink && !$this->file->getRealObject())
{
$this->errorCollection->addOne(new Error(Loc::getMessage('DISK_DOWNLOAD_CONTROLLER_ERROR_COULD_NOT_FIND_FILE'), self::ERROR_COULD_NOT_FIND_REAL_FILE));
return new EventResult(EventResult::ERROR);
}
//todo refactor. The method send response. It's bad.
$this->checkPermissions();
return new EventResult(EventResult::SUCCESS);
}
protected function checkPermissions()
{
$securityContext = $this->file->getStorage()->getCurrentUserSecurityContext();
if(!$this->file->canRead($securityContext))
{
$this->errorCollection->addOne(new Error(Loc::getMessage('DISK_DOWNLOAD_CONTROLLER_ERROR_BAD_RIGHTS'), self::ERROR_BAD_RIGHTS));
if(Desktop::getDiskVersion())
{
$this->sendJsonErrorResponse();
}
//general for user we show simple message
$this->sendResponse(Loc::getMessage('DISK_DOWNLOAD_CONTROLLER_ERROR_BAD_RIGHTS'));
}
}
protected function processActionDownloadFile()
{
$fileData = $this->file->getFile();
\CFile::viewByUser($fileData, array('force_download' => true, 'cache_time' => Configuration::DEFAULT_CACHE_TIME, 'attachment_name' => $this->file->getName()));
}
protected function processActionShowViewHtml($pathToView, $mode = '', $print = '', $preview = '', $sizeType = '', $printUrl = '', $autostart = 'Y')
{
$printParam = $iframe = 'N';
if($mode === 'iframe')
{
$iframe = 'Y';
if($print === 'Y')
{
$printParam = 'Y';
}
}
$elementId = 'bx_ajaxelement_' . $this->file->getId() . '_' . randString(4);
$this->version = $this->file->getVersion($this->versionId);
if($this->version)
{
$view = $this->version->getView();
}
else
{
$view = $this->file->getView();
}
$html = $view->render(array(
'PATH' => $pathToView,
'IFRAME' => $iframe,
'ID' => $elementId,
'PRINT' => $printParam,
'PREVIEW' => $preview,
'SIZE_TYPE' => $sizeType,
'PRINT_URL' => $printUrl,
'AUTOSTART' => ($autostart !== 'Y' ? 'N' : 'Y'),
));
if($iframe == 'Y')
{
echo $html;
}
else
{
$result = array('html' => $html, 'innerElementId' => $elementId);
$result = array_merge($result, $view->getJsViewerAdditionalJsonParams());
$this->sendJsonResponse($result);
}
$this->end();
}
protected function processActionShowVersionView()
{
$this->version = $this->file->getVersion($this->versionId);
if(!$this->version)
{
$this->errorCollection[] = new Error(Loc::getMessage('DISK_DOWNLOAD_CONTROLLER_ERROR_COULD_NOT_FIND_VERSION'), self::ERROR_COULD_NOT_FIND_VERSION);
$this->sendJsonErrorResponse();
}
$fileName = $this->version->getView()->getName();
$fileData = $this->version->getView()->getData();
$cacheTime = Configuration::DEFAULT_CACHE_TIME;
$this->showFileByArray($fileName, $fileData, $cacheTime);
}
protected function processActionShowView()
{
if(!$this->file->getView()->getId())
{
$this->end();
}
$fileName = $this->file->getView()->getName();
$fileData = $this->file->getView()->getData();
$cacheTime = Configuration::DEFAULT_CACHE_TIME;
$this->showFileByArray($fileName, $fileData, $cacheTime);
}
protected function processActionTransformOnOpen()
{
$result = $this->file->getView()->transformOnOpen($this->file);
$this->sendJsonResponse($result);
}
protected function processActionShowTransformationInfo($noError = '')
{
if(!$this->file->getView()->isShowTransformationInfo())
{
$this->sendJsonErrorResponse();
}
if($noError == 'y' && $this->file->getView()->isLastTransformationFailed())
{
$this->sendJsonErrorResponse();
}
$urlManager = Driver::getInstance()->getUrlManager();
$params = array(
'TRANSFORM_URL' => $urlManager->getUrlForTransformOnOpen($this->file),
'REFRESH_URL' => $urlManager->getUrlForShowViewHtml($this->file, array('autostart' => 'N', 'sizeType' => 'adjust')),
);
$result = array(
'html' => $this->file->getView()->renderTransformationInProcessMessage($params)
);
$this->sendJsonSuccessResponse($result);
}
protected function processActionShowFile()
{
$fileName = $this->file->getName();
$fileData = $this->file->getFile();
if(!$fileData)
{
$this->end();
}
$isImage = TypeFile::isImage($fileData['ORIGINAL_NAME']) || TypeFile::isImage($fileName);
$cacheTime = $isImage? 86400 : Configuration::DEFAULT_CACHE_TIME;
$this->showFileByArray($fileName, $fileData, $cacheTime);
}
protected function processActionShowPreview()
{
if(!$this->file->getPreviewId())
{
$this->end();
}
$fileName = $this->file->getView()->getPreviewName();
$fileData = $this->file->getView()->getPreviewData();
$cacheTime = 86400;
$this->showFileByArray($fileName, $fileData, $cacheTime);
}
private function showFileByArray($fileName, $fileData = array(), $cacheTime = 86400, $forceDownload = false)
{
if (empty($fileName) || !is_array($fileData) || empty($fileData))
{
$this->end();
}
if (TypeFile::isImage($fileData['ORIGINAL_NAME']) || TypeFile::isImage($fileName))
{
$fileData = $this->resizeImage($fileData, $this->file->getId());
}
\CFile::viewByUser($fileData, array('force_download' => $forceDownload, 'cache_time' => $cacheTime, 'attachment_name' => $fileName));
}
protected function processActionDownloadVersion()
{
$this->version = $this->file->getVersion($this->versionId);
if(!$this->version)
{
$this->errorCollection->addOne(new Error(Loc::getMessage('DISK_DOWNLOAD_CONTROLLER_ERROR_COULD_NOT_FIND_VERSION'), self::ERROR_COULD_NOT_FIND_VERSION));
$this->sendJsonErrorResponse();
}
$fileData = $this->version->getFile();
\CFile::viewByUser($fileData, array('force_download' => false, 'cache_time' => Configuration::DEFAULT_CACHE_TIME, 'attachment_name' => $this->file->getName()));
}
protected function processActionCopyToMe()
{
$userStorage = Driver::getInstance()->getStorageByUserId($this->getUser()->getId());
if(!$userStorage)
{
$this->errorCollection->addOne(new Error('Could not find storage for current user'));
$this->sendJsonErrorResponse();
}
$folder = $userStorage->getFolderForSavedFiles();
if(!$folder)
{
$this->errorCollection->addOne(new Error('Could not find folder for created files'));
$this->sendJsonErrorResponse();
}
//so, now we don't copy links in the method copyTo. But here we have to copy content.
//And after we set name to new object as it was on link.
$newFile = $this->file->getRealObject()->copyTo($folder, $this->getUser()->getId(), true);
if ($this->file->getRealObject()->getName() != $this->file->getName())
{
$newFile->renameInternal($this->file->getName(), true);
}
if(!$newFile)
{
$this->errorCollection->addOne(new Error('Could not copy file to storage for current user'));
$this->sendJsonErrorResponse();
}
$urlManager = Driver::getInstance()->getUrlManager();
$viewUrl = $urlManager->encodeUrn(
$urlManager->getUrlFocusController('showObjectInGrid', array(
'objectId' => $newFile->getId(),
))
);
$runViewerUrl = $urlManager->encodeUrn(
$urlManager->getUrlFocusController('showObjectInGrid', array(
'objectId' => $newFile->getId(),
'cmd' => 'show',
))
);
$this->sendJsonSuccessResponse(array(
'newId' => $newFile->getId(),
'viewUrl' => $viewUrl,
'runViewUrl' => $runViewerUrl,
));
}
protected function processActionDownloadArchive($signature, array $objectIds)
{
if(!ParameterSigner::validateArchiveSignature($signature, $objectIds))
{
$this->sendJsonInvalidSignResponse('Invalid signature');
}
if(!ZipNginx\Configuration::isEnabled())
{
$this->errorCollection[] = new Error('Work with mod_zip is disabled in module settings.');
$this->sendJsonErrorResponse();
}
$zipArchive = new ZipNginx\Archive('archive' . date('y-m-d') . '.zip');
foreach($objectIds as $id)
{
//now we can't allow to download whole folder.
$file = File::loadById($id);
if(!$file)
{
continue;
}
$storage = $file->getStorage();
if(!$storage)
{
continue;
}
$securityContext = $storage->getCurrentUserSecurityContext();
if(!$file->canRead($securityContext))
{
continue;
}
$zipArchive->addEntry(
ZipNginx\ArchiveEntry::createFromFileModel($file)
);
}
if($zipArchive->isEmpty())
{
$this->errorCollection[] = new Error('Archive is empty');
$this->sendJsonErrorResponse();
}
$zipArchive->send();
$this->end();
}
protected function processActionDownloadByExternalLink($externalLink)
{
if(!ExternalLink::isValidValueForField('HASH', $externalLink, $this->errorCollection))
{
$this->sendJsonErrorResponse();
}
/** @var ExternalLink $externalLink */
$externalLink = ExternalLink::load(
array(
'=HASH' => $externalLink,
),
array(
'FILE'
)
);
if (!$externalLink || $externalLink->isExpired() || $externalLink->isSpecificVersion() || !$externalLink->getFile())
{
$this->errorCollection[] = new Error('Could not find external link');
$this->sendJsonErrorResponse();
}
if ($externalLink->hasPassword())
{
$this->errorCollection[] = new Error('Could not use external link with password');
$this->sendJsonErrorResponse();
}
$file = $externalLink->getFile();
$fileData = $file->getFile();
\CFile::viewByUser($fileData, array('force_download' => false, 'attachment_name' => $file->getName()));
}
}