diff --git a/web/app.js b/web/app.js index 3a66de48a..99a943d8f 100644 --- a/web/app.js +++ b/web/app.js @@ -949,62 +949,59 @@ const PDFViewerApplication = { ); }, - download({ sourceEventType = "download" } = {}) { - function downloadByUrl() { - downloadManager.downloadUrl(url, filename); - } - - const downloadManager = this.downloadManager, - url = this.baseUrl, - filename = this._docFilename; - - // When the PDF document isn't ready, or the PDF file is still downloading, - // simply download using the URL. - if (!this.pdfDocument || !this.downloadComplete) { - downloadByUrl(); + /** + * @private + */ + _ensureDownloadComplete() { + if (this.pdfDocument && this.downloadComplete) { return; } + throw new Error("PDF document not downloaded."); + }, - this.pdfDocument - .getData() - .then(function (data) { - const blob = new Blob([data], { type: "application/pdf" }); - downloadManager.download(blob, url, filename, sourceEventType); - }) - .catch(downloadByUrl); // Error occurred, try downloading with the URL. + async download({ sourceEventType = "download" } = {}) { + const url = this.baseUrl, + filename = this._docFilename; + try { + this._ensureDownloadComplete(); + + const data = await this.pdfDocument.getData(); + const blob = new Blob([data], { type: "application/pdf" }); + + await this.downloadManager.download(blob, url, filename, sourceEventType); + } catch (reason) { + // When the PDF document isn't ready, or the PDF file is still + // downloading, simply download using the URL. + await this.downloadManager.downloadUrl(url, filename); + } }, async save({ sourceEventType = "download" } = {}) { if (this._saveInProgress) { return; } - - const downloadManager = this.downloadManager, - url = this.baseUrl, - filename = this._docFilename; - - // When the PDF document isn't ready, or the PDF file is still downloading, - // simply download using the URL. - if (!this.pdfDocument || !this.downloadComplete) { - this.download({ sourceEventType }); - return; - } this._saveInProgress = true; await this.pdfScriptingManager.dispatchWillSave(); - this.pdfDocument - .saveDocument(this.pdfDocument.annotationStorage) - .then(data => { - const blob = new Blob([data], { type: "application/pdf" }); - downloadManager.download(blob, url, filename, sourceEventType); - }) - .catch(() => { - this.download({ sourceEventType }); - }) - .finally(async () => { - await this.pdfScriptingManager.dispatchDidSave(); - this._saveInProgress = false; - }); + const url = this.baseUrl, + filename = this._docFilename; + try { + this._ensureDownloadComplete(); + + const data = await this.pdfDocument.saveDocument( + this.pdfDocument.annotationStorage + ); + const blob = new Blob([data], { type: "application/pdf" }); + + await this.downloadManager.download(blob, url, filename, sourceEventType); + } catch (reason) { + // When the PDF document isn't ready, or the PDF file is still + // downloading, simply fallback to a "regular" download. + await this.download({ sourceEventType }); + } finally { + await this.pdfScriptingManager.dispatchDidSave(); + this._saveInProgress = false; + } }, downloadOrSave(options) {