initial version of the headless download manager delegate

headless/lib/browser/headless_browser_impl.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "headless/lib/browser/headless_browser_impl.h"
 
 #include <string>
 #include <utility>
 #include <vector>
 
@@ skipping 143 matching lines @@
       HeadlessBrowserContextImpl::Create(builder);
 
   if (!browser_context) {
     return nullptr;
   }
 
   HeadlessBrowserContext* result = browser_context.get();
 
   browser_contexts_[browser_context->Id()] = std::move(browser_context);
 
+  HeadlessDownloadManagerDelegate* download_delegate =
+      static_cast<HeadlessDownloadManagerDelegate*>(
+          browser_context->GetDownloadManagerDelegate());
+
+  if (download_behavior_.has_value())
+    download_delegate->SetDownloadBehavior(download_behavior_.value());
+
+  if (download_path_.has_value())
+    download_delegate->SetDownloadDirectory(download_path_.value());
+
   return result;
 }
 
 void HeadlessBrowserImpl::DestroyBrowserContext(
     HeadlessBrowserContextImpl* browser_context) {
   auto it = browser_contexts_.find(browser_context->Id());
   DCHECK(it != browser_contexts_.end());
   browser_contexts_.erase(it);
   if (default_browser_context_ == browser_context)
     SetDefaultBrowserContext(nullptr);
 }
 
 void HeadlessBrowserImpl::SetDefaultBrowserContext(
     HeadlessBrowserContext* browser_context) {
   DCHECK(!browser_context ||
          this == HeadlessBrowserContextImpl::From(browser_context)->browser());
   default_browser_context_ = browser_context;
 }
 
 HeadlessBrowserContext* HeadlessBrowserImpl::GetDefaultBrowserContext() {
   return default_browser_context_;
 }
 
 base::WeakPtr<HeadlessBrowserImpl> HeadlessBrowserImpl::GetWeakPtr() {
   DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
   return weak_ptr_factory_.GetWeakPtr();
 }
 
+void HeadlessBrowserImpl::SetDownloadBehavior(
+    HeadlessDownloadManagerDelegate::DownloadBehavior behavior) {
+  download_behavior_ = behavior;
+
+  for (HeadlessBrowserContext* hbc : GetAllBrowserContexts()) {
+    HeadlessBrowserContextImpl* hbci =
+        static_cast<HeadlessBrowserContextImpl*>(hbc);
+
+    HeadlessDownloadManagerDelegate* download_delegate =
+        static_cast<HeadlessDownloadManagerDelegate*>(
+            hbci->GetDownloadManagerDelegate());
+
+    download_delegate->SetDownloadBehavior(behavior);
+  }
+}
+
+void HeadlessBrowserImpl::SetDownloadDirectory(const base::FilePath& path) {
+  download_path_ = path;
+
+  for (HeadlessBrowserContext* hbc : GetAllBrowserContexts()) {
+    HeadlessBrowserContextImpl* hbci =
+        static_cast<HeadlessBrowserContextImpl*>(hbc);
+
+    HeadlessDownloadManagerDelegate* download_delegate =
+        static_cast<HeadlessDownloadManagerDelegate*>(
+            hbci->GetDownloadManagerDelegate());
+
+    download_delegate->SetDownloadDirectory(path);
+  }
+}
+
 HeadlessWebContents* HeadlessBrowserImpl::GetWebContentsForDevToolsAgentHostId(
     const std::string& devtools_agent_host_id) {
   for (HeadlessBrowserContext* context : GetAllBrowserContexts()) {
     HeadlessWebContents* web_contents =
         context->GetWebContentsForDevToolsAgentHostId(devtools_agent_host_id);
     if (web_contents)
       return web_contents;
   }
   return nullptr;
 }
@@ skipping 85 matching lines @@
 
   // Child processes should not end up here.
   DCHECK(!base::CommandLine::ForCurrentProcess()->HasSwitch(
       ::switches::kProcessType));
 #endif
   return RunContentMain(std::move(options),
                         std::move(on_browser_start_callback));
 }
 
 }  // namespace headless