Do not load PDF viewer if public share has a download limit

The PDF viewer needs to download the file in order to render it in the
client. Due to this, if a public share has a download limit, loading the
PDF viewer automatically reduces the number of available downloads by
one, even if the user does not download the file manually. To prevent
that now the PDF viewer is not loaded in public shares if the share has
a download limit.

Signed-off-by: Daniel Calviño Sánchez <[email protected]>

Author: danxuliu

lib/Listeners/LoadPublicViewerListener.php

@@ -28,11 +28,21 @@
 
 use OCA\Files_PDFViewer\AppInfo\Application;
 use OCA\Files_Sharing\Event\BeforeTemplateRenderedEvent;
+use OCP\AppFramework\QueryException;
 use OCP\EventDispatcher\Event;
 use OCP\EventDispatcher\IEventListener;
+use OCP\IServerContainer;
 use OCP\Util;
 
 class LoadPublicViewerListener implements IEventListener {
+
+	/** @var IServerContainer */
+	private $serverContainer;
+
+	public function __construct(IServerContainer $serverContainer) {
+		$this->serverContainer = $serverContainer;
+	}
+
 	public function handle(Event $event): void {
 		if (!$event instanceof BeforeTemplateRenderedEvent) {
 			return;
@@ -44,6 +54,27 @@ public function handle(Event $event): void {
 			return;
 		}
 
+		// Do not load the viewer if there is a download limit
+		if ($this->getDownloadLimit($event->getShare()->getToken()) >= 0) {
+			return;
+		}
+
 		Util::addScript(Application::APP_ID, 'files_pdfviewer-public');
 	}
+
+	private function getDownloadLimit(string $shareToken): int {
+		try {
+			$limitMapper = $this->serverContainer->get('\OCA\Files_DownloadLimit\Db\LimitMapper');
+		} catch (QueryException $e) {
+			return -1;
+		}
+
+		try {
+			$shareLimit = $limitMapper->get($shareToken);
+		} catch (\Exception $e) {
+			return -1;
+		}
+
+		return $shareLimit->getLimit();
+	}
 }