[headless] Add command line flag for default background color.

headless/app/headless_shell.cc

 // Copyright 2016 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 <memory>
 #include <sstream>
 #include <string>
 #include <utility>
 
 #include "base/base64.h"
@@ skipping 118 matching lines @@
   web_contents_ = nullptr;
   browser_context_->Close();
   browser_->Shutdown();
 }
 
 void HeadlessShell::DevToolsTargetReady() {
   web_contents_->GetDevToolsTarget()->AttachClient(devtools_client_.get());
   devtools_client_->GetInspector()->GetExperimental()->AddObserver(this);
   devtools_client_->GetPage()->GetExperimental()->AddObserver(this);
   devtools_client_->GetPage()->Enable();
-  // Check if the document had already finished loading by the time we
-  // attached.
 
   devtools_client_->GetEmulation()->GetExperimental()->AddObserver(this);
 
   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
           switches::kDeterministicFetch)) {
     devtools_client_->GetPage()->GetExperimental()->SetControlNavigations(
         headless::page::SetControlNavigationsParams::Builder()
             .SetEnabled(true)
             .Build());
   }
+  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
+          switches::kDefaultBackgroundColor)) {
+    std::string color_hex =
+        base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
+            switches::kDefaultBackgroundColor);
+    uint32_t color;
+    CHECK(base::HexStringToUInt(color_hex, &color))
+        << "Expected a hex value for --default-background-color=";
+    auto rgba = headless::dom::RGBA::Builder()
+                    .SetR((color & 0xff000000) >> 24)
+                    .SetG((color & 0x00ff0000) >> 16)
+                    .SetB((color & 0x0000ff00) >> 8)
+                    .SetA(color & 0x000000ff)
+                    .Build();
+    devtools_client_->GetEmulation()
+        ->GetExperimental()
+        ->SetDefaultBackgroundColorOverride(
+            headless::emulation::SetDefaultBackgroundColorOverrideParams::
+                Builder()
+                    .SetColor(std::move(rgba))
+                    .Build());
+  }
 
   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
           switches::kVirtualTimeBudget)) {
     std::string budget_ms_ascii =
         base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
             switches::kVirtualTimeBudget);
     int budget_ms;
     CHECK(base::StringToInt(budget_ms_ascii, &budget_ms))
         << "Expected an integer value for --virtual-time-budget=";
     devtools_client_->GetEmulation()->GetExperimental()->SetVirtualTimePolicy(
         emulation::SetVirtualTimePolicyParams::Builder()
             .SetPolicy(
                 emulation::VirtualTimePolicy::PAUSE_IF_NETWORK_FETCHES_PENDING)
             .SetBudget(budget_ms)
             .Build());
   } else {
+    // Check if the document had already finished loading by the time we
+    // attached.
     PollReadyState();
   }
 
   if (base::CommandLine::ForCurrentProcess()->HasSwitch(switches::kTimeout)) {
     std::string timeout_ms_ascii =
         base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
             switches::kTimeout);
     int timeout_ms;
     CHECK(base::StringToInt(timeout_ms_ascii, &timeout_ms))
         << "Expected an integer value for --timeout=";
@@ skipping 254 matching lines @@
 }
 
 bool ValidateCommandLine(const base::CommandLine& command_line) {
   if (!command_line.HasSwitch(switches::kRemoteDebuggingPort)) {
     if (command_line.GetArgs().size() <= 1)
       return true;
     LOG(ERROR) << "Open multiple tabs is only supported when the "
                << "remote debug port is set.";
     return false;
   }
+  if (command_line.HasSwitch(switches::kDefaultBackgroundColor)) {
+    LOG(ERROR) << "Setting default background color is disabled "
+               << "when remote debugging is enabled.";
+    return false;
+  }
   if (command_line.HasSwitch(switches::kDumpDom)) {
     LOG(ERROR) << "Dump DOM is disabled when remote debugging is enabled.";
     return false;
   }
+  if (command_line.HasSwitch(switches::kPrintToPDF)) {
+    LOG(ERROR) << "Print to PDF is disabled "
+               << "when remote debugging is enabled.";
+    return false;
+  }
   if (command_line.HasSwitch(switches::kRepl)) {
     LOG(ERROR) << "Evaluate Javascript is disabled "
                << "when remote debugging is enabled.";
     return false;
   }
   if (command_line.HasSwitch(switches::kScreenshot)) {
     LOG(ERROR) << "Capture screenshot is disabled "
                << "when remote debugging is enabled.";
     return false;
   }
-  if (command_line.HasSwitch(switches::kPrintToPDF)) {
-    LOG(ERROR) << "Print to PDF is disabled "
-               << "when remote debugging is enabled.";
-    return false;
-  }
   if (command_line.HasSwitch(switches::kTimeout)) {
     LOG(ERROR) << "Navigation timeout is disabled "
                << "when remote debugging is enabled.";
     return false;
   }
   if (command_line.HasSwitch(switches::kVirtualTimeBudget)) {
     LOG(ERROR) << "Virtual time budget is disabled "
                << "when remote debugging is enabled.";
     return false;
   }
@@ skipping 94 matching lines @@
     if (net::HttpUtil::IsValidHeaderValue(ua))
       builder.SetUserAgent(ua);
   }
 
   return HeadlessBrowserMain(
       builder.Build(),
       base::Bind(&HeadlessShell::OnStart, base::Unretained(&shell)));
 }
 
 }  // namespace headless