Fix download & print view

Signed-off-by: John Molakvoæ (skjnldsv) <skjnldsv@protonmail.com>

Author: skjnldsv

src/public.js

@@ -34,6 +34,13 @@ window.addEventListener('DOMContentLoaded', function() {
 		isSecureViewerAvailable: isSecureViewerAvailable(),
 	})
 
+	// If we display a folder, we don't have anything more to do here
+	if (isPublicPage() && !isPdf()) {
+		logger.debug('But this is not a single pdf share')
+		return
+	}
+
+	// If we display a single PDF and we don't use the richdocument secureViewer
 	if (isPublicPage() && isPdf() && !isSecureViewerAvailable()) {
 		const page = location.hash.split('page=')[1] || 0
 		const contentElmt = document.getElementById('files-public-content')
@@ -42,7 +49,11 @@ window.addEventListener('DOMContentLoaded', function() {
 
 		const sharingToken = sharingTokenElmt.value
 		const downloadUrl = generateUrl('/s/{token}/download', { token: sharingToken })
-		const viewerUrl = generateUrl('/apps/files_pdfviewer/?file={downloadUrl}#page={page}', { downloadUrl, page })
+		const viewerUrl = generateUrl('/apps/files_pdfviewer/?file={downloadUrl}&canDownload={canDownload}#page={page}', {
+			canDownload: canDownload() ? 1 : 0,
+			downloadUrl,
+			page,
+		})
 
 		// Create viewer frame
 		const viewerNode = document.createElement('iframe')
@@ -59,46 +70,6 @@ window.addEventListener('DOMContentLoaded', function() {
 		} else {
 			logger.error('Unable to inject the PDF Viewer')
 		}
-
-		// When pdf viewer is loaded
-		addEventListener('load', function() {
-			// If we forbid download, prevent interaction
-			if (!canDownload()) {
-				const pdfViewer = viewerNode.contentDocument.querySelector('.pdfViewer')
-				const PDFViewerApplication = viewerNode.contentWindow.PDFViewerApplication
-
-				if (pdfViewer) {
-					pdfViewer.classList.add('disabledTextSelection')
-				}
-
-				if (PDFViewerApplication) {
-					// Disable download function when downloads are hidden, as even if the
-					// buttons in the UI are hidden the download could still be triggered
-					// with Ctrl|Meta+S.
-					PDFViewerApplication.download = function() {
-					}
-
-					// Disable printing service when downloads are hidden, as even if the
-					// buttons in the UI are hidden the printing could still be triggered
-					// with Ctrl|Meta+P.
-					// Abuse the "supportsPrinting" parameter, which signals that the
-					// browser does not fully support printing, to make PDFViewer disable
-					// the printing service.
-					// "supportsPrinting" is a getter function, so it needs to be deleted
-					// before replacing it with a simple value.
-					delete PDFViewerApplication.supportsPrinting
-					PDFViewerApplication.supportsPrinting = false
-
-					// When printing is not supported a warning is shown by the default
-					// "beforePrint" function when trying to print. That function needs to
-					// be replaced with an empty one to prevent that warning to be shown.
-					PDFViewerApplication.beforePrint = function() {
-					}
-				}
-
-				logger.info('Download, printing and user interaction disabled')
-			}
-		})
 	} else {
 		logger.error('But this does not appear to be a public page')
 	}

src/views/PDFView.vue

@@ -26,13 +26,15 @@
 
 <script>
 import { generateUrl } from '@nextcloud/router'
+import canDownload from '../utils/canDownload'
 
 export default {
 	name: 'PDFView',
 
 	computed: {
 		iframeSrc() {
-			return generateUrl('/apps/files_pdfviewer/?file={file}', {
+			return generateUrl('/apps/files_pdfviewer/?file={file}&canDownload={canDownload}', {
+				canDownload: canDownload() ? 1 : 0,
 				file: this.davPath,
 			})
 		},

src/workersrc.js

@@ -28,6 +28,12 @@ import redirectIfNotIframe from './utils/redirectIfNotIframe'
 // Checks if the page is displayed in an iframe. If not redirect to /.
 redirectIfNotIframe()
 
+// Retrieve the canDownload from the url, this is
+// the most easy way to pass the prop to this iframe
+const queryString = window.location.search
+const urlParams = new URLSearchParams(queryString)
+const canDownload = urlParams.get('canDownload')
+
 // When "PDFViewerApplication.webViewerInitialized" is executed (once
 // "PDFViewerApplication.initialize" is done) it opens the PDF file via URL,
 // which requires the PDFViewerApplication to be properly configured, so the
@@ -45,6 +51,43 @@ function initializeCustomPDFViewerApplication() {
 	PDFViewerApplicationOptions.set('cMapUrl', document.getElementsByTagName('head')[0].getAttribute('data-cmapurl'))
 	PDFViewerApplicationOptions.set('enablePermissions', true)
 
+	if (canDownload === '0') {
+		const pdfViewer = window.document.querySelector('.pdfViewer')
+
+		if (pdfViewer) {
+			pdfViewer.classList.add('disabledTextSelection')
+		}
+
+		if (PDFViewerApplication) {
+			// Disable download function when downloads are hidden, as even if the
+			// buttons in the UI are hidden the download could still be triggered
+			// with Ctrl|Meta+S.
+			PDFViewerApplication.download = function() {
+			}
+
+			// Disable printing service when downloads are hidden, as even if the
+			// buttons in the UI are hidden the printing could still be triggered
+			// with Ctrl|Meta+P.
+			// Abuse the "supportsPrinting" parameter, which signals that the
+			// browser does not fully support printing, to make PDFViewer disable
+			// the printing service.
+			// "supportsPrinting" is a getter function, so it needs to be deleted
+			// before replacing it with a simple value.
+			delete PDFViewerApplication.supportsPrinting
+			PDFViewerApplication.supportsPrinting = false
+
+			// When printing is not supported a warning is shown by the default
+			// "beforePrint" function when trying to print. That function needs to
+			// be replaced with an empty one to prevent that warning to be shown.
+			PDFViewerApplication.beforePrint = function() {
+			}
+		}
+
+		logger.info('Download, print and user interaction disabled')
+	} else {
+		logger.info('Download and print available')
+	}
+
 	logger.debug('Initialized files_pdfviewer', PDFViewerApplicationOptions.getAll())
 }