From f3dc4206ee29c28badceccffb3f4c4cfd709687c Mon Sep 17 00:00:00 2001
From: Vivien Nicolas <21@vingtetun.org>
Date: Wed, 19 Oct 2011 06:29:14 +0200
Subject: [PATCH] Make the 'download' button behavior more robust

---
 extensions/firefox/components/pdfContentHandler.js | 10 +++-------
 web/viewer.js                                      | 10 +++++-----
 2 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/extensions/firefox/components/pdfContentHandler.js b/extensions/firefox/components/pdfContentHandler.js
index a04b717a2..d5f466e3f 100644
--- a/extensions/firefox/components/pdfContentHandler.js
+++ b/extensions/firefox/components/pdfContentHandler.js
@@ -48,16 +48,12 @@ pdfContentHandler.prototype = {
       throw NS_ERROR_WONT_HANDLE_CONTENT;
     }
 
-    // To allow a Download feature we need to ensure pdf.js
-    // is not opened again if the request for opening an
-    // application/pdf document has been done by itself.
-    let location = window.location.toString();
-    if (location.indexOf(url.replace('%s', '')) == 0)
+    let targetUrl = aRequest.URI.spec;
+    if (targetUrl.indexOf('?pdfjs.action=download') >= 0)
       throw NS_ERROR_WONT_HANDLE_CONTENT;
 
     aRequest.cancel(Cr.NS_BINDING_ABORTED);
-    let uri = aRequest.URI;
-    window.location = url.replace('%s', uri.spec);
+    window.location = url.replace('%s', targetUrl);
   },
 
   classID: Components.ID('{2278dfd0-b75c-11e0-8257-1ba3d93c9f1a}'),
diff --git a/web/viewer.js b/web/viewer.js
index 27211ce1c..12d0a3f37 100644
--- a/web/viewer.js
+++ b/web/viewer.js
@@ -85,10 +85,6 @@ var PDFView = {
     this.setScale(newScale, true);
   },
 
-  download: function pdfViewDownload() {
-    window.open(document.title, '_parent', 'pdf=yes');
-  },
-
   set page(val) {
     var pages = this.pages;
     var input = document.getElementById('pageNumber');
@@ -119,7 +115,7 @@ var PDFView = {
   },
 
   open: function pdfViewOpen(url, scale) {
-    document.title = url;
+    document.title = this.url = url;
 
     getPdf(
       {
@@ -135,6 +131,10 @@ var PDFView = {
       });
   },
 
+  download: function pdfViewDownload() {
+    window.open(this.url + '?pdfjs.action=download', '_parent');
+  },
+
   navigateTo: function pdfViewNavigateTo(dest) {
     if (typeof dest === 'string')
       dest = this.destinations[dest];