headless: expose window size through headless_shell switch.

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 <iostream>
 #include <memory>
 #include <string>
 
 #include "base/base64.h"
 #include "base/bind.h"
@@ skipping 24 matching lines @@
 using headless::HeadlessDevToolsClient;
 using headless::HeadlessWebContents;
 namespace page = headless::page;
 namespace runtime = headless::runtime;
 
 namespace {
 // Address where to listen to incoming DevTools connections.
 const char kDevToolsHttpServerAddress[] = "127.0.0.1";
 // Default file name for screenshot. Can be overriden by "--screenshot" switch.
 const char kDefaultScreenshotFileName[] = "screenshot.png";
+
+bool ParseWindowSize(std::string window_size, gfx::Size* parsed_window_size) {
+  int width, height = 0;
+  if (sscanf(window_size.c_str(), "%dx%d", &width, &height) >= 2 &&
+      width >= 0 && height >= 0) {
+    parsed_window_size->set_width(width);
+    parsed_window_size->set_height(height);
+    return true;
+  }
+  return false;
 }
+}  // namespace
 
 // A sample application which demonstrates the use of the headless API.
 class HeadlessShell : public HeadlessWebContents::Observer, page::Observer {
  public:
   HeadlessShell()
       : browser_(nullptr),
         devtools_client_(HeadlessDevToolsClient::Create()),
         web_contents_(nullptr),
         processed_page_ready_(false),
         browser_context_(nullptr) {}
@@ skipping 289 matching lines @@
     builder.SetGLImplementation(
         command_line.GetSwitchValueASCII(headless::switches::kUseGL));
   }
 
   if (command_line.HasSwitch(headless::switches::kUserDataDir)) {
     builder.SetUserDataDir(
         command_line.GetSwitchValuePath(headless::switches::kUserDataDir));
     builder.SetIncognitoMode(false);
   }
 
+  if (command_line.HasSwitch(headless::switches::kWindowSize)) {
+    std::string window_size =
+        command_line.GetSwitchValueASCII(headless::switches::kWindowSize);
+    gfx::Size parsed_window_size;
+    if (!ParseWindowSize(window_size, &parsed_window_size)) {
+      LOG(ERROR) << "Malformed window size";
+      return EXIT_FAILURE;
+    }
+    builder.SetWindowSize(parsed_window_size);
+  }
+
   return HeadlessBrowserMain(
       builder.Build(),
       base::Bind(&HeadlessShell::OnStart, base::Unretained(&shell)));
 }