Implement window management devtools commands for headless.

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 182 matching lines @@
     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;
 }
 
+HeadlessWebContentsImpl* HeadlessBrowserImpl::GetWebContentsForWindowId(
+    const int window_id) {
+  for (HeadlessBrowserContext* context : GetAllBrowserContexts()) {
+    for (HeadlessWebContents* web_contents : context->GetAllWebContents()) {
+      auto* contents = HeadlessWebContentsImpl::From(web_contents);
+      if (contents->window_id() == window_id) {
+        return contents;
+      }
+    }
+  }
+  return nullptr;
+}
+
 HeadlessBrowserContext* HeadlessBrowserImpl::GetBrowserContextForId(
     const std::string& id) {
   auto find_it = browser_contexts_.find(id);
   if (find_it == browser_contexts_.end())
     return nullptr;
   return find_it->second.get();
 }
 
 HeadlessDevToolsTarget* HeadlessBrowserImpl::GetDevToolsTarget() {
   return agent_host_ ? this : nullptr;
@@ skipping 61 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