2018-04-13 15:04:41 +03:00
|
|
|
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
/**
|
|
|
|
* @copyright 2018, Roeland Jago Douma <roeland@famdouma.nl>
|
|
|
|
*
|
|
|
|
* @author Roeland Jago Douma <roeland@famdouma.nl>
|
|
|
|
*
|
|
|
|
* @license GNU AGPL version 3 or any later version
|
|
|
|
*
|
|
|
|
* This program is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU Affero General Public License as
|
|
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
|
|
* License, or (at your option) any later version.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU Affero General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
|
|
|
namespace OCA\DAV\Controller;
|
|
|
|
|
2018-04-13 17:54:24 +03:00
|
|
|
use OCA\DAV\Db\Direct;
|
2018-04-13 15:04:41 +03:00
|
|
|
use OCA\DAV\Db\DirectMapper;
|
2018-04-13 17:54:24 +03:00
|
|
|
use OCP\AppFramework\Http\DataResponse;
|
|
|
|
use OCP\AppFramework\OCS\OCSNotFoundException;
|
2018-04-13 15:04:41 +03:00
|
|
|
use OCP\AppFramework\OCSController;
|
2018-04-13 17:54:24 +03:00
|
|
|
use OCP\AppFramework\Utility\ITimeFactory;
|
2018-04-13 15:04:41 +03:00
|
|
|
use OCP\Files\IRootFolder;
|
|
|
|
use OCP\IRequest;
|
2018-04-13 17:54:24 +03:00
|
|
|
use OCP\IURLGenerator;
|
|
|
|
use OCP\Security\ISecureRandom;
|
2018-04-13 15:04:41 +03:00
|
|
|
|
|
|
|
class DirectController extends OCSController {
|
|
|
|
|
|
|
|
/** @var IRootFolder */
|
|
|
|
private $rootFolder;
|
|
|
|
|
|
|
|
/** @var string */
|
|
|
|
private $userId;
|
|
|
|
|
|
|
|
/** @var DirectMapper */
|
|
|
|
private $mapper;
|
|
|
|
|
2018-04-13 17:54:24 +03:00
|
|
|
/** @var ISecureRandom */
|
|
|
|
private $random;
|
|
|
|
|
|
|
|
/** @var ITimeFactory */
|
|
|
|
private $timeFactory;
|
|
|
|
|
|
|
|
/** @var IURLGenerator */
|
|
|
|
private $urlGenerator;
|
|
|
|
|
2018-04-13 15:04:41 +03:00
|
|
|
|
|
|
|
public function __construct(string $appName,
|
|
|
|
IRequest $request,
|
|
|
|
IRootFolder $rootFolder,
|
|
|
|
string $userId,
|
2018-04-13 17:54:24 +03:00
|
|
|
DirectMapper $mapper,
|
|
|
|
ISecureRandom $random,
|
|
|
|
ITimeFactory $timeFactory,
|
|
|
|
IURLGenerator $urlGenerator) {
|
2018-04-13 15:04:41 +03:00
|
|
|
parent::__construct($appName, $request);
|
|
|
|
|
|
|
|
$this->rootFolder = $rootFolder;
|
|
|
|
$this->userId = $userId;
|
|
|
|
$this->mapper = $mapper;
|
2018-04-13 17:54:24 +03:00
|
|
|
$this->random = $random;
|
|
|
|
$this->timeFactory = $timeFactory;
|
|
|
|
$this->urlGenerator = $urlGenerator;
|
2018-04-13 15:04:41 +03:00
|
|
|
}
|
|
|
|
|
2018-04-13 17:54:24 +03:00
|
|
|
/**
|
|
|
|
* @NoAdminRequired
|
|
|
|
*/
|
|
|
|
public function getUrl(int $fileId): DataResponse {
|
|
|
|
$userFolder = $this->rootFolder->getUserFolder($this->userId);
|
|
|
|
|
|
|
|
$files = $userFolder->getById($fileId);
|
|
|
|
|
|
|
|
if ($files === []) {
|
|
|
|
throw new OCSNotFoundException();
|
|
|
|
}
|
|
|
|
|
|
|
|
$file = array_shift($files);
|
2018-04-13 18:40:52 +03:00
|
|
|
$storage = $file->getStorage();
|
|
|
|
$directDownload = $storage->getDirectDownload($file->getInternalPath());
|
2018-04-13 17:54:24 +03:00
|
|
|
|
2018-04-13 18:40:52 +03:00
|
|
|
if (isset($directDownload['url'])) {
|
|
|
|
$url = $directDownload['url'];
|
|
|
|
} else {
|
|
|
|
// Fallback to our default implemenation
|
|
|
|
$direct = new Direct();
|
|
|
|
$direct->setUserId($this->userId);
|
|
|
|
$direct->setFileId($fileId);
|
2018-04-13 17:54:24 +03:00
|
|
|
|
2018-04-13 18:40:52 +03:00
|
|
|
$token = $this->random->generate(60, ISecureRandom::CHAR_UPPER . ISecureRandom::CHAR_LOWER . ISecureRandom::CHAR_DIGITS);
|
|
|
|
$direct->setToken($token);
|
|
|
|
$direct->setExpiration($this->timeFactory->getTime() + 60 * 60 * 8);
|
2018-04-13 17:54:24 +03:00
|
|
|
|
2018-04-13 18:40:52 +03:00
|
|
|
$this->mapper->insert($direct);
|
2018-04-13 17:54:24 +03:00
|
|
|
|
2018-04-13 18:40:52 +03:00
|
|
|
$url = $this->urlGenerator->getAbsoluteURL('remote.php/direct/'.$token);
|
|
|
|
}
|
2018-04-13 15:04:41 +03:00
|
|
|
|
2018-04-13 17:54:24 +03:00
|
|
|
return new DataResponse([
|
2018-04-13 18:40:52 +03:00
|
|
|
'url' => $url,
|
2018-04-13 17:54:24 +03:00
|
|
|
]);
|
2018-04-13 15:04:41 +03:00
|
|
|
}
|
|
|
|
}
|