initial version of the headless download manager delegate

headless/lib/browser/headless_devtools_manager_delegate.cc

Index: headless/lib/browser/headless_devtools_manager_delegate.cc
diff --git a/headless/lib/browser/headless_devtools_manager_delegate.cc b/headless/lib/browser/headless_devtools_manager_delegate.cc
index 1be2c696618d3b67b18d6e7e8c2820e3a2a59b2f..530bdc4d60a6ffbb0b6cd0872102f82db2e435f5 100644
--- a/headless/lib/browser/headless_devtools_manager_delegate.cc
+++ b/headless/lib/browser/headless_devtools_manager_delegate.cc
@@ -15,6 +15,7 @@
 #include "headless/grit/headless_lib_resources.h"
 #include "headless/lib/browser/headless_browser_context_impl.h"
 #include "headless/lib/browser/headless_browser_impl.h"
+#include "headless/lib/browser/headless_download_manager_delegate.h"
 #include "headless/lib/browser/headless_web_contents_impl.h"
 #include "headless/public/devtools/domains/target.h"
 #include "printing/units.h"
@@ -47,6 +48,10 @@ std::unique_ptr<base::DictionaryValue> CreateSuccessResponse(
   return response;
 }
 
+std::unique_ptr<base::DictionaryValue> CreateSuccessResponse(int command_id) {
+  return CreateSuccessResponse(command_id, nullptr);
+}
+
 std::unique_ptr<base::DictionaryValue> CreateErrorResponse(
     int command_id,
     int error_code,
@@ -192,6 +197,12 @@ HeadlessDevToolsManagerDelegate::HeadlessDevToolsManagerDelegate(
   command_map_["Browser.setWindowBounds"] =
       base::Bind(&HeadlessDevToolsManagerDelegate::SetWindowBounds,
                  base::Unretained(this));
+  command_map_["Browser.setDownloadBehavior"] =
+      base::Bind(&HeadlessDevToolsManagerDelegate::SetDownloadBehavior,
+                 base::Unretained(this));
+  command_map_["Browser.setDownloadDirectory"] =
+      base::Bind(&HeadlessDevToolsManagerDelegate::SetDownloadDirectory,
+                 base::Unretained(this));
 
   async_command_map_["Page.printToPDF"] = base::Bind(
       &HeadlessDevToolsManagerDelegate::PrintToPDF, base::Unretained(this));
@@ -518,4 +529,39 @@ HeadlessDevToolsManagerDelegate::SetWindowBounds(
   return CreateSuccessResponse(command_id, nullptr);
 }
 
+std::unique_ptr<base::DictionaryValue>
+HeadlessDevToolsManagerDelegate::SetDownloadBehavior(
+    int command_id,
+    const base::DictionaryValue* params) {
+  std::string behavior;
+  if (!params || !params->GetString("behavior", &behavior))
+    return CreateInvalidParamResponse(command_id, "behavior");
+
+  HeadlessDownloadManagerDelegate::DownloadBehavior download_behaviour =
+      HeadlessDownloadManagerDelegate::DownloadBehavior::DENY;
+
+  if (behavior == "allow") {
+    download_behaviour =
+        HeadlessDownloadManagerDelegate::DownloadBehavior::ALLOW;
+  } else if (behavior == "deny") {
+    download_behaviour =
+        HeadlessDownloadManagerDelegate::DownloadBehavior::DENY;
+  }
+
+  browser_->SetDownloadBehavior(download_behaviour);
+  return CreateSuccessResponse(command_id);
+}
+
+std::unique_ptr<base::DictionaryValue>
+HeadlessDevToolsManagerDelegate::SetDownloadDirectory(
+    int command_id,
+    const base::DictionaryValue* params) {
+  std::string path;
+  if (!params || !params->GetString("path", &path))
+    return CreateInvalidParamResponse(command_id, "path");
+
+  browser_->SetDownloadDirectory(base::FilePath(path));
+  return CreateSuccessResponse(command_id);
+}
+
 }  // namespace headless