%PDF-
%PDF-
Mini Shell
Mini Shell
<?php
declare(strict_types=1);
/**
* SPDX-FileCopyrightText: 2024 Nextcloud GmbH and Nextcloud contributors
* SPDX-License-Identifier: AGPL-3.0-or-later
*/
namespace OCA\Files_DownloadLimit\Dav;
use OCA\DAV\Connector\Sabre\File as SabreFile;
use OCA\Files_DownloadLimit\Db\Limit;
use OCA\Files_DownloadLimit\Db\LimitMapper;
use OCA\Files_DownloadLimit\LimitList;
use OCP\AppFramework\Db\DoesNotExistException;
use OCP\Files\File;
use OCP\IUser;
use OCP\IUserSession;
use OCP\Share\IManager as IShareManager;
use OCP\Share\IShare;
use Psr\Log\LoggerInterface;
use Sabre\DAV\INode;
use Sabre\DAV\PropFind;
use Sabre\DAV\Server;
use Sabre\DAV\ServerPlugin;
class PropFindPlugin extends ServerPlugin {
public const DOWNLOAD_LIMITS_PROPERTY = '{http://nextcloud.org/ns}share-download-limits';
public function __construct(
private IShareManager $shareManager,
private IUserSession $userSession,
private LimitMapper $limitMapper,
private LoggerInterface $logger,
) {
}
public function initialize(Server $server): void {
$server->on('propFind', [$this, 'propFind']);
}
public function propFind(PropFind $propFind, INode $node) {
if (!in_array(static::DOWNLOAD_LIMITS_PROPERTY, $propFind->getRequestedProperties(), true)) {
return;
}
if (!($node instanceof SabreFile)) { // Only allowed on files
return;
}
$propFind->handle(
static::DOWNLOAD_LIMITS_PROPERTY,
function () use ($node) {
$user = $this->userSession->getUser();
if (!($user instanceof IUser)) {
return new LimitList([]);
}
$externalShares = $this->getSharesForTypes(
$user,
$node->getNode(),
[
IShare::TYPE_LINK,
IShare::TYPE_EMAIL,
],
);
/** @var Limit[] $limits */
$limits = array_values(array_filter(array_map(function (IShare $share) {
try {
return $this->limitMapper->get($share->getToken());
} catch (DoesNotExistException $e) {
// Allow as no limit has been set
return null;
}
}, $externalShares)));
return new LimitList($limits);
},
);
}
/**
* @param int[] $types
* @return IShare[]
*/
private function getSharesForTypes(IUser $user, File $file, array $types) {
/** @var IShare[] $shares */
$shares = [];
foreach ($types as $type) {
$shares = array_merge($shares, $this->shareManager->getSharesBy($user->getUID(), $type, $file, false, -1, 0));
}
return $shares;
}
}
Zerion Mini Shell 1.0