From 6c4ab914c09fe62a27a58cff9f6ecbca6d19960b Mon Sep 17 00:00:00 2001
From: Milan Burda <milan.burda@gmail.com>
Date: Wed, 21 Aug 2019 20:31:59 +0200
Subject: [PATCH] feat: add session.downloadURL()

---
 docs/api/session.md                   |  8 ++++++++
 shell/browser/api/atom_api_session.cc | 10 ++++++++++
 shell/browser/api/atom_api_session.h  |  1 +
 spec-main/api-session-spec.js         | 12 ++++++++++++
 4 files changed, 31 insertions(+)

diff --git a/docs/api/session.md b/docs/api/session.md
index 82da931e92814..b7e52bc6f4470 100644
--- a/docs/api/session.md
+++ b/docs/api/session.md
@@ -386,6 +386,14 @@ Returns `String` - The user agent for this session.
 
 Returns `Promise<Buffer>` - resolves with blob data.
 
+#### `ses.downloadURL(url)`
+
+* `url` String
+
+Initiates a download of the resource at `url`.
+The API will generate a [DownloadItem](download-item.md) that can be accessed
+with the [will-download](#event-will-download) event.
+
 #### `ses.createInterruptedDownload(options)`
 
 * `options` Object
diff --git a/shell/browser/api/atom_api_session.cc b/shell/browser/api/atom_api_session.cc
index 19cae79881289..f1dee882166c5 100644
--- a/shell/browser/api/atom_api_session.cc
+++ b/shell/browser/api/atom_api_session.cc
@@ -18,6 +18,7 @@
 #include "chrome/browser/browser_process.h"
 #include "chrome/common/pref_names.h"
 #include "components/download/public/common/download_danger_type.h"
+#include "components/download/public/common/download_url_parameters.h"
 #include "components/prefs/pref_service.h"
 #include "components/prefs/value_map_pref_store.h"
 #include "components/proxy_config/proxy_config_dictionary.h"
@@ -506,6 +507,14 @@ v8::Local<v8::Promise> Session::GetBlobData(v8::Isolate* isolate,
   return handle;
 }
 
+void Session::DownloadURL(const GURL& url) {
+  auto* download_manager =
+      content::BrowserContext::GetDownloadManager(browser_context());
+  auto download_params = std::make_unique<download::DownloadUrlParameters>(
+      url, MISSING_TRAFFIC_ANNOTATION);
+  download_manager->DownloadUrl(std::move(download_params));
+}
+
 void Session::CreateInterruptedDownload(const mate::Dictionary& options) {
   int64_t offset = 0, length = 0;
   double start_time = 0.0;
@@ -654,6 +663,7 @@ void Session::BuildPrototype(v8::Isolate* isolate,
       .SetMethod("setUserAgent", &Session::SetUserAgent)
       .SetMethod("getUserAgent", &Session::GetUserAgent)
       .SetMethod("getBlobData", &Session::GetBlobData)
+      .SetMethod("downloadURL", &Session::DownloadURL)
       .SetMethod("createInterruptedDownload",
                  &Session::CreateInterruptedDownload)
       .SetMethod("setPreloads", &Session::SetPreloads)
diff --git a/shell/browser/api/atom_api_session.h b/shell/browser/api/atom_api_session.h
index 07322b85a01b1..8adfb2b7208e6 100644
--- a/shell/browser/api/atom_api_session.h
+++ b/shell/browser/api/atom_api_session.h
@@ -79,6 +79,7 @@ class Session : public mate::TrackableObject<Session>,
   std::string GetUserAgent();
   v8::Local<v8::Promise> GetBlobData(v8::Isolate* isolate,
                                      const std::string& uuid);
+  void DownloadURL(const GURL& url);
   void CreateInterruptedDownload(const mate::Dictionary& options);
   void SetPreloads(const std::vector<base::FilePath::StringType>& preloads);
   std::vector<base::FilePath::StringType> GetPreloads() const;
diff --git a/spec-main/api-session-spec.js b/spec-main/api-session-spec.js
index 611eee4fde783..2e58cf305f0cf 100644
--- a/spec-main/api-session-spec.js
+++ b/spec-main/api-session-spec.js
@@ -595,6 +595,18 @@ describe('session module', () => {
       fs.unlinkSync(downloadFilePath)
     }
 
+    it('can download using session.downloadURL', (done) => {
+      const port = downloadServer.address().port
+      session.defaultSession.once('will-download', function (e, item) {
+        item.savePath = downloadFilePath
+        item.on('done', function (e, state) {
+          assertDownload(state, item)
+          done()
+        })
+      })
+      session.defaultSession.downloadURL(`${url}:${port}`)
+    })
+
     it('can download using WebContents.downloadURL', (done) => {
       const port = downloadServer.address().port
       w.webContents.session.once('will-download', function (e, item) {