123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146 |
- <?php
- declare(strict_types=1);
- namespace OCA\DAV\Controller;
- use OCA\DAV\Db\Direct;
- use OCA\DAV\Db\DirectMapper;
- use OCP\AppFramework\Http;
- use OCP\AppFramework\Http\DataResponse;
- use OCP\AppFramework\OCS\OCSBadRequestException;
- use OCP\AppFramework\OCS\OCSForbiddenException;
- use OCP\AppFramework\OCS\OCSNotFoundException;
- use OCP\AppFramework\OCSController;
- use OCP\AppFramework\Utility\ITimeFactory;
- use OCP\EventDispatcher\IEventDispatcher;
- use OCP\Files\Events\BeforeDirectFileDownloadEvent;
- use OCP\Files\File;
- use OCP\Files\IRootFolder;
- use OCP\IRequest;
- use OCP\IURLGenerator;
- use OCP\Security\ISecureRandom;
- class DirectController extends OCSController {
-
- private $rootFolder;
-
- private $userId;
-
- private $mapper;
-
- private $random;
-
- private $timeFactory;
-
- private $urlGenerator;
-
- private $eventDispatcher;
- public function __construct(string $appName,
- IRequest $request,
- IRootFolder $rootFolder,
- string $userId,
- DirectMapper $mapper,
- ISecureRandom $random,
- ITimeFactory $timeFactory,
- IURLGenerator $urlGenerator,
- IEventDispatcher $eventDispatcher) {
- parent::__construct($appName, $request);
- $this->rootFolder = $rootFolder;
- $this->userId = $userId;
- $this->mapper = $mapper;
- $this->random = $random;
- $this->timeFactory = $timeFactory;
- $this->urlGenerator = $urlGenerator;
- $this->eventDispatcher = $eventDispatcher;
- }
-
- public function getUrl(int $fileId, int $expirationTime = 60 * 60 * 8): DataResponse {
- $userFolder = $this->rootFolder->getUserFolder($this->userId);
- $files = $userFolder->getById($fileId);
- if ($files === []) {
- throw new OCSNotFoundException();
- }
- if ($expirationTime <= 0 || $expirationTime > (60 * 60 * 24)) {
- throw new OCSBadRequestException('Expiration time should be greater than 0 and less than or equal to ' . (60 * 60 * 24));
- }
- $file = array_shift($files);
- if (!($file instanceof File)) {
- throw new OCSBadRequestException('Direct download only works for files');
- }
- $event = new BeforeDirectFileDownloadEvent($userFolder->getRelativePath($file->getPath()));
- $this->eventDispatcher->dispatchTyped($event);
- if ($event->isSuccessful() === false) {
- throw new OCSForbiddenException('Permission denied to download file');
- }
-
- $direct = new Direct();
- $direct->setUserId($this->userId);
- $direct->setFileId($fileId);
- $token = $this->random->generate(60, ISecureRandom::CHAR_ALPHANUMERIC);
- $direct->setToken($token);
- $direct->setExpiration($this->timeFactory->getTime() + $expirationTime);
- $this->mapper->insert($direct);
- $url = $this->urlGenerator->getAbsoluteURL('remote.php/direct/'.$token);
- return new DataResponse([
- 'url' => $url,
- ]);
- }
- }
|